org.neo4j.server.extension.script.JRubyExtensionInitializer.java Source code

Introduction

Here is the source code for org.neo4j.server.extension.script.JRubyExtensionInitializer.java

Source

/**
 * Copyright (c) 2002-2011 "Neo Technology,"
 * Network Engine for Objects in Lund AB [http://neotechnology.com]
 *
 * This file is part of Neo4j.
 *
 * Neo4j is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as
 * published by the Free Software Foundation, either version 3 of the
 * License, or (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program. If not, see <http://www.gnu.org/licenses/>.
 */
package org.neo4j.server.extension.script;

import org.apache.commons.configuration.Configuration;
import org.mortbay.jetty.Server;
import org.neo4j.graphdb.GraphDatabaseService;
import org.neo4j.kernel.GraphDatabaseAPI;
import org.neo4j.server.NeoServer;
import org.neo4j.server.NeoServerWithEmbeddedWebServer;
import org.neo4j.server.logging.Logger;
import org.neo4j.server.plugins.Injectable;
import org.neo4j.server.plugins.SPIPluginLifecycle;

import java.io.IOException;
import java.util.Arrays;
import java.util.Collection;

import static org.neo4j.server.extension.script.Util.listMountPoints;
import static org.neo4j.server.extension.script.Util.locateGemHome;
import static org.neo4j.server.plugins.TypedInjectable.injectable;

public class JRubyExtensionInitializer implements SPIPluginLifecycle {

    public static final String CONFIG_PREFIX = JRubyExtensionInitializer.class.getPackage().getName() + ".";
    private static final Logger LOG = new Logger(JRubyExtensionInitializer.class);

    @Override
    public Collection<Injectable<?>> start(GraphDatabaseService graphDatabaseService, Configuration config) {
        throw new IllegalAccessError();
    }

    public void stop() {
    }

    @Override
    public Collection<Injectable<?>> start(final NeoServer neoServer) {
        LOG.info("START " + JRubyExtensionInitializer.class.toString());

        final Server jetty = getJetty(neoServer);
        final GraphDatabaseAPI gds = neoServer.getDatabase().graph;
        final Configuration configuration = neoServer.getConfiguration();

        final String gemHome = locateGemHome(configuration);

        final EmbeddedRackApplicationFactory factory;
        try {
            factory = new EmbeddedRackApplicationFactory(gds, gemHome);
        } catch (IOException e) {
            throw new RuntimeException(e);
        }

        JRubyRackContextMap contextMap = new JRubyRackContextMap();

        for (String mount : listMountPoints(configuration)) {
            try {
                LOG.info("creating jruby-endpoint at " + mount);
                final JRubyRackContext context = new JRubyRackContext(mount, gds, configuration, factory);
                contextMap.addEndpoint(mount, context);
                jetty.addHandler(context);
            } catch (IOException e) {
                throw new RuntimeException(e);
            }
        }

        return Arrays.<Injectable<?>>asList(injectable(contextMap), injectable(factory));
    }

    private Server getJetty(final NeoServer neoServer) {
        if (neoServer instanceof NeoServerWithEmbeddedWebServer) {
            final NeoServerWithEmbeddedWebServer server = (NeoServerWithEmbeddedWebServer) neoServer;
            return server.getWebServer().getJetty();
        } else {
            throw new IllegalArgumentException("expected NeoServerWithEmbeddedWebServer");
        }
    }
}