java/sql-dk/src/info/globalcode/sql/dk/CLIParser.java
author František Kučera <franta-hg@frantovo.cz>
Sun, 15 Dec 2013 23:58:58 +0100
branchv_0
changeset 9 2ec52027b97f
parent 8 4507cb9a0cf1
child 10 f528406f33f4
permissions -rw-r--r--
better exceptions
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     1
package info.globalcode.sql.dk;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
     2
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     3
import java.sql.Types;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     4
import java.util.ArrayList;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     5
import java.util.Collections;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     6
import java.util.HashMap;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     7
import java.util.List;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     8
import java.util.Map;
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
     9
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    10
/**
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    11
 *
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    12
 * @author Ing. František Kučera (frantovo.cz)
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    13
 */
5
26223eb63851 more OOP
František Kučera <franta-hg@frantovo.cz>
parents: 4
diff changeset
    14
public class CLIParser {
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    15
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    16
	public static final String TYPE_NAME_SEPARATOR = ":";
5
26223eb63851 more OOP
František Kučera <franta-hg@frantovo.cz>
parents: 4
diff changeset
    17
	private final Map<String, Integer> types;
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    18
5
26223eb63851 more OOP
František Kučera <franta-hg@frantovo.cz>
parents: 4
diff changeset
    19
	public CLIParser() {
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    20
		Map<String, Integer> m = new HashMap<>();
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    21
		m.put("int", Types.INTEGER);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    22
		m.put("string", Types.VARCHAR);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    23
		m.put("boolean", Types.BOOLEAN);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    24
		/**
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    25
		 * TODO: more types
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    26
		 */
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    27
		types = Collections.unmodifiableMap(m);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    28
	}
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    29
9
2ec52027b97f better exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 8
diff changeset
    30
	public CLIOptions parseOptions(String[] args) throws CLIParserException {
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    31
		CLIOptions options = new CLIOptions();
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    32
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    33
		List<Integer> numberedTypes = new ArrayList<>();
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    34
		Map<String, Integer> namedTypes = new HashMap<>();
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    35
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    36
		for (int i = 0; i < args.length; i++) {
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    37
			String arg = args[i];
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    38
			switch (arg) {
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    39
				case Tokens.TYPES:
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    40
					String typesString = fetchNext(args, ++i);
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    41
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    42
					for (String oneType : typesString.split("\\s*,\\s*")) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    43
						int sepatratorIndex = oneType.indexOf(TYPE_NAME_SEPARATOR);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    44
						if (sepatratorIndex == -1) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    45
							numberedTypes.add(getType(oneType));
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    46
						} else {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    47
							String namePart = oneType.substring(0, sepatratorIndex);
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    48
							String typePart = oneType.substring(sepatratorIndex + TYPE_NAME_SEPARATOR.length(), oneType.length());
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    49
							namedTypes.put(namePart, getType(typePart));
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    50
						}
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    51
					}
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    52
					break;
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    53
				case Tokens.NAME_PREFIX:
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    54
					options.setNamePrefix(fetchNext(args, ++i));
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    55
					break;
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    56
				case Tokens.DB:
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    57
					options.setDatabaseName(fetchNext(args, ++i));
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    58
					break;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    59
				case Tokens.SQL:
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    60
					options.setSql(fetchNext(args, ++i));
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    61
					options.setCommandType(CLIOptions.COMMAND_TYPE.QUERY);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    62
					break;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    63
				case Tokens.SQL_UPDATE:
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    64
				case Tokens.SQL_INSERT:
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    65
					options.setSql(fetchNext(args, ++i));
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    66
					options.setCommandType(CLIOptions.COMMAND_TYPE.UPDATE);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    67
					break;
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    68
				case Tokens.BATCH:
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
    69
					options.setBatch(true);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    70
					break;
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    71
				case Tokens.DATA: // --data is the last option
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    72
					for (i++; i < args.length; i++) {
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    73
						arg = args[i];
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    74
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    75
						if (arg.startsWith(options.getNamePrefix())) { // Named parameters:
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    76
							String paramName = arg.substring(options.getNamePrefix().length());
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    77
							String paramValue = fetchNext(args, ++i);
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    78
							options.addNamedParameter(new NamedParameter(paramName, paramValue, namedTypes.get(paramName)));
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    79
						} else { // Numbered parameters:
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    80
							Parameter parameter;
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    81
							if (numberedTypes.isEmpty()) {
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    82
								parameter = new Parameter(arg, null);
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    83
							} else {
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    84
								int paramIndex = options.getNumberedParameters().size();
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    85
								int paramType;
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    86
								try {
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    87
									paramType = numberedTypes.get(paramIndex);
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    88
								} catch (IndexOutOfBoundsException e) {
9
2ec52027b97f better exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 8
diff changeset
    89
									throw new CLIParserException("Missing type for parameter #" + paramIndex, e);
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    90
								} catch (NullPointerException e) {
9
2ec52027b97f better exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 8
diff changeset
    91
									throw new CLIParserException("Invalid type definition for parameter #" + paramIndex, e);
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    92
								}
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    93
								parameter = new Parameter(arg, paramType);
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    94
							}
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
    95
							options.addNumberedParameter(parameter);
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    96
						}
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    97
					}
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
    98
					break;
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
    99
				default:
9
2ec52027b97f better exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 8
diff changeset
   100
					throw new CLIParserException("Unknown option: " + arg);
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   101
			}
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   102
		}
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   103
		return options;
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   104
	}
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   105
9
2ec52027b97f better exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 8
diff changeset
   106
	private String fetchNext(String[] args, int index) throws CLIParserException {
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   107
		if (index < args.length) {
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   108
			return args[index];
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   109
		} else {
9
2ec52027b97f better exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 8
diff changeset
   110
			throw new CLIParserException("Expecting value for option: " + args[index - 1]);
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   111
		}
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   112
	}
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   113
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   114
	public static class Tokens {
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   115
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   116
		public static final String DB = "--db";
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   117
		public static final String SQL = "--sql";
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   118
		public static final String SQL_UPDATE = "--sql-update";
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   119
		public static final String SQL_INSERT = "--sql-insert";
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   120
		public static final String BATCH = "--batch";
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   121
		public static final String DATA = "--data";
2
72da10f632b5 more code
František Kučera <franta-hg@frantovo.cz>
parents: 1
diff changeset
   122
		public static final String NAME_PREFIX = "--name-prefix";
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   123
		public static final String TYPES = "--types";
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   124
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   125
		private Tokens() {
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   126
		}
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   127
	}
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   128
9
2ec52027b97f better exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 8
diff changeset
   129
	private int getType(String typeString) throws CLIParserException {
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   130
		Integer type = types.get(typeString);
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   131
		if (type == null) {
9
2ec52027b97f better exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 8
diff changeset
   132
			throw new CLIParserException("Unsupported type: " + typeString);
8
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   133
		} else {
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   134
			return type;
4507cb9a0cf1 more exceptions
František Kučera <franta-hg@frantovo.cz>
parents: 5
diff changeset
   135
		}
4
f5c3350f3d78 data/types CLI options parsing
František Kučera <franta-hg@frantovo.cz>
parents: 3
diff changeset
   136
	}
1
f32dac78d13a WOW some classes LOL; TODO: refactor
František Kučera <franta-hg@frantovo.cz>
parents:
diff changeset
   137
}