src/main/java/bjc/rgens/parser/elements/vars/VariableElement.java


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

package bjc.rgens.parser.elements.vars;

import bjc.utils.funcutils.StringUtils;

import bjc.rgens.parser.GenerationState;
import bjc.rgens.parser.GrammarException;

import java.util.ArrayList;
import java.util.List;

public abstract class VariableElement {
	public boolean forbidSpaces;

	protected VariableElement(boolean forbidSpacing) {
		forbidSpaces = forbidSpacing;
	}

	public abstract void generate(GenerationState state);

	public static List<VariableElement> parseElementString(String varElm) {
		boolean forbidSpaces = StringUtils.levelContains(varElm, "-", "+");

		String[] parts;

		if(forbidSpaces) {
			parts = StringUtils.levelSplit(varElm, true, "-", "+").toArray(new String[0]);
		} else {
			parts = new String[] { varElm };
		}

		return parseElementString(forbidSpaces, parts);
	}

	public static List<VariableElement> parseElementString(boolean forbidSpaces, String... parts) {
		List<VariableElement> elms = new ArrayList<>(parts.length);

		VariableElement prevElement = null;

		for (String part : parts) {
			VariableElement elm = null;

			if(part.startsWith("$")) {
				elm = new VRefVariableElement(forbidSpaces, part.substring(1));
			} else if (part.startsWith("@")) {
				if(forbidSpaces)
					throw new GrammarException("Arrays references aren't allowed in rule names");

				elm = new ARefVariableElement(part.substring(1));
			} else if (part.startsWith("%")) {
				elm = new RRefVariableElement(forbidSpaces, part.substring(1));
			} else if (part.startsWith("/")) {
				throw new GrammarException("Template variables aren't implemented yet");
			} else {
				if(prevElement != null && prevElement instanceof LiteralVariableElement) {
					/* Aggregate chain literals together */
					((LiteralVariableElement)prevElement).val += part;
				} else {
					elm = new LiteralVariableElement(forbidSpaces, part);
				}
			}

			if(elm != null) {
				elms.add(elm);

				prevElement = elm;
			}
		}

		return elms;
	}
}