Mercurialsql-dk / file revision
summary | shortlog | changelog | graph | tags | bookmarks | branches | files | changeset | file | latest | revisions | annotate | diff | raw | help
java/sql-dk/src/info/globalcode/sql/dk/CLIParser.java
author František Kučera <franta-hg@frantovo.cz>
Mon, 16 Dec 2013 15:15:32 +0100
branchv_0
changeset 14 189b1260b942
parent 12 2dcb67d90fd2
child 15 bbd335b5410c
permissions -rw-r--r--
show info basics 
     1 package info.globalcode.sql.dk;

     2 

     3 import java.sql.Types;

     4 import java.util.ArrayList;

     5 import java.util.Collections;

     6 import java.util.HashMap;

     7 import java.util.List;

     8 import java.util.Map;

     9 

    10 /**

    11  *

    12  * @author Ing. František Kučera (frantovo.cz)

    13  */

    14 public class CLIParser {

    15 

    16 	public static final String TYPE_NAME_SEPARATOR = ":";

    17 	private final Map<String, Integer> types;

    18 

    19 	public CLIParser() {

    20 		Map<String, Integer> m = new HashMap<>();

    21 		m.put("int", Types.INTEGER);

    22 		m.put("string", Types.VARCHAR);

    23 		m.put("boolean", Types.BOOLEAN);

    24 		/**

    25 		 * TODO: more types

    26 		 */

    27 		types = Collections.unmodifiableMap(m);

    28 	}

    29 

    30 	public CLIOptions parseOptions(String[] args) throws CLIParserException {

    31 		CLIOptions options = new CLIOptions();

    32 

    33 		List<Integer> numberedTypes = new ArrayList<>();

    34 		Map<String, Integer> namedTypes = new HashMap<>();

    35 

    36 		for (int i = 0; i < args.length; i++) {

    37 			String arg = args[i];

    38 			switch (arg) {

    39 				case Tokens.TYPES:

    40 					String typesString = fetchNext(args, ++i);

    41 

    42 					for (String oneType : typesString.split(",")) {

    43 						int sepatratorIndex = oneType.indexOf(TYPE_NAME_SEPARATOR);

    44 						if (sepatratorIndex == -1) {

    45 							numberedTypes.add(getType(oneType));

    46 						} else {

    47 							String namePart = oneType.substring(0, sepatratorIndex).trim();

    48 							String typePart = oneType.substring(sepatratorIndex + TYPE_NAME_SEPARATOR.length(), oneType.length());

    49 							namedTypes.put(namePart, getType(typePart));

    50 						}

    51 					}

    52 					break;

    53 				case Tokens.NAME_PREFIX:

    54 					options.setNamePrefix(fetchNext(args, ++i));

    55 					break;

    56 				case Tokens.DB:

    57 					options.setDatabaseName(fetchNext(args, ++i));

    58 					break;

    59 				case Tokens.SQL:

    60 					options.setSql(fetchNext(args, ++i));

    61 					options.setCommandType(CLIOptions.COMMAND_TYPE.QUERY);

    62 					break;

    63 				case Tokens.SQL_UPDATE:

    64 				case Tokens.SQL_INSERT:

    65 					options.setSql(fetchNext(args, ++i));

    66 					options.setCommandType(CLIOptions.COMMAND_TYPE.UPDATE);

    67 					break;

    68 				case Tokens.BATCH:

    69 					options.setBatch(true);

    70 					break;

    71 				case Tokens.DATA: // --data is the last option

    72 					for (i++; i < args.length; i++) {

    73 						arg = args[i];

    74 

    75 						if (arg.startsWith(options.getNamePrefix())) { // Named parameters:

    76 							String paramName = arg.substring(options.getNamePrefix().length());

    77 							String paramValue = fetchNext(args, ++i);

    78 							options.addNamedParameter(new NamedParameter(paramName, paramValue, namedTypes.get(paramName)));

    79 						} else { // Numbered parameters:

    80 							Parameter parameter;

    81 							if (numberedTypes.isEmpty()) {

    82 								parameter = new Parameter(arg, null);

    83 							} else {

    84 								int paramIndex = options.getNumberedParameters().size();

    85 								int paramType;

    86 								try {

    87 									paramType = numberedTypes.get(paramIndex);

    88 								} catch (IndexOutOfBoundsException e) {

    89 									throw new CLIParserException("Missing type for parameter #" + paramIndex, e);

    90 								} catch (NullPointerException e) {

    91 									throw new CLIParserException("Invalid type definition for parameter #" + paramIndex, e);

    92 								}

    93 								parameter = new Parameter(arg, paramType);

    94 							}

    95 							options.addNumberedParameter(parameter);

    96 						}

    97 					}

    98 					break;

    99 				case Tokens.FORMATTER:

   100 					options.setFormatterName(fetchNext(args, ++i));

   101 					break;

   102 				case Tokens.INFO_HELP:

   103 					options.addShowInfo(CLIOptions.INFO_TYPE.HELP);

   104 					break;

   105 				case Tokens.INFO_FORMATTERS:

   106 					options.addShowInfo(CLIOptions.INFO_TYPE.FORMATTERS);

   107 					break;

   108 				case Tokens.INFO_LICENSE:

   109 					options.addShowInfo(CLIOptions.INFO_TYPE.LICENSE);

   110 					break;

   111 				case Tokens.INFO_TYPES:

   112 					options.addShowInfo(CLIOptions.INFO_TYPE.TYPES);

   113 					break;

   114 				case Tokens.INFO_VERSION:

   115 					options.addShowInfo(CLIOptions.INFO_TYPE.VERSION);

   116 					break;

   117 				default:

   118 					throw new CLIParserException("Unknown option: " + arg);

   119 			}

   120 		}

   121 		return options;

   122 	}

   123 

   124 	private String fetchNext(String[] args, int index) throws CLIParserException {

   125 		if (index < args.length) {

   126 			return args[index];

   127 		} else {

   128 			throw new CLIParserException("Expecting value for option: " + args[index - 1]);

   129 		}

   130 	}

   131 

   132 	public static class Tokens {

   133 

   134 		public static final String DB = "--db";

   135 		public static final String SQL = "--sql";

   136 		public static final String SQL_UPDATE = "--sql-update";

   137 		public static final String SQL_INSERT = "--sql-insert";

   138 		public static final String BATCH = "--batch";

   139 		public static final String DATA = "--data";

   140 		public static final String NAME_PREFIX = "--name-prefix";

   141 		public static final String TYPES = "--types";

   142 		public static final String FORMATTER = "--formatter";

   143 		public static final String INFO_HELP = "--help";

   144 		public static final String INFO_VERSION = "--version";

   145 		public static final String INFO_LICENSE = "--license";

   146 		public static final String INFO_FORMATTERS = "--list-formatters";

   147 		public static final String INFO_TYPES = "--list-types";

   148 

   149 		private Tokens() {

   150 		}

   151 	}

   152 

   153 	private int getType(String typeString) throws CLIParserException {

   154 		Integer type = types.get(typeString.trim());

   155 		if (type == null) {

   156 			throw new CLIParserException("Unsupported type: " + typeString);

   157 		} else {

   158 			return type;

   159 		}

   160 	}

   161 }
sql-dk