package net.filebot.ui.rename;
import static java.awt.Font.*;
import static java.util.Collections.*;
import static javax.swing.BorderFactory.*;
import static net.filebot.Logging.*;
import static net.filebot.Settings.*;
import static net.filebot.util.ui.SwingUI.*;
import java.awt.Font;
import java.awt.Window;
import java.io.File;
import java.util.List;
import java.util.logging.Level;
import javax.swing.Action;
import javax.swing.ButtonGroup;
import javax.swing.JButton;
import javax.swing.JComboBox;
import javax.swing.JComponent;
import javax.swing.JDialog;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JPopupMenu;
import javax.swing.JRadioButton;
import javax.swing.JTextField;
import javax.swing.ListCellRenderer;
import org.fife.ui.rsyntaxtextarea.RSyntaxDocument;
import org.fife.ui.rsyntaxtextarea.RSyntaxTextArea;
import org.fife.ui.rsyntaxtextarea.SyntaxConstants;
import org.fife.ui.rtextarea.RTextScrollPane;
import net.filebot.Language;
import net.filebot.RenameAction;
import net.filebot.ResourceManager;
import net.filebot.StandardRenameAction;
import net.filebot.UserFiles;
import net.filebot.WebServices;
import net.filebot.format.ExpressionFileFormat;
import net.filebot.format.ExpressionFilter;
import net.filebot.format.MediaBindingBean;
import net.filebot.platform.mac.MacAppUtilities;
import net.filebot.ui.HeaderPanel;
import net.filebot.util.FileUtilities.ExtensionFileFilter;
import net.filebot.web.Datasource;
import net.filebot.web.EpisodeListProvider;
import net.filebot.web.MovieIdentificationService;
import net.filebot.web.SortOrder;
import net.miginfocom.swing.MigLayout;
public class PresetEditor extends JDialog {
enum Result {
SET, DELETE, CANCEL;
}
private Result result = Result.CANCEL;
private HeaderPanel presetNameHeader;
private JTextField pathInput;
private RSyntaxTextArea filterEditor;
private RSyntaxTextArea formatEditor;
private JComboBox<RenameAction> actionCombo;
private JComboBox<Datasource> providerCombo;
private JComboBox<SortOrder> sortOrderCombo;
private JComboBox<String> matchModeCombo;
private JComboBox<Language> languageCombo;
private JRadioButton selectRadio;
private JRadioButton inheritRadio;
private JPanel inputPanel;
public PresetEditor(Window owner) {
super(owner, "Preset Editor", ModalityType.APPLICATION_MODAL);
JComponent c = (JComponent) getContentPane();
presetNameHeader = new HeaderPanel();
inheritRadio = new JRadioButton("<html>Use <b>Original Files</b> selection</html>");
selectRadio = new JRadioButton("<html>Do <b>Select</b> files</html>");
pathInput = new JTextField(40);
filterEditor = createEditor();
formatEditor = createEditor();
actionCombo = createRenameActionCombo();
providerCombo = createDataProviderCombo();
sortOrderCombo = new JComboBox<SortOrder>(SortOrder.values());
matchModeCombo = new JComboBox<String>(Preset.getSupportedMatchModes());
languageCombo = createLanguageCombo();
inputPanel = new JPanel(new MigLayout("insets 0, fill"));
inputPanel.setOpaque(false);
inputPanel.add(new JLabel("Input Folder:"), "gap indent");
inputPanel.add(pathInput, "growx, gap rel");
inputPanel.add(createImageButton(selectInputFolder), "gap 0px, wrap");
inputPanel.add(new JLabel("Includes:"), "gap indent, skip 1, split 3");
inputPanel.add(wrapEditor(filterEditor), "growx, gap rel");
inputPanel.add(createImageButton(listFiles), "gap rel");
JPanel inputGroup = createGroupPanel("Files");
inputGroup.add(selectRadio);
inputGroup.add(inheritRadio, "wrap");
inputGroup.add(inputPanel);
JPanel formatGroup = createGroupPanel("Format");
formatGroup.add(new JLabel("Expression:"));
formatGroup.add(wrapEditor(formatEditor), "growx, gap rel");
formatGroup.add(createImageButton(editFormatExpression), "gap 10px");
JPanel searchGroup = createGroupPanel("Options");
searchGroup.add(new JLabel("Datasource:"), "sg label");
searchGroup.add(providerCombo, "sg combo");
searchGroup.add(new JLabel("Episode Order:"), "sg label, gap indent");
searchGroup.add(sortOrderCombo, "sg combo, wrap");
searchGroup.add(new JLabel("Language:"), "sg label");
searchGroup.add(languageCombo, "sg combo");
searchGroup.add(new JLabel("Match Mode:"), "sg label, gap indent");
searchGroup.add(matchModeCombo, "sg combo, wrap");
searchGroup.add(new JLabel("Rename Action:"), "sg label");
searchGroup.add(actionCombo, "sg combo, wrap");
c.setLayout(new MigLayout("insets dialog, hidemode 3, nogrid, fill"));
c.add(presetNameHeader, "wmin 150px, hmin 75px, growx, dock north");
c.add(inputGroup, "growx, wrap");
c.add(formatGroup, "growx, wrap");
c.add(searchGroup, "growx, wrap push");
c.add(new JButton(ok), "tag apply");
c.add(new JButton(delete), "tag cancel");
ButtonGroup inputButtonGroup = new ButtonGroup();
inputButtonGroup.add(inheritRadio);
inputButtonGroup.add(selectRadio);
inheritRadio.setSelected(true);
selectRadio.addItemListener((evt) -> updateComponentStates());
providerCombo.addItemListener((evt) -> updateComponentStates());
updateComponentStates();
setSize(730, 570);
// add helpful tooltips
filterEditor.setToolTipText(FILE_FILTER_TOOLTIP);
}
public void updateComponentStates() {
Datasource ds = (Datasource) providerCombo.getSelectedItem();
boolean input = selectRadio.isSelected();
inputPanel.setVisible(input);
sortOrderCombo.setEnabled(ds instanceof EpisodeListProvider);
languageCombo.setEnabled(ds instanceof EpisodeListProvider || ds instanceof MovieIdentificationService);
matchModeCombo.setEnabled(ds instanceof EpisodeListProvider || ds instanceof MovieIdentificationService);
}
public void setPreset(Preset p) {
presetNameHeader.getTitleLabel().setText(p.getName());
pathInput.setText(p.getInputFolder() == null ? "" : p.getInputFolder().getPath());
filterEditor.setText(p.getIncludeFilter() == null ? "" : p.getIncludeFilter().getExpressionFilter().getExpression());
formatEditor.setText(p.getFormat() == null ? "" : p.getFormat().getExpression());
providerCombo.setSelectedItem(p.getDatasource() == null ? WebServices.TheTVDB : p.getDatasource());
sortOrderCombo.setSelectedItem(p.getSortOrder() == null ? SortOrder.Airdate : p.getSortOrder());
matchModeCombo.setSelectedItem(p.getMatchMode() == null ? RenamePanel.MATCH_MODE_OPPORTUNISTIC : p.getMatchMode());
actionCombo.setSelectedItem(p.getRenameAction() == null ? StandardRenameAction.MOVE : p.getRenameAction());
// ugly hack, since Language objects only do object equality
if (p.getLanguage() != null && !p.getLanguage().getCode().equals(((Language) languageCombo.getSelectedItem()).getCode())) {
for (int i = 0; i < languageCombo.getModel().getSize(); i++) {
if (p.getLanguage().getCode().equals(languageCombo.getModel().getElementAt(i).getCode())) {
languageCombo.setSelectedIndex(i);
break;
}
}
}
selectRadio.setSelected(p.getInputFolder() != null);
updateComponentStates();
}
public Preset getPreset() throws Exception {
String name = presetNameHeader.getTitleLabel().getText();
File path = inheritRadio.isSelected() ? null : new File(pathInput.getText());
ExpressionFilter includes = inheritRadio.isSelected() ? null : new ExpressionFilter(filterEditor.getText());
ExpressionFileFormat format = formatEditor.getText().trim().isEmpty() ? null : new ExpressionFileFormat(formatEditor.getText());
Datasource database = ((Datasource) providerCombo.getSelectedItem());
SortOrder sortOrder = sortOrderCombo.isEnabled() ? ((SortOrder) sortOrderCombo.getSelectedItem()) : null;
String matchMode = matchModeCombo.isEnabled() ? (String) matchModeCombo.getSelectedItem() : null;
Language language = languageCombo.isEnabled() ? ((Language) languageCombo.getSelectedItem()) : null;
StandardRenameAction action = actionCombo.isEnabled() ? (StandardRenameAction) actionCombo.getSelectedItem() : null;
return new Preset(name, path, includes, format, database, sortOrder, matchMode, language, action);
}
private JPanel createGroupPanel(String title) {
JPanel inputGroup = new JPanel(new MigLayout("insets dialog, hidemode 3, nogrid, fill"));
inputGroup.setBorder(createTitledBorder(title));
return inputGroup;
}
private RSyntaxTextArea createEditor() {
RSyntaxTextArea editor = new RSyntaxTextArea(new RSyntaxDocument(SyntaxConstants.SYNTAX_STYLE_GROOVY), "", 1, 80);
editor.setAntiAliasingEnabled(true);
editor.setAnimateBracketMatching(false);
editor.setAutoIndentEnabled(false);
editor.setClearWhitespaceLinesEnabled(false);
editor.setBracketMatchingEnabled(true);
editor.setCloseCurlyBraces(false);
editor.setCodeFoldingEnabled(false);
editor.setHyperlinksEnabled(false);
editor.setUseFocusableTips(false);
editor.setHighlightCurrentLine(false);
editor.setLineWrap(false);
editor.setPaintMarkOccurrencesBorder(false);
editor.setPaintTabLines(false);
editor.setMarkOccurrences(false);
editor.setFont(new Font(MONOSPACED, PLAIN, 14));
return editor;
}
private RTextScrollPane wrapEditor(RSyntaxTextArea editor) {
RTextScrollPane scroll = new RTextScrollPane(editor, false);
scroll.setLineNumbersEnabled(false);
scroll.setFoldIndicatorEnabled(false);
scroll.setIconRowHeaderEnabled(false);
scroll.setVerticalScrollBarPolicy(RTextScrollPane.VERTICAL_SCROLLBAR_NEVER);
scroll.setHorizontalScrollBarPolicy(RTextScrollPane.HORIZONTAL_SCROLLBAR_NEVER);
scroll.setBackground(editor.getBackground());
scroll.setOpaque(true);
scroll.setBorder(pathInput.getBorder());
return scroll;
}
private JComboBox<Datasource> createDataProviderCombo() {
JComboBox<Datasource> combo = new JComboBox<Datasource>(Preset.getSupportedServices());
ListCellRenderer<? super Datasource> renderer = combo.getRenderer();
combo.setRenderer((list, value, index, isSelected, cellHasFocus) -> {
JLabel label = (JLabel) renderer.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);
if (value instanceof Datasource) {
Datasource provider = (Datasource) value;
label.setText(provider.getName());
label.setIcon(provider.getIcon());
}
return label;
});
return combo;
}
private JComboBox<Language> createLanguageCombo() {
JComboBox<Language> combo = new JComboBox<Language>(Preset.getSupportedLanguages());
ListCellRenderer<? super Language> renderer = combo.getRenderer();
combo.setRenderer((list, value, index, isSelected, cellHasFocus) -> {
JLabel label = (JLabel) renderer.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);
if (value instanceof Language) {
Language it = value;
label.setText(it.getName());
label.setIcon(ResourceManager.getFlagIcon(it.getCode()));
}
return label;
});
return combo;
}
private JComboBox<RenameAction> createRenameActionCombo() {
JComboBox<RenameAction> combo = new JComboBox<RenameAction>(Preset.getSupportedActions());
ListCellRenderer<? super RenameAction> renderer = combo.getRenderer();
combo.setRenderer((list, value, index, isSelected, cellHasFocus) -> {
JLabel label = (JLabel) renderer.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);
if (value instanceof StandardRenameAction) {
StandardRenameAction it = (StandardRenameAction) value;
label.setText(it.getDisplayName());
label.setIcon(ResourceManager.getIcon("rename.action." + it.toString().toLowerCase()));
}
return label;
});
return combo;
}
public Result getResult() {
return result;
}
private final Action selectInputFolder = newAction("Select Input Folder", ResourceManager.getIcon("action.load"), evt -> {
File f = UserFiles.showOpenDialogSelectFolder(null, "Select Input Folder", evt);
if (f != null) {
pathInput.setText(f.getAbsolutePath());
}
});
private final Action editFormatExpression = newAction("Open Format Editor", ResourceManager.getIcon("action.format"), evt -> {
FormatDialog.Mode mode = FormatDialog.Mode.getMode((Datasource) providerCombo.getSelectedItem());
Object sample = mode.getDefaultSampleObject();
File file = null;
if (mode == FormatDialog.Mode.File) {
List<File> files = UserFiles.showLoadDialogSelectFiles(false, false, null, new ExtensionFileFilter(ExtensionFileFilter.WILDCARD), "Select Sample File", evt);
if (files.isEmpty()) {
return;
}
sample = file = files.get(0);
}
FormatDialog dialog = new FormatDialog(getWindow(evt.getSource()), mode, new MediaBindingBean(sample, file, singletonMap(file, sample)), false);
dialog.setFormatCode(formatEditor.getText());
dialog.setLocation(getOffsetLocation(dialog.getOwner()));
dialog.setVisible(true);
if (dialog.submit()) {
formatEditor.setText(dialog.getFormat().getExpression());
}
});
private final Action listFiles = newAction("List Files", ResourceManager.getIcon("action.search"), evt -> {
withWaitCursor(evt.getSource(), () -> {
try {
Preset preset = getPreset();
if (preset.getInputFolder() == null) {
return;
}
if (isMacSandbox()) {
if (!MacAppUtilities.askUnlockFolders(getWindow(evt.getSource()), singleton(preset.getInputFolder()))) {
return;
}
}
List<File> files = preset.selectFiles();
// display selected files as popup with easy access to more binding info
JPopupMenu popup = new JPopupMenu();
if (files.size() > 0) {
for (File f : files) {
popup.add(newAction(f.getPath(), e -> {
BindingDialog dialog = new BindingDialog(getWindow(evt.getSource()), "File Bindings", FormatDialog.Mode.File.getFormat(), false);
dialog.setLocation(getOffsetLocation(getWindow(evt.getSource())));
dialog.setSample(new MediaBindingBean(f, f));
dialog.setVisible(true);
}));
}
} else {
popup.add("No files selected").setEnabled(false);
}
JComponent source = (JComponent) evt.getSource();
popup.show(source, -3, source.getHeight() + 4);
} catch (Exception e) {
log.log(Level.WARNING, "Invalid preset settings: " + e.getMessage(), e);
}
});
});
private final Action ok = newAction("Save Preset", ResourceManager.getIcon("dialog.continue"), evt -> {
try {
Preset preset = getPreset();
if (preset != null) {
result = Result.SET;
setVisible(false);
}
} catch (Exception e) {
log.log(Level.WARNING, "Invalid preset settings: " + e.getMessage(), e);
}
});
private final Action delete = newAction("Delete Preset", ResourceManager.getIcon("dialog.cancel"), evt -> {
result = Result.DELETE;
setVisible(false);
});
private static final String FILE_FILTER_TOOLTIP = "<html>File Selector Expression<br><hr noshade>e.g.<br>• fn =~ /alias/<br>• ext =~ /mp4/<br>• minutes > 100<br>• age < 7<br>• file.isEpisode()<br>• …<br></html>";
}