mirror of https://github.com/octoleo/plantuml.git
410 lines
12 KiB
Java
410 lines
12 KiB
Java
/* ========================================================================
|
|
* PlantUML : a free UML diagram generator
|
|
* ========================================================================
|
|
*
|
|
* (C) Copyright 2009-2023, 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.swing;
|
|
|
|
import java.awt.BorderLayout;
|
|
import java.awt.Frame;
|
|
import java.awt.event.ActionEvent;
|
|
import java.awt.event.ActionListener;
|
|
import java.awt.event.FocusEvent;
|
|
import java.awt.event.FocusListener;
|
|
import java.awt.event.KeyAdapter;
|
|
import java.awt.event.KeyEvent;
|
|
import java.awt.event.MouseAdapter;
|
|
import java.awt.event.MouseEvent;
|
|
import java.awt.event.MouseListener;
|
|
import java.io.File;
|
|
import java.io.IOException;
|
|
import java.util.ArrayList;
|
|
import java.util.Collections;
|
|
import java.util.HashSet;
|
|
import java.util.Iterator;
|
|
import java.util.List;
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.Vector;
|
|
import java.util.concurrent.ExecutionException;
|
|
import java.util.concurrent.Future;
|
|
import java.util.prefs.Preferences;
|
|
import java.util.regex.Matcher;
|
|
import java.util.regex.Pattern;
|
|
|
|
import javax.swing.BorderFactory;
|
|
import javax.swing.JButton;
|
|
import javax.swing.JFileChooser;
|
|
import javax.swing.JFrame;
|
|
import javax.swing.JLabel;
|
|
import javax.swing.JList;
|
|
import javax.swing.JMenu;
|
|
import javax.swing.JMenuBar;
|
|
import javax.swing.JMenuItem;
|
|
import javax.swing.JPanel;
|
|
import javax.swing.JScrollPane;
|
|
import javax.swing.JTextField;
|
|
import javax.swing.ListModel;
|
|
import javax.swing.SwingUtilities;
|
|
import javax.swing.Timer;
|
|
import javax.swing.border.CompoundBorder;
|
|
|
|
import net.sourceforge.plantuml.DirWatcher2;
|
|
import net.sourceforge.plantuml.GeneratedImage;
|
|
import net.sourceforge.plantuml.Log;
|
|
import net.sourceforge.plantuml.Option;
|
|
import net.sourceforge.plantuml.log.Logger;
|
|
import net.sourceforge.plantuml.version.PSystemVersion;
|
|
|
|
public class MainWindow extends JFrame {
|
|
|
|
final private static Preferences prefs = Preferences.userNodeForPackage(MainWindow.class);
|
|
final private static String KEY_DIR = "cur";
|
|
final private static String KEY_PATTERN = "pat";
|
|
|
|
private final JList jList1 = new JList();
|
|
private final JScrollPane scrollPane;
|
|
private final JButton changeDirButton = new JButton("Change Directory");
|
|
private final JTextField extensions = new JTextField();
|
|
private final int period = 300;
|
|
|
|
final private List<SimpleLine> currentDirectoryListing2 = new ArrayList<>();
|
|
final private Set<ImageWindow> openWindows2 = new HashSet<>();
|
|
final private Option option;
|
|
|
|
private DirWatcher2 dirWatcher;
|
|
|
|
private String getExtensions() {
|
|
return prefs.get(KEY_PATTERN, getDefaultFileExtensions());
|
|
}
|
|
|
|
private String getDefaultFileExtensions() {
|
|
return "txt, tex, java, htm, html, c, h, cpp, apt, pu, puml, hpp, hh, md";
|
|
}
|
|
|
|
private void changeExtensions(String ext) {
|
|
if (ext.equals(getExtensions())) {
|
|
return;
|
|
}
|
|
final Pattern p = Pattern.compile("\\w+");
|
|
final Matcher m = p.matcher(ext);
|
|
final StringBuilder sb = new StringBuilder();
|
|
|
|
while (m.find()) {
|
|
final String value = m.group();
|
|
if (sb.length() > 0) {
|
|
sb.append(", ");
|
|
}
|
|
sb.append(value);
|
|
|
|
}
|
|
ext = sb.toString();
|
|
if (ext.length() == 0) {
|
|
ext = getDefaultFileExtensions();
|
|
}
|
|
extensions.setText(ext);
|
|
prefs.put(KEY_PATTERN, ext);
|
|
changeDir(dirWatcher.getDir());
|
|
}
|
|
|
|
private String getRegexpPattern(String ext) {
|
|
final Pattern p = Pattern.compile("\\w+");
|
|
final Matcher m = p.matcher(ext);
|
|
final StringBuilder filePattern = new StringBuilder("(?i)^.*\\.(");
|
|
|
|
while (m.find()) {
|
|
final String value = m.group();
|
|
if (filePattern.toString().endsWith("(") == false) {
|
|
filePattern.append("|");
|
|
}
|
|
filePattern.append(value);
|
|
}
|
|
if (filePattern.toString().endsWith("(") == false) {
|
|
filePattern.append(")$");
|
|
return filePattern.toString();
|
|
}
|
|
return Option.getPattern();
|
|
}
|
|
|
|
public MainWindow(Option option, File arg) {
|
|
super(getDirectory(arg).getAbsolutePath());
|
|
System.setProperty("PLANTUML_SECURITY_PROFILE", "UNSECURE");
|
|
final File dir = getDirectory(arg);
|
|
setIconImage(PSystemVersion.getPlantumlSmallIcon2());
|
|
this.option = option;
|
|
dirWatcher = new DirWatcher2(dir, option, getRegexpPattern(getExtensions()));
|
|
|
|
Log.info("Showing MainWindow");
|
|
scrollPane = new JScrollPane(jList1);
|
|
scrollPane.setBorder(BorderFactory.createEmptyBorder(10, 10, 10, 10));
|
|
|
|
final JPanel south = new JPanel(new BorderLayout());
|
|
final JLabel labelFileExtensions = new JLabel("File extensions: ");
|
|
extensions.setText(getExtensions());
|
|
|
|
labelFileExtensions.setBorder(BorderFactory.createEmptyBorder(6, 6, 6, 6));
|
|
CompoundBorder border = BorderFactory.createCompoundBorder(BorderFactory.createEmptyBorder(10, 10, 10, 10),
|
|
BorderFactory.createEtchedBorder());
|
|
border = BorderFactory.createCompoundBorder(border, BorderFactory.createEmptyBorder(5, 5, 5, 5));
|
|
south.setBorder(border);
|
|
south.add(labelFileExtensions, BorderLayout.WEST);
|
|
south.add(extensions, BorderLayout.CENTER);
|
|
|
|
south.add(changeDirButton, BorderLayout.SOUTH);
|
|
|
|
getContentPane().add(south, BorderLayout.SOUTH);
|
|
getContentPane().add(scrollPane, BorderLayout.CENTER);
|
|
|
|
final MouseListener mouseListener = new MouseAdapter() {
|
|
@Override
|
|
public void mouseClicked(MouseEvent e) {
|
|
try {
|
|
if (e.getClickCount() == 2) {
|
|
final int index = jList1.locationToIndex(e.getPoint());
|
|
doubleClick((SimpleLine) jList1.getModel().getElementAt(index), jList1.getModel(), index);
|
|
}
|
|
} catch (Exception ex) {
|
|
|
|
}
|
|
}
|
|
};
|
|
jList1.addMouseListener(mouseListener);
|
|
changeDirButton.addActionListener(new ActionListener() {
|
|
public void actionPerformed(ActionEvent e) {
|
|
System.err.println("Opening Directory Window");
|
|
displayDialogChangeDir();
|
|
}
|
|
});
|
|
jList1.addKeyListener(new KeyAdapter() {
|
|
@Override
|
|
public void keyPressed(KeyEvent e) {
|
|
if (e.getKeyCode() == KeyEvent.VK_ENTER) {
|
|
final int index = jList1.getSelectedIndex();
|
|
doubleClick((SimpleLine) jList1.getModel().getElementAt(index), jList1.getModel(), index);
|
|
}
|
|
}
|
|
|
|
});
|
|
|
|
extensions.addActionListener(new ActionListener() {
|
|
public void actionPerformed(ActionEvent e) {
|
|
changeExtensions(extensions.getText());
|
|
}
|
|
});
|
|
extensions.addFocusListener(new FocusListener() {
|
|
|
|
public void focusGained(FocusEvent e) {
|
|
}
|
|
|
|
public void focusLost(FocusEvent e) {
|
|
changeExtensions(extensions.getText());
|
|
}
|
|
});
|
|
|
|
final JMenuBar menuBar = new JMenuBar();
|
|
final JMenu mFile = new JMenu("File");
|
|
menuBar.add(mFile);
|
|
setJMenuBar(menuBar);
|
|
|
|
final JMenuItem sprite = new JMenuItem("Open Sprite Window");
|
|
mFile.add(sprite);
|
|
sprite.addActionListener(new ActionListener() {
|
|
public void actionPerformed(ActionEvent e) {
|
|
new SpriteWindow();
|
|
}
|
|
});
|
|
|
|
final JMenuItem about = new JMenuItem("About");
|
|
mFile.add(about);
|
|
about.addActionListener(new ActionListener() {
|
|
public void actionPerformed(ActionEvent e) {
|
|
new AboutWindow();
|
|
}
|
|
});
|
|
|
|
final JMenuItem exit = new JMenuItem("Exit");
|
|
mFile.add(exit);
|
|
exit.addActionListener(new ActionListener() {
|
|
public void actionPerformed(ActionEvent e) {
|
|
System.exit(0);
|
|
}
|
|
});
|
|
|
|
setSize(640, 400);
|
|
this.setLocationRelativeTo(this.getParent());
|
|
setVisible(true);
|
|
setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
|
|
|
|
startTimer();
|
|
}
|
|
|
|
private static File getDirectory(File arg) {
|
|
if (arg != null && arg.exists() && arg.isDirectory()) {
|
|
return arg;
|
|
}
|
|
return new File(prefs.get(KEY_DIR, "."));
|
|
}
|
|
|
|
private void startTimer() {
|
|
Log.info("Init done");
|
|
final Timer timer = new Timer(period, new ActionListener() {
|
|
public void actionPerformed(ActionEvent e) {
|
|
tick();
|
|
}
|
|
});
|
|
timer.setInitialDelay(0);
|
|
timer.start();
|
|
Log.info("Timer started");
|
|
}
|
|
|
|
private void displayDialogChangeDir() {
|
|
final JFileChooser chooser = new JFileChooser();
|
|
chooser.setDialogType(JFileChooser.CUSTOM_DIALOG);
|
|
chooser.setDialogTitle("Directory to watch:");
|
|
chooser.setFileSelectionMode(JFileChooser.DIRECTORIES_ONLY);
|
|
chooser.setAcceptAllFileFilterUsed(false);
|
|
final String currentPath = prefs.get(KEY_DIR, ".");
|
|
chooser.setCurrentDirectory(new File(currentPath));
|
|
Log.info("Showing OpenDialog");
|
|
final int returnVal = chooser.showOpenDialog(this);
|
|
Log.info("Closing OpenDialog");
|
|
if (returnVal == JFileChooser.APPROVE_OPTION) {
|
|
final File dir = chooser.getSelectedFile();
|
|
changeDir(dir);
|
|
}
|
|
|
|
}
|
|
|
|
private void changeDir(File dir) {
|
|
prefs.put(KEY_DIR, dir.getAbsolutePath());
|
|
dirWatcher.cancel();
|
|
dirWatcher = new DirWatcher2(dir, option, getRegexpPattern(getExtensions()));
|
|
setTitle(dir.getAbsolutePath());
|
|
Log.info("Creating DirWatcher");
|
|
currentDirectoryListing2.clear();
|
|
jList1.setListData(new Vector<>(currentDirectoryListing2));
|
|
jList1.setVisible(true);
|
|
}
|
|
|
|
private void doubleClick(SimpleLine simpleLine, ListModel listModel, int index) {
|
|
for (ImageWindow win : openWindows2) {
|
|
if (win.getSimpleLine().equals(simpleLine)) {
|
|
win.setVisible(true);
|
|
win.setExtendedState(Frame.NORMAL);
|
|
return;
|
|
}
|
|
}
|
|
if (simpleLine.getGeneratedImage() != null) {
|
|
openWindows2.add(new ImageWindow(simpleLine, this, listModel, index));
|
|
}
|
|
}
|
|
|
|
private void tick() {
|
|
SwingUtilities.invokeLater(new Runnable() {
|
|
public void run() {
|
|
try {
|
|
final boolean changed = refreshDir();
|
|
if (changed) {
|
|
jList1.setListData(new Vector<>(currentDirectoryListing2));
|
|
jList1.setVisible(true);
|
|
}
|
|
} catch (IOException | InterruptedException | ExecutionException e) {
|
|
Logger.error(e);
|
|
}
|
|
}
|
|
});
|
|
}
|
|
|
|
private boolean refreshDir() throws IOException, InterruptedException, ExecutionException {
|
|
final Map<File, Future<List<GeneratedImage>>> createdFiles2 = dirWatcher.buildCreatedFiles();
|
|
boolean changed = false;
|
|
|
|
for (Map.Entry<File, Future<List<GeneratedImage>>> ent : createdFiles2.entrySet()) {
|
|
final File file = ent.getKey();
|
|
removeAllThatUseThisFile(file);
|
|
final Future<List<GeneratedImage>> future = ent.getValue();
|
|
final SimpleLine simpleLine = SimpleLine.fromFuture(file, future);
|
|
currentDirectoryListing2.add(simpleLine);
|
|
changed = true;
|
|
}
|
|
|
|
for (SimpleLine line : new ArrayList<>(currentDirectoryListing2)) {
|
|
if (line.pendingAndFinished()) {
|
|
currentDirectoryListing2.remove(line);
|
|
changed = true;
|
|
final Future<List<GeneratedImage>> future = line.getFuture();
|
|
for (GeneratedImage im : future.get()) {
|
|
mayRefreshImageWindow(im.getPngFile());
|
|
final SimpleLine simpleLine = SimpleLine.fromGeneratedImage(line.getFile(), im);
|
|
currentDirectoryListing2.add(simpleLine);
|
|
}
|
|
}
|
|
}
|
|
Collections.sort(currentDirectoryListing2);
|
|
return changed;
|
|
}
|
|
|
|
private void removeAllThatUseThisFile(File file) {
|
|
for (final Iterator<SimpleLine> it = currentDirectoryListing2.iterator(); it.hasNext();) {
|
|
final SimpleLine line = it.next();
|
|
if (line.getFile().equals(file)) {
|
|
it.remove();
|
|
}
|
|
}
|
|
}
|
|
|
|
private void mayRefreshImageWindow(File pngFile) {
|
|
for (ImageWindow win : openWindows2) {
|
|
if (win.getSimpleLine().getGeneratedImage() == null) {
|
|
continue;
|
|
}
|
|
if (pngFile.equals(win.getSimpleLine().getGeneratedImage().getPngFile())) {
|
|
win.refreshImage(true);
|
|
}
|
|
}
|
|
|
|
}
|
|
|
|
public void closing(ImageWindow imageWindow) {
|
|
final boolean ok = openWindows2.remove(imageWindow);
|
|
if (ok == false) {
|
|
throw new IllegalStateException();
|
|
}
|
|
}
|
|
|
|
public List<SimpleLine> getCurrentDirectoryListing2() {
|
|
return Collections.unmodifiableList(currentDirectoryListing2);
|
|
}
|
|
|
|
}
|