org.callimachusproject.cli.CommandSet.java Source code

Introduction

Here is the source code for org.callimachusproject.cli.CommandSet.java

Source

/*
 * Copyright (c) 2013 3 Round Stones Inc., Some Rights Reserved
 *
 * 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 org.callimachusproject.cli;

import org.apache.commons.cli.CommandLine;
import org.apache.commons.cli.GnuParser;
import org.apache.commons.cli.Option;
import org.apache.commons.cli.Options;
import org.apache.commons.cli.ParseException;

public class CommandSet {
    private final String name;
    private final Options options = new Options();
    private String otherArgName;
    private CommandLine line;

    public CommandSet(String version) {
        this.name = version;
    }

    public CommandOption require(String opt) {
        Option option = new Option(opt, opt);
        option.setRequired(true);
        return new CommandOption(option);
    }

    public CommandOption require(String opt, String longOpt) {
        Option option = new Option(opt, longOpt);
        if (!opt.equals(longOpt)) {
            option.setLongOpt(longOpt);
        }
        option.setRequired(true);
        return new CommandOption(option);
    }

    public CommandOption option(String opt) {
        Option option = new Option(opt, opt);
        return new CommandOption(option);
    }

    public CommandOption option(String opt, String longOpt) {
        Option option = new Option(opt, longOpt);
        if (!opt.equals(longOpt)) {
            option.setLongOpt(longOpt);
        }
        return new CommandOption(option);
    }

    public void other(String argName) {
        this.otherArgName = argName;
    }

    public synchronized Command parse(String[] args) {
        try {
            line = new GnuParser() {
                public CommandLine parse(Options options, String[] arguments) throws ParseException {
                    try {
                        return super.parse(options, arguments);
                    } catch (ParseException e) {
                        line = this.cmd;
                        throw e;
                    }
                };
            }.parse(options, args);
            return new Command(name, options, otherArgName, line);
        } catch (ParseException e) {
            return new Command(name, options, otherArgName, line, e);
        } finally {
            line = null;
        }
    }

    public class CommandOption {
        private final Option option;

        protected CommandOption(Option option) {
            this.option = option;
        }

        public CommandOption arg(String argName) {
            option.setArgName(argName);
            option.setArgs(1);
            option.setOptionalArg(false);
            return this;
        }

        public CommandOption optional(String argName) {
            option.setArgName(argName);
            option.setArgs(1);
            option.setOptionalArg(true);
            return this;
        }

        public void desc(String description) {
            option.setDescription(description);
            options.addOption(option);
        }
    }
}