1
0
mirror of https://github.com/mitb-archive/filebot synced 2024-08-13 17:03:45 -04:00
filebot/source/net/filebot/ui/rename/FormatDialog.java
2019-02-28 16:17:22 +07:00

700 lines
23 KiB
Java

package net.filebot.ui.rename;
import static java.awt.Font.*;
import static java.awt.event.KeyEvent.*;
import static java.util.Collections.*;
import static java.util.stream.Collectors.*;
import static javax.swing.BorderFactory.*;
import static javax.swing.KeyStroke.*;
import static net.filebot.Logging.*;
import static net.filebot.Settings.*;
import static net.filebot.ui.ThemeSupport.*;
import static net.filebot.util.ExceptionUtilities.*;
import static net.filebot.util.FileUtilities.*;
import static net.filebot.util.ui.SwingUI.*;
import java.awt.Component;
import java.awt.Cursor;
import java.awt.Dimension;
import java.awt.Font;
import java.awt.Rectangle;
import java.awt.Window;
import java.awt.event.WindowAdapter;
import java.awt.event.WindowEvent;
import java.io.File;
import java.text.Format;
import java.util.LinkedHashSet;
import java.util.Map;
import java.util.MissingResourceException;
import java.util.Objects;
import java.util.ResourceBundle;
import java.util.Set;
import java.util.TreeMap;
import java.util.concurrent.CancellationException;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Future;
import java.util.concurrent.LinkedBlockingQueue;
import java.util.concurrent.RunnableFuture;
import java.util.concurrent.ThreadPoolExecutor;
import java.util.concurrent.ThreadPoolExecutor.DiscardOldestPolicy;
import java.util.concurrent.TimeUnit;
import java.util.logging.Level;
import java.util.prefs.Preferences;
import javax.script.ScriptException;
import javax.swing.Action;
import javax.swing.JButton;
import javax.swing.JComponent;
import javax.swing.JDialog;
import javax.swing.JLabel;
import javax.swing.JMenuItem;
import javax.swing.JPanel;
import javax.swing.JPopupMenu;
import javax.swing.JTextField;
import javax.swing.SwingUtilities;
import javax.swing.SwingWorker;
import javax.swing.Timer;
import javax.swing.event.PopupMenuEvent;
import javax.swing.event.PopupMenuListener;
import org.fife.ui.rtextarea.RTextScrollPane;
import net.filebot.ResourceManager;
import net.filebot.Settings;
import net.filebot.UserFiles;
import net.filebot.format.BindingException;
import net.filebot.format.ExpressionFileFormat;
import net.filebot.format.ExpressionFormat;
import net.filebot.format.MediaBindingBean;
import net.filebot.format.SuppressedThrowables;
import net.filebot.media.MetaAttributes;
import net.filebot.platform.mac.MacAppUtilities;
import net.filebot.util.DefaultThreadFactory;
import net.filebot.util.PreferencesList;
import net.filebot.util.PreferencesMap.PreferencesEntry;
import net.filebot.util.ui.LinkButton;
import net.filebot.util.ui.ProgressIndicator;
import net.filebot.util.ui.notification.SeparatorBorder.Position;
import net.filebot.web.AudioTrackFormat;
import net.filebot.web.Datasource;
import net.filebot.web.EpisodeFormat;
import net.filebot.web.EpisodeListProvider;
import net.filebot.web.MovieFormat;
import net.filebot.web.MovieIdentificationService;
import net.filebot.web.MusicIdentificationService;
import net.miginfocom.swing.MigLayout;
public class FormatDialog extends JDialog {
private boolean submit = false;
private ExpressionFileFormat format;
private Mode mode;
private boolean locked = false;
private MediaBindingBean sample = null;
private ExecutorService executor = createExecutor();
private RunnableFuture<String> currentPreviewFuture;
private JLabel preview = new JLabel();
private JLabel status = new JLabel();
private FormatExpressionTextArea editor = new FormatExpressionTextArea();
private ProgressIndicator progressIndicator = getProgressIndicator();
private JLabel title = new JLabel();
private JPanel help = new JPanel(new MigLayout("insets 0, nogrid, novisualpadding, fillx"));
private static final PreferencesEntry<String> persistentSampleFile = Settings.forPackage(FormatDialog.class).entry("format.sample.file");
public enum Mode {
Episode, Movie, Music, File;
public Mode next() {
// cycle through Episode -> Movie -> Music (but ignore generic File mode)
return values()[(ordinal() + 1) % File.ordinal()];
}
public String key() {
return name().toLowerCase();
}
public Format getFormat() {
switch (this) {
case Episode:
return new EpisodeFormat();
case Movie:
return new MovieFormat(true, true, false);
case Music:
return new AudioTrackFormat();
default:
return new FileNameFormat();
}
}
public PreferencesEntry<String> persistentSample() {
return Settings.forPackage(FormatDialog.class).entry("format.sample." + key());
}
public PreferencesList<String> persistentFormatHistory() {
return Settings.forPackage(FormatDialog.class).node("format.recent." + key()).asList();
}
public Object getDefaultSampleObject() {
try {
ResourceBundle bundle = ResourceBundle.getBundle(FormatDialog.class.getName());
String json = bundle.getString(key() + ".sample");
return MetaAttributes.toObject(json);
} catch (MissingResourceException e) {
// ignore
} catch (Exception e) {
debug.log(Level.SEVERE, "Illegal Sample", e);
}
return null;
}
public String getDefaultFormatExpression() {
return getSampleExpressions().iterator().next();
}
public Iterable<String> getSampleExpressions() {
ResourceBundle bundle = ResourceBundle.getBundle(FormatDialog.class.getName());
Map<String, String> examples = new TreeMap<String, String>();
// extract all example entries and sort by key
String prefix = key() + ".example";
for (String key : bundle.keySet()) {
if (key.startsWith(prefix))
examples.put(key, bundle.getString(key));
}
return examples.values();
}
public static Mode getMode(Datasource datasource) {
if (datasource instanceof MovieIdentificationService)
return Mode.Movie;
if (datasource instanceof EpisodeListProvider)
return Mode.Episode;
if (datasource instanceof MusicIdentificationService)
return Mode.Music;
return Mode.File;
}
}
public FormatDialog(Window owner, Mode initMode, MediaBindingBean bindings, boolean locked) {
super(owner, ModalityType.DOCUMENT_MODAL);
// initialize hidden
progressIndicator.setVisible(false);
// bold title label in header
title.setFont(title.getFont().deriveFont(BOLD));
JPanel header = new JPanel(new MigLayout("insets dialog, nogrid, novisualpadding"));
header.setBackground(getPanelBackground());
header.setBorder(getSeparatorBorder(Position.BOTTOM));
header.add(progressIndicator, "pos 1al 0al, hidemode 3");
header.add(title, "wmin 150px, wrap unrel:push");
header.add(preview, "wmin 150px, hmin 16px, gap indent, hidemode 3, wmax 90%");
header.add(status, "wmin 150px, hmin 16px, gap indent, hidemode 3, wmax 90%, newline");
JPanel content = new JPanel(new MigLayout("insets dialog, nogrid, fill, hidemode 1"));
RTextScrollPane editorScrollPane = new RTextScrollPane(editor, false);
editorScrollPane.setLineNumbersEnabled(false);
editorScrollPane.setFoldIndicatorEnabled(false);
editorScrollPane.setIconRowHeaderEnabled(false);
editorScrollPane.setVerticalScrollBarPolicy(RTextScrollPane.VERTICAL_SCROLLBAR_NEVER);
editorScrollPane.setHorizontalScrollBarPolicy(RTextScrollPane.HORIZONTAL_SCROLLBAR_NEVER);
editorScrollPane.setViewportBorder(createEmptyBorder(7, 2, 7, 2));
editorScrollPane.setOpaque(true);
// ensure that the editor area looks like a simple text field
JTextField template = new JTextField();
editorScrollPane.setBorder(template.getBorder());
editorScrollPane.setBackground(template.getBackground());
editor.setBackground(template.getBackground());
content.add(editorScrollPane, "w 120px:min(pref, 420px), h pref, growx, wrap 4px, id editor");
content.add(createImageButton(changeSampleAction), "sg action, w 28!, h 24!, pos n editor.y2+2 editor.x2 n");
content.add(createImageButton(selectFolderAction), "sg action, w 28!, h 24!, pos n editor.y2+2 editor.x2-(30*1) n");
content.add(createImageButton(showRecentAction), "sg action, w 28!, h 24!, pos n editor.y2+2 editor.x2-(30*2) n");
content.add(help, "growx, wrap 25px:push");
if (bindings == null) {
content.add(new JButton(switchEditModeAction), "tag left");
}
content.add(new JButton(approveFormatAction), "tag apply");
content.add(new JButton(cancelAction), "tag cancel");
JComponent pane = (JComponent) getContentPane();
pane.setLayout(new MigLayout("insets 0, fill"));
pane.add(header, "h 60px, growx, dock north");
pane.add(content, "grow");
// copy text to clipboard when clicked
preview.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));
preview.addMouseListener(mouseClicked(evt -> {
copyToClipboard(preview.getText());
log.info("Format value has been copied to clipboard");
}));
status.setCursor(Cursor.getPredefinedCursor(Cursor.HAND_CURSOR));
status.addMouseListener(mouseClicked(evt -> {
copyToClipboard(status.getText());
log.info("Error message has been copied to clipboard");
}));
// update format on change
editor.onChange(evt -> {
checkFormatInBackground();
});
// update preview if sample has changed
addPropertyChangeListener("sample", evt -> {
if (isMacSandbox()) {
if (sample != null && sample.getFileObject() != null && sample.getFileObject().exists()) {
MacAppUtilities.askUnlockFolders(getWindow(evt.getSource()), singleton(sample.getFileObject()));
}
}
checkFormatInBackground();
});
// focus editor by default and finish dialog and close window manually
addWindowListener(new WindowAdapter() {
@Override
public void windowGainedFocus(WindowEvent e) {
SwingUtilities.invokeLater(editor::requestFocusInWindow);
}
@Override
public void windowActivated(WindowEvent e) {
SwingUtilities.invokeLater(() -> revalidate());
}
@Override
public void windowClosing(WindowEvent e) {
SwingUtilities.invokeLater(() -> finish(false));
}
});
// install editor suggestions popup
editor.setComponentPopupMenu(createRecentFormatPopup());
// initialize window properties
setDefaultCloseOperation(DO_NOTHING_ON_CLOSE);
setMinimumSize(new Dimension(650, 520));
// restore sample file if necessary
if (bindings == null) {
bindings = restoreSample(initMode);
} else if (bindings.getFileObject() == null && !locked) {
bindings = new MediaBindingBean(bindings.getInfoObject(), restoreSample(initMode).getFileObject());
}
// install show / hide help keyboard shortcut
installAction(this.getRootPane(), getKeyStroke(VK_F1, 0), newAction("Help", evt -> help.setVisible(!help.isVisible())));
// initialize data
setState(initMode, bindings, locked);
}
public void setState(Mode mode, MediaBindingBean bindings, boolean locked) {
this.mode = mode;
this.locked = locked;
if (locked) {
this.setTitle(String.format("%s Format", mode));
title.setText(String.format("%s ⇔ %s", mode, bindings.getInfoObject(), bindings.getFileObject() == null ? null : bindings.getFileObject().getName()));
} else {
this.setTitle(String.format("%s Format", mode));
title.setText(String.format("%s Format", mode));
}
status.setVisible(false);
switchEditModeAction.putValue(Action.NAME, String.format("Switch to %s Format", mode.next()));
switchEditModeAction.setEnabled(!locked);
updateHelpPanel(mode);
// update preview to current format
sample = bindings;
// restore editor state
setFormatCode(mode.persistentFormatHistory().isEmpty() ? mode.getDefaultFormatExpression() : mode.persistentFormatHistory().get(0));
// update examples
fireSampleChanged();
}
private JComponent updateHelpPanel(Mode mode) {
help.removeAll();
if (help.isVisible()) {
help.add(new JLabel("Syntax"), "gap indent+unrel, wrap 0");
help.add(createSyntaxPanel(mode), "gapx indent indent, wrap 8px");
help.add(new JLabel("Examples"), "gap indent+unrel, wrap 0");
help.add(createExamplesPanel(mode), "growx, h pref!, gapx indent indent");
}
return help;
}
public void setFormatCode(String text) {
try {
// update format code
editor.setText(text);
// scroll to last character and focus
if (text != null && text.length() > 0) {
editor.scrollRectToVisible(new Rectangle(0, 0)); // reset scroll
editor.setCaretPosition(editor.getText().length()); // scroll to end of format
editor.requestFocusInWindow();
}
} catch (Exception e) {
debug.warning(e::toString);
}
}
private JComponent createSyntaxPanel(Mode mode) {
JPanel panel = new JPanel(new MigLayout("fill, nogrid, novisualpadding", "[pref]", "[fill, min]"));
panel.setBorder(getHelpPanelBorder());
panel.setBackground(getHelpPanelBackground());
panel.setOpaque(true);
panel.add(new LinkButton(newAction(ResourceBundle.getBundle(FormatDialog.class.getName()).getString(mode.key() + ".syntax"), evt -> {
openURI(ResourceBundle.getBundle(FormatDialog.class.getName()).getString("help.url"));
})), "h min!");
return panel;
}
private JComponent createExamplesPanel(Mode mode) {
JPanel panel = new JPanel(new MigLayout("fill, wrap 3"));
panel.setBorder(getHelpPanelBorder());
panel.setBackground(getHelpPanelBackground());
panel.setOpaque(true);
for (String format : mode.getSampleExpressions()) {
LinkButton formatLink = new LinkButton(newAction(format, e -> setFormatCode(format)));
formatLink.setFont(new Font(MONOSPACED, PLAIN, 11));
// compute format label in background
JLabel formatExample = new JLabel("[evaluate]");
// bind text to preview
addPropertyChangeListener("sample", evt -> {
newSwingWorker(() -> {
return new ExpressionFileFormat(format).format(sample);
}, s -> {
formatExample.setText(s);
}).execute();
});
panel.add(formatLink);
panel.add(new JLabel(""));
panel.add(formatExample, "wmin 150px");
}
return panel;
}
protected MediaBindingBean restoreSample(Mode mode) {
Object info = null;
File media = null;
Map<File, ?> context = null;
try {
// restore sample from user preferences
info = MetaAttributes.toObject(mode.persistentSample().getValue());
} catch (Exception e) {
debug.log(Level.WARNING, e, e::toString);
}
// restore sample from application properties if necessary
if (info == null) {
info = mode.getDefaultSampleObject();
}
// restore media file
String path = persistentSampleFile.getValue();
if (path != null && !path.isEmpty()) {
media = new File(path);
context = singletonMap(media, info);
}
return new MediaBindingBean(info, media, context);
}
private ExecutorService createExecutor() {
// only keep the latest task in the queue
ThreadPoolExecutor executor = new ThreadPoolExecutor(1, 1, 0L, TimeUnit.SECONDS, new LinkedBlockingQueue<Runnable>(1), new DefaultThreadFactory("PreviewFormatter"));
executor.setRejectedExecutionHandler(new DiscardOldestPolicy());
return executor;
}
private void checkFormatInBackground() {
try {
// check syntax in foreground
ExpressionFileFormat format = new ExpressionFileFormat(editor.getText().trim());
// activate delayed to avoid flickering when formatting takes only a couple of milliseconds
Timer progressIndicatorTimer = invokeLater(400, () -> progressIndicator.setVisible(true));
// cancel old worker later
Future<String> obsoletePreviewFuture = currentPreviewFuture;
// create new worker
currentPreviewFuture = new SwingWorker<String, Void>() {
@Override
protected String doInBackground() throws Exception {
return format.format(sample);
}
@Override
protected void done() {
try {
preview.setText(get());
// check internal script exception
if (format.suppressed() != null) {
throw format.suppressed();
}
// no warning or error
status.setText(null);
status.setVisible(false);
} catch (CancellationException e) {
// ignore, cancelled tasks are obsolete anyway
} catch (Exception execution) {
SuppressedThrowables suppressed = findCause(execution, SuppressedThrowables.class);
if (suppressed != null) {
BindingException bindingIssue = findCause(suppressed, BindingException.class);
if (bindingIssue != null && getMessage(bindingIssue).contains(MediaBindingBean.EXCEPTION_SAMPLE_FILE_NOT_SET)) {
// exception caused by file bindings because sample file has not been set
status.setText(getMessage(bindingIssue));
status.setIcon(ResourceManager.getIcon("action.variables"));
} else if (bindingIssue != null) {
// default binding exception handler
status.setText(getMessage(bindingIssue));
status.setIcon(ResourceManager.getIcon("status.info"));
} else if (suppressed.getCause() != null && suppressed.getCause().getClass() == Exception.class) {
// ScriptShellMethods throws Exception type exceptions which are not unexpected
status.setText(execution.getCause().getMessage());
status.setIcon(ResourceManager.getIcon("status.info"));
} else {
// default exception handler
status.setText(suppressed.getMessage());
status.setIcon(ResourceManager.getIcon("status.warning"));
}
} else {
// unexpected Exception
status.setText(execution.toString());
status.setIcon(ResourceManager.getIcon("status.warning"));
}
status.setVisible(true);
} finally {
preview.setVisible(preview.getText().trim().length() > 0);
editor.setForeground(preview.getForeground());
// stop progress indicator from becoming visible, if we have been fast enough
progressIndicatorTimer.stop();
// hide progress indicator, if this still is the current worker
if (this == currentPreviewFuture) {
progressIndicator.setVisible(false);
}
}
}
};
// cancel old worker, after new worker has been created, because done() might be called from within cancel()
if (obsoletePreviewFuture != null) {
obsoletePreviewFuture.cancel(true);
}
// submit new worker
executor.execute(currentPreviewFuture);
} catch (ScriptException e) {
// incorrect syntax
status.setText(getRootCauseMessage(e));
status.setIcon(ResourceManager.getIcon("status.error"));
status.setVisible(true);
preview.setVisible(false);
editor.setForeground(getErrorColor());
}
}
public boolean submit() {
return submit;
}
public Mode getMode() {
return mode;
}
public ExpressionFormat getFormat() {
return format;
}
private void finish(boolean submit) {
this.submit = submit;
// force shutdown
executor.shutdownNow();
setVisible(false);
dispose();
}
private JPopupMenu createRecentFormatPopup() {
JPopupMenu popup = new JPopupMenu();
popup.addPopupMenuListener(new PopupMenuListener() {
@Override
public void popupMenuWillBecomeVisible(PopupMenuEvent evt) {
// make sure to reset state
popupMenuWillBecomeInvisible(evt);
JPopupMenu popup = (JPopupMenu) evt.getSource();
for (String expression : mode.persistentFormatHistory()) {
JMenuItem item = popup.add(newAction(expression, e -> setFormatCode(expression)));
item.setFont(new Font(MONOSPACED, PLAIN, 11));
}
}
@Override
public void popupMenuWillBecomeInvisible(PopupMenuEvent evt) {
JPopupMenu popup = (JPopupMenu) evt.getSource();
popup.removeAll();
}
@Override
public void popupMenuCanceled(PopupMenuEvent evt) {
popupMenuWillBecomeInvisible(evt);
}
});
return popup;
}
protected final Action changeSampleAction = newAction("Change Sample", ResourceManager.getIcon("action.variables"), evt -> {
BindingDialog dialog = new BindingDialog(getWindow(evt.getSource()), String.format("%s Bindings", mode), mode.getFormat(), !locked);
dialog.setSample(sample);
// open dialog
dialog.setLocationRelativeTo((Component) evt.getSource());
dialog.setVisible(true);
if (dialog.submit()) {
Object info = dialog.getInfoObject();
File file = dialog.getMediaFile();
// change sample
sample = new MediaBindingBean(info, file);
// remember sample
try {
mode.persistentSample().setValue(info == null ? "" : MetaAttributes.toJson(info));
persistentSampleFile.setValue(file == null ? "" : sample.getFileObject().getAbsolutePath());
} catch (Exception e) {
debug.log(Level.WARNING, e.getMessage(), e);
}
// reevaluate everything
fireSampleChanged();
}
});
protected final Action selectFolderAction = newAction("Change Folder", ResourceManager.getIcon("action.load"), evt -> {
String relativeFormat = editor.getText().trim();
File absoluteFolder = null;
if (relativeFormat.length() > 0) {
File templatePath = new File(relativeFormat);
if (templatePath.isAbsolute()) {
File existingPath = null;
for (File next : listPath(templatePath)) {
if (existingPath != null && !next.exists()) {
absoluteFolder = existingPath;
relativeFormat = relativeFormat.substring(existingPath.getPath().length() + 1); // account for file separator
break;
}
existingPath = next;
}
}
}
File selectedFolder = UserFiles.showOpenDialogSelectFolder(absoluteFolder, "Select Folder", evt);
if (selectedFolder != null) {
editor.setText(normalizePathSeparators(selectedFolder.getAbsolutePath()) + "/" + relativeFormat);
}
});
protected final Action showRecentAction = newAction("Change Format", ResourceManager.getIcon("action.menu"), evt -> {
// display popup below format editor
JComponent c = (JComponent) evt.getSource();
editor.getComponentPopupMenu().show(c, 0, c.getHeight() + 3);
});
protected final Action cancelAction = newAction("Cancel", ResourceManager.getIcon("dialog.cancel"), evt -> {
finish(false);
});
protected final Action switchEditModeAction = newAction("Switch Mode", ResourceManager.getIcon("dialog.switch"), evt -> {
Mode next = mode.next();
setState(next, restoreSample(next), false);
});
protected final Action approveFormatAction = newAction("Use Format", ResourceManager.getIcon("dialog.continue"), evt -> {
try {
// check syntax
format = new ExpressionFileFormat(editor.getText().trim());
if (format.getExpression().isEmpty()) {
throw new ScriptException("Expression is empty");
}
if (format.getExpression().length() > Preferences.MAX_VALUE_LENGTH) {
throw new ScriptException("Expression is limited to " + Preferences.MAX_VALUE_LENGTH + " characters");
}
// create new recent history and ignore duplicates
Set<String> recent = new LinkedHashSet<String>();
// remember the 8 most recent formats
recent.add(format.getExpression());
recent.addAll(mode.persistentFormatHistory());
// update persistent history
mode.persistentFormatHistory().set(recent.stream().filter(Objects::nonNull).limit(8).collect(toList()));
finish(true);
} catch (ScriptException e) {
log.log(Level.WARNING, getRootCauseMessage(e));
} catch (Exception e) {
log.log(Level.WARNING, e.getMessage(), e);
}
});
protected void fireSampleChanged() {
firePropertyChange("sample", null, sample);
}
}