diff options
Diffstat (limited to 'src/net/sourceforge/plantuml/preproc/Preprocessor.java')
| -rw-r--r-- | src/net/sourceforge/plantuml/preproc/Preprocessor.java | 161 |
1 files changed, 161 insertions, 0 deletions
diff --git a/src/net/sourceforge/plantuml/preproc/Preprocessor.java b/src/net/sourceforge/plantuml/preproc/Preprocessor.java new file mode 100644 index 0000000..5cc73df --- /dev/null +++ b/src/net/sourceforge/plantuml/preproc/Preprocessor.java @@ -0,0 +1,161 @@ +/* ========================================================================
+ * PlantUML : a free UML diagram generator
+ * ========================================================================
+ *
+ * (C) Copyright 2009-2014, Arnaud Roques
+ *
+ * Project Info: http://plantuml.sourceforge.net
+ *
+ * This file is part of PlantUML.
+ *
+ * Licensed under The MIT License (Massachusetts Institute of Technology License)
+ *
+ * See http://opensource.org/licenses/MIT
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining
+ * a copy of this software and associated documentation files (the "Software"),
+ * to deal in the Software without restriction, including without limitation
+ * the rights to use, copy, modify, merge, publish, distribute, sublicense,
+ * and/or sell copies of the Software, and to permit persons to whom the
+ * Software is furnished to do so, subject to the following conditions:
+ *
+ * The above copyright notice and this permission notice shall be included
+ * in all copies or substantial portions of the Software.
+ *
+ * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
+ * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+ * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+ * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY,
+ * WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR
+ * IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
+ *
+ *
+ * Original Author: Arnaud Roques
+ */
+package net.sourceforge.plantuml.preproc;
+
+import java.io.File;
+import java.io.IOException;
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Set;
+import java.util.regex.Matcher;
+import java.util.regex.Pattern;
+
+import net.sourceforge.plantuml.command.regex.MyPattern;
+import net.sourceforge.plantuml.utils.StartUtils;
+import net.sourceforge.plantuml.StringUtils;
+
+public class Preprocessor implements ReadLine {
+
+ private static final String ID = "[A-Za-z_][A-Za-z_0-9]*";
+ private static final String ARG = "(?:\\(" + ID + "(?:," + ID + ")*?\\))?";
+ private static final Pattern definePattern = MyPattern.cmpile("^[%s]*!define[%s]+(" + ID + ARG + ")"
+ + "(?:[%s]+(.*))?$");
+ private static final Pattern undefPattern = MyPattern.cmpile("^[%s]*!undef[%s]+(" + ID + ")$");
+ private static final Pattern definelongPattern = MyPattern.cmpile("^[%s]*!definelong[%s]+(" + ID + ARG + ")");
+ private static final Pattern enddefinelongPattern = MyPattern.cmpile("^[%s]*!enddefinelong[%s]*$");
+
+ private final Defines defines;
+ private final PreprocessorInclude rawSource;
+ private final ReadLineInsertable source;
+
+ public Preprocessor(ReadLine reader, String charset, Defines defines, Set<File> filesUsed, File newCurrentDir) {
+ this.defines = defines;
+ this.defines.saveState();
+ this.rawSource = new PreprocessorInclude(reader, defines, charset, filesUsed, newCurrentDir);
+ this.source = new ReadLineInsertable(new IfManager(rawSource, defines));
+ }
+
+ public String readLine() throws IOException {
+ final String s = source.readLine();
+ if (s == null) {
+ return null;
+ }
+ if (StartUtils.isArobaseStartDiagram(s)) {
+ this.defines.restoreState();
+ }
+
+ Matcher m = definePattern.matcher(s);
+ if (m.find()) {
+ return manageDefine(m);
+ }
+
+ m = definelongPattern.matcher(s);
+ if (m.find()) {
+ return manageDefineLong(m);
+ }
+
+ m = undefPattern.matcher(s);
+ if (m.find()) {
+ return manageUndef(m);
+ }
+
+ if (ignoreDefineDuringSeveralLines > 0) {
+ ignoreDefineDuringSeveralLines--;
+ return s;
+ }
+
+ final List<String> result = defines.applyDefines(s);
+ if (result.size() > 1) {
+ ignoreDefineDuringSeveralLines = result.size() - 2;
+ source.insert(result.subList(1, result.size() - 1));
+ }
+ return result.get(0);
+ }
+
+ private int ignoreDefineDuringSeveralLines = 0;
+
+ private String manageUndef(Matcher m) throws IOException {
+ defines.undefine(m.group(1));
+ return this.readLine();
+ }
+
+ private String manageDefineLong(Matcher m) throws IOException {
+ final String group1 = m.group(1);
+ final List<String> def = new ArrayList<String>();
+ while (true) {
+ final String read = this.readLine();
+ if (read == null) {
+ return null;
+ }
+ def.add(read);
+ if (enddefinelongPattern.matcher(read).find()) {
+ defines.define(group1, def);
+ return this.readLine();
+ }
+ }
+ }
+
+ private String manageDefine(Matcher m) throws IOException {
+ final String group1 = m.group(1);
+ final String group2 = m.group(2);
+ if (group2 == null) {
+ defines.define(group1, null);
+ } else {
+ final List<String> strings = defines.applyDefines(group2);
+ if (strings.size() > 1) {
+ throw new UnsupportedOperationException();
+ }
+ final StringBuilder value = new StringBuilder(strings.get(0));
+ while (StringUtils.endsWithBackslash(value.toString())) {
+ value.setLength(value.length() - 1);
+ final String read = this.readLine();
+ value.append(read);
+ }
+ final List<String> li = new ArrayList<String>();
+ li.add(value.toString());
+ defines.define(group1, li);
+ }
+ return this.readLine();
+ }
+
+ public int getLineNumber() {
+ return rawSource.getLineNumber();
+ }
+
+ public void close() throws IOException {
+ rawSource.close();
+ }
+
+}
|