private void findArtwork( TvShow show, List<File> directoryContents, Pattern searchPattern, MediaFileType type) { for (File file : directoryContents) { Matcher matcher = searchPattern.matcher(file.getName()); if (matcher.matches() && !file.getName().startsWith("._")) { // MacOS ignore // false positive exclusion: poster regexp would also find season posters.. if (type == MediaFileType.POSTER && file.getName().matches(seasonPattern.pattern())) { continue; } MediaFile mf = new MediaFile(file, type); show.addToMediaFiles(mf); LOGGER.debug("found " + mf.getType().name().toLowerCase() + ": " + file.getPath()); break; } } }
@Override public Object getColumnValue(MediaFile mediaFile, int column) { switch (column) { case 0: if (mediaFile.isVideo()) { return IconManager.PLAY_SMALL; } if (mediaFile.isGraphic()) { return IconManager.SEARCH; } return null; case 1: return mediaFile.getFilename(); case 2: return mediaFile.getFilesizeInMegabytes(); case 3: return getMediaFileTypeLocalized(mediaFile.getType()); case 4: return mediaFile.getCombinedCodecs(); case 5: return mediaFile.getVideoResolution(); case 6: return mediaFile.getDurationHM(); case 7: return mediaFile.getSubtitlesAsString(); } throw new IllegalStateException(); }
/** * Find additional episode files.<br> * adds everything which starts with "videoFile name"<br> * scans subs/subtitle directories aswell * * @param episode the episode * @param videoFile the video file * @param directoryContents the directory contents * @return indicator whether something new has been found or not */ private boolean findAdditionalEpisodeFiles( TvShowEpisode episode, File videoFile, File[] directoryContents) { boolean newFileFound = false; List<MediaFile> existingMediaFiles = episode.getMediaFiles(); for (File file : directoryContents) { if (file.isFile()) { MediaFile mf = new MediaFile(file); if (existingMediaFiles.contains(mf)) { continue; } if (mf.getType().equals(MediaFileType.VIDEO) || !mf.getBasename().startsWith(FilenameUtils.getBaseName(videoFile.getName())) || file.getName().startsWith(skipFilesStartingWith)) { // MacOS ignore) continue; } if (mf.getType() == MediaFileType.SUBTITLE) { episode.setSubtitles(true); } // check if it is a poster if (mf.getType() == MediaFileType.GRAPHIC) { LOGGER.debug("parsing unknown graphic " + mf.getFilename()); String vfilename = FilenameUtils.getBaseName(videoFile.getName()); if (vfilename.equals(FilenameUtils.getBaseName(mf.getFilename())) // basename match || Utils.cleanStackingMarkers(vfilename) .trim() .equals(FilenameUtils.getBaseName(mf.getFilename())) // basename w/o stacking || episode .getTitle() .equals(FilenameUtils.getBaseName(mf.getFilename()))) { // title match mf.setType(MediaFileType.THUMB); } } episode.addToMediaFiles(mf); newFileFound = true; } else { if (file.getName().equalsIgnoreCase("subs") || file.getName().equalsIgnoreCase("subtitle")) { File[] subDirContent = file.listFiles(); if (subDirContent == null) { LOGGER.error("Whops. Cannot access directory: " + file.getName()); } else { for (File subDirFile : subDirContent) { if (FilenameUtils.getBaseName(subDirFile.getName()) .startsWith(FilenameUtils.getBaseName(videoFile.getName()))) { MediaFile mf = new MediaFile(subDirFile); if (existingMediaFiles.contains(mf)) { continue; } if (mf.getType() == MediaFileType.SUBTITLE) { episode.setSubtitles(true); } episode.addToMediaFiles(mf); newFileFound = true; } } } } } } return newFileFound; }
/** * Find tv episodes. * * @param tvShow the tv show * @param dir the dir */ private void findTvEpisodes(TvShow tvShow, File dir) { LOGGER.debug("parsing " + dir.getPath()); // crawl this folder and try to find every episode and its corresponding files in it File[] content = dir.listFiles(); if (content == null) { LOGGER.error("Whops. Cannot access directory: " + dir.getName()); return; } Arrays.sort(content); for (File file : content) { if (file.isFile()) { if (!file.getName().startsWith(skipFilesStartingWith)) { MediaFile mf = new MediaFile(file); // check filetype - we only proceed here if it's a video file if (!mf.getType().equals(MediaFileType.VIDEO)) { continue; } // is this file already assigned to another episode? List<TvShowEpisode> episodes = tvShowList.getTvEpisodesByFile(tvShow, file); if (episodes.size() == 0) { // try to check what episode//season // EpisodeMatchingResult result = // TvShowEpisodeAndSeasonParser.detectEpisodeFromFilename(file); String relativePath = new File(tvShow.getPath()).toURI().relativize(file.toURI()).getPath(); EpisodeMatchingResult result = TvShowEpisodeAndSeasonParser.detectEpisodeFromFilenameAlternative( relativePath, tvShow.getTitle()); // second check: is the detected episode (>-1; season >-1) already in tmm and any valid // stacking markers found? if (result.episodes.size() == 1 && result.season > -1 && result.stackingMarkerFound) { // get any assigned episode TvShowEpisode ep = tvShow.getEpisode(result.season, result.episodes.get(0)); if (ep != null) { ep.setNewlyAdded(true); ep.addToMediaFiles(mf); continue; } } if (result.episodes.size() == 0) { // try to parse out episodes/season from parent directory result = TvShowEpisodeAndSeasonParser.detectEpisodeFromDirectory(dir, tvShow.getPath()); } 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); } List<TvShowEpisode> episodesInNfo = TvShowEpisode.parseNFO(file); // did we find any episodes in the NFO? if (episodesInNfo.size() > 0) { // these have priority! for (TvShowEpisode e : episodesInNfo) { e.setPath(dir.getPath()); e.setTvShow(tvShow); e.addToMediaFiles(mf); e.setDateAddedFromMediaFile(mf); findAdditionalEpisodeFiles(e, file, content); e.setNewlyAdded(true); e.saveToDb(); tvShow.addEpisode(e); } } else if (result.episodes.size() > 0) { // something found with the season detection? for (int ep : result.episodes) { TvShowEpisode episode = new TvShowEpisode(); episode.setDvdOrder(Globals.settings.getTvShowSettings().isDvdOrder()); episode.setEpisode(ep); episode.setSeason(result.season); episode.setFirstAired(result.date); if (result.name.isEmpty()) { result.name = FilenameUtils.getBaseName(file.getName()); } episode.setTitle(result.name); episode.setPath(dir.getPath()); episode.setTvShow(tvShow); episode.addToMediaFiles(mf); episode.setDateAddedFromMediaFile(mf); findAdditionalEpisodeFiles(episode, file, content); episode.setNewlyAdded(true); episode.saveToDb(); tvShow.addEpisode(episode); } } else { // episode detection found nothing - simply add this file TvShowEpisode episode = new TvShowEpisode(); episode.setDvdOrder(Globals.settings.getTvShowSettings().isDvdOrder()); episode.setEpisode(-1); episode.setSeason(-1); episode.setPath(dir.getPath()); episode.setTitle(FilenameUtils.getBaseName(file.getName())); episode.setTvShow(tvShow); episode.setFirstAired(result.date); episode.addToMediaFiles(mf); episode.setDateAddedFromMediaFile(mf); findAdditionalEpisodeFiles(episode, file, content); episode.setNewlyAdded(true); 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, file, content)) { episode.saveToDb(); } } } } // end skipFilesStartingWith } // end isFile if (file.isDirectory() && !skipFolders.contains(file.getName().toUpperCase()) && !file.getName().matches(skipFoldersRegex) && !TvShowModuleManager.TV_SHOW_SETTINGS .getTvShowSkipFolders() .contains(file.getAbsolutePath())) { // check if that directory contains a .tmmignore file File tmmIgnore = new File(file, ".tmmignore"); if (!tmmIgnore.exists()) { // dig deeper if (file.getName().toUpperCase().equals("VIDEO_TS")) { findTvEpisodesAsDisc(tvShow, file); } else if (file.getName().toUpperCase().equals("BDMV")) { findTvEpisodesAsDisc(tvShow, file); } else { findTvEpisodes(tvShow, file); } } } } }
private static String generateName( String template, TvShow tvShow, MediaFile mf, boolean forFile) { String filename = ""; List<TvShowEpisode> eps = TvShowList.getInstance().getTvEpisodesByFile(tvShow, 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 ex) { } eps = TvShowList.getInstance().getTvEpisodesByFile(tvShow, mf.getFile()); } if (eps == null || eps.size() == 0) { return ""; } if (StringUtils.isBlank(template)) { filename = createDestination(SETTINGS.getRenamerFilename(), tvShow, eps); } else { filename = createDestination(template, tvShow, eps); } // since we can use this method for folders too, use the next options solely for files if (forFile) { if (mf.getType().equals(MediaFileType.THUMB)) { if (SETTINGS.isUseRenamerThumbPostfix()) { filename = filename + "-thumb"; } // else let the filename as is } if (mf.getType().equals(MediaFileType.FANART)) { filename = filename + "-fanart"; } if (mf.getType().equals(MediaFileType.TRAILER)) { filename = filename + "-trailer"; } if (mf.getType().equals(MediaFileType.VIDEO_EXTRA)) { String name = mf.getBasename(); Pattern p = Pattern.compile("(?i).*([ _.-]extras[ _.-]).*"); Matcher m = p.matcher(name); if (m.matches()) { name = name.substring(m.end(1)); // everything behind } // if not, MF must be within /extras/ folder - use name 1:1 filename = filename + "-extras-" + name; } if (mf.getType().equals(MediaFileType.SUBTITLE)) { List<MediaFileSubtitle> subtitles = mf.getSubtitles(); if (subtitles != null && subtitles.size() > 0) { MediaFileSubtitle mfs = mf.getSubtitles().get(0); if (mfs != null) { if (!mfs.getLanguage().isEmpty()) { filename = filename + "." + mfs.getLanguage(); } if (mfs.isForced()) { filename = filename + ".forced"; } } else { // TODO: meh, we didn't have an actual MF yet - need to parse filename ourselves (like // movie). But with a recent scan of files/DB this // should not occur. } } } } // end forFile // ASCII replacement if (SETTINGS.isAsciiReplacement()) { filename = StrgUtils.convertToAscii(filename, false); } filename = filename + "." + mf.getExtension(); // readd original extension return filename; }
/** * 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()})); } } } }