nl.mikero.turntopassage.commandline.Application.java Source code

Introduction

Here is the source code for nl.mikero.turntopassage.commandline.Application.java

Source

package nl.mikero.turntopassage.commandline;

import com.google.inject.Guice;
import com.google.inject.Inject;
import com.google.inject.Injector;
import nl.mikero.turntopassage.commandline.inject.TwineModule;
import nl.mikero.turntopassage.core.service.TwineService;
import org.apache.commons.cli.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.*;

/**
 * Provides a commandline interface to the TurnToPassage.Transformer transformation features.
 * <p>
 * Basic commandline usage:
 * </p>
 * <pre>
 * {@code
 *  usage: application.jar [-help | -version] | [[-f <file>] [-o <file>]]
 *  -f,--file <file>     location of input HTML file
 *  -help                display this help and exit
 *  -o,--output <file>   location of output EPUB file
 *  -version             output version information and exit
 * }
 * </pre>
 *
 * @author Mike Rombout
 */
public class Application {

    private static final Logger LOGGER = LoggerFactory.getLogger(Application.class);

    private static final String OPT_FILE = "file";
    private static final String OPT_OUTPUT = "output";
    private static final String OPT_HELP = "help";
    private static final String OPT_VERSION = "version";

    private final TwineService twineService;

    @Inject
    public Application(TwineService twineService) {
        this.twineService = twineService;
    }

    private void execute(String[] args) {
        // Options
        final Options options = new Options();

        Option input = OptionBuilder.hasArg().withArgName("file").withDescription("location of input HTML file")
                .withLongOpt(OPT_FILE).create('f');

        Option output = OptionBuilder.hasArg().withArgName("file").withDescription("location of output EPUB file")
                .withLongOpt(OPT_OUTPUT).create('o');
        options.addOption(input);
        options.addOption(output);

        OptionGroup infoGroup = new OptionGroup();
        infoGroup.addOption(new Option(OPT_HELP, "display this help and exit"));
        infoGroup.addOption(new Option(OPT_VERSION, "output version information and exit"));
        options.addOptionGroup(infoGroup);

        // Parse
        CommandLineParser parser = new BasicParser();
        try {
            CommandLine cmd = parser.parse(options, args);

            if (cmd.hasOption(OPT_VERSION)) {
                System.out.println("TurnToPassage.Transformer 0.0.1");
                System.out.println("Copyright (C) 2015 Mike Rombout");
                System.out.println("License GPLv3+: GNU GPL version 3 or later <http://gnu.org/licenses/gpl.html>");
                System.out.println("This is free software: you are free to change and redistribute it.");
                System.out.println("There is NO WARRANTY, to the extend permitted by law.");
                System.exit(0);
            } else if (cmd.hasOption(OPT_HELP)) {
                HelpFormatter formatter = new HelpFormatter();
                formatter.printHelp("turntopassage", options, true);
                System.exit(0);
            } else {
                InputStream inputStream = System.in;
                OutputStream outputStream = System.out;

                if (cmd.hasOption(OPT_FILE)) {
                    String fileArg = cmd.getOptionValue(OPT_FILE);
                    try {
                        inputStream = new BufferedInputStream(new FileInputStream(new File(fileArg)));
                    } catch (FileNotFoundException e) {
                        LOGGER.error("Input file {} could not be found.", fileArg, e);
                        System.exit(1);
                    }
                }
                if (cmd.hasOption(OPT_OUTPUT)) {
                    String outputArg = cmd.getOptionValue(OPT_OUTPUT);
                    try {
                        outputStream = new BufferedOutputStream(new FileOutputStream(new File(outputArg)));
                    } catch (FileNotFoundException e) {
                        LOGGER.error("Output file {} could not be found.", outputArg, e);
                        System.exit(1);
                    }
                }

                twineService.transform(inputStream, outputStream);
                System.exit(0);
            }
        } catch (ParseException e) {
            LOGGER.error("Error: {}", e.getMessage(), e);
        }

        System.exit(1);
    }

    public static void main(String[] args) {
        Injector injector = Guice.createInjector(new TwineModule());

        Application application = injector.getInstance(Application.class);
        application.execute(args);
    }

}