org.apache.oodt.commons.option.parser.CmdLineOptionParser Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of oodt-commons Show documentation
Apache OODT Common Utilities Project
There is a newer version: 1.9.1
/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You 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.apache.oodt.commons.option.parser;

//JDK imports
import java.io.IOException;
import java.io.PrintStream;
import java.util.LinkedList;
import java.util.List;

//OODT imports
import org.apache.oodt.commons.option.CmdLineOption;
import org.apache.oodt.commons.option.CmdLineOptionInstance;
import org.apache.oodt.commons.option.OptionHelpException;
import org.apache.oodt.commons.option.util.CmdLineOptionUtils;
import org.apache.oodt.commons.option.util.CmdLineOptionsUsagePrinter;

/**
 * 
 * @author bfoster
 * @version $Revision$
 *
 * Describe your class here.
 */
public class CmdLineOptionParser {

    private List validOptions;

    public List getValidOptions() {
        return validOptions;
    }

    public void setValidOptions(List validOptions) {
        this.validOptions = validOptions;
    }

    public List parse(String[] args) throws IOException {
        List optionInstances = new LinkedList();
        if (args.length < 1)
            throw new OptionHelpException(
                    "Must specify options : type -h or --help for info");

        for (int j = 0; j < args.length; j++) {
            String curArg = args[j];

            // print usage and quit if -h or --help is given
            if (curArg.equals("-h") || curArg.equals("--help")) {
            	if (args.length > j+1) {
            		String[] helpArgs = new String[args.length - j - 1];
            		System.arraycopy(args, j + 1, helpArgs, 0, helpArgs.length);
            		CmdLineOptionsUsagePrinter.printUsage(new PrintStream(
                            System.out), this.validOptions, this.parse(helpArgs).get(0));
            	}else { 
            		CmdLineOptionsUsagePrinter.printUsage(new PrintStream(
                        System.out), this.validOptions);
            	}
                System.exit(0);
            }


            if (curArg.startsWith("-")) {
                // check if long or short version was used
                if (curArg.startsWith("--"))
                    curArg = curArg.substring(2);
                else
                    curArg = curArg.substring(1);

                // check if option is a valid one
                CmdLineOption curOption = CmdLineOptionUtils.getOptionByName(
                        curArg, validOptions);
                if (curOption == null)
                    throw new IOException("Invalid option '" + curArg + "'");

                // check if option has arguments
                List values = new LinkedList();
                if (curOption.hasArgs()) {
                    while (j + 1 < args.length && !args[j + 1].startsWith("-"))
                        values.add(args[++j]);
                    if (values.size() < 1)
                        throw new IOException("Option " + curArg
                                + " should have at least one argument");
                }

                //check if is a perform and quit option
                if (curOption.isPerformAndQuit()) {
                	curOption.getHandler().handleOption(curOption, values);
                	System.exit(0);
                }
                
                // add to list of option instances
                optionInstances
                        .add(new CmdLineOptionInstance(curOption, values));
                
            } else {
                throw new IOException("Argument with no option flag '" + curArg
                        + "'");
            }
        }
        return optionInstances;
    }

}