summaryrefslogtreecommitdiff
path: root/BJC-Utils2/src/main/java/bjc/utils/ioutils/RuleBasedReaderPragmas.java
blob: e26a7eec8a47a6fec83811ffd394c13a72f10e22 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
package bjc.utils.ioutils;

import java.util.function.BiConsumer;

import bjc.utils.exceptions.PragmaFormatException;
import bjc.utils.funcdata.FunctionalStringTokenizer;
import bjc.utils.funcutils.ListUtils;

/**
 * Contains factory methods for common pragma types
 *
 * @author ben
 *
 */
public class RuleBasedReaderPragmas {

	/**
	 * Creates a pragma that takes a single integer argument
	 *
	 * @param <StateType>
	 *                The type of state that goes along with this pragma
	 * @param name
	 *                The name of this pragma, for error message purpose
	 * @param consumer
	 *                The function to invoke with the parsed integer
	 * @return A pragma that functions as described above.
	 */
	public static <StateType> BiConsumer<FunctionalStringTokenizer, StateType> buildInteger(final String name,
			final BiConsumer<Integer, StateType> consumer) {
		return (tokenizer, state) -> {
			/*
			 * Check our input is correct
			 */
			if (!tokenizer.hasMoreTokens()) {
				String fmt = "Pragma %s requires one integer argument";

				throw new PragmaFormatException(String.format(fmt, name));
			}

			/*
			 * Read the argument
			 */
			final String token = tokenizer.nextToken();

			try {
				/*
				 * Run the pragma
				 */
				consumer.accept(Integer.parseInt(token), state);
			} catch (final NumberFormatException nfex) {
				/*
				 * Tell the user their argument isn't correct
				 */
				String fmt = "Argument %s to %s pragma isn't a valid integer, and this pragma requires an integer argument.";

				final PragmaFormatException pfex = new PragmaFormatException(String.format(fmt, token, name));

				pfex.initCause(nfex);

				throw pfex;
			}
		};
	}

	/**
	 * Creates a pragma that takes any number of arguments and collapses
	 * them all into a single string
	 *
	 * @param <StateType>
	 *                The type of state that goes along with this pragma
	 * @param name
	 *                The name of this pragma, for error message purpose
	 * @param consumer
	 *                The function to invoke with the parsed string
	 * @return A pragma that functions as described above.
	 */
	public static <StateType> BiConsumer<FunctionalStringTokenizer, StateType> buildStringCollapser(
			final String name, final BiConsumer<String, StateType> consumer) {
		return (tokenizer, state) -> {
			/*
			 * Check our input
			 */
			if (!tokenizer.hasMoreTokens()) {
				String fmt = "Pragma %s requires one or more string arguments.";

				throw new PragmaFormatException(String.format(fmt, name));
			}

			/*
			 * Build our argument
			 */
			final String collapsed = ListUtils.collapseTokens(tokenizer.toList());

			/*
			 * Run the pragma
			 */
			consumer.accept(collapsed, state);
		};
	}
}