mirror of https://github.com/octoleo/plantuml.git
201 lines
6.2 KiB
Java
201 lines
6.2 KiB
Java
/* ========================================================================
|
|
* PlantUML : a free UML diagram generator
|
|
* ========================================================================
|
|
*
|
|
* (C) Copyright 2009-2017, Arnaud Roques
|
|
*
|
|
* Project Info: http://plantuml.com
|
|
*
|
|
* If you like this project or if you find it useful, you can support us at:
|
|
*
|
|
* http://plantuml.com/patreon (only 1$ per month!)
|
|
* http://plantuml.com/paypal
|
|
*
|
|
* This file is part of PlantUML.
|
|
*
|
|
* PlantUML is free software; you can redistribute it and/or modify it
|
|
* under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation, either version 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* PlantUML distributed in the hope that it will be useful, but
|
|
* WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
|
|
* or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public
|
|
* License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301,
|
|
* USA.
|
|
*
|
|
*
|
|
* Original Author: Arnaud Roques
|
|
*
|
|
*
|
|
*/
|
|
package net.sourceforge.plantuml.preproc;
|
|
|
|
import java.io.IOException;
|
|
import java.util.ArrayList;
|
|
import java.util.List;
|
|
|
|
import net.sourceforge.plantuml.CharSequence2;
|
|
import net.sourceforge.plantuml.CharSequence2Impl;
|
|
import net.sourceforge.plantuml.StringUtils;
|
|
import net.sourceforge.plantuml.command.regex.Matcher2;
|
|
import net.sourceforge.plantuml.command.regex.MyPattern;
|
|
import net.sourceforge.plantuml.command.regex.Pattern2;
|
|
import net.sourceforge.plantuml.utils.StartUtils;
|
|
|
|
public class PreprocessorDefine extends ReadLineInstrumented implements ReadLine {
|
|
|
|
private static final String END_DEFINE_LONG = "!enddefinelong";
|
|
private static final String ID = "[A-Za-z_][A-Za-z_0-9]*";
|
|
private static final String ID_ARG = "\\s*[A-Za-z_][A-Za-z_0-9]*\\s*(?:=\\s*(?:\"[^\"]*\"|'[^']*')\\s*)?";
|
|
private static final String ARG = "(?:\\(" + ID_ARG + "(?:," + ID_ARG + ")*?\\))?";
|
|
private static final Pattern2 definePattern = MyPattern.cmpile("^[%s]*!define[%s]+(" + ID + ARG + ")"
|
|
+ "(?:[%s]+(.*))?$");
|
|
private static final Pattern2 filenamePattern = MyPattern.cmpile("^[%s]*!filename[%s]+(.+)$");
|
|
private static final Pattern2 undefPattern = MyPattern.cmpile("^[%s]*!undef[%s]+(" + ID + ")$");
|
|
private static final Pattern2 definelongPattern = MyPattern.cmpile("^[%s]*!definelong[%s]+(" + ID + ARG + ")");
|
|
private static final Pattern2 enddefinelongPattern = MyPattern.cmpile("^[%s]*" + END_DEFINE_LONG + "[%s]*$");
|
|
|
|
private final Defines defines;
|
|
|
|
private final ReadLineInsertable source;
|
|
|
|
public PreprocessorDefine(Defines defines, ReadLineInsertable source) {
|
|
this.defines = defines;
|
|
this.defines.saveState();
|
|
this.source = source;
|
|
}
|
|
|
|
@Override
|
|
CharSequence2 readLineInst() throws IOException {
|
|
final CharSequence2 s = source.readLine();
|
|
if (s == null) {
|
|
return null;
|
|
}
|
|
if (StartUtils.isArobaseStartDiagram(s)) {
|
|
this.defines.restoreState();
|
|
}
|
|
|
|
Matcher2 m = filenamePattern.matcher(s);
|
|
if (m.find()) {
|
|
return manageFilename(m);
|
|
}
|
|
m = definePattern.matcher(s);
|
|
if (m.find()) {
|
|
return manageDefine(m, s.toString().trim().endsWith("()"));
|
|
}
|
|
|
|
m = definelongPattern.matcher(s);
|
|
if (m.find()) {
|
|
return manageDefineLong(m, s.toString().trim().endsWith("()"));
|
|
}
|
|
|
|
m = undefPattern.matcher(s);
|
|
if (m.find()) {
|
|
return manageUndef(m);
|
|
}
|
|
|
|
if (ignoreDefineDuringSeveralLines > 0) {
|
|
ignoreDefineDuringSeveralLines--;
|
|
return s;
|
|
}
|
|
|
|
List<String> result = defines.applyDefines(s.toString2());
|
|
if (result.size() > 1) {
|
|
result = cleanEndDefineLong(result);
|
|
final List<String> inserted = cleanEndDefineLong(result.subList(1, result.size()));
|
|
ignoreDefineDuringSeveralLines = inserted.size();
|
|
source.insert(inserted, s.getLocation());
|
|
}
|
|
return new CharSequence2Impl(result.get(0), s.getLocation(), s.getPreprocessorError());
|
|
}
|
|
|
|
private List<String> cleanEndDefineLong(List<String> data) {
|
|
final List<String> result = new ArrayList<String>();
|
|
for (String s : data) {
|
|
final String clean = cleanEndDefineLong(s);
|
|
if (clean != null) {
|
|
result.add(clean);
|
|
}
|
|
}
|
|
return result;
|
|
|
|
}
|
|
|
|
private String cleanEndDefineLong(String s) {
|
|
if (s.trim().startsWith(END_DEFINE_LONG)) {
|
|
s = s.trim().substring(END_DEFINE_LONG.length());
|
|
if (s.length() == 0) {
|
|
return null;
|
|
}
|
|
}
|
|
return s;
|
|
}
|
|
|
|
private int ignoreDefineDuringSeveralLines = 0;
|
|
|
|
private CharSequence2 manageUndef(Matcher2 m) throws IOException {
|
|
defines.undefine(m.group(1));
|
|
return this.readLine();
|
|
}
|
|
|
|
private CharSequence2 manageDefineLong(Matcher2 m, boolean emptyParentheses) throws IOException {
|
|
final String group1 = m.group(1);
|
|
final List<String> def = new ArrayList<String>();
|
|
while (true) {
|
|
final CharSequence2 read = this.readLine();
|
|
if (read == null) {
|
|
return null;
|
|
}
|
|
if (enddefinelongPattern.matcher(read).find()) {
|
|
defines.define(group1, def, emptyParentheses);
|
|
return this.readLine();
|
|
}
|
|
def.add(read.toString2());
|
|
}
|
|
}
|
|
|
|
private CharSequence2 manageFilename(Matcher2 m) throws IOException {
|
|
final String group1 = m.group(1);
|
|
this.defines.overrideFilename(group1);
|
|
return this.readLine();
|
|
}
|
|
|
|
private CharSequence2 manageDefine(Matcher2 m, boolean emptyParentheses) throws IOException {
|
|
final String group1 = m.group(1);
|
|
final String group2 = m.group(2);
|
|
if (group2 == null) {
|
|
defines.define(group1, null, emptyParentheses);
|
|
} else {
|
|
final List<String> strings = defines.applyDefines(group2);
|
|
if (strings.size() > 1) {
|
|
defines.define(group1, strings, emptyParentheses);
|
|
} else {
|
|
final StringBuilder value = new StringBuilder(strings.get(0));
|
|
while (StringUtils.endsWithBackslash(value.toString())) {
|
|
value.setLength(value.length() - 1);
|
|
final CharSequence2 read = this.readLine();
|
|
value.append(read.toString2());
|
|
}
|
|
final List<String> li = new ArrayList<String>();
|
|
li.add(value.toString());
|
|
defines.define(group1, li, emptyParentheses);
|
|
}
|
|
}
|
|
return this.readLine();
|
|
}
|
|
|
|
// public int getLineNumber() {
|
|
// return source.getLineNumber();
|
|
// }
|
|
|
|
@Override
|
|
void closeInst() throws IOException {
|
|
source.close();
|
|
}
|
|
|
|
} |