net.sourceforge.plantuml.preproc.StartDiagramExtractReader Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of plantuml-gplv2 Show documentation
Show all versions of plantuml-gplv2 Show documentation
PlantUML is a component that allows to quickly write diagrams from text.
// THIS FILE HAS BEEN GENERATED BY A PREPROCESSOR.
/* +=======================================================================
* |
* | PlantUML : a free UML diagram generator
* |
* +=======================================================================
*
* (C) Copyright 2009-2024, Arnaud Roques
*
* Project Info: https://plantuml.com
*
* If you like this project or if you find it useful, you can support us at:
*
* https://plantuml.com/patreon (only 1$ per month!)
* https://plantuml.com/liberapay (only 1€ per month!)
* https://plantuml.com/paypal
*
*
* PlantUML is free software; you can redistribute it and/or modify it
* under the terms of the GNU General Public License V2.
*
* THE ACCOMPANYING PROGRAM IS PROVIDED UNDER THE TERMS OF THIS ECLIPSE PUBLIC
* LICENSE ("AGREEMENT"). [GNU General Public License V2]
*
* ANY USE, REPRODUCTION OR DISTRIBUTION OF THE PROGRAM CONSTITUTES
* RECIPIENT'S ACCEPTANCE OF THIS AGREEMENT.
*
* You may obtain a copy of the License at
*
* https://www.gnu.org/licenses/old-licenses/gpl-2.0.html
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*
* PlantUML can occasionally display sponsored or advertising messages. Those
* messages are usually generated on welcome or error images and never on
* functional diagrams.
* See https://plantuml.com/professional if you want to remove them
*
* Images (whatever their format : PNG, SVG, EPS...) generated by running PlantUML
* are owned by the author of their corresponding sources code (that is, their
* textual description in PlantUML language). Those images are not covered by
* this GPL v2 license.
*
* The generated images can then be used without any reference to the GPL v2 license.
* It is not even necessary to stipulate that they have been generated with PlantUML,
* although this will be appreciated by the PlantUML team.
*
* There is an exception : if the textual description in PlantUML language is also covered
* by any license, then the generated images are logically covered
* by the very same license.
*
* This is the IGY distribution (Install GraphViz by Yourself).
* You have to install GraphViz and to setup the GRAPHVIZ_DOT environment variable
* (see https://plantuml.com/graphviz-dot )
*
* Icons provided by OpenIconic : https://useiconic.com/open
* Archimate sprites provided by Archi : http://www.archimatetool.com
* Stdlib AWS provided by https://github.com/milo-minderbinder/AWS-PlantUML
* Stdlib Icons provided https://github.com/tupadr3/plantuml-icon-font-sprites
* ASCIIMathML (c) Peter Jipsen http://www.chapman.edu/~jipsen
* ASCIIMathML (c) David Lippman http://www.pierce.ctc.edu/dlippman
* CafeUndZopfli ported by Eugene Klyuchnikov https://github.com/eustas/CafeUndZopfli
* Brotli (c) by the Brotli Authors https://github.com/google/brotli
* Themes (c) by Brett Schwarz https://github.com/bschwarz/puml-themes
* Twemoji (c) by Twitter at https://twemoji.twitter.com/
*
*/
package net.sourceforge.plantuml.preproc;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.io.Reader;
import java.nio.charset.Charset;
import net.sourceforge.plantuml.log.Logme;
import net.sourceforge.plantuml.preproc2.ReadFilterMergeLines;
import net.sourceforge.plantuml.security.SURL;
import net.sourceforge.plantuml.text.StringLocated;
import net.sourceforge.plantuml.utils.Log;
import net.sourceforge.plantuml.utils.StartUtils;
public class StartDiagramExtractReader implements ReadLine {
private final ReadLine raw;
private boolean finished = false;
public static StartDiagramExtractReader build(FileWithSuffix f2, StringLocated s, Charset charset) {
return new StartDiagramExtractReader(getReadLine(f2, s, charset), f2.getSuffix());
}
public static StartDiagramExtractReader build(SURL url, StringLocated s, String uid, Charset charset) {
return new StartDiagramExtractReader(getReadLine(url, s, charset), uid);
}
public static StartDiagramExtractReader build(InputStream is, StringLocated s, String desc) {
return new StartDiagramExtractReader(getReadLine(is, s, desc), null);
}
private StartDiagramExtractReader(ReadLine raw, String suf) {
int bloc = 0;
String uid = null;
if (suf != null && suf.matches("\\d+"))
bloc = Integer.parseInt(suf);
else
uid = suf;
if (bloc < 0)
bloc = 0;
this.raw = raw;
StringLocated s = null;
try {
while ((s = raw.readLine()) != null) {
if (StartUtils.isArobaseStartDiagram(s.getString()) && checkUid(uid, s)) {
if (bloc == 0)
return;
bloc--;
}
}
} catch (IOException e) {
Logme.error(e);
Log.error("Error " + e);
}
finished = true;
}
private boolean checkUid(String uid, StringLocated s) {
if (uid == null)
return true;
if (s.toString().matches(".*id=" + uid + "\\W.*"))
return true;
return false;
}
private static ReadLine getReadLine(FileWithSuffix f2, StringLocated s, Charset charset) {
try {
final Reader tmp1 = f2.getReader(charset);
if (tmp1 == null)
return new ReadLineSimple(s, "Cannot open " + f2.getDescription());
return uncommentAndMerge(ReadLineReader.create(tmp1, f2.getDescription()));
} catch (IOException e) {
return new ReadLineSimple(s, e.toString());
}
}
private static ReadLine getReadLine(InputStream is, StringLocated s, String description) {
return uncommentAndMerge(ReadLineReader.create(new InputStreamReader(is), description));
}
private static ReadLine getReadLine(SURL url, StringLocated s, Charset charset) {
final InputStream tmp = url.openStream();
if (tmp == null)
return new ReadLineSimple(s, "Cannot connect");
return uncommentAndMerge(ReadLineReader.create(new InputStreamReader(tmp, charset), url.toString()));
}
private static ReadLine uncommentAndMerge(ReadLine reader) {
return new UncommentReadLine(new ReadFilterMergeLines().applyFilter(reader));
}
static public boolean containsStartDiagram(FileWithSuffix f2, StringLocated s, Charset charset) throws IOException {
final ReadLine r = getReadLine(f2, s, charset);
return containsStartDiagram(r);
}
static public boolean containsStartDiagram(SURL url, StringLocated s, Charset charset) throws IOException {
final ReadLine r = getReadLine(url, s, charset);
return containsStartDiagram(r);
}
static public boolean containsStartDiagram(InputStream is, StringLocated s, String description) throws IOException {
final ReadLine r = getReadLine(is, s, description);
return containsStartDiagram(r);
}
private static boolean containsStartDiagram(final ReadLine r) throws IOException {
try {
StringLocated s = null;
while ((s = r.readLine()) != null)
if (StartUtils.isArobaseStartDiagram(s.getString()))
return true;
} finally {
if (r != null)
r.close();
}
return false;
}
public StringLocated readLine() throws IOException {
if (finished)
return null;
final StringLocated result = raw.readLine();
if (result != null && StartUtils.isArobaseEndDiagram(result.getString())) {
finished = true;
return null;
}
return result;
}
public void close() throws IOException {
raw.close();
}
}