RedPenRunner.java Source code

Introduction

Here is the source code for RedPenRunner.java

Source

/**
 * redpen: a text inspection tool
 * Copyright (c) 2014-2015 Recruit Technologies Co., Ltd. and contributors
 * (see CONTRIBUTORS.md)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

import cc.redpen.RedPen;
import org.apache.commons.cli.*;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.handler.HandlerList;
import org.eclipse.jetty.server.handler.ShutdownHandler;
import org.eclipse.jetty.webapp.WebAppContext;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.URL;
import java.security.ProtectionDomain;
import java.util.Locale;

public class RedPenRunner {

    public static void main(String[] args) throws Exception {
        Options options = new Options();
        options.addOption("h", "help", false, "help");
        options.addOption("v", "version", false, "print the version information and exit");

        OptionBuilder.withLongOpt("port");
        OptionBuilder.withDescription("port number");
        OptionBuilder.hasArg();
        OptionBuilder.withArgName("PORT");
        options.addOption(OptionBuilder.create("p"));

        OptionBuilder.withLongOpt("key");
        OptionBuilder.withDescription("stop key");
        OptionBuilder.hasArg();
        OptionBuilder.withArgName("STOP_KEY");
        options.addOption(OptionBuilder.create("k"));

        OptionBuilder.withLongOpt("conf");
        OptionBuilder.withDescription("configuration file");
        OptionBuilder.hasArg();
        OptionBuilder.withArgName("CONFIG_FILE");
        options.addOption(OptionBuilder.create("c"));

        OptionBuilder.withLongOpt("lang");
        OptionBuilder.withDescription("Language of error messages");
        OptionBuilder.hasArg();
        OptionBuilder.withArgName("LANGUAGE");
        options.addOption(OptionBuilder.create("L"));

        CommandLineParser parser = new BasicParser();
        CommandLine commandLine = null;

        try {
            commandLine = parser.parse(options, args);
        } catch (Exception e) {
            printHelp(options);
            System.exit(-1);
        }

        int portNum = 8080;
        String language = "en";
        if (commandLine.hasOption("h")) {
            printHelp(options);
            System.exit(0);
        }
        if (commandLine.hasOption("v")) {
            System.out.println(RedPen.VERSION);
            System.exit(0);
        }
        if (commandLine.hasOption("p")) {
            portNum = Integer.parseInt(commandLine.getOptionValue("p"));
        }
        if (commandLine.hasOption("L")) {
            language = commandLine.getOptionValue("L");
        }
        if (isPortTaken(portNum)) {
            System.err.println("port is taken...");
            System.exit(1);
        }

        // set language
        if (language.equals("ja")) {
            Locale.setDefault(new Locale("ja", "JA"));
        } else {
            Locale.setDefault(new Locale("en", "EN"));
        }

        final String contextPath = System.getProperty("redpen.home", "/");

        Server server = new Server(portNum);
        ProtectionDomain domain = RedPenRunner.class.getProtectionDomain();
        URL location = domain.getCodeSource().getLocation();

        HandlerList handlerList = new HandlerList();
        if (commandLine.hasOption("key")) {
            // Add Shutdown handler only when STOP_KEY is specified
            ShutdownHandler shutdownHandler = new ShutdownHandler(commandLine.getOptionValue("key"));
            handlerList.addHandler(shutdownHandler);
        }

        WebAppContext webapp = new WebAppContext();
        webapp.setContextPath(contextPath);
        if (location.toExternalForm().endsWith("redpen-server/target/classes/")) {
            // use redpen-server/target/redpen-server instead, because target/classes doesn't contain web resources.
            webapp.setWar(location.toExternalForm() + "../redpen-server/");
        } else {
            webapp.setWar(location.toExternalForm());
        }
        if (commandLine.hasOption("c")) {
            webapp.setInitParameter("redpen.conf.path", commandLine.getOptionValue("c"));
        }

        handlerList.addHandler(webapp);
        server.setHandler(handlerList);

        server.start();
        server.join();
    }

    private static void printHelp(Options opt) {
        HelpFormatter formatter = new HelpFormatter();
        formatter.printHelp("redpen-server", opt);
    }

    private static boolean isPortTaken(int portNum) {
        boolean portTaken = false;
        try (ServerSocket socket = new ServerSocket(portNum)) {
            // do nothing
        } catch (IOException e) {
            System.err.println("Detect: port is taken"); // TODO: use logger
            portTaken = true;
        }
        return portTaken;
    }
}