mirror of https://github.com/octoleo/plantuml.git
190 lines
6.1 KiB
Java
190 lines
6.1 KiB
Java
/* ========================================================================
|
|
* PlantUML : a free UML diagram generator
|
|
* ========================================================================
|
|
*
|
|
* (C) Copyright 2009-2020, 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.style;
|
|
|
|
import java.io.IOException;
|
|
import java.io.InputStream;
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.Collections;
|
|
import java.util.EnumMap;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
|
|
import net.sourceforge.plantuml.FileSystem;
|
|
import net.sourceforge.plantuml.LineLocationImpl;
|
|
import net.sourceforge.plantuml.Log;
|
|
import net.sourceforge.plantuml.SkinParam;
|
|
import net.sourceforge.plantuml.StringLocated;
|
|
import net.sourceforge.plantuml.command.BlocLines;
|
|
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.security.SFile;
|
|
|
|
public class StyleLoader {
|
|
|
|
private final SkinParam skinParam;
|
|
|
|
public StyleLoader(SkinParam skinParam) {
|
|
this.skinParam = skinParam;
|
|
}
|
|
|
|
private StyleBuilder styleBuilder;
|
|
|
|
public StyleBuilder loadSkin(String filename) throws IOException {
|
|
this.styleBuilder = new StyleBuilder(skinParam);
|
|
|
|
InputStream internalIs = null;
|
|
SFile localFile = new SFile(filename);
|
|
Log.info("Trying to load style " + filename);
|
|
if (localFile.exists() == false) {
|
|
localFile = FileSystem.getInstance().getFile(filename);
|
|
}
|
|
if (localFile.exists()) {
|
|
Log.info("File found : " + localFile.getPrintablePath());
|
|
internalIs = localFile.openFile();
|
|
} else {
|
|
Log.info("File not found : " + localFile.getPrintablePath());
|
|
final String res = "/skin/" + filename;
|
|
internalIs = StyleLoader.class.getResourceAsStream(res);
|
|
if (internalIs != null) {
|
|
Log.info("... but " + filename + " found inside the .jar");
|
|
}
|
|
}
|
|
if (internalIs == null) {
|
|
Log.error("No .skin file seems to be available");
|
|
throw new NoStyleAvailableException();
|
|
}
|
|
final BlocLines lines2 = BlocLines.load(internalIs, new LineLocationImpl(filename, null));
|
|
loadSkinInternal(lines2);
|
|
if (this.styleBuilder == null) {
|
|
Log.error("No .skin file seems to be available");
|
|
throw new NoStyleAvailableException();
|
|
}
|
|
return this.styleBuilder;
|
|
}
|
|
|
|
private void loadSkinInternal(final BlocLines lines) {
|
|
for (Style newStyle : getDeclaredStyles(lines, styleBuilder)) {
|
|
this.styleBuilder.loadInternal(newStyle.getSignature(), newStyle);
|
|
}
|
|
}
|
|
|
|
private static final String NAME_USER = "[\\w()]+?";
|
|
private final static Pattern2 userName = MyPattern.cmpile("^[.:]?(" + NAME_USER + ")([%s]+\\*)?[%s]*\\{$");
|
|
private final static Pattern2 propertyAndValue = MyPattern.cmpile("^([\\w]+):?[%s]+(.*?);?$");
|
|
private final static Pattern2 closeBracket = MyPattern.cmpile("^\\}$");
|
|
|
|
public static Collection<Style> getDeclaredStyles(BlocLines lines, AutomaticCounter counter) {
|
|
lines = lines.eventuallyMoveAllEmptyBracket();
|
|
final List<Style> result = new ArrayList<>();
|
|
|
|
final List<String> context = new ArrayList<>();
|
|
final List<Map<PName, Value>> maps = new ArrayList<Map<PName, Value>>();
|
|
boolean inComment = false;
|
|
for (StringLocated s : lines) {
|
|
String trimmed = s.getTrimmed().getString();
|
|
if (trimmed.startsWith("/*") || trimmed.startsWith("/'")) {
|
|
inComment = true;
|
|
continue;
|
|
}
|
|
if (trimmed.endsWith("*/") || trimmed.endsWith("'/")) {
|
|
inComment = false;
|
|
continue;
|
|
}
|
|
if (inComment) {
|
|
continue;
|
|
}
|
|
final int x = trimmed.lastIndexOf("//");
|
|
if (x != -1) {
|
|
trimmed = trimmed.substring(0, x).trim();
|
|
}
|
|
final Matcher2 mUserName = userName.matcher(trimmed);
|
|
if (mUserName.find()) {
|
|
String n = mUserName.group(1);
|
|
final boolean isRecurse = mUserName.group(2) != null;
|
|
if (isRecurse) {
|
|
n += "*";
|
|
}
|
|
context.add(n);
|
|
maps.add(new EnumMap<PName, Value>(PName.class));
|
|
continue;
|
|
}
|
|
final Matcher2 mPropertyAndValue = propertyAndValue.matcher(trimmed);
|
|
if (mPropertyAndValue.find()) {
|
|
final PName key = PName.getFromName(mPropertyAndValue.group(1));
|
|
final String value = mPropertyAndValue.group(2);
|
|
if (key != null && maps.size() > 0) {
|
|
maps.get(maps.size() - 1).put(key, new ValueImpl(value, counter));
|
|
}
|
|
continue;
|
|
}
|
|
final Matcher2 mCloseBracket = closeBracket.matcher(trimmed);
|
|
if (mCloseBracket.find()) {
|
|
if (context.size() > 0) {
|
|
final StyleSignature signature = contextToSignature(context);
|
|
final Style style = new Style(signature, maps.get(maps.size() - 1));
|
|
result.add(style);
|
|
context.remove(context.size() - 1);
|
|
maps.remove(maps.size() - 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
return Collections.unmodifiableList(result);
|
|
|
|
}
|
|
|
|
private static StyleSignature contextToSignature(List<String> context) {
|
|
StyleSignature result = StyleSignature.empty();
|
|
boolean star = false;
|
|
for (Iterator<String> it = context.iterator(); it.hasNext();) {
|
|
String s = it.next();
|
|
if (s.endsWith("*")) {
|
|
star = true;
|
|
s = s.substring(0, s.length() - 1);
|
|
}
|
|
result = result.add(s);
|
|
}
|
|
if (star) {
|
|
result = result.addStar();
|
|
}
|
|
return result;
|
|
}
|
|
|
|
} |