// filebot -script "fn:utorrent-postprocess" --output "X:/media" --action copy --conflict override --def subtitles=true artwork=true xbmc=localhost plex=10.0.0.1 gmail=username:password "ut_dir=%D" "ut_file=%F" "ut_kind=%K" "ut_title=%N" "ut_label=%L" "ut_state=%S" def input = [] def failOnError = _args.conflict == 'fail' // print input parameters _args.bindings?.each{ _log.finest("Parameter: $it.key = $it.value") } args.each{ _log.finest("Argument: $it") } // check user-defined pre-condition if (tryQuietly{ ut_state != ut_state_allow }) { throw new Exception("Invalid state: ut_state = $ut_state (expected $ut_state_allow)") } // enable/disable features as specified via --def parameters def music = tryQuietly{ music.toBoolean() } def subtitles = tryQuietly{ subtitles.toBoolean() ? ['en'] : subtitles.split(/[ ,|]+/).findAll{ it.length() >= 2 } } def artwork = tryQuietly{ artwork.toBoolean() } def backdrops = tryQuietly{ backdrops.toBoolean() } def clean = tryQuietly{ clean.toBoolean() } // array of xbmc/plex hosts def xbmc = tryQuietly{ xbmc.split(/[ ,|]+/) } def plex = tryQuietly{ plex.split(/[ ,|]+/) } // myepisodes updates and email notifications def myepisodes = tryQuietly { myepisodes.split(':', 2) } def gmail = tryQuietly{ gmail.split(':', 2) } def pushover = tryQuietly{ pushover } // series/anime/movie format expressions def format = [ tvs: tryQuietly{ seriesFormat } ?: '''TV Shows/{n}/{episode.special ? "Special" : "Season "+s}/{n} - {episode.special ? "S00E"+special.pad(2) : s00e00} - {t}{".$lang"}''', anime: tryQuietly{ animeFormat } ?: '''Anime/{n}/{n} - {sxe} - {t}''', mov: tryQuietly{ movieFormat } ?: '''Movies/{n} ({y})/{n} ({y}){" CD$pi"}{".$lang"}''', music: tryQuietly{ musicFormat } ?: '''Music/{n}/{album}/{n} - {t}''' ] // force movie/series/anime logic def forceMovie(f) { tryQuietly{ ut_label } =~ /^(?i:Movie|Couch.Potato)/ } def forceSeries(f) { parseEpisodeNumber(f) || parseDate(f) || tryQuietly{ ut_label } =~ /^(?i:TV|Kids.Shows)/ } def forceAnime(f) { tryQuietly{ ut_label } =~ /^(?i:Anime)/ || (f.isVideo() && (f.name =~ "[\\(\\[]\\p{XDigit}{8}[\\]\\)]" || getMediaInfo(file:f, format:'''{media.AudioLanguageList} {media.TextCodecList}''').tokenize().containsAll(['Japanese', 'ASS']))) } def forceIgnore(f) { tryQuietly{ ut_label } =~ /^(?i:ebook|other|ignore)/ || f.path =~ tryQuietly{ ignore } } // specify how to resolve input folders, e.g. grab files from all folders except disk folders def resolveInput(f) { if (f.isDirectory() && !f.isDisk()) return f.listFiles().toList().findResults{ resolveInput(it) } else return f } // collect input fileset as specified by the given --def parameters if (args.empty) { // assume we're called with utorrent parameters (account for older and newer versions of uTorrents) if (ut_kind == 'single' || (ut_kind != 'multi' && ut_dir && ut_file)) { input += new File(ut_dir, ut_file) // single-file torrent } else { input += resolveInput(ut_dir as File) // multi-file torrent } } else { // assume we're called normally with arguments input += args.findResults{ resolveInput(it) } } // flatten nested file structure input = input.flatten() // extract archives (zip, rar, etc) that contain at least one video file def extractedFiles = extract(file: input.findAll{ it.isArchive() }, output: null, conflict: 'override', filter: { it.isVideo() || (music && it.isAudio()) }, forceExtractAll: true) input += extractedFiles // sanitize input input = input.findAll{ it?.exists() }.collect{ it.canonicalFile }.unique() // process only media files input = input.findAll{ it.isVideo() || it.isSubtitle() || it.isDisk() || (music && it.isAudio()) } // ignore clutter files input = input.findAll{ !(it.path =~ /\b(?i:sample|trailer|extras|deleted.scenes|music.video|scrapbook|behind.the.scenes)\b/) } // print input fileset input.each{ f -> _log.finest("Input: $f") } // artwork/nfo utility include('fn:lib/htpc') // group episodes/movies and rename according to XBMC standards def groups = input.groupBy{ f -> // skip auto-detection if possible if (forceIgnore(f)) return [] if (f.isAudio() && !f.isVideo()) // PROCESS MUSIC FOLDER BY FOLDER return [music: f.dir.name] if (forceMovie(f)) return [mov: detectMovie(f, false)] if (forceSeries(f)) return [tvs: detectSeriesName(f) ?: detectSeriesName(f.dir.listFiles{ it.isVideo() })] if (forceAnime(f)) return [anime: detectSeriesName(f) ?: detectSeriesName(f.dir.listFiles{ it.isVideo() })] def tvs = detectSeriesName(f) def mov = detectMovie(f, false) println "$f.name [series: $tvs, movie: $mov]" // DECIDE EPISODE VS MOVIE (IF NOT CLEAR) if (tvs && mov) { def norm = { s -> s.lower().space(' ') } def dn = norm(guessMovieFolder(f)?.name ?: '') def fn = norm(f.nameWithoutExtension) def sn = norm(tvs) def mn = norm(mov.name) // S00E00 | 2012.07.21 | One Piece 217 | Firefly - Serenity | [Taken 1, Taken 2, Taken 3, Taken 4, ..., Taken 10] if (parseEpisodeNumber(fn, true) || parseDate(fn) || (fn =~ sn && parseEpisodeNumber(fn.after(sn), false)) || fn.after(sn) =~ / - .+/ || f.dir.listFiles{ it.isVideo() && norm(it.name) =~ sn && it.name =~ /\b\d{1,3}\b/}.size() >= 10) { println "Exclude Movie: $mov" mov = null } else if ((detectMovie(f, true) && [dn, fn].find{ it =~ /(19|20)\d{2}/ }) || [dn, fn].find{ it =~ mn && !(it.after(mn) =~ /\b\d{1,3}\b/) }) { println "Exclude Series: $tvs" tvs = null } } // CHECK CONFLICT if (((mov && tvs) || (!mov && !tvs)) && failOnError) { throw new Exception("Media detection failed") } return [tvs: tvs, mov: mov, anime: null] } // log movie/series/anime detection results groups.each{ group, files -> _log.finest("Group: $group => ${files*.name}") } // process each batch groups.each{ group, files -> // fetch subtitles (but not for anime) if (subtitles && !group.anime) { subtitles.each{ languageCode -> files += getMissingSubtitles(file:files, output:'srt', encoding:'UTF-8', lang:languageCode) } } // EPISODE MODE if ((group.tvs || group.anime) && !group.mov) { // choose series / anime config def config = group.tvs ? [name:group.tvs, format:format.tvs, db:'TheTVDB', seasonFolder:true ] : [name:group.anime, format:format.anime, db:'AniDB', seasonFolder:false] def dest = rename(file: files, format: config.format, db: config.db) if (dest && artwork) { dest.mapByFolder().each{ dir, fs -> println "Fetching artwork for $dir from TheTVDB" def sxe = fs.findResult{ eps -> parseEpisodeNumber(eps) } def options = TheTVDB.search(config.name) if (options.isEmpty()) { println "TV Series not found: $config.name" return } options = options.sortBySimilarity(config.name, { s -> s.name }) fetchSeriesArtworkAndNfo(config.seasonFolder ? dir.dir : dir, dir, options[0], sxe && sxe.season > 0 ? sxe.season : 1) } } if (dest == null && failOnError) { throw new Exception("Failed to rename series: $config.name") } } // MOVIE MODE if (group.mov && !group.tvs && !group.anime) { def dest = rename(file:files, format:format.mov, db:'TheMovieDB') if (dest && artwork) { dest.mapByFolder().each{ dir, fs -> println "Fetching artwork for $dir from TheMovieDB" fetchMovieArtworkAndNfo(dir, group.mov, fs.findAll{ it.isVideo() }.sort{ it.length() }.reverse().findResult{ it }, backdrops) } } if (dest == null && failOnError) { throw new Exception("Failed to rename movie: $group.mov") } } // MUSIC MODE if (group.music) { def dest = rename(file:files, format:format.music, db:'AcoustID') if (dest == null && failOnError) { throw new Exception("Failed to rename music: $group.music") } } } // skip notifications if nothing was renamed anyway if (getRenameLog().isEmpty()) { return } // make xbmc or plex scan for new content xbmc?.each{ println "Notify XBMC: $it" invokeScanVideoLibrary(it) } plex?.each{ println "Notify Plex: $it" refreshPlexLibrary(it) } // mark episodes as 'acquired' if (myepisodes) { println 'Update MyEpisodes' include('fn:update-mes', [login:myepisodes.join(':'), addshows:false], getRenameLog().values()) } if (pushover) { // include webservice utility include('fn:lib/ws') println('Sending Pushover notification') Pushover(pushover).send("Finished processing ${tryQuietly { ut_title } ?: input*.dir.name.unique()} (${getRenameLog().size()} files).") } // send status email if (gmail) { // ant/mail utility include('fn:lib/ant') // send html mail def renameLog = getRenameLog() def emailTitle = tryQuietly { ut_title } ?: input*.dir.name.unique() sendGmail( subject: "[FileBot] ${emailTitle}", message: XML { html { body { p("FileBot finished processing ${emailTitle} (${renameLog.size()} files)."); hr(); table { th("Parameter"); th("Value") _args.bindings.findAll{ param -> param.key =~ /^ut_/ }.each{ param -> tr { [param.key, param.value].each{ td(it)} } } } hr(); table { th("Original Name"); th("New Name"); th("New Location") renameLog.each{ from, to -> tr { [from.name, to.name, to.parent].each{ cell -> td{ nobr{ code(cell) } } } } } } hr(); small("// Generated by ${net.sourceforge.filebot.Settings.applicationIdentifier} on ${new Date().dateString} at ${new Date().timeString}") } } }, messagemimetype: 'text/html', to: tryQuietly{ mailto } ?: gmail[0] + '@gmail.com', // mail to self by default user: gmail[0], password: gmail[1] ) } // clean empty folders, clutter files, etc after move if (clean) { if (['COPY', 'HARDLINK'].find{ it.equalsIgnoreCase(_args.action) } && extractedFiles?.size() > 0) { println 'Clean temporary extracted files' // delete extracted files extractedFiles.each{ if(it.isFile()) { println "Delete $it" it.delete() } } // delete remaining empty folders extractedFiles*.dir.unique().findAll{ it.listFiles().length == 0 }.each{ if(it.isDirectory()) { println "Delete $it" it.delete() } } } // deleting remaining files only makes sense after moving files if ('MOVE'.equalsIgnoreCase(_args.action)) { println 'Clean clutter files and empty folders' include('fn:cleaner', [:], !args.empty ? args : ut_kind == 'multi' && ut_dir ? [ut_dir as File] : []) } }