419 lines
15 KiB
Java
419 lines
15 KiB
Java
|
|
package net.sourceforge.filebot.ui.rename;
|
|
|
|
|
|
import static javax.swing.JOptionPane.*;
|
|
import static javax.swing.SwingUtilities.*;
|
|
import static net.sourceforge.filebot.ui.NotificationLogging.*;
|
|
import static net.sourceforge.tuned.ui.LoadingOverlayPane.*;
|
|
import static net.sourceforge.tuned.ui.TunedUtilities.*;
|
|
|
|
import java.awt.Component;
|
|
import java.awt.Insets;
|
|
import java.awt.event.ActionEvent;
|
|
import java.beans.PropertyChangeEvent;
|
|
import java.beans.PropertyChangeListener;
|
|
import java.io.File;
|
|
import java.util.ArrayList;
|
|
import java.util.LinkedList;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.logging.Level;
|
|
|
|
import javax.swing.AbstractAction;
|
|
import javax.swing.Action;
|
|
import javax.swing.DefaultListCellRenderer;
|
|
import javax.swing.Icon;
|
|
import javax.swing.JButton;
|
|
import javax.swing.JComboBox;
|
|
import javax.swing.JComponent;
|
|
import javax.swing.JLabel;
|
|
import javax.swing.JList;
|
|
import javax.swing.JOptionPane;
|
|
import javax.swing.JPanel;
|
|
import javax.swing.JScrollPane;
|
|
import javax.swing.SwingConstants;
|
|
import javax.swing.SwingWorker;
|
|
import javax.swing.border.CompoundBorder;
|
|
import javax.swing.border.TitledBorder;
|
|
|
|
import ca.odell.glazedlists.ListSelection;
|
|
import ca.odell.glazedlists.swing.EventSelectionModel;
|
|
|
|
import net.miginfocom.swing.MigLayout;
|
|
import net.sourceforge.filebot.ResourceManager;
|
|
import net.sourceforge.filebot.Settings;
|
|
import net.sourceforge.filebot.WebServices;
|
|
import net.sourceforge.filebot.similarity.Match;
|
|
import net.sourceforge.filebot.ui.Language;
|
|
import net.sourceforge.filebot.ui.rename.RenameModel.FormattedFuture;
|
|
import net.sourceforge.filebot.web.Episode;
|
|
import net.sourceforge.filebot.web.EpisodeFormat;
|
|
import net.sourceforge.filebot.web.EpisodeListProvider;
|
|
import net.sourceforge.filebot.web.Movie;
|
|
import net.sourceforge.filebot.web.MovieFormat;
|
|
import net.sourceforge.filebot.web.MovieIdentificationService;
|
|
import net.sourceforge.filebot.web.SortOrder;
|
|
import net.sourceforge.tuned.ExceptionUtilities;
|
|
import net.sourceforge.tuned.PreferencesMap.PreferencesEntry;
|
|
import net.sourceforge.tuned.ui.ActionPopup;
|
|
import net.sourceforge.tuned.ui.LoadingOverlayPane;
|
|
|
|
|
|
public class RenamePanel extends JComponent {
|
|
|
|
protected final RenameModel renameModel = new RenameModel();
|
|
|
|
protected final RenameList<FormattedFuture> namesList = new RenameList<FormattedFuture>(renameModel.names());
|
|
|
|
protected final RenameList<File> filesList = new RenameList<File>(renameModel.files());
|
|
|
|
protected final MatchAction matchAction = new MatchAction(renameModel);
|
|
|
|
protected final RenameAction renameAction = new RenameAction(renameModel);
|
|
|
|
private static final PreferencesEntry<String> persistentPreserveExtension = Settings.forPackage(RenamePanel.class).entry("rename.extension.preserve").defaultValue("true");
|
|
private static final PreferencesEntry<String> persistentEpisodeFormat = Settings.forPackage(RenamePanel.class).entry("rename.format.episode");
|
|
private static final PreferencesEntry<String> persistentMovieFormat = Settings.forPackage(RenamePanel.class).entry("rename.format.movie");
|
|
private static final PreferencesEntry<String> persistentPreferredLanguage = Settings.forPackage(RenamePanel.class).entry("rename.language").defaultValue("en");
|
|
private static final PreferencesEntry<String> persistentPreferredEpisodeOrder = Settings.forPackage(RenamePanel.class).entry("rename.episode.order").defaultValue("Airdate");
|
|
|
|
|
|
public RenamePanel() {
|
|
namesList.setTitle("New Names");
|
|
namesList.setTransferablePolicy(new NamesListTransferablePolicy(renameModel.values()));
|
|
|
|
filesList.setTitle("Original Files");
|
|
filesList.setTransferablePolicy(new FilesListTransferablePolicy(renameModel.files()));
|
|
|
|
// restore state
|
|
renameModel.setPreserveExtension(Boolean.parseBoolean(persistentPreserveExtension.getValue()));
|
|
|
|
// filename formatter
|
|
renameModel.useFormatter(File.class, new FileNameFormatter(renameModel.preserveExtension()));
|
|
|
|
// movie formatter
|
|
renameModel.useFormatter(Movie.class, new MovieFormatter());
|
|
|
|
try {
|
|
// restore custom episode formatter
|
|
renameModel.useFormatter(Episode.class, new ExpressionFormatter(persistentEpisodeFormat.getValue(), EpisodeFormat.SeasonEpisode, Episode.class));
|
|
} catch (Exception e) {
|
|
// illegal format, ignore
|
|
}
|
|
|
|
try {
|
|
// restore custom movie formatter
|
|
renameModel.useFormatter(Movie.class, new ExpressionFormatter(persistentMovieFormat.getValue(), MovieFormat.NameYear, Movie.class));
|
|
} catch (Exception e) {
|
|
// illegal format, ignore
|
|
}
|
|
|
|
RenameListCellRenderer cellrenderer = new RenameListCellRenderer(renameModel);
|
|
|
|
namesList.getListComponent().setCellRenderer(cellrenderer);
|
|
filesList.getListComponent().setCellRenderer(cellrenderer);
|
|
|
|
EventSelectionModel<Match<Object, File>> selectionModel = new EventSelectionModel<Match<Object, File>>(renameModel.matches());
|
|
selectionModel.setSelectionMode(ListSelection.SINGLE_SELECTION);
|
|
|
|
// use the same selection model for both lists to synchronize selection
|
|
namesList.getListComponent().setSelectionModel(selectionModel);
|
|
filesList.getListComponent().setSelectionModel(selectionModel);
|
|
|
|
// synchronize viewports
|
|
new ScrollPaneSynchronizer(namesList, filesList);
|
|
|
|
// create Match button
|
|
JButton matchButton = new JButton(matchAction);
|
|
matchButton.setVerticalTextPosition(SwingConstants.BOTTOM);
|
|
matchButton.setHorizontalTextPosition(SwingConstants.CENTER);
|
|
|
|
// create Rename button
|
|
JButton renameButton = new JButton(renameAction);
|
|
renameButton.setVerticalTextPosition(SwingConstants.BOTTOM);
|
|
renameButton.setHorizontalTextPosition(SwingConstants.CENTER);
|
|
|
|
// create fetch popup
|
|
ActionPopup fetchPopup = createFetchPopup();
|
|
|
|
namesList.getListComponent().setComponentPopupMenu(fetchPopup);
|
|
matchButton.setComponentPopupMenu(fetchPopup);
|
|
matchButton.addActionListener(showPopupAction);
|
|
|
|
// create settings popup
|
|
renameButton.setComponentPopupMenu(createSettingsPopup());
|
|
|
|
setLayout(new MigLayout("fill, insets dialog, gapx 10px", "[fill][align center, pref!][fill]", "align 33%"));
|
|
|
|
add(filesList, "grow, sizegroupx list");
|
|
|
|
// make buttons larger
|
|
matchButton.setMargin(new Insets(3, 14, 2, 14));
|
|
renameButton.setMargin(new Insets(6, 11, 2, 11));
|
|
|
|
add(matchButton, "split 2, flowy, sizegroupx button");
|
|
add(renameButton, "gapy 30px, sizegroupx button");
|
|
|
|
add(new LoadingOverlayPane(namesList, namesList, "32px", "30px"), "grow, sizegroupx list");
|
|
}
|
|
|
|
|
|
protected ActionPopup createFetchPopup() {
|
|
final ActionPopup actionPopup = new ActionPopup("Fetch Episode List", ResourceManager.getIcon("action.fetch"));
|
|
|
|
// create actions for match popup episode list completion
|
|
for (EpisodeListProvider provider : WebServices.getEpisodeListProviders()) {
|
|
actionPopup.add(new AutoCompleteAction(provider.getName(), provider.getIcon(), new EpisodeListMatcher(provider)));
|
|
}
|
|
|
|
actionPopup.addSeparator();
|
|
actionPopup.addDescription(new JLabel("Movie Mode:"));
|
|
|
|
// create action for movie name completion
|
|
for (MovieIdentificationService it : WebServices.getMovieIdentificationServices()) {
|
|
actionPopup.add(new AutoCompleteAction(it.getName(), it.getIcon(), new MovieHashMatcher(it)));
|
|
}
|
|
|
|
actionPopup.addSeparator();
|
|
actionPopup.addDescription(new JLabel("Options:"));
|
|
|
|
actionPopup.add(new AbstractAction("Edit Format", ResourceManager.getIcon("action.format")) {
|
|
|
|
@Override
|
|
public void actionPerformed(ActionEvent evt) {
|
|
FormatDialog dialog = new FormatDialog(getWindowAncestor(RenamePanel.this));
|
|
dialog.setLocation(getOffsetLocation(dialog.getOwner()));
|
|
dialog.setVisible(true);
|
|
|
|
if (dialog.submit()) {
|
|
switch (dialog.getMode()) {
|
|
case Episode:
|
|
renameModel.useFormatter(Episode.class, new ExpressionFormatter(dialog.getFormat().getExpression(), EpisodeFormat.SeasonEpisode, Episode.class));
|
|
persistentEpisodeFormat.setValue(dialog.getFormat().getExpression());
|
|
break;
|
|
case Movie:
|
|
renameModel.useFormatter(Movie.class, new ExpressionFormatter(dialog.getFormat().getExpression(), MovieFormat.NameYear, Movie.class));
|
|
persistentMovieFormat.setValue(dialog.getFormat().getExpression());
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
});
|
|
|
|
actionPopup.add(new AbstractAction("Preferences", ResourceManager.getIcon("action.preferences")) {
|
|
|
|
@Override
|
|
public void actionPerformed(ActionEvent evt) {
|
|
List<Language> languages = new ArrayList<Language>();
|
|
languages.addAll(Language.preferredLanguages()); // add preferred languages first
|
|
languages.addAll(Language.availableLanguages()); // then others
|
|
|
|
JComboBox orderCombo = new JComboBox(SortOrder.values());
|
|
JList languageList = new JList(languages.toArray());
|
|
languageList.setCellRenderer(new DefaultListCellRenderer() {
|
|
|
|
@Override
|
|
public Component getListCellRendererComponent(JList list, Object value, int index, boolean isSelected, boolean cellHasFocus) {
|
|
super.getListCellRendererComponent(list, value, index, isSelected, cellHasFocus);
|
|
setIcon(ResourceManager.getFlagIcon(((Language) value).getCode()));
|
|
return this;
|
|
}
|
|
});
|
|
|
|
// pre-select current preferences
|
|
try {
|
|
orderCombo.setSelectedItem(SortOrder.forName(persistentPreferredEpisodeOrder.getValue()));
|
|
} catch (IllegalArgumentException e) {
|
|
// ignore
|
|
}
|
|
for (Language language : languages) {
|
|
if (language.getCode().equals(persistentPreferredLanguage.getValue())) {
|
|
languageList.setSelectedValue(language, true);
|
|
break;
|
|
}
|
|
}
|
|
|
|
JScrollPane spLanguageList = new JScrollPane(languageList);
|
|
spLanguageList.setBorder(new CompoundBorder(new TitledBorder("Preferred Language"), spLanguageList.getBorder()));
|
|
JScrollPane spOrderCombo = new JScrollPane(orderCombo, JScrollPane.VERTICAL_SCROLLBAR_NEVER, JScrollPane.HORIZONTAL_SCROLLBAR_NEVER);
|
|
spOrderCombo.setBorder(new CompoundBorder(new TitledBorder("Preferred Episode Order"), spOrderCombo.getBorder()));
|
|
|
|
JPanel message = new JPanel(new MigLayout("fill, flowy, insets 0"));
|
|
message.add(spLanguageList, "grow");
|
|
message.add(spOrderCombo, "grow, hmin 24px");
|
|
JOptionPane pane = new JOptionPane(message, PLAIN_MESSAGE, OK_CANCEL_OPTION);
|
|
pane.createDialog(getWindowAncestor(RenamePanel.this), "Preferences").setVisible(true);
|
|
|
|
if (pane.getValue() != null && pane.getValue().equals(OK_OPTION)) {
|
|
persistentPreferredLanguage.setValue(((Language) languageList.getSelectedValue()).getCode());
|
|
persistentPreferredEpisodeOrder.setValue(((SortOrder) orderCombo.getSelectedItem()).name());
|
|
}
|
|
}
|
|
});
|
|
|
|
return actionPopup;
|
|
}
|
|
|
|
|
|
protected ActionPopup createSettingsPopup() {
|
|
ActionPopup actionPopup = new ActionPopup("Rename Options", ResourceManager.getIcon("action.rename.small"));
|
|
|
|
actionPopup.addDescription(new JLabel("Extension:"));
|
|
|
|
actionPopup.add(new PreserveExtensionAction(true, "Preserve", ResourceManager.getIcon("action.extension.preserve")));
|
|
actionPopup.add(new PreserveExtensionAction(false, "Override", ResourceManager.getIcon("action.extension.override")));
|
|
|
|
actionPopup.addSeparator();
|
|
actionPopup.addDescription(new JLabel("History:"));
|
|
|
|
actionPopup.add(new AbstractAction("Open History", ResourceManager.getIcon("action.report")) {
|
|
|
|
@Override
|
|
public void actionPerformed(ActionEvent e) {
|
|
History model = HistorySpooler.getInstance().getCompleteHistory();
|
|
|
|
HistoryDialog dialog = new HistoryDialog(getWindow(RenamePanel.this));
|
|
dialog.setLocationRelativeTo(RenamePanel.this);
|
|
dialog.setModel(model);
|
|
|
|
// show and block
|
|
dialog.setVisible(true);
|
|
|
|
if (!model.equals(dialog.getModel())) {
|
|
// model was changed, switch to the new model
|
|
HistorySpooler.getInstance().commit(dialog.getModel());
|
|
}
|
|
}
|
|
});
|
|
|
|
return actionPopup;
|
|
}
|
|
|
|
|
|
protected final Action showPopupAction = new AbstractAction("Show Popup") {
|
|
|
|
@Override
|
|
public void actionPerformed(ActionEvent e) {
|
|
// show popup on actionPerformed only when names list is empty
|
|
if (renameModel.size() > 0 && !renameModel.hasComplement(0)) {
|
|
JComponent source = (JComponent) e.getSource();
|
|
|
|
// display popup below component
|
|
source.getComponentPopupMenu().show(source, -3, source.getHeight() + 4);
|
|
}
|
|
}
|
|
};
|
|
|
|
|
|
protected class PreserveExtensionAction extends AbstractAction {
|
|
|
|
private final boolean activate;
|
|
|
|
|
|
private PreserveExtensionAction(boolean activate, String name, Icon icon) {
|
|
super(name, icon);
|
|
this.activate = activate;
|
|
}
|
|
|
|
|
|
@Override
|
|
public void actionPerformed(ActionEvent evt) {
|
|
renameModel.setPreserveExtension(activate);
|
|
|
|
// use different file name formatter
|
|
renameModel.useFormatter(File.class, new FileNameFormatter(renameModel.preserveExtension()));
|
|
|
|
// display changed state
|
|
filesList.repaint();
|
|
|
|
// save state
|
|
persistentPreserveExtension.setValue(Boolean.toString(activate));
|
|
}
|
|
}
|
|
|
|
|
|
protected class AutoCompleteAction extends AbstractAction {
|
|
|
|
private final AutoCompleteMatcher matcher;
|
|
|
|
|
|
public AutoCompleteAction(String name, Icon icon, AutoCompleteMatcher matcher) {
|
|
super(name, icon);
|
|
|
|
this.matcher = matcher;
|
|
|
|
// disable action while episode list matcher is working
|
|
namesList.addPropertyChangeListener(LOADING_PROPERTY, new PropertyChangeListener() {
|
|
|
|
@Override
|
|
public void propertyChange(PropertyChangeEvent evt) {
|
|
// disable action while loading is in progress
|
|
setEnabled(!(Boolean) evt.getNewValue());
|
|
}
|
|
});
|
|
}
|
|
|
|
|
|
@Override
|
|
public void actionPerformed(final ActionEvent evt) {
|
|
if (renameModel.files().isEmpty()) {
|
|
return;
|
|
}
|
|
|
|
// auto-match in progress
|
|
namesList.firePropertyChange(LOADING_PROPERTY, false, true);
|
|
|
|
// clear names list
|
|
renameModel.values().clear();
|
|
|
|
SwingWorker<List<Match<File, ?>>, Void> worker = new SwingWorker<List<Match<File, ?>>, Void>() {
|
|
|
|
private final List<File> remainingFiles = new LinkedList<File>(renameModel.files());
|
|
private final SortOrder order = SortOrder.forName(persistentPreferredEpisodeOrder.getValue());
|
|
private final Locale locale = new Locale(persistentPreferredLanguage.getValue());
|
|
private final boolean autodetection = !isShiftDown(evt); // skip name auto-detection if SHIFT is pressed
|
|
|
|
|
|
@Override
|
|
protected List<Match<File, ?>> doInBackground() throws Exception {
|
|
List<Match<File, ?>> matches = matcher.match(remainingFiles, order, locale, autodetection, getWindow(RenamePanel.this));
|
|
|
|
// remove matched files
|
|
for (Match<File, ?> match : matches) {
|
|
remainingFiles.remove(match.getValue());
|
|
}
|
|
|
|
return matches;
|
|
}
|
|
|
|
|
|
@Override
|
|
protected void done() {
|
|
try {
|
|
List<Match<Object, File>> matches = new ArrayList<Match<Object, File>>();
|
|
|
|
for (Match<File, ?> match : get()) {
|
|
matches.add(new Match<Object, File>(match.getCandidate(), match.getValue()));
|
|
}
|
|
|
|
renameModel.clear();
|
|
renameModel.addAll(matches);
|
|
|
|
// add remaining file entries
|
|
renameModel.files().addAll(remainingFiles);
|
|
} catch (Exception e) {
|
|
UILogger.log(Level.WARNING, ExceptionUtilities.getRootCauseMessage(e), e);
|
|
} finally {
|
|
// auto-match finished
|
|
namesList.firePropertyChange(LOADING_PROPERTY, true, false);
|
|
}
|
|
}
|
|
};
|
|
|
|
worker.execute();
|
|
}
|
|
}
|
|
|
|
}
|