net.sourceforge.plantuml.ftp.FtpConnexion Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of plantuml-lgpl Show documentation
Show all versions of plantuml-lgpl 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 Lesser 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 Lesser General Public
* License for more details.
*
* You should have received a copy of the GNU Lesser General Public License
* along with this library. If not, see .
*
* 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 LGPL license.
*
* The generated images can then be used without any reference to the LGPL 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.ftp;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import net.sourceforge.plantuml.BlockUml;
import net.sourceforge.plantuml.FileFormat;
import net.sourceforge.plantuml.FileFormatOption;
import net.sourceforge.plantuml.SourceStringReader;
import net.sourceforge.plantuml.core.DiagramDescription;
public class FtpConnexion {
// ::remove folder when __CORE__
private final String user;
private final Map incoming = new HashMap();
private final Map outgoing = new HashMap();
private final Set futureOutgoing = new HashSet<>();
private FileFormat fileFormat;
public FtpConnexion(String user, FileFormat defaultfileFormat) {
this.user = user;
this.fileFormat = defaultfileFormat;
}
public synchronized void addIncoming(String fileName, String data) {
if (fileName.startsWith("/"))
throw new IllegalArgumentException();
incoming.put(fileName, data);
}
public synchronized void futureOutgoing(String fileName) {
outgoing.remove(fileName);
futureOutgoing.add(fileName);
}
public synchronized Collection getFiles() {
final List result = new ArrayList<>(incoming.keySet());
result.addAll(outgoing.keySet());
result.addAll(futureOutgoing);
return Collections.unmodifiableCollection(result);
}
public synchronized boolean willExist(String fileName) {
if (incoming.containsKey(fileName))
return true;
if (outgoing.containsKey(fileName))
return true;
if (futureOutgoing.contains(fileName))
return true;
return false;
}
public synchronized boolean doesExist(String fileName) {
if (incoming.containsKey(fileName))
return true;
if (outgoing.containsKey(fileName))
return true;
return false;
}
public synchronized byte[] getData(String fileName) throws InterruptedException {
if (fileName.startsWith("/"))
throw new IllegalArgumentException();
final String data = incoming.get(fileName);
if (data != null)
return data.getBytes();
final byte data2[] = outgoing.get(fileName);
if (data2 == null)
return new byte[1];
return data2;
}
public synchronized int getSize(String fileName) {
if (fileName.startsWith("/"))
throw new IllegalArgumentException();
final String data = incoming.get(fileName);
if (data != null)
return data.length();
final byte data2[] = outgoing.get(fileName);
if (data2 != null)
return data2.length;
return 0;
}
public void processImage(String fileName) throws IOException {
if (fileName.startsWith("/"))
throw new IllegalArgumentException();
final String pngFileName = getFutureFileName(fileName);
boolean done = false;
try {
final SourceStringReader sourceStringReader = new SourceStringReader(incoming.get(fileName));
final ByteArrayOutputStream baos = new ByteArrayOutputStream();
final FileFormat format = getFileFormat();
final FileFormatOption fileFormatOption = new FileFormatOption(format);
final DiagramDescription desc = sourceStringReader.generateDiagramDescription(fileFormatOption);
final List blocks = sourceStringReader.getBlocks();
if (blocks.size() > 0)
blocks.get(0).getDiagram().exportDiagram(baos, 0, fileFormatOption);
final String errorFileName = pngFileName.substring(0, pngFileName.length() - 4) + ".err";
synchronized (this) {
outgoing.remove(pngFileName);
futureOutgoing.remove(pngFileName);
outgoing.remove(errorFileName);
if (desc != null && desc.getDescription() != null) {
outgoing.put(pngFileName, baos.toByteArray());
done = true;
if (desc.getDescription().startsWith("(Error)")) {
final ByteArrayOutputStream errBaos = new ByteArrayOutputStream();
sourceStringReader.outputImage(errBaos, new FileFormatOption(FileFormat.ATXT));
errBaos.close();
outgoing.put(errorFileName, errBaos.toByteArray());
}
}
}
} finally {
if (done == false)
outgoing.put(pngFileName, new byte[0]);
}
}
public String getFutureFileName(String fileName) {
return getFileFormat().changeName(fileName, 0);
}
private FileFormat getFileFormat() {
return fileFormat;
}
public synchronized void delete(String fileName) {
if (fileName.contains("*")) {
incoming.clear();
outgoing.clear();
futureOutgoing.clear();
} else {
incoming.remove(fileName);
outgoing.remove(fileName);
futureOutgoing.add(fileName);
}
}
public void setFileFormat(FileFormat fileFormat) {
this.fileFormat = fileFormat;
}
}