mirror of
https://github.com/mitb-archive/filebot
synced 2024-11-13 12:55:00 -05:00
1214 lines
44 KiB
Java
1214 lines
44 KiB
Java
package net.filebot.cli;
|
|
|
|
import static java.nio.charset.StandardCharsets.*;
|
|
import static java.util.Arrays.*;
|
|
import static java.util.Collections.*;
|
|
import static java.util.stream.Collectors.*;
|
|
import static net.filebot.Logging.*;
|
|
import static net.filebot.MediaTypes.*;
|
|
import static net.filebot.Settings.*;
|
|
import static net.filebot.WebServices.*;
|
|
import static net.filebot.hash.VerificationUtilities.*;
|
|
import static net.filebot.media.MediaDetection.*;
|
|
import static net.filebot.media.XattrMetaInfo.*;
|
|
import static net.filebot.subtitle.SubtitleUtilities.*;
|
|
import static net.filebot.util.FileUtilities.*;
|
|
|
|
import java.io.File;
|
|
import java.io.FileFilter;
|
|
import java.io.IOException;
|
|
import java.nio.ByteBuffer;
|
|
import java.nio.charset.Charset;
|
|
import java.util.AbstractMap;
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.HashMap;
|
|
import java.util.HashSet;
|
|
import java.util.LinkedHashMap;
|
|
import java.util.LinkedHashSet;
|
|
import java.util.List;
|
|
import java.util.Locale;
|
|
import java.util.Map;
|
|
import java.util.Map.Entry;
|
|
import java.util.Objects;
|
|
import java.util.Set;
|
|
import java.util.SortedSet;
|
|
import java.util.TreeMap;
|
|
import java.util.TreeSet;
|
|
import java.util.logging.Level;
|
|
import java.util.stream.IntStream;
|
|
import java.util.stream.Stream;
|
|
|
|
import net.filebot.HistorySpooler;
|
|
import net.filebot.Language;
|
|
import net.filebot.RenameAction;
|
|
import net.filebot.StandardRenameAction;
|
|
import net.filebot.archive.Archive;
|
|
import net.filebot.archive.FileMapper;
|
|
import net.filebot.format.ExpressionFileFormat;
|
|
import net.filebot.format.ExpressionFilter;
|
|
import net.filebot.format.ExpressionFormat;
|
|
import net.filebot.format.MediaBindingBean;
|
|
import net.filebot.hash.HashType;
|
|
import net.filebot.hash.VerificationFileReader;
|
|
import net.filebot.hash.VerificationFileWriter;
|
|
import net.filebot.media.AutoDetection;
|
|
import net.filebot.media.AutoDetection.Group;
|
|
import net.filebot.media.AutoDetection.Type;
|
|
import net.filebot.media.LocalDatasource;
|
|
import net.filebot.media.VideoQuality;
|
|
import net.filebot.similarity.CommonSequenceMatcher;
|
|
import net.filebot.similarity.EpisodeMatcher;
|
|
import net.filebot.similarity.Match;
|
|
import net.filebot.subtitle.SubtitleFormat;
|
|
import net.filebot.subtitle.SubtitleNaming;
|
|
import net.filebot.util.EntryList;
|
|
import net.filebot.util.FileUtilities.ParentFilter;
|
|
import net.filebot.vfs.FileInfo;
|
|
import net.filebot.vfs.MemoryFile;
|
|
import net.filebot.vfs.SimpleFileInfo;
|
|
import net.filebot.web.AudioTrack;
|
|
import net.filebot.web.Datasource;
|
|
import net.filebot.web.Episode;
|
|
import net.filebot.web.EpisodeListProvider;
|
|
import net.filebot.web.Movie;
|
|
import net.filebot.web.MovieIdentificationService;
|
|
import net.filebot.web.MoviePart;
|
|
import net.filebot.web.MusicIdentificationService;
|
|
import net.filebot.web.OpenSubtitlesClient;
|
|
import net.filebot.web.SearchResult;
|
|
import net.filebot.web.SortOrder;
|
|
import net.filebot.web.SubtitleDescriptor;
|
|
import net.filebot.web.SubtitleProvider;
|
|
import net.filebot.web.VideoHashSubtitleService;
|
|
|
|
public class CmdlineOperations implements CmdlineInterface {
|
|
|
|
@Override
|
|
public List<File> rename(Collection<File> files, RenameAction action, ConflictAction conflict, File output, ExpressionFileFormat format, Datasource db, String query, SortOrder order, ExpressionFilter filter, Locale locale, boolean strict, ExecCommand exec) throws Exception {
|
|
// movie mode
|
|
if (db instanceof MovieIdentificationService) {
|
|
return renameMovie(files, action, conflict, output, format, (MovieIdentificationService) db, query, filter, locale, strict, exec);
|
|
}
|
|
|
|
// series mode
|
|
if (db instanceof EpisodeListProvider) {
|
|
return renameSeries(files, action, conflict, output, format, (EpisodeListProvider) db, query, order, filter, locale, strict, exec);
|
|
}
|
|
|
|
// music mode
|
|
if (db instanceof MusicIdentificationService) {
|
|
return renameMusic(files, action, conflict, output, format, singletonList((MusicIdentificationService) db), exec);
|
|
}
|
|
|
|
// photo / xattr / plain file mode
|
|
if (db instanceof LocalDatasource) {
|
|
return renameFiles(files, action, conflict, output, format, (LocalDatasource) db, filter, strict, exec);
|
|
}
|
|
|
|
// auto-detect mode for each fileset
|
|
AutoDetection auto = new AutoDetection(files, false, locale);
|
|
List<File> results = new ArrayList<File>();
|
|
|
|
for (Entry<Group, Set<File>> it : auto.group().entrySet()) {
|
|
if (it.getKey().types().length == 1) {
|
|
for (Type key : it.getKey().types()) {
|
|
switch (key) {
|
|
case Movie:
|
|
results.addAll(renameMovie(it.getValue(), action, conflict, output, format, TheMovieDB, query, filter, locale, strict, exec));
|
|
break;
|
|
case Series:
|
|
results.addAll(renameSeries(it.getValue(), action, conflict, output, format, TheTVDB, query, order, filter, locale, strict, exec));
|
|
break;
|
|
case Anime:
|
|
results.addAll(renameSeries(it.getValue(), action, conflict, output, format, AniDB, query, order, filter, locale, strict, exec));
|
|
break;
|
|
case Music:
|
|
results.addAll(renameMusic(it.getValue(), action, conflict, output, format, asList(MediaInfoID3, AcoustID), exec)); // prefer existing ID3 tags and use acoustid only when necessary
|
|
break;
|
|
}
|
|
}
|
|
} else {
|
|
debug.warning(format("Failed to process group: %s => %s", it.getKey(), it.getValue()));
|
|
}
|
|
}
|
|
|
|
if (results.isEmpty()) {
|
|
throw new CmdlineException("Failed to identify or process any files");
|
|
}
|
|
|
|
return results;
|
|
}
|
|
|
|
@Override
|
|
public List<File> rename(EpisodeListProvider db, String query, ExpressionFileFormat format, ExpressionFilter filter, SortOrder order, Locale locale, boolean strict, List<File> files, RenameAction action, ConflictAction conflict, File outputDir, ExecCommand exec) throws Exception {
|
|
// match files and episodes in linear order
|
|
List<Episode> episodes = fetchEpisodeList(db, query, filter, order, locale, strict);
|
|
|
|
List<Match<File, ?>> matches = new ArrayList<Match<File, ?>>();
|
|
for (int i = 0; i < files.size() && i < episodes.size(); i++) {
|
|
matches.add(new Match<File, Episode>(files.get(i), episodes.get(i)));
|
|
}
|
|
|
|
// rename episodes
|
|
return renameAll(formatMatches(matches, format, outputDir), action, conflict, matches, exec);
|
|
}
|
|
|
|
@Override
|
|
public List<File> rename(Map<File, File> renameMap, RenameAction renameAction, ConflictAction conflict) throws Exception {
|
|
// generic rename function that can be passed any set of files
|
|
return renameAll(renameMap, renameAction, conflict, null, null);
|
|
}
|
|
|
|
public List<File> renameSeries(Collection<File> files, RenameAction renameAction, ConflictAction conflictAction, File outputDir, ExpressionFileFormat format, EpisodeListProvider db, String query, SortOrder sortOrder, ExpressionFilter filter, Locale locale, boolean strict, ExecCommand exec) throws Exception {
|
|
log.config(format("Rename episodes using [%s]", db.getName()));
|
|
|
|
// ignore sample files
|
|
List<File> fileset = sortByUniquePath(filter(files, not(getClutterFileFilter())));
|
|
|
|
List<File> mediaFiles = filter(fileset, VIDEO_FILES, SUBTITLE_FILES);
|
|
if (mediaFiles.isEmpty()) {
|
|
throw new CmdlineException("No media files: " + files);
|
|
}
|
|
|
|
// similarity metrics for matching
|
|
List<Match<File, ?>> matches = new ArrayList<Match<File, ?>>();
|
|
|
|
// auto-determine optimal batch sets
|
|
for (Entry<Set<File>, Set<String>> sameSeriesGroup : mapSeriesNamesByFiles(mediaFiles, locale, db == AniDB).entrySet()) {
|
|
List<List<File>> batchSets = new ArrayList<List<File>>();
|
|
|
|
if (sameSeriesGroup.getValue() != null && sameSeriesGroup.getValue().size() > 0) {
|
|
// handle series name batch set all at once
|
|
batchSets.add(new ArrayList<File>(sameSeriesGroup.getKey()));
|
|
} else {
|
|
// these files don't seem to belong to any series -> handle folder per folder
|
|
batchSets.addAll(mapByFolder(sameSeriesGroup.getKey()).values());
|
|
}
|
|
|
|
for (List<File> batch : batchSets) {
|
|
// fetch episode data
|
|
List<Episode> episodes;
|
|
|
|
if (query == null) {
|
|
Collection<String> seriesNames = detectSeriesNames(batch, db == AniDB, locale); // detect series name by common word sequence
|
|
log.config("Auto-detected query: " + seriesNames);
|
|
|
|
if (seriesNames.size() == 0) {
|
|
log.warning("Failed to detect query for files: " + batch);
|
|
continue;
|
|
}
|
|
|
|
if (strict && seriesNames.size() > 1) {
|
|
throw new CmdlineException("Multiple queries: Processing multiple shows at once requires -non-strict matching: " + seriesNames);
|
|
}
|
|
|
|
episodes = fetchEpisodeSet(db, seriesNames, sortOrder, locale, strict, 5); // consider episodes of up to N search results for each query
|
|
} else {
|
|
if (isSeriesID(query)) {
|
|
episodes = db.getEpisodeList(Integer.parseInt(query), sortOrder, locale);
|
|
} else {
|
|
episodes = fetchEpisodeSet(db, singleton(query), sortOrder, locale, false, 1); // use --q option and pick first result
|
|
}
|
|
}
|
|
|
|
if (episodes.isEmpty()) {
|
|
continue;
|
|
}
|
|
|
|
// filter episodes
|
|
episodes = applyExpressionFilter(episodes, filter);
|
|
|
|
for (List<File> filesPerType : mapByMediaExtension(filter(batch, VIDEO_FILES, SUBTITLE_FILES)).values()) {
|
|
matches.addAll(matchEpisodes(filesPerType, episodes, strict));
|
|
}
|
|
}
|
|
}
|
|
|
|
if (matches.isEmpty()) {
|
|
throw new CmdlineException("Failed to match files to episode data");
|
|
}
|
|
|
|
// handle derived files
|
|
List<Match<File, ?>> derivateMatches = new ArrayList<Match<File, ?>>();
|
|
SortedSet<File> derivateFiles = new TreeSet<File>(fileset);
|
|
derivateFiles.removeAll(mediaFiles);
|
|
|
|
for (File file : derivateFiles) {
|
|
for (Match<File, ?> match : matches) {
|
|
if (file.getPath().startsWith(match.getValue().getParentFile().getPath()) && isDerived(file, match.getValue()) && match.getCandidate() instanceof Episode) {
|
|
derivateMatches.add(new Match<File, Object>(file, ((Episode) match.getCandidate()).clone()));
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
// add matches from other files that are linked via filenames
|
|
matches.addAll(derivateMatches);
|
|
|
|
// rename episodes
|
|
return renameAll(formatMatches(matches, format, outputDir), renameAction, conflictAction, matches, exec);
|
|
}
|
|
|
|
private List<Match<File, Object>> matchEpisodes(Collection<File> files, Collection<Episode> episodes, boolean strict) throws Exception {
|
|
// always use strict fail-fast matcher
|
|
EpisodeMatcher matcher = new EpisodeMatcher(files, episodes, strict);
|
|
List<Match<File, Object>> matches = matcher.match();
|
|
|
|
for (File failedMatch : matcher.remainingValues()) {
|
|
log.warning("No matching episode: " + failedMatch.getName());
|
|
}
|
|
|
|
// in non-strict mode just pass back results as we got it from the matcher
|
|
if (!strict) {
|
|
return matches;
|
|
}
|
|
|
|
// in strict mode sanity check the result and only pass back good matches
|
|
List<Match<File, Object>> validMatches = new ArrayList<Match<File, Object>>();
|
|
for (Match<File, Object> it : matches) {
|
|
if (isEpisodeNumberMatch(it.getValue(), (Episode) it.getCandidate())) {
|
|
validMatches.add(it);
|
|
}
|
|
}
|
|
return validMatches;
|
|
}
|
|
|
|
private List<Episode> fetchEpisodeSet(EpisodeListProvider db, Collection<String> names, SortOrder sortOrder, Locale locale, boolean strict, int limit) throws Exception {
|
|
Set<SearchResult> shows = new LinkedHashSet<SearchResult>();
|
|
Set<Episode> episodes = new LinkedHashSet<Episode>();
|
|
|
|
// detect series names and create episode list fetch tasks
|
|
for (String query : names) {
|
|
List<SearchResult> results = db.search(query, locale);
|
|
|
|
// select search result
|
|
if (results.size() > 0) {
|
|
List<SearchResult> selectedSearchResults = selectSearchResult(query, results, true, true, strict, limit);
|
|
|
|
if (selectedSearchResults != null) {
|
|
for (SearchResult it : selectedSearchResults) {
|
|
if (shows.add(it)) {
|
|
try {
|
|
log.fine(format("Fetching episode data for [%s]", it.getName()));
|
|
episodes.addAll(db.getEpisodeList(it, sortOrder, locale));
|
|
} catch (IOException e) {
|
|
throw new CmdlineException(String.format("Failed to fetch episode data for [%s]: %s", it, e.getMessage()), e);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (episodes.isEmpty()) {
|
|
log.warning("Failed to fetch episode data: " + names);
|
|
}
|
|
|
|
return new ArrayList<Episode>(episodes);
|
|
}
|
|
|
|
public List<File> renameMovie(Collection<File> files, RenameAction renameAction, ConflictAction conflictAction, File outputDir, ExpressionFileFormat format, MovieIdentificationService service, String query, ExpressionFilter filter, Locale locale, boolean strict, ExecCommand exec) throws Exception {
|
|
log.config(format("Rename movies using [%s]", service.getName()));
|
|
|
|
// ignore sample files
|
|
List<File> fileset = sortByUniquePath(filter(files, not(getClutterFileFilter())));
|
|
|
|
// handle movie files
|
|
Set<File> movieFiles = new TreeSet<File>(filter(fileset, VIDEO_FILES));
|
|
Set<File> nfoFiles = new TreeSet<File>(filter(fileset, NFO_FILES));
|
|
|
|
List<File> orphanedFiles = new ArrayList<File>(filter(fileset, FILES));
|
|
orphanedFiles.removeAll(movieFiles);
|
|
orphanedFiles.removeAll(nfoFiles);
|
|
|
|
Map<File, List<File>> derivatesByMovieFile = new HashMap<File, List<File>>();
|
|
for (File movieFile : movieFiles) {
|
|
derivatesByMovieFile.put(movieFile, new ArrayList<File>());
|
|
}
|
|
for (File file : orphanedFiles) {
|
|
List<File> orphanParent = listPath(file);
|
|
for (File movieFile : movieFiles) {
|
|
if (orphanParent.contains(movieFile.getParentFile()) && isDerived(file, movieFile)) {
|
|
derivatesByMovieFile.get(movieFile).add(file);
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
for (List<File> derivates : derivatesByMovieFile.values()) {
|
|
orphanedFiles.removeAll(derivates);
|
|
}
|
|
|
|
// match movie hashes online
|
|
Map<File, Movie> movieByFile = new TreeMap<File, Movie>();
|
|
if (query == null) {
|
|
// collect useful nfo files even if they are not part of the selected fileset
|
|
Set<File> effectiveNfoFileSet = new TreeSet<File>(nfoFiles);
|
|
for (File dir : mapByFolder(movieFiles).keySet()) {
|
|
effectiveNfoFileSet.addAll(getChildren(dir, NFO_FILES));
|
|
}
|
|
for (File dir : filter(fileset, FOLDERS)) {
|
|
effectiveNfoFileSet.addAll(getChildren(dir, NFO_FILES));
|
|
}
|
|
|
|
for (File nfo : effectiveNfoFileSet) {
|
|
try {
|
|
Movie movie = grepMovie(nfo, service, locale);
|
|
|
|
// ignore illegal nfos
|
|
if (movie == null) {
|
|
continue;
|
|
}
|
|
|
|
if (nfoFiles.contains(nfo)) {
|
|
movieByFile.put(nfo, movie);
|
|
}
|
|
|
|
if (isDiskFolder(nfo.getParentFile())) {
|
|
// special handling for disk folders
|
|
for (File folder : fileset) {
|
|
if (nfo.getParentFile().equals(folder)) {
|
|
movieByFile.put(folder, movie);
|
|
}
|
|
}
|
|
} else {
|
|
// match movie info to movie files that match the nfo file name
|
|
SortedSet<File> siblingMovieFiles = new TreeSet<File>(filter(movieFiles, new ParentFilter(nfo.getParentFile())));
|
|
String baseName = stripReleaseInfo(getName(nfo)).toLowerCase();
|
|
|
|
for (File movieFile : siblingMovieFiles) {
|
|
if (!baseName.isEmpty() && stripReleaseInfo(getName(movieFile)).toLowerCase().startsWith(baseName)) {
|
|
movieByFile.put(movieFile, movie);
|
|
}
|
|
}
|
|
}
|
|
} catch (Exception e) {
|
|
log.log(Level.WARNING, "Failed to grep IMDbID: " + nfo.getName(), e);
|
|
}
|
|
}
|
|
} else {
|
|
log.fine(format("Looking up movie by query [%s]", query));
|
|
List<Movie> results = service.searchMovie(query, locale);
|
|
List<Movie> options = applyExpressionFilter(results, filter);
|
|
if (options.isEmpty()) {
|
|
throw new CmdlineException("Failed to find a valid match: " + results);
|
|
}
|
|
|
|
// force all mappings
|
|
Movie movie = selectSearchResult(query, options);
|
|
for (File file : files) {
|
|
movieByFile.put(file, movie);
|
|
}
|
|
}
|
|
|
|
// collect files that will be matched one by one
|
|
List<File> movieMatchFiles = new ArrayList<File>();
|
|
movieMatchFiles.addAll(movieFiles);
|
|
movieMatchFiles.addAll(nfoFiles);
|
|
movieMatchFiles.addAll(filter(files, FOLDERS));
|
|
movieMatchFiles.addAll(filter(orphanedFiles, SUBTITLE_FILES)); // run movie detection only on orphaned subtitle files
|
|
|
|
// sanity check that we have something to do
|
|
if (fileset.isEmpty() || movieMatchFiles.isEmpty()) {
|
|
throw new CmdlineException("No media files: " + files);
|
|
}
|
|
|
|
// map movies to (possibly multiple) files (in natural order)
|
|
Map<Movie, SortedSet<File>> filesByMovie = new HashMap<Movie, SortedSet<File>>();
|
|
|
|
// map all files by movie
|
|
for (File file : movieMatchFiles) {
|
|
Movie movie = movieByFile.get(file);
|
|
|
|
// unknown hash, try via imdb id from nfo file
|
|
if (movie == null) {
|
|
log.fine(format("Auto-detect movie from context: [%s]", file));
|
|
List<Movie> options = detectMovieWithYear(file, service, locale, strict);
|
|
|
|
// ignore files that cannot yield any acceptable matches (e.g. movie files without year in strict mode)
|
|
if (options == null) {
|
|
continue;
|
|
}
|
|
|
|
// apply filter if defined
|
|
options = applyExpressionFilter(options, filter);
|
|
|
|
// reduce options to perfect matches if possible
|
|
List<Movie> perfectMatches = matchMovieByWordSequence(getName(file), options, 0);
|
|
|
|
// narrow down options if possible
|
|
if (perfectMatches.size() > 0) {
|
|
options = perfectMatches;
|
|
}
|
|
|
|
try {
|
|
// select first element if matches are reliable
|
|
if (options.size() > 0) {
|
|
movie = selectSearchResult(stripReleaseInfo(getName(file)), options);
|
|
|
|
// make sure to get the language-specific movie object for the selected option
|
|
movie = getLocalizedMovie(service, movie, locale);
|
|
}
|
|
} catch (Exception e) {
|
|
log.warning(cause(e));
|
|
}
|
|
}
|
|
|
|
// check if we managed to lookup the movie descriptor
|
|
if (movie != null) {
|
|
// add to file list for movie
|
|
filesByMovie.computeIfAbsent(movie, k -> new TreeSet<File>()).add(file);
|
|
}
|
|
}
|
|
|
|
// collect all File/MoviePart matches
|
|
List<Match<File, ?>> matches = new ArrayList<Match<File, ?>>();
|
|
|
|
filesByMovie.forEach((movie, fs) -> {
|
|
groupByMediaCharacteristics(fs).forEach(moviePartFiles -> {
|
|
// resolve movie parts
|
|
for (int i = 0; i < moviePartFiles.size(); i++) {
|
|
Movie moviePart = moviePartFiles.size() == 1 ? movie : new MoviePart(movie, i + 1, moviePartFiles.size());
|
|
matches.add(new Match<File, Movie>(moviePartFiles.get(i), moviePart.clone()));
|
|
|
|
// automatically add matches for derived files
|
|
List<File> derivates = derivatesByMovieFile.get(moviePartFiles.get(i));
|
|
if (derivates != null) {
|
|
for (File derivate : derivates) {
|
|
matches.add(new Match<File, Movie>(derivate, moviePart.clone()));
|
|
}
|
|
}
|
|
}
|
|
});
|
|
});
|
|
|
|
// rename movies
|
|
return renameAll(formatMatches(matches, format, outputDir), renameAction, conflictAction, matches, exec);
|
|
}
|
|
|
|
public List<File> renameMusic(Collection<File> files, RenameAction renameAction, ConflictAction conflictAction, File outputDir, ExpressionFileFormat format, List<MusicIdentificationService> services, ExecCommand exec) throws Exception {
|
|
List<File> audioFiles = sortByUniquePath(filter(files, AUDIO_FILES, VIDEO_FILES));
|
|
|
|
// check audio files against all services if necessary
|
|
List<Match<File, ?>> matches = new ArrayList<Match<File, ?>>();
|
|
LinkedHashSet<File> remaining = new LinkedHashSet<File>(audioFiles);
|
|
|
|
// check audio files against all services
|
|
for (MusicIdentificationService service : services) {
|
|
if (remaining.size() > 0) {
|
|
log.config(format("Rename music using %s", service.getIdentifier()));
|
|
service.lookup(remaining).forEach((file, music) -> {
|
|
if (music != null) {
|
|
matches.add(new Match<File, AudioTrack>(file, music.clone()));
|
|
remaining.remove(file);
|
|
}
|
|
});
|
|
}
|
|
}
|
|
|
|
// error logging
|
|
remaining.forEach(f -> log.warning(format("Failed to process music file: %s", f)));
|
|
|
|
// rename movies
|
|
return renameAll(formatMatches(matches, format, outputDir), renameAction, conflictAction, null, exec);
|
|
}
|
|
|
|
public List<File> renameFiles(Collection<File> files, RenameAction renameAction, ConflictAction conflictAction, File outputDir, ExpressionFileFormat format, LocalDatasource service, ExpressionFilter filter, boolean strict, ExecCommand exec) throws Exception {
|
|
log.config(format("Rename files using [%s]", service.getName()));
|
|
|
|
Map<File, File> renameMap = new LinkedHashMap<File, File>();
|
|
|
|
// match to xattr metadata object or the file itself
|
|
Map<File, Object> matches = service.match(files, strict);
|
|
|
|
service.match(files, strict).forEach((k, v) -> {
|
|
MediaBindingBean bindingBean = new MediaBindingBean(v, k, matches);
|
|
|
|
if (filter == null || filter.matches(bindingBean)) {
|
|
String destinationPath = format != null ? format.format(bindingBean) : v instanceof File ? v.toString() : validateFileName(v.toString());
|
|
renameMap.put(k, getDestinationFile(k, destinationPath, outputDir));
|
|
}
|
|
});
|
|
|
|
return renameAll(renameMap, renameAction, conflictAction, null, exec);
|
|
}
|
|
|
|
private Map<File, Object> getContext(List<Match<File, ?>> matches) {
|
|
return new AbstractMap<File, Object>() {
|
|
|
|
@Override
|
|
public Set<Entry<File, Object>> entrySet() {
|
|
return matches.stream().collect(toMap(it -> it.getValue(), it -> (Object) it.getCandidate(), (a, b) -> a, LinkedHashMap::new)).entrySet();
|
|
}
|
|
};
|
|
}
|
|
|
|
private File getDestinationFile(File original, String newName, File outputDir) {
|
|
String extension = getExtension(original);
|
|
File newFile = new File(extension != null ? newName + '.' + extension.toLowerCase() : newName);
|
|
|
|
// resolve against output dir
|
|
if (outputDir != null && !newFile.isAbsolute()) {
|
|
newFile = new File(outputDir, newFile.getPath());
|
|
}
|
|
|
|
if (isInvalidFilePath(newFile) && !isUnixFS()) {
|
|
log.config("Stripping invalid characters from new path: " + newName);
|
|
newFile = validateFilePath(newFile);
|
|
}
|
|
|
|
return newFile;
|
|
}
|
|
|
|
private Map<File, File> formatMatches(List<Match<File, ?>> matches, ExpressionFileFormat format, File outputDir) throws Exception {
|
|
// map old files to new paths by applying formatting and validating filenames
|
|
Map<File, File> renameMap = new LinkedHashMap<File, File>();
|
|
|
|
for (Match<File, ?> match : matches) {
|
|
File file = match.getValue();
|
|
Object object = match.getCandidate();
|
|
String destinationPath = format != null ? format.format(new MediaBindingBean(object, file, getContext(matches))) : validateFileName(object.toString());
|
|
|
|
renameMap.put(file, getDestinationFile(file, destinationPath, outputDir));
|
|
}
|
|
|
|
return renameMap;
|
|
}
|
|
|
|
protected List<File> renameAll(Map<File, File> renameMap, RenameAction renameAction, ConflictAction conflictAction, List<Match<File, ?>> matches, ExecCommand exec) throws Exception {
|
|
if (renameMap.isEmpty()) {
|
|
throw new CmdlineException("Failed to identify or process any files");
|
|
}
|
|
|
|
// allow --action test for evaluation purposes
|
|
if (renameAction != StandardRenameAction.TEST) {
|
|
LICENSE.check();
|
|
}
|
|
|
|
// rename files
|
|
Map<File, File> renameLog = new LinkedHashMap<File, File>();
|
|
|
|
try {
|
|
for (Entry<File, File> it : renameMap.entrySet()) {
|
|
try {
|
|
File source = it.getKey();
|
|
File destination = it.getValue();
|
|
|
|
// resolve destination
|
|
if (!destination.isAbsolute()) {
|
|
// same folder, different name
|
|
destination = resolve(source, destination);
|
|
}
|
|
|
|
if (!destination.equals(source) && existsNoFollowLinks(destination)) {
|
|
if (conflictAction == ConflictAction.FAIL) {
|
|
throw new CmdlineException(String.format("Failed to process [%s] because [%s] already exists", source, destination));
|
|
}
|
|
|
|
// do not allow abuse of online databases by repeatedly processing the same files
|
|
if (matches != null && renameAction.canRevert() && source.length() > 0 && equalsFileContent(source, destination)) {
|
|
throw new CmdlineException(String.format("Failed to process [%s] because [%s] is an exact copy and already exists", source, destination));
|
|
}
|
|
|
|
// delete existing destination path if necessary
|
|
if (conflictAction == ConflictAction.OVERRIDE || (conflictAction == ConflictAction.AUTO && VideoQuality.isBetter(source, destination))) {
|
|
// do not delete files in test mode
|
|
if (renameAction.canRevert()) {
|
|
try {
|
|
log.fine(format("[%s] Delete [%s]", conflictAction, destination));
|
|
delete(destination);
|
|
} catch (Exception e) {
|
|
log.warning(format("[%s] Failed to delete [%s]: %s", conflictAction, destination, e));
|
|
}
|
|
}
|
|
}
|
|
|
|
// generate indexed destination path if necessary
|
|
if (conflictAction == ConflictAction.INDEX) {
|
|
destination = nextAvailableIndexedName(destination);
|
|
}
|
|
}
|
|
|
|
// rename file, throw exception on failure
|
|
if (!destination.equals(source) && !destination.exists()) {
|
|
log.info(format("[%s] from [%s] to [%s]", renameAction, source, destination));
|
|
destination = renameAction.rename(source, destination);
|
|
|
|
// remember successfully renamed matches for history entry and possible revert
|
|
renameLog.put(source, destination);
|
|
} else {
|
|
log.info(format("Skipped [%s] because [%s] already exists", source, destination));
|
|
}
|
|
} catch (IOException e) {
|
|
log.warning(format("[%s] Failure: %s", renameAction, e));
|
|
throw e;
|
|
}
|
|
}
|
|
} finally {
|
|
// update history and xattr metadata
|
|
if (renameLog.size() > 0) {
|
|
writeHistory(renameAction, renameLog, matches);
|
|
}
|
|
|
|
// print number of processed files
|
|
log.fine(format("Processed %d files", renameLog.size()));
|
|
}
|
|
|
|
// execute command
|
|
if (exec != null) {
|
|
try {
|
|
execute(renameLog.values(), Objects::nonNull, exec); // destination files may include null values
|
|
} catch (Exception e) {
|
|
log.warning(message("Execute", e.getMessage()));
|
|
}
|
|
}
|
|
|
|
return new ArrayList<File>(renameLog.values());
|
|
}
|
|
|
|
protected void writeHistory(RenameAction action, Map<File, File> log, List<Match<File, ?>> matches) {
|
|
// write rename history
|
|
if (action.canRevert()) {
|
|
HistorySpooler.getInstance().append(log.entrySet());
|
|
}
|
|
|
|
// write xattr metadata
|
|
if (matches != null) {
|
|
for (Match<File, ?> match : matches) {
|
|
if (match.getCandidate() != null) {
|
|
File destination = log.get(match.getValue());
|
|
if (destination != null && destination.isFile()) {
|
|
xattr.setMetaInfo(destination, match.getCandidate(), match.getValue().getName());
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
protected File nextAvailableIndexedName(File file) {
|
|
File parent = file.getParentFile();
|
|
String name = getName(file);
|
|
String ext = getExtension(file);
|
|
return IntStream.range(1, 100).mapToObj(i -> new File(parent, name + '.' + i + '.' + ext)).filter(f -> !f.exists()).findFirst().get();
|
|
}
|
|
|
|
@Override
|
|
public List<File> getSubtitles(Collection<File> files, String query, Language language, SubtitleFormat output, Charset encoding, SubtitleNaming format, boolean strict) throws Exception {
|
|
// ignore anything that is not a video
|
|
files = filter(files, VIDEO_FILES);
|
|
|
|
// ignore sample files
|
|
files = sortByUniquePath(filter(files, not(getClutterFileFilter())));
|
|
|
|
// try to find subtitles for each video file
|
|
List<File> remainingVideos = new ArrayList<File>(files);
|
|
|
|
// parallel download
|
|
List<File> subtitleFiles = new ArrayList<File>();
|
|
|
|
log.finest(format("Get [%s] subtitles for %d files", language.getName(), remainingVideos.size()));
|
|
if (remainingVideos.isEmpty()) {
|
|
throw new CmdlineException("No video files: " + files);
|
|
}
|
|
|
|
// lookup subtitles by hash
|
|
for (VideoHashSubtitleService service : getVideoHashSubtitleServices(language.getLocale())) {
|
|
if (remainingVideos.isEmpty() || !requireLogin(service)) {
|
|
continue;
|
|
}
|
|
|
|
try {
|
|
log.fine("Looking up subtitles by hash via " + service.getName());
|
|
Map<File, List<SubtitleDescriptor>> options = lookupSubtitlesByHash(service, remainingVideos, language.getLocale(), false, strict);
|
|
Map<File, File> downloads = downloadSubtitleBatch(service, options, output, encoding, format);
|
|
remainingVideos.removeAll(downloads.keySet());
|
|
subtitleFiles.addAll(downloads.values());
|
|
} catch (Exception e) {
|
|
log.warning("Lookup by hash failed: " + e.getMessage());
|
|
}
|
|
}
|
|
|
|
for (SubtitleProvider service : getSubtitleProviders(language.getLocale())) {
|
|
if (strict || remainingVideos.isEmpty() || !requireLogin(service)) {
|
|
continue;
|
|
}
|
|
|
|
try {
|
|
log.fine(format("Looking up subtitles by name via %s", service.getName()));
|
|
Map<File, List<SubtitleDescriptor>> options = findSubtitlesByName(service, remainingVideos, language.getLocale(), query, false, strict);
|
|
Map<File, File> downloads = downloadSubtitleBatch(service, options, output, encoding, format);
|
|
remainingVideos.removeAll(downloads.keySet());
|
|
subtitleFiles.addAll(downloads.values());
|
|
} catch (Exception e) {
|
|
log.warning(format("Search by name failed: %s", e.getMessage()));
|
|
}
|
|
}
|
|
|
|
// no subtitles for remaining video files
|
|
for (File it : remainingVideos) {
|
|
log.warning("No matching subtitles found: " + it);
|
|
}
|
|
|
|
return subtitleFiles;
|
|
}
|
|
|
|
protected static boolean requireLogin(Object service) {
|
|
if (service instanceof OpenSubtitlesClient) {
|
|
OpenSubtitlesClient osdb = (OpenSubtitlesClient) service;
|
|
if (osdb.isAnonymous()) {
|
|
throw new CmdlineException(String.format("%s: Please enter your login details by calling `filebot -script fn:configure`", osdb.getName()));
|
|
}
|
|
}
|
|
return true; // no login => logged in by default
|
|
}
|
|
|
|
@Override
|
|
public List<File> getMissingSubtitles(Collection<File> files, String query, Language language, SubtitleFormat output, Charset encoding, SubtitleNaming format, boolean strict) throws Exception {
|
|
List<File> videoFiles = filter(filter(files, VIDEO_FILES), new FileFilter() {
|
|
|
|
// save time on repeating filesystem calls
|
|
private Map<File, List<File>> cache = new HashMap<File, List<File>>();
|
|
|
|
public boolean matchesLanguageCode(File f) {
|
|
Language languageSuffix = Language.getLanguage(releaseInfo.getSubtitleLanguageTag(getName(f)));
|
|
if (languageSuffix != null) {
|
|
return languageSuffix.getCode().equals(language.getCode());
|
|
}
|
|
return false;
|
|
}
|
|
|
|
@Override
|
|
public boolean accept(File video) {
|
|
if (!video.isFile()) {
|
|
return false;
|
|
}
|
|
|
|
List<File> subtitleFiles = cache.computeIfAbsent(video.getParentFile(), parent -> {
|
|
return getChildren(parent, SUBTITLE_FILES);
|
|
});
|
|
|
|
// can't tell which subtitle belongs to which file -> if any subtitles exist skip the whole folder
|
|
if (format == SubtitleNaming.ORIGINAL) {
|
|
return subtitleFiles.size() == 0;
|
|
}
|
|
|
|
return subtitleFiles.stream().allMatch(f -> {
|
|
if (isDerived(f, video)) {
|
|
return format != SubtitleNaming.MATCH_VIDEO && !matchesLanguageCode(f);
|
|
}
|
|
return true;
|
|
});
|
|
}
|
|
});
|
|
|
|
if (videoFiles.isEmpty()) {
|
|
log.info("No missing subtitles");
|
|
return emptyList();
|
|
}
|
|
|
|
return getSubtitles(videoFiles, query, language, output, encoding, format, strict);
|
|
}
|
|
|
|
private Map<File, File> downloadSubtitleBatch(Datasource service, Map<File, List<SubtitleDescriptor>> subtitles, SubtitleFormat outputFormat, Charset outputEncoding, SubtitleNaming naming) {
|
|
Map<File, File> downloads = new LinkedHashMap<File, File>();
|
|
|
|
// fetch subtitle
|
|
subtitles.forEach((movie, options) -> {
|
|
if (options.size() > 0) {
|
|
SubtitleDescriptor subtitle = options.get(0);
|
|
try {
|
|
downloads.put(movie, downloadSubtitle(service, subtitle, movie, outputFormat, outputEncoding, naming));
|
|
} catch (Exception e) {
|
|
log.warning(format("Failed to download %s: %s", subtitle, e));
|
|
}
|
|
}
|
|
});
|
|
|
|
return downloads;
|
|
}
|
|
|
|
private File downloadSubtitle(Datasource service, SubtitleDescriptor descriptor, File movieFile, SubtitleFormat outputFormat, Charset outputEncoding, SubtitleNaming naming) throws Exception {
|
|
// fetch subtitle archive
|
|
log.config(format("Fetching [%s] subtitles [%s] from [%s]", descriptor.getLanguageName(), descriptor.getPath(), service.getName()));
|
|
MemoryFile subtitleFile = fetchSubtitle(descriptor);
|
|
|
|
// subtitle filename is based on movie filename
|
|
String extension = getExtension(subtitleFile.getName());
|
|
ByteBuffer data = subtitleFile.getData();
|
|
|
|
if (outputFormat != null || outputEncoding != null) {
|
|
// adjust extension of the output file
|
|
if (outputFormat != null) {
|
|
extension = outputFormat.getFilter().extension();
|
|
}
|
|
|
|
// default to UTF-8 if no other encoding is given
|
|
if (outputEncoding == null) {
|
|
outputEncoding = UTF_8;
|
|
}
|
|
|
|
log.finest(format("Export [%s] as [%s / %s]", subtitleFile.getName(), outputFormat, outputEncoding));
|
|
data = exportSubtitles(subtitleFile, outputFormat, 0, outputEncoding);
|
|
}
|
|
|
|
File destination = new File(movieFile.getParentFile(), naming.format(movieFile, descriptor, extension));
|
|
log.info(format("Writing [%s] to [%s]", subtitleFile.getName(), destination.getName()));
|
|
|
|
writeFile(data, destination);
|
|
return destination;
|
|
}
|
|
|
|
protected <T> List<T> applyExpressionFilter(List<T> input, ExpressionFilter filter) {
|
|
if (filter == null) {
|
|
return input;
|
|
}
|
|
|
|
log.fine(format("Apply filter [%s] on [%d] items", filter.getExpression(), input.size()));
|
|
|
|
return input.stream().filter(it -> {
|
|
if (filter.matches(new MediaBindingBean(it, null, new EntryList<File, T>(null, input)))) {
|
|
log.finest(format("Include [%s]", it));
|
|
return true;
|
|
}
|
|
return false;
|
|
}).collect(toList());
|
|
}
|
|
|
|
protected <T extends SearchResult> T selectSearchResult(String query, Collection<T> options) throws Exception {
|
|
List<T> matches = selectSearchResult(query, options, false, false, false, 1);
|
|
return matches.size() > 0 ? matches.get(0) : null;
|
|
}
|
|
|
|
protected <T extends SearchResult> List<T> selectSearchResult(String query, Collection<T> options, boolean sort, boolean alias, boolean strict, int limit) throws Exception {
|
|
List<T> probableMatches = getProbableMatches(sort ? query : null, options, alias, strict);
|
|
|
|
if (probableMatches.isEmpty() || (strict && probableMatches.size() != 1)) {
|
|
// allow single search results to just pass through in non-strict mode even if match confidence is low
|
|
if (options.size() == 1 && !strict) {
|
|
return options.stream().collect(toList());
|
|
}
|
|
|
|
if (strict) {
|
|
throw new CmdlineException("Multiple options: Advanced auto-selection requires -non-strict matching: " + probableMatches);
|
|
}
|
|
|
|
// just pick the best N matches
|
|
if (sort) {
|
|
probableMatches = sortBySimilarity(options, singleton(query), getSeriesMatchMetric()).stream().collect(toList());
|
|
}
|
|
}
|
|
|
|
// return first and only value
|
|
return probableMatches.size() <= limit ? probableMatches : probableMatches.subList(0, limit); // trust that the correct match is in the Top N
|
|
}
|
|
|
|
@Override
|
|
public boolean check(Collection<File> files) throws Exception {
|
|
// only check existing hashes
|
|
boolean result = true;
|
|
|
|
for (File it : filter(files, VERIFICATION_FILES)) {
|
|
result &= check(it, it.getParentFile());
|
|
}
|
|
|
|
return result;
|
|
}
|
|
|
|
@Override
|
|
public File compute(Collection<File> files, File output, HashType hash, Charset encoding) throws Exception {
|
|
// ignore folders and any sort of special files
|
|
files = filter(files, FILES);
|
|
|
|
if (files.isEmpty()) {
|
|
throw new CmdlineException("No files: " + files);
|
|
}
|
|
|
|
// find common parent folder of all files
|
|
File[] fileList = files.toArray(new File[0]);
|
|
File[][] pathArray = new File[fileList.length][];
|
|
for (int i = 0; i < fileList.length; i++) {
|
|
pathArray[i] = listPath(fileList[i].getParentFile()).toArray(new File[0]);
|
|
}
|
|
|
|
CommonSequenceMatcher csm = new CommonSequenceMatcher(null, 0, true);
|
|
File[] common = csm.matchFirstCommonSequence(pathArray);
|
|
|
|
if (common == null) {
|
|
throw new CmdlineException("All paths must be on the same filesystem: " + files);
|
|
}
|
|
|
|
// last element in the common sequence must be the root folder
|
|
File root = common[common.length - 1];
|
|
|
|
if (output == null) {
|
|
output = new File(root, root.getName() + '.' + hash.getFilter().extension());
|
|
} else if (!output.isAbsolute()) {
|
|
output = new File(root, output.getPath());
|
|
}
|
|
|
|
log.info(format("Compute %s hash for %s files [%s]", hash, files.size(), output));
|
|
compute(root, files, output, hash, encoding);
|
|
|
|
return output;
|
|
}
|
|
|
|
private boolean check(File verificationFile, File root) throws Exception {
|
|
HashType type = getHashType(verificationFile);
|
|
|
|
// check if type is supported
|
|
if (type == null) {
|
|
throw new CmdlineException("Unsupported format: " + verificationFile);
|
|
}
|
|
|
|
// add all file names from verification file
|
|
log.fine(format("Checking [%s]", verificationFile.getName()));
|
|
VerificationFileReader parser = new VerificationFileReader(createTextReader(verificationFile), type.getFormat());
|
|
boolean status = true;
|
|
|
|
try {
|
|
while (parser.hasNext()) {
|
|
try {
|
|
Entry<File, String> it = parser.next();
|
|
|
|
File file = new File(root, it.getKey().getPath()).getAbsoluteFile();
|
|
String current = computeHash(new File(root, it.getKey().getPath()), type);
|
|
log.info(format("%s %s", current, file));
|
|
|
|
if (current.compareToIgnoreCase(it.getValue()) != 0) {
|
|
throw new IOException(String.format("Corrupted file found: %s [hash mismatch: %s vs %s]", it.getKey(), current, it.getValue()));
|
|
}
|
|
} catch (IOException e) {
|
|
status = false;
|
|
log.warning(e.getMessage());
|
|
}
|
|
}
|
|
} finally {
|
|
parser.close();
|
|
}
|
|
|
|
return status;
|
|
}
|
|
|
|
private void compute(File root, Collection<File> files, File outputFile, HashType hashType, Charset encoding) throws IOException, Exception {
|
|
// compute hashes recursively and write to file
|
|
VerificationFileWriter out = new VerificationFileWriter(outputFile, hashType.getFormat(), encoding != null ? encoding : UTF_8);
|
|
|
|
try {
|
|
for (File it : files) {
|
|
if (it.isHidden() || VERIFICATION_FILES.accept(it)) {
|
|
continue;
|
|
}
|
|
|
|
String relativePath = normalizePathSeparators(it.getPath().substring(root.getPath().length() + 1)); // skip root and first slash
|
|
String hash = computeHash(it, hashType);
|
|
log.info(format("%s %s", hash, relativePath));
|
|
|
|
out.write(relativePath, hash);
|
|
}
|
|
} catch (Exception e) {
|
|
outputFile.deleteOnExit(); // delete only partially written files
|
|
throw e;
|
|
} finally {
|
|
out.close();
|
|
}
|
|
}
|
|
|
|
private List<Episode> fetchEpisodeList(EpisodeListProvider db, String query, ExpressionFilter filter, SortOrder order, Locale locale, boolean strict) throws Exception {
|
|
// sanity check
|
|
if (query == null) {
|
|
throw new CmdlineException(String.format("%s: query parameter is required", db.getName()));
|
|
}
|
|
|
|
// collect all episode objects first
|
|
List<Episode> episodes = new ArrayList<Episode>();
|
|
|
|
if (isSeriesID(query)) {
|
|
// lookup by id
|
|
episodes.addAll(db.getEpisodeList(Integer.parseInt(query), order, locale));
|
|
} else {
|
|
// search by name and select search result
|
|
List<SearchResult> options = selectSearchResult(query, db.search(query, locale), false, false, false, strict ? 1 : 5);
|
|
|
|
// fetch episodes
|
|
for (SearchResult option : options) {
|
|
episodes.addAll(db.getEpisodeList(option, order, locale));
|
|
}
|
|
}
|
|
|
|
// sanity check
|
|
if (episodes.isEmpty()) {
|
|
throw new CmdlineException(String.format("%s: no results", db.getName()));
|
|
}
|
|
|
|
// apply filter
|
|
return applyExpressionFilter(episodes, filter);
|
|
}
|
|
|
|
private boolean isSeriesID(String query) {
|
|
return query.matches("\\d{5,9}");
|
|
}
|
|
|
|
@Override
|
|
public Stream<String> fetchEpisodeList(EpisodeListProvider db, String query, ExpressionFormat format, ExpressionFilter filter, SortOrder order, Locale locale, boolean strict) throws Exception {
|
|
// collect all episode objects first
|
|
List<Episode> episodes = fetchEpisodeList(db, query, filter, order, locale, strict);
|
|
|
|
// instant format
|
|
if (format == null) {
|
|
return episodes.stream().map(Episode::toString);
|
|
}
|
|
|
|
// lazy format
|
|
return episodes.stream().map(episode -> {
|
|
try {
|
|
return format.format(new MediaBindingBean(episode, null, new EntryList<File, Episode>(null, episodes)));
|
|
} catch (Exception e) {
|
|
debug.warning(e::getMessage);
|
|
}
|
|
return null;
|
|
}).filter(Objects::nonNull);
|
|
}
|
|
|
|
@Override
|
|
public Stream<String> getMediaInfo(Collection<File> files, FileFilter filter, ExpressionFormat format) throws Exception {
|
|
// use default expression format if not set
|
|
if (format == null) {
|
|
return getMediaInfo(files, filter, new ExpressionFormat("{fn} [{resolution} {vc} {channels} {ac} {hours}]"));
|
|
}
|
|
|
|
return files.stream().filter(filter::accept).map(f -> {
|
|
try {
|
|
return format.format(new MediaBindingBean(xattr.getMetaInfo(f), f));
|
|
} catch (Exception e) {
|
|
debug.warning(e::getMessage);
|
|
}
|
|
return null;
|
|
}).filter(Objects::nonNull);
|
|
}
|
|
|
|
@Override
|
|
public boolean execute(Collection<File> files, FileFilter filter, ExecCommand exec) throws Exception {
|
|
// collect files
|
|
List<File> f = filter(files, filter);
|
|
|
|
if (f.isEmpty()) {
|
|
return false;
|
|
}
|
|
|
|
// collect object metadata
|
|
List<Object> m = f.stream().map(xattr::getMetaInfo).collect(toList());
|
|
|
|
// build and execute commands
|
|
MediaBindingBean[] group = IntStream.range(0, f.size()).mapToObj(i -> new MediaBindingBean(m.get(i), f.get(i), new EntryList<File, Object>(f, m))).toArray(MediaBindingBean[]::new);
|
|
exec.execute(group);
|
|
|
|
return true;
|
|
}
|
|
|
|
@Override
|
|
public List<File> revert(Collection<File> files, FileFilter filter, RenameAction action) throws Exception {
|
|
if (files.isEmpty()) {
|
|
throw new CmdlineException("Expecting at least one input path");
|
|
}
|
|
|
|
Set<File> whitelist = new HashSet<File>(files);
|
|
Map<File, File> history = HistorySpooler.getInstance().getCompleteHistory().getRenameMap();
|
|
|
|
return history.entrySet().stream().filter(it -> {
|
|
File original = it.getKey();
|
|
File current = it.getValue();
|
|
return Stream.of(current, original).flatMap(f -> listPath(f).stream()).anyMatch(whitelist::contains) && current.exists() && filter.accept(current);
|
|
}).map(it -> {
|
|
File original = it.getKey();
|
|
File current = it.getValue();
|
|
|
|
log.info(format("Revert [%s] to [%s]", current, original));
|
|
if (action.canRevert()) {
|
|
try {
|
|
return StandardRenameAction.revert(current, original);
|
|
} catch (Exception e) {
|
|
log.warning("Failed to revert file: " + e);
|
|
}
|
|
}
|
|
return null;
|
|
}).filter(Objects::nonNull).collect(toList());
|
|
}
|
|
|
|
@Override
|
|
public List<File> extract(Collection<File> files, File output, ConflictAction conflict, FileFilter filter, boolean forceExtractAll) throws Exception {
|
|
// only keep single-volume archives or first part of multi-volume archives
|
|
List<File> archiveFiles = filter(files, Archive.VOLUME_ONE_FILTER);
|
|
List<File> extractedFiles = new ArrayList<File>();
|
|
|
|
for (File file : archiveFiles) {
|
|
Archive archive = Archive.open(file);
|
|
try {
|
|
File outputFolder = output;
|
|
|
|
if (outputFolder == null || !outputFolder.isAbsolute()) {
|
|
outputFolder = new File(file.getParentFile(), outputFolder == null ? getName(file) : outputFolder.getPath()).getCanonicalFile();
|
|
}
|
|
|
|
log.info(format("Read archive [%s] and extract to [%s]", file.getName(), outputFolder));
|
|
FileMapper outputMapper = new FileMapper(outputFolder);
|
|
|
|
List<FileInfo> outputMapping = new ArrayList<FileInfo>();
|
|
for (FileInfo it : archive.listFiles()) {
|
|
File outputPath = outputMapper.getOutputFile(it.toFile());
|
|
outputMapping.add(new SimpleFileInfo(outputPath.getPath(), it.getLength()));
|
|
}
|
|
|
|
Set<FileInfo> selection = new TreeSet<FileInfo>();
|
|
for (FileInfo future : outputMapping) {
|
|
if (filter == null || filter.accept(future.toFile())) {
|
|
selection.add(future);
|
|
}
|
|
}
|
|
|
|
// check if there is anything to extract at all
|
|
if (selection.isEmpty()) {
|
|
continue;
|
|
}
|
|
|
|
boolean skip = true;
|
|
for (FileInfo future : filter == null || forceExtractAll ? outputMapping : selection) {
|
|
if (conflict == ConflictAction.AUTO) {
|
|
skip &= (future.toFile().exists() && future.getLength() == future.toFile().length());
|
|
} else {
|
|
skip &= (future.toFile().exists());
|
|
}
|
|
}
|
|
|
|
if (!skip || conflict == ConflictAction.OVERRIDE) {
|
|
if (filter == null || forceExtractAll) {
|
|
log.finest("Extracting files " + outputMapping);
|
|
|
|
// extract all files
|
|
archive.extract(outputMapper.getOutputDir());
|
|
|
|
for (FileInfo it : outputMapping) {
|
|
extractedFiles.add(it.toFile());
|
|
}
|
|
} else {
|
|
log.finest("Extracting files " + selection);
|
|
|
|
// extract files selected by the given filter
|
|
archive.extract(outputMapper.getOutputDir(), outputMapper.newPathFilter(selection));
|
|
|
|
for (FileInfo it : selection) {
|
|
extractedFiles.add(it.toFile());
|
|
}
|
|
}
|
|
} else {
|
|
log.finest("Skipped extracting files " + selection);
|
|
}
|
|
} finally {
|
|
archive.close();
|
|
}
|
|
}
|
|
|
|
return extractedFiles;
|
|
}
|
|
|
|
}
|