view cmdline/org/tmatesoft/hg/console/Options.java @ 394:f52ca9530774 v0.8.0

Resolve FIXMEs: more consistent exceptions
author Artem Tikhomirov <tikhomirov.artem@gmail.com>
date Wed, 22 Feb 2012 03:10:55 +0100
parents 45dc79e545f5
children
line wrap: on
line source
/*
 * Copyright (c) 2011 TMate Software Ltd
 *  
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; version 2 of the License.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * For information on how to redistribute this software under
 * the terms of a license other than GNU General Public License
 * contact TMate Software at support@hg4j.com
 */
package org.tmatesoft.hg.console;

import java.util.Collections;
import java.util.HashMap;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;

import org.tmatesoft.hg.repo.HgLookup;
import org.tmatesoft.hg.repo.HgRepository;

/**
 * Parse command-line options. Primitive implementation that recognizes options with 0 or 1 argument.
 * 
 * @author Artem Tikhomirov
 * @author TMate Software Ltd.
 */
class Options {

	public final Map<String,List<String>> opt2values = new HashMap<String, List<String>>();

	public boolean getBoolean(String... aliases) {
		return getBoolean(false, aliases);
	}

	public boolean getBoolean(boolean def, String... aliases) {
		for (String s : aliases) {
			if (opt2values.containsKey(s)) {
				return true;
			}
		}
		return def;
	}

	public String getSingle(String... aliases) {
		String rv = null;
		for (String s : aliases) {
			List<String> values = opt2values.get(s);
			if (values != null && values.size() > 0) {
				rv = values.get(values.size() - 1); // take last one, most recent
			}
		}
		return rv;
	}
	
	public int getSingleInt(int def, String... aliases) {
		String r = getSingle(aliases);
		if (r == null) {
			return def;
		}
		return Integer.parseInt(r);
	}

	public List<String> getList(String... aliases) {
		LinkedList<String> rv = new LinkedList<String>();
		for (String s : aliases) {
			List<String> values = opt2values.get(s);
			if (values != null) {
				rv.addAll(values);
			}
		}
		return rv;
	}
	
	public HgRepository findRepository() throws Exception {
		String repoLocation = getSingle("-R", "--repository");
		if (repoLocation != null) {
			return new HgLookup().detect(repoLocation);
		}
		return new HgLookup().detectFromWorkingDir();
	}


	public static Options parse(String[] commandLineArgs, Set<String> flagOptions) {
		Options rv = new Options();
		List<String> values = new LinkedList<String>();
		rv.opt2values.put("", values); // values with no options
		for (String arg : commandLineArgs) {
			if (arg.charAt(0) == '-') {
				// option
				if (arg.length() == 1) {
					throw new IllegalArgumentException("Bad option: -");
				}
				if (flagOptions.contains(arg)) {
					rv.opt2values.put(arg, Collections.<String>emptyList());
					values = rv.opt2values.get("");
				} else {
					values = rv.opt2values.get(arg);
					if (values == null) {
						rv.opt2values.put(arg, values = new LinkedList<String>());
					}
				}
				// next value, if any, gets into the values list for arg option.
			} else {
				values.add(arg);
				values = rv.opt2values.get("");
			}
		}
		return rv;
	}

	public static Set<String> asSet(String... ss) {
		TreeSet<String> rv = new TreeSet<String>();
		for (String s : ss) {
			rv.add(s);
		}
		return rv;
	}
}