com.ctriposs.rest4j.tools.snapshot.gen.Rest4JSnapshotExporterCmdLineApp.java Source code

Introduction

Here is the source code for com.ctriposs.rest4j.tools.snapshot.gen.Rest4JSnapshotExporterCmdLineApp.java

Source

/**
 * Copyright (C) 2014 the original author or authors.
 * See the notice.md file distributed with this work for additional
 * information regarding copyright ownership.
 *
 * 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.
 */
package com.ctriposs.rest4j.tools.snapshot.gen;

import com.ctriposs.data.schema.generator.AbstractGenerator;
import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.CommandLineParser;
import org.apache.commons.cli.GnuParser;
import org.apache.commons.cli.HelpFormatter;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.OptionBuilder;
import org.apache.commons.cli.OptionGroup;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class Rest4JSnapshotExporterCmdLineApp {
    private static final Logger log = LoggerFactory.getLogger(Rest4JSnapshotExporterCmdLineApp.class);

    private static final Options OPTIONS = new Options();
    static {
        OPTIONS.addOption(OptionBuilder.isRequired().withArgName("sourcepath").hasArgs().withDescription(
                "Space-delimited list of directories in which to find resource Java source files\nIf neither -resourcepackages nor -resourcepackages is provided, all classes defined in the directories will be scanned")
                .create("sourcepath"));
        OPTIONS.addOption(
                OptionBuilder.withArgName("name").hasArg().withDescription("Name of the API").create("name"));
        OPTIONS.addOption(OptionBuilder.withArgName("outdir").hasArg()
                .withDescription(
                        "Directory in which to output the generated Snapshot files (default=current working dir)")
                .create("outdir"));

        final OptionGroup sourceGroup = new OptionGroup();
        final Option sourcePkgs = OptionBuilder.withArgName("resourcepackages").hasArgs()
                .withDescription("Space-delimited list of packages to scan for resource classes")
                .create("resourcepackages");
        final Option sourceClasses = OptionBuilder.withArgName("resourceclasses").hasArgs()
                .withDescription("space-delimited list of resource classes to scan").create("resourceclasses");
        sourceGroup.addOption(sourcePkgs);
        sourceGroup.addOption(sourceClasses);
        OPTIONS.addOptionGroup(sourceGroup);
    }

    /**
     * @param args rest4jexporter -sourcepath sourcepath -resourcepackages packagenames [-name api_name] [-outdir outdir]
     */
    public static void main(String[] args) {
        CommandLine cl = null;
        try {
            final CommandLineParser parser = new GnuParser();
            cl = parser.parse(OPTIONS, args);
        } catch (ParseException e) {
            System.err.println("Invalid arguments: " + e.getMessage());
            final HelpFormatter formatter = new HelpFormatter();
            formatter.printHelp(
                    "rest4jexporter -sourcepath sourcepath [-resourcepackages packagenames] [-resourceclasses classnames]"
                            + "[-name api_name] [-outdir outdir]",
                    OPTIONS);
            System.exit(0);
        }

        final String resolverPath = System.getProperty(AbstractGenerator.GENERATOR_RESOLVER_PATH);

        try {
            final Rest4JSnapshotExporter exporter = new Rest4JSnapshotExporter();
            exporter.setResolverPath(resolverPath);
            exporter.export(cl.getOptionValue("name"), null, cl.getOptionValues("sourcepath"),
                    cl.getOptionValues("resourcepackages"), cl.getOptionValues("resourceClasses"),
                    cl.getOptionValue("outdir", "."));
        } catch (Throwable e) {
            log.error("Error writing Snapshot files", e);
            System.out.println("Error writing Snapshot files:\n" + e);
            System.exit(1);
        }
    }
}