mirror of
https://github.com/mitb-archive/filebot
synced 2024-08-13 17:03:45 -04:00
94 lines
2.7 KiB
Java
94 lines
2.7 KiB
Java
package net.filebot.media;
|
|
|
|
import static java.util.Arrays.*;
|
|
import static java.util.stream.Collectors.*;
|
|
import static net.filebot.WebServices.*;
|
|
import static net.filebot.similarity.Normalization.*;
|
|
import static net.filebot.util.FileUtilities.*;
|
|
import static net.filebot.web.EpisodeFormat.*;
|
|
|
|
import java.util.Objects;
|
|
|
|
import net.filebot.web.AudioTrack;
|
|
import net.filebot.web.Episode;
|
|
import net.filebot.web.EpisodeFormat;
|
|
import net.filebot.web.Movie;
|
|
import net.filebot.web.MoviePart;
|
|
import net.filebot.web.MultiEpisode;
|
|
|
|
public enum NamingStandard {
|
|
|
|
Plex;
|
|
|
|
public String getPath(Object o) {
|
|
if (o instanceof Episode)
|
|
return getPath((Episode) o);
|
|
if (o instanceof Movie)
|
|
return getPath((Movie) o);
|
|
if (o instanceof AudioTrack)
|
|
return getPath((AudioTrack) o);
|
|
|
|
return null;
|
|
}
|
|
|
|
public String getPath(Episode e) {
|
|
// enforce title length limit by default
|
|
String episodeTitle = truncateText(e instanceof MultiEpisode ? SeasonEpisode.formatMultiTitle(((MultiEpisode) e).getEpisodes()) : e.getTitle(), 150);
|
|
|
|
// Anime
|
|
if (isAnime(e)) {
|
|
String primaryTitle = e.getSeriesInfo().getName();
|
|
String episode = String.join(" - ", primaryTitle, EpisodeFormat.SeasonEpisode.formatSxE(e), episodeTitle);
|
|
return path("Anime", primaryTitle, episode);
|
|
}
|
|
|
|
// TV Series
|
|
String episode = String.join(" - ", e.getSeriesName(), EpisodeFormat.SeasonEpisode.formatS00E00(e), episodeTitle);
|
|
String season = e.getSeason() == null ? null : e.getSpecial() == null ? String.format("Season %02d", e.getSeason()) : "Special";
|
|
return path("TV Shows", e.getSeriesName(), season, episode);
|
|
}
|
|
|
|
public String getPath(Movie m) {
|
|
// Movie
|
|
String name = m.getNameWithYear();
|
|
|
|
// Movie (multi-part)
|
|
if (m instanceof MoviePart) {
|
|
name = String.format("%s CD%d", name, ((MoviePart) m).getPartIndex());
|
|
}
|
|
|
|
return path("Movies", m.getNameWithYear(), name);
|
|
}
|
|
|
|
public String getPath(AudioTrack a) {
|
|
// Music
|
|
String name = String.join(" - ", a.getArtist(), first(a.getTrackTitle(), a.getTitle()));
|
|
|
|
// prepend track number
|
|
if (a.getTrack() != null) {
|
|
name = String.format("%02d. %s", a.getTrack(), name);
|
|
}
|
|
|
|
return path("Music", first(a.getAlbumArtist(), a.getArtist()), a.getAlbum(), name);
|
|
}
|
|
|
|
private static String path(String... name) {
|
|
return stream(name).filter(Objects::nonNull).map(s -> {
|
|
s = replacePathSeparators(s, " ");
|
|
s = replaceSpace(s, " ");
|
|
s = normalizeQuotationMarks(s);
|
|
s = trimTrailingPunctuation(s);
|
|
return s;
|
|
}).collect(joining("/"));
|
|
}
|
|
|
|
private static String first(String... options) {
|
|
return stream(options).filter(Objects::nonNull).findFirst().get();
|
|
}
|
|
|
|
private static boolean isAnime(Episode e) {
|
|
return AniDB.getIdentifier().equals(e.getSeriesInfo().getDatabase());
|
|
}
|
|
|
|
}
|