mirror of
https://github.com/octoleo/plantuml.git
synced 2024-06-14 22:22:24 +00:00
179 lines
5.7 KiB
Java
179 lines
5.7 KiB
Java
/* ========================================================================
|
|
* PlantUML : a free UML diagram generator
|
|
* ========================================================================
|
|
*
|
|
* (C) Copyright 2009, Arnaud Roques
|
|
*
|
|
* Project Info: http://plantuml.sourceforge.net
|
|
*
|
|
* 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 Lesser 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.
|
|
*
|
|
* [Java is a trademark or registered trademark of Sun Microsystems, Inc.
|
|
* in the United States and other countries.]
|
|
*
|
|
* Original Author: Arnaud Roques
|
|
*
|
|
* Revision $Revision: 5391 $
|
|
*
|
|
*/
|
|
package net.sourceforge.plantuml;
|
|
|
|
import java.awt.GraphicsEnvironment;
|
|
import java.io.BufferedReader;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.io.InputStreamReader;
|
|
import java.io.PrintStream;
|
|
import java.util.List;
|
|
|
|
import javax.swing.UIManager;
|
|
|
|
import net.sourceforge.plantuml.code.Transcoder;
|
|
import net.sourceforge.plantuml.png.MetadataTag;
|
|
import net.sourceforge.plantuml.preproc.Defines;
|
|
import net.sourceforge.plantuml.swing.MainWindow;
|
|
|
|
public class Run {
|
|
|
|
public static void main(String[] argsArray) throws IOException, InterruptedException {
|
|
final Option option = new Option(argsArray);
|
|
if (OptionFlags.getInstance().isVerbose()) {
|
|
Log.info("GraphicsEnvironment.isHeadless() " + GraphicsEnvironment.isHeadless());
|
|
}
|
|
if (OptionFlags.getInstance().isGui()) {
|
|
try {
|
|
UIManager.setLookAndFeel("com.sun.java.swing.plaf.windows.WindowsLookAndFeel");
|
|
} catch (Exception e) {
|
|
}
|
|
new MainWindow(option);
|
|
} else if (option.isPipe() || option.isSyntax()) {
|
|
managePipe(option);
|
|
} else {
|
|
manageFiles(option);
|
|
}
|
|
}
|
|
|
|
private static void managePipe(Option option) throws IOException {
|
|
final String charset = option.getCharset();
|
|
final BufferedReader br;
|
|
if (charset == null) {
|
|
br = new BufferedReader(new InputStreamReader(System.in));
|
|
} else {
|
|
br = new BufferedReader(new InputStreamReader(System.in, charset));
|
|
}
|
|
managePipe(option, br, System.out);
|
|
}
|
|
|
|
static void managePipe(Option option, final BufferedReader br, final PrintStream ps) throws IOException {
|
|
final StringBuilder sb = new StringBuilder();
|
|
String s = null;
|
|
while ((s = br.readLine()) != null) {
|
|
sb.append(s);
|
|
sb.append("\n");
|
|
}
|
|
String source = sb.toString();
|
|
if (source.contains("@startuml") == false) {
|
|
source = "@startuml\n" + source + "\n@enduml";
|
|
}
|
|
final SourceStringReader sourceStringReader = new SourceStringReader(new Defines(), source, option.getConfig());
|
|
|
|
if (option.isSyntax()) {
|
|
try {
|
|
final PSystem system = sourceStringReader.getBlocks().get(0).getSystem();
|
|
if (system instanceof UmlDiagram) {
|
|
ps.println(((UmlDiagram) system).getUmlDiagramType().name());
|
|
ps.println(system.getDescription());
|
|
} else if (system instanceof PSystemError) {
|
|
ps.println("ERROR");
|
|
final PSystemError sys = (PSystemError) system;
|
|
ps.println(sys.getHigherErrorPosition());
|
|
for (String er : sys.getErrs()) {
|
|
ps.println(er);
|
|
}
|
|
} else {
|
|
ps.println("OTHER");
|
|
ps.println(system.getDescription());
|
|
}
|
|
} catch (InterruptedException e) {
|
|
Log.error("InterruptedException " + e);
|
|
}
|
|
} else if (option.isPipe()) {
|
|
final String result = sourceStringReader.generateImage(ps, 0, option.getFileFormat());
|
|
}
|
|
}
|
|
|
|
private static void manageFile(File f, Option option) throws IOException, InterruptedException {
|
|
if (OptionFlags.getInstance().isMetadata()) {
|
|
System.out.println("------------------------");
|
|
System.out.println(f);
|
|
// new Metadata().readAndDisplayMetadata(f);
|
|
System.out.println();
|
|
System.out.println(new MetadataTag(f, "plantuml").getData());
|
|
System.out.println("------------------------");
|
|
} else {
|
|
final SourceFileReader sourceFileReader = new SourceFileReader(option.getDefaultDefines(), f, option
|
|
.getOutputDir(), option.getConfig(), option.getCharset(), option.getFileFormat());
|
|
if (option.isComputeurl()) {
|
|
final List<String> urls = sourceFileReader.getEncodedUrl();
|
|
for (String s : urls) {
|
|
System.out.println(s);
|
|
}
|
|
} else {
|
|
sourceFileReader.getGeneratedImages();
|
|
|
|
}
|
|
}
|
|
}
|
|
|
|
private static void manageFiles(Option option) throws IOException, InterruptedException {
|
|
|
|
File lockFile = null;
|
|
try {
|
|
if (OptionFlags.getInstance().isWord()) {
|
|
final File dir = new File(option.getResult().get(0));
|
|
final File javaIsRunningFile = new File(dir, "javaisrunning.tmp");
|
|
javaIsRunningFile.delete();
|
|
lockFile = new File(dir, "javaumllock.tmp");
|
|
}
|
|
processArgs(option);
|
|
} finally {
|
|
if (lockFile != null) {
|
|
lockFile.delete();
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
private static void processArgs(Option option) throws IOException, InterruptedException {
|
|
for (String s : option.getResult()) {
|
|
if (option.isDecodeurl()) {
|
|
final Transcoder transcoder = new Transcoder();
|
|
System.out.println("@startuml");
|
|
System.out.println(transcoder.decode(s));
|
|
System.out.println("@enduml");
|
|
} else {
|
|
final FileGroup group = new FileGroup(s, option.getExcludes(), option);
|
|
for (File f : group.getFiles()) {
|
|
manageFile(f, option);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
}
|