public static String generateSeasonDir(String template, TvShowEpisode episode) {
String seasonDir = template;
// replace $1 and $2 as the only episode specific tokens
seasonDir = seasonDir.replace("$1", String.valueOf(episode.getSeason()));
seasonDir = seasonDir.replace("$2", lz(episode.getSeason()));
seasonDir = seasonDir.replace("$1", String.valueOf(episode.getDvdSeason()));
seasonDir = seasonDir.replace("$2", lz(episode.getDvdSeason()));
// replace all other tokens
seasonDir = createDestination(seasonDir, episode.getTvShow(), new ArrayList<TvShowEpisode>());
// only allow empty season dir if the season is in the filename
if (StringUtils.isBlank(seasonDir)
&& !(SETTINGS.getRenamerFilename().contains("$1")
|| SETTINGS.getRenamerFilename().contains("$2")
|| SETTINGS.getRenamerFilename().contains("$3")
|| SETTINGS.getRenamerFilename().contains("$4"))) {
seasonDir = "Season " + String.valueOf(episode.getSeason());
}
return seasonDir;
}
/**
* Rename Episode (PLUS all Episodes having the same MediaFile!!!).
*
* @param episode the Episode
*/
public static void renameEpisode(TvShowEpisode episode) {
// test for valid season/episode number
if (episode.getSeason() < 0 || episode.getEpisode() < 0) {
LOGGER.warn(
"failed to rename episode "
+ episode.getTitle()
+ " (TV show "
+ episode.getTvShow().getTitle()
+ ") - invalid season/episode number");
MessageManager.instance.pushMessage(
new Message(
MessageLevel.ERROR,
episode.getTvShow().getTitle(),
"tvshow.renamer.failedrename",
new String[] {episode.getTitle()}));
return;
}
LOGGER.info(
"Renaming TvShow '" + episode.getTvShow().getTitle() + "' Episode " + episode.getEpisode());
for (MediaFile mf : new ArrayList<MediaFile>(episode.getMediaFiles())) {
renameMediaFile(mf, episode.getTvShow());
}
}
/**
* Find tv episodes.
*
* @param tvShow the tv show
* @param dir the dir
*/
private void findTvEpisodesAsDisc(TvShow tvShow, File dir) {
String parentDir = dir.getParent();
LOGGER.debug("parsing disc structure in " + dir.getPath() + " parent: " + parentDir);
// crawl this folder and try to find every episode in it
List<MediaFile> videoFiles = new ArrayList<MediaFile>();
File firstVideoFile = null;
File[] content = dir.listFiles();
if (content == null) {
LOGGER.error("Whops. Cannot access directory: " + dir.getName());
return;
}
for (File file : content) {
if (file.isFile()) {
// check filetype
if (!Globals.settings
.getVideoFileType()
.contains("." + FilenameUtils.getExtension(file.getName()).toLowerCase())
|| file.getName().startsWith(skipFilesStartingWith)) { // MacOS ignore
continue;
}
videoFiles.add(new MediaFile(file));
if (firstVideoFile == null) {
firstVideoFile = file;
}
}
}
List<TvShowEpisode> episodes = tvShowList.getTvEpisodesByFile(tvShow, firstVideoFile);
if (episodes.size() == 0) {
String relativePath =
new File(tvShow.getPath()).toURI().relativize(firstVideoFile.toURI()).getPath();
EpisodeMatchingResult result =
TvShowEpisodeAndSeasonParser.detectEpisodeFromFilenameAlternative(
relativePath, tvShow.getTitle());
if (result.season == -1) {
// did the search find a season?
// no -> search for it in the folder name (relative path between tv show root and the
// current dir)
result.season = TvShowEpisodeAndSeasonParser.detectSeason(relativePath);
}
if (result.episodes.size() == 0) {
// try to parse out episodes/season from parent directory
result =
TvShowEpisodeAndSeasonParser.detectEpisodeFromDirectory(
dir.getParentFile(), tvShow.getPath());
}
List<TvShowEpisode> episodesInNfo = TvShowEpisode.parseNFO(firstVideoFile);
// FIXME: Episode root is outside of disc folders ?!
while (dir.getPath().toUpperCase().contains("BDMV")
|| dir.getPath().toUpperCase().contains("VIDEO_TS")) {
dir = dir.getParentFile();
}
if (result.episodes.size() > 0) {
// add it
for (int ep : result.episodes) {
TvShowEpisode episode = null;
// search in the NFO list if an episode has been found
for (int i = episodesInNfo.size() - 1; i >= 0; i--) {
TvShowEpisode e = episodesInNfo.get(i);
if (e.getSeason() == result.season && e.getEpisode() == ep) {
episode = e;
episodesInNfo.remove(i);
break;
}
}
if (episode == null) {
episode = new TvShowEpisode();
episode.setDvdOrder(Globals.settings.getTvShowSettings().isDvdOrder());
episode.setEpisode(ep);
episode.setSeason(result.season);
episode.setNewlyAdded(true);
episode.setFirstAired(result.date);
}
episode.setPath(dir.getPath());
episode.setTvShow(tvShow);
episode.setDisc(true);
episode.setNewlyAdded(true);
episode.addToMediaFiles(videoFiles);
episode.setDateAddedFromMediaFile(new MediaFile(firstVideoFile));
findAdditionalEpisodeFiles(episode, firstVideoFile, content);
episode.saveToDb();
tvShow.addEpisode(episode);
}
} else {
// episode detection found nothing - simply add this file
if (episodesInNfo.size() > 0) {
for (TvShowEpisode e : episodesInNfo) {
e.setPath(dir.getPath());
e.setTvShow(tvShow);
e.addToMediaFiles(videoFiles);
e.setNewlyAdded(true);
// e.findImages();
e.setDateAddedFromMediaFile(new MediaFile(firstVideoFile));
findAdditionalEpisodeFiles(e, firstVideoFile, content);
e.saveToDb();
tvShow.addEpisode(e);
}
} else {
TvShowEpisode episode = new TvShowEpisode();
episode.setPath(dir.getPath());
episode.setDvdOrder(Globals.settings.getTvShowSettings().isDvdOrder());
episode.setEpisode(-1);
episode.setSeason(-1);
episode.setTvShow(tvShow);
episode.setFirstAired(result.date);
episode.setDisc(true);
episode.setNewlyAdded(true);
episode.addToMediaFiles(videoFiles);
episode.setDateAddedFromMediaFile(new MediaFile(firstVideoFile));
findAdditionalEpisodeFiles(episode, firstVideoFile, content);
episode.saveToDb();
tvShow.addEpisode(episode);
}
}
} else {
// episode already added; look if any new additional files have been added
for (TvShowEpisode episode : episodes) {
if (findAdditionalEpisodeFiles(episode, firstVideoFile, content)) {
episode.saveToDb();
}
}
}
}
/**
* Creates the new file/folder name according to template string
*
* @param template the template
* @param show the TV show
* @param episodes the TV show episodes; nullable for TV show root foldername
* @return the string
*/
public static String createDestination(
String template, TvShow show, List<TvShowEpisode> episodes) {
String newDestination = template;
TvShowEpisode firstEp = null;
// replace token show title ($N)
if (newDestination.contains("$N")) {
newDestination = replaceToken(newDestination, "$N", show.getTitle());
}
// parse out episode depended tokens - for multi EP naming
if (!episodes.isEmpty()) {
Matcher matcher = multiEpisodeTokenPattern.matcher(template);
String episodeTokens = "";
if (matcher.find()) {
episodeTokens = matcher.group(0);
}
String combinedEpisodeParts = "";
for (TvShowEpisode episode : episodes) {
String episodePart = episodeTokens;
// remember first episode for media file tokens
if (firstEp == null) {
firstEp = episode;
}
// Season w/o leading zeros ($1)
if (episodePart.contains("$1")) {
episodePart = replaceToken(episodePart, "$1", String.valueOf(episode.getSeason()));
}
// Season leading zeros ($2)
if (episodePart.contains("$2")) {
episodePart = replaceToken(episodePart, "$2", lz(episode.getSeason()));
}
// DVD-Season w/o leading zeros ($3)
if (episodePart.contains("$3")) {
episodePart = replaceToken(episodePart, "$3", String.valueOf(episode.getDvdSeason()));
}
// DVD-Season leading zeros ($4)
if (episodePart.contains("$4")) {
episodePart = replaceToken(episodePart, "$4", lz(episode.getDvdSeason()));
}
// episode number
if (episodePart.contains("$E")) {
episodePart = replaceToken(episodePart, "$E", lz(episode.getEpisode()));
}
// DVD-episode number
if (episodePart.contains("$D")) {
episodePart = replaceToken(episodePart, "$D", lz(episode.getDvdEpisode()));
}
// episode title
if (episodePart.contains("$T")) {
episodePart = replaceToken(episodePart, "$T", episode.getTitle());
}
combinedEpisodeParts += episodePart + " ";
}
// and now fill in the (multiple) episode parts
if (StringUtils.isNotBlank(episodeTokens)) {
newDestination = newDestination.replace(episodeTokens, combinedEpisodeParts);
}
} else {
// we're in either TV show folder or season folder generation;
// strip out episode tokens
newDestination = newDestination.replace("$E", "");
newDestination = newDestination.replace("$T", "");
}
// replace token year ($Y)
if (newDestination.contains("$Y")) {
if (show.getYear().equals("0")) {
newDestination = newDestination.replace("$Y", "");
} else {
newDestination = replaceToken(newDestination, "$Y", show.getYear());
}
}
if (firstEp != null && firstEp.getMediaFiles(MediaFileType.VIDEO).size() > 0) {
MediaFile mf = firstEp.getMediaFiles(MediaFileType.VIDEO).get(0);
// replace token resolution ($R)
if (newDestination.contains("$R")) {
newDestination = replaceToken(newDestination, "$R", mf.getVideoResolution());
}
// replace token audio codec + channels ($A)
if (newDestination.contains("$A")) {
newDestination =
replaceToken(
newDestination,
"$A",
mf.getAudioCodec()
+ (mf.getAudioCodec().isEmpty() ? "" : "-")
+ mf.getAudioChannels());
}
// replace token video codec + format ($V)
if (newDestination.contains("$V")) {
newDestination =
replaceToken(
newDestination,
"$V",
mf.getVideoCodec()
+ (mf.getVideoCodec().isEmpty() ? "" : "-")
+ mf.getVideoFormat());
}
// replace token video format ($F)
if (newDestination.contains("$F")) {
newDestination = replaceToken(newDestination, "$F", mf.getVideoFormat());
}
} else {
// no mediafiles; remove at least token (if available)
newDestination = newDestination.replace("$R", "");
newDestination = newDestination.replace("$A", "");
newDestination = newDestination.replace("$V", "");
newDestination = newDestination.replace("$F", "");
}
// replace empty brackets
newDestination = newDestination.replaceAll("\\(\\)", "");
newDestination = newDestination.replaceAll("\\[\\]", "");
// if there are multiple file separators in a row - strip them out
if (SystemUtils.IS_OS_WINDOWS) {
// we need to mask it in windows
newDestination = newDestination.replaceAll("\\\\{2,}", "\\\\");
newDestination = newDestination.replaceAll("^\\\\", "");
} else {
newDestination = newDestination.replaceAll(File.separator + "{2,}", File.separator);
newDestination = newDestination.replaceAll("^" + File.separator, "");
}
// ASCII replacement
if (SETTINGS.isAsciiReplacement()) {
newDestination = StrgUtils.convertToAscii(newDestination, false);
}
// trim out unnecessary whitespaces
newDestination = newDestination.trim();
// any whitespace replacements?
if (SETTINGS.isRenamerSpaceSubstitution()) {
newDestination = newDestination.replaceAll(" ", SETTINGS.getRenamerSpaceReplacement());
}
// replace trailing dots and spaces
newDestination = newDestination.replaceAll("[ \\.]+$", "");
return newDestination.trim();
}
/**
* Renames a MediaFiles<br>
* gets all episodes of it, creates season folder, updates MFs & DB
*
* @param mf the MediaFile
* @param show the tvshow (only needed for path)
*/
public static void renameMediaFile(MediaFile mf, TvShow show) {
// #######################################################
// Assumption: all multi-episodes share the same season!!!
// #######################################################
List<TvShowEpisode> eps = TvShowList.getInstance().getTvEpisodesByFile(show, mf.getFile());
if (eps == null || eps.size() == 0) {
// this should not happen, but unluckily ODB does it sometimes; try a second time to get the
// episode
try {
Thread.sleep(250);
} catch (Exception e) {
}
eps = TvShowList.getInstance().getTvEpisodesByFile(show, mf.getFile());
}
if (eps == null || eps.size() == 0) {
// FIXME: workaround for r1972
// when moving video file, all NFOs get deleted and a new gets created.
// so this OLD NFO is not found anylonger - just delete it
if (mf.getType() == MediaFileType.NFO) {
FileUtils.deleteQuietly(mf.getFile());
return;
}
LOGGER.warn("No episodes found for file '" + mf.getFilename() + "' - skipping");
return;
}
// get first, for isDisc and season
TvShowEpisode ep = eps.get(0);
// test access rights or return
LOGGER.debug(
"testing file S:"
+ ep.getSeason()
+ " E:"
+ ep.getEpisode()
+ " MF:"
+ mf.getFile().getAbsolutePath());
File f = mf.getFile();
boolean testRenameOk = false;
for (int i = 0; i < 5; i++) {
testRenameOk = f.renameTo(f); // haahaa, try to rename to itself :P
if (testRenameOk) {
break; // ok it worked, step out
}
try {
if (!f.exists()) {
LOGGER.debug("Hmmm... file " + f + " does not even exists; delete from DB");
// delete from MF
for (TvShowEpisode e : eps) {
e.removeFromMediaFiles(mf);
e.saveToDb();
e.writeNFO();
}
return;
}
LOGGER.debug("rename did not work - sleep a while and try again...");
Thread.sleep(1000);
} catch (InterruptedException e) {
LOGGER.warn("I'm so excited - could not sleep");
}
}
if (!testRenameOk) {
LOGGER.warn("File " + mf.getFile().getAbsolutePath() + " is not accessible!");
MessageManager.instance.pushMessage(
new Message(MessageLevel.ERROR, mf.getFilename(), "message.renamer.failedrename"));
return;
}
// create SeasonDir
// String seasonName = "Season " + String.valueOf(ep.getSeason());
String seasonName = generateSeasonDir(SETTINGS.getRenamerSeasonFoldername(), ep);
File seasonDir = null;
if (StringUtils.isNotBlank(seasonName)) {
seasonDir = new File(show.getPath(), seasonName);
if (!seasonDir.exists()) {
seasonDir.mkdir();
}
} else {
seasonDir = new File(show.getPath());
}
// rename epFolder accordingly
if (ep.isDisc() || mf.isDiscFile()) {
// \Season 1\S01E02E03\VIDEO_TS\VIDEO_TS.VOB
// ........ \epFolder \disc... \ file
File disc = mf.getFile().getParentFile();
File epFolder = disc.getParentFile();
// sanity check
if (!disc.getName().equalsIgnoreCase("BDMV")
&& !disc.getName().equalsIgnoreCase("VIDEO_TS")) {
LOGGER.error(
"Episode is labeled as 'on BD/DVD', but structure seems not to match. Better exit and do nothing... o_O");
return;
}
String newFoldername =
FilenameUtils.getBaseName(generateFolderename(show, mf)); // w/o extension
if (newFoldername != null && !newFoldername.isEmpty()) {
File newEpFolder = new File(seasonDir + File.separator + newFoldername);
File newDisc = new File(newEpFolder + File.separator + disc.getName()); // old disc name
try {
// if (!epFolder.equals(newEpFolder)) {
if (!epFolder.getAbsolutePath().equals(newEpFolder.getAbsolutePath())) {
boolean ok = false;
try {
ok = Utils.moveDirectorySafe(epFolder, newEpFolder);
} catch (Exception e) {
LOGGER.error(e.getMessage());
MessageManager.instance.pushMessage(
new Message(
MessageLevel.ERROR,
epFolder.getName(),
"message.renamer.failedrename",
new String[] {":", e.getLocalizedMessage()}));
}
if (ok) {
// iterate over all EPs & MFs and fix new path
LOGGER.debug("updating *all* MFs for new path -> " + newEpFolder);
for (TvShowEpisode e : eps) {
e.updateMediaFilePath(disc, newDisc);
e.setPath(newEpFolder.getPath());
e.saveToDb();
e.writeNFO();
}
}
// and cleanup
cleanEmptyDir(epFolder);
} else {
// old and new folder are equal, do nothing
}
} catch (Exception e) {
LOGGER.error("error moving video file " + disc.getName() + " to " + newFoldername, e);
MessageManager.instance.pushMessage(
new Message(
MessageLevel.ERROR,
mf.getFilename(),
"message.renamer.failedrename",
new String[] {":", e.getLocalizedMessage()}));
}
}
} // end isDisc
else {
MediaFile newMF = new MediaFile(mf); // clone MF
if (mf.getType().equals(MediaFileType.TRAILER)) {
// move trailer into separate dir - not supported by XBMC
File sample = new File(seasonDir, "sample");
if (!sample.exists()) {
sample.mkdir();
}
seasonDir = sample; // change directory storage
}
String filename = generateFilename(show, mf);
LOGGER.debug("new filename should be " + filename);
if (filename != null && !filename.isEmpty()) {
File newFile = new File(seasonDir, filename);
try {
// if (!mf.getFile().equals(newFile)) {
if (!mf.getFile().getAbsolutePath().equals(newFile.getAbsolutePath())) {
File oldMfFile = mf.getFile();
boolean ok = false;
try {
ok = Utils.moveFileSafe(oldMfFile, newFile);
} catch (Exception e) {
LOGGER.error(e.getMessage());
MessageManager.instance.pushMessage(
new Message(
MessageLevel.ERROR,
oldMfFile.getPath(),
"message.renamer.failedrename",
new String[] {":", e.getLocalizedMessage()}));
}
if (ok) {
newMF.setPath(seasonDir.getAbsolutePath());
newMF.setFilename(filename);
// iterate over all EPs and delete old / set new MF
for (TvShowEpisode e : eps) {
e.removeFromMediaFiles(mf);
e.addToMediaFiles(newMF);
e.setPath(seasonDir.getAbsolutePath());
e.saveToDb();
e.writeNFO();
}
}
// and cleanup
cleanEmptyDir(oldMfFile.getParentFile());
} else {
// old and new file are equal, keep MF
}
} catch (Exception e) {
LOGGER.error(
"error moving video file " + mf.getFilename() + " to " + newFile.getPath(), e);
MessageManager.instance.pushMessage(
new Message(
MessageLevel.ERROR,
mf.getFilename(),
"message.renamer.failedrename",
new String[] {":", e.getLocalizedMessage()}));
}
}
}
}
