* include(script) method for scripting
* fn:script handler for scripting * completed the utorrent-postprocess script with artwork/nfo/subtitles/etc * minor bugfixes
This commit is contained in:
parent
de73d1ac31
commit
077db7af74
|
@ -7,7 +7,14 @@ import static net.sourceforge.tuned.ExceptionUtilities.*;
|
||||||
import static net.sourceforge.tuned.FileUtilities.*;
|
import static net.sourceforge.tuned.FileUtilities.*;
|
||||||
|
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
|
import java.io.FileInputStream;
|
||||||
|
import java.io.FileNotFoundException;
|
||||||
|
import java.io.IOException;
|
||||||
|
import java.io.InputStreamReader;
|
||||||
import java.net.URI;
|
import java.net.URI;
|
||||||
|
import java.net.URL;
|
||||||
|
import java.nio.ByteBuffer;
|
||||||
|
import java.nio.charset.Charset;
|
||||||
import java.security.AccessController;
|
import java.security.AccessController;
|
||||||
import java.util.ArrayList;
|
import java.util.ArrayList;
|
||||||
import java.util.Collection;
|
import java.util.Collection;
|
||||||
|
@ -25,6 +32,8 @@ import org.kohsuke.args4j.CmdLineParser;
|
||||||
|
|
||||||
import net.sourceforge.filebot.Analytics;
|
import net.sourceforge.filebot.Analytics;
|
||||||
import net.sourceforge.filebot.MediaTypes;
|
import net.sourceforge.filebot.MediaTypes;
|
||||||
|
import net.sourceforge.filebot.cli.ScriptShell.ScriptProvider;
|
||||||
|
import net.sourceforge.filebot.web.CachedResource;
|
||||||
|
|
||||||
|
|
||||||
public class ArgumentProcessor {
|
public class ArgumentProcessor {
|
||||||
|
@ -111,11 +120,11 @@ public class ArgumentProcessor {
|
||||||
Bindings bindings = new SimpleBindings();
|
Bindings bindings = new SimpleBindings();
|
||||||
bindings.put("args", args.getFiles(false));
|
bindings.put("args", args.getFiles(false));
|
||||||
|
|
||||||
ScriptShell shell = new ScriptShell(cli, args, args.parameters, args.trustScript, AccessController.getContext());
|
ScriptProvider scriptProvider = new DefaultScriptProvider();
|
||||||
URI script = shell.getScriptLocation(args.script);
|
Analytics.trackEvent("CLI", "ExecuteScript", scriptProvider.getScriptLocation(args.script).getScheme());
|
||||||
|
|
||||||
Analytics.trackEvent("CLI", "ExecuteScript", script.getScheme());
|
ScriptShell shell = new ScriptShell(cli, args, args.parameters, args.trustScript, AccessController.getContext(), scriptProvider);
|
||||||
shell.run(script, bindings);
|
shell.runScript(args.script, bindings);
|
||||||
}
|
}
|
||||||
|
|
||||||
CLILogger.finest("Done ヾ(@⌒ー⌒@)ノ");
|
CLILogger.finest("Done ヾ(@⌒ー⌒@)ノ");
|
||||||
|
@ -133,4 +142,67 @@ public class ArgumentProcessor {
|
||||||
System.out.println(" -X<name>=<value> : Define script variable");
|
System.out.println(" -X<name>=<value> : Define script variable");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
public static class DefaultScriptProvider implements ScriptProvider {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public URI getScriptLocation(String input) {
|
||||||
|
try {
|
||||||
|
return new URL(input).toURI();
|
||||||
|
} catch (Exception eu) {
|
||||||
|
try {
|
||||||
|
// fn:sortivo
|
||||||
|
if (input.startsWith("fn:")) {
|
||||||
|
return new URI("http", "filebot.sourceforge.net", "/scripts/" + input.substring(3) + ".groovy", null);
|
||||||
|
}
|
||||||
|
|
||||||
|
// script:println 'hello world'
|
||||||
|
if (input.startsWith("script:")) {
|
||||||
|
return new URI("script", input.substring(7), null, null, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
// system:in
|
||||||
|
if (input.equals("system:in")) {
|
||||||
|
return new URI("system", "in", null, null, null);
|
||||||
|
}
|
||||||
|
|
||||||
|
// X:/foo/bar.groovy
|
||||||
|
File file = new File(input);
|
||||||
|
if (!file.isFile()) {
|
||||||
|
throw new FileNotFoundException(file.getPath());
|
||||||
|
}
|
||||||
|
return file.toURI();
|
||||||
|
} catch (Exception e) {
|
||||||
|
throw new IllegalArgumentException(e);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String fetchScript(URI uri) throws IOException {
|
||||||
|
if (uri.getScheme().equals("file")) {
|
||||||
|
return readAll(new InputStreamReader(new FileInputStream(new File(uri)), "UTF-8"));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (uri.getScheme().equals("system")) {
|
||||||
|
return readAll(new InputStreamReader(System.in));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (uri.getScheme().equals("script")) {
|
||||||
|
return uri.getAuthority();
|
||||||
|
}
|
||||||
|
|
||||||
|
// fetch remote script only if modified
|
||||||
|
CachedResource<String> script = new CachedResource<String>(uri.toString(), String.class, 24 * 60 * 60 * 1000) {
|
||||||
|
|
||||||
|
@Override
|
||||||
|
public String process(ByteBuffer data) {
|
||||||
|
return Charset.forName("UTF-8").decode(data).toString();
|
||||||
|
}
|
||||||
|
};
|
||||||
|
return script.get();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -498,7 +498,7 @@ public class CmdlineOperations implements CmdlineInterface {
|
||||||
HistorySpooler.getInstance().append(renameMap.entrySet());
|
HistorySpooler.getInstance().append(renameMap.entrySet());
|
||||||
|
|
||||||
// printer number of renamed files if any
|
// printer number of renamed files if any
|
||||||
CLILogger.fine(format("Renamed %d files", renameLog.size()));
|
CLILogger.fine(format("Processed %d files", renameLog.size()));
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -3,23 +3,14 @@ package net.sourceforge.filebot.cli;
|
||||||
|
|
||||||
|
|
||||||
import static net.sourceforge.filebot.cli.CLILogging.*;
|
import static net.sourceforge.filebot.cli.CLILogging.*;
|
||||||
import static net.sourceforge.tuned.FileUtilities.*;
|
|
||||||
|
|
||||||
import java.awt.AWTPermission;
|
import java.awt.AWTPermission;
|
||||||
import java.io.File;
|
import java.io.File;
|
||||||
import java.io.FileInputStream;
|
|
||||||
import java.io.FileNotFoundException;
|
|
||||||
import java.io.FilePermission;
|
import java.io.FilePermission;
|
||||||
import java.io.InputStreamReader;
|
import java.io.InputStreamReader;
|
||||||
import java.io.Reader;
|
|
||||||
import java.io.StringReader;
|
|
||||||
import java.lang.reflect.ReflectPermission;
|
import java.lang.reflect.ReflectPermission;
|
||||||
import java.net.SocketPermission;
|
import java.net.SocketPermission;
|
||||||
import java.net.URI;
|
import java.net.URI;
|
||||||
import java.net.URISyntaxException;
|
|
||||||
import java.net.URL;
|
|
||||||
import java.nio.ByteBuffer;
|
|
||||||
import java.nio.charset.Charset;
|
|
||||||
import java.security.AccessControlContext;
|
import java.security.AccessControlContext;
|
||||||
import java.security.AccessController;
|
import java.security.AccessController;
|
||||||
import java.security.Permissions;
|
import java.security.Permissions;
|
||||||
|
@ -45,7 +36,6 @@ import net.sourceforge.filebot.WebServices;
|
||||||
import net.sourceforge.filebot.format.AssociativeScriptObject;
|
import net.sourceforge.filebot.format.AssociativeScriptObject;
|
||||||
import net.sourceforge.filebot.format.ExpressionFormat;
|
import net.sourceforge.filebot.format.ExpressionFormat;
|
||||||
import net.sourceforge.filebot.format.PrivilegedInvocation;
|
import net.sourceforge.filebot.format.PrivilegedInvocation;
|
||||||
import net.sourceforge.filebot.web.CachedResource;
|
|
||||||
import net.sourceforge.filebot.web.EpisodeListProvider;
|
import net.sourceforge.filebot.web.EpisodeListProvider;
|
||||||
import net.sourceforge.filebot.web.MovieIdentificationService;
|
import net.sourceforge.filebot.web.MovieIdentificationService;
|
||||||
|
|
||||||
|
@ -55,8 +45,11 @@ class ScriptShell {
|
||||||
private final ScriptEngine engine = new GroovyScriptEngineFactory().getScriptEngine();
|
private final ScriptEngine engine = new GroovyScriptEngineFactory().getScriptEngine();
|
||||||
private final boolean trustScript;
|
private final boolean trustScript;
|
||||||
|
|
||||||
|
private final ScriptProvider scriptProvider;
|
||||||
|
|
||||||
public ScriptShell(CmdlineInterface cli, ArgumentBean args, Map<String, ?> parameters, boolean trustScript, AccessControlContext acc) throws ScriptException {
|
|
||||||
|
public ScriptShell(CmdlineInterface cli, ArgumentBean args, Map<String, ?> parameters, boolean trustScript, AccessControlContext acc, ScriptProvider scriptProvider) throws ScriptException {
|
||||||
|
this.scriptProvider = scriptProvider;
|
||||||
this.trustScript = trustScript;
|
this.trustScript = trustScript;
|
||||||
|
|
||||||
// setup script context
|
// setup script context
|
||||||
|
@ -70,100 +63,19 @@ class ScriptShell {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
protected Bindings initializeBindings(CmdlineInterface cli, ArgumentBean args, Map<String, ?> parameters, AccessControlContext acc) {
|
public static interface ScriptProvider {
|
||||||
Bindings bindings = new SimpleBindings();
|
|
||||||
|
|
||||||
// bind external parameters
|
public URI getScriptLocation(String input);
|
||||||
for (Entry<String, ?> it : parameters.entrySet()) {
|
|
||||||
bindings.put(it.getKey(), it.getValue());
|
|
||||||
}
|
|
||||||
|
|
||||||
// bind API objects
|
|
||||||
bindings.put("_cli", PrivilegedInvocation.newProxy(CmdlineInterface.class, cli, acc));
|
|
||||||
bindings.put("_script", new File(args.script));
|
|
||||||
bindings.put("_args", args);
|
|
||||||
|
|
||||||
bindings.put("_types", MediaTypes.getDefault());
|
public String fetchScript(URI uri) throws Exception;
|
||||||
bindings.put("_log", CLILogger);
|
|
||||||
|
|
||||||
// bind Java properties and environment variables
|
|
||||||
bindings.put("_parameter", new AssociativeScriptObject(parameters));
|
|
||||||
bindings.put("_system", new AssociativeScriptObject(System.getProperties()));
|
|
||||||
bindings.put("_environment", new AssociativeScriptObject(System.getenv()));
|
|
||||||
|
|
||||||
// bind console object
|
|
||||||
bindings.put("console", System.console());
|
|
||||||
|
|
||||||
// bind Episode data providers
|
|
||||||
for (EpisodeListProvider service : WebServices.getEpisodeListProviders()) {
|
|
||||||
bindings.put(service.getName(), service);
|
|
||||||
}
|
|
||||||
|
|
||||||
// bind Movie data providers
|
|
||||||
for (MovieIdentificationService service : WebServices.getMovieIdentificationServices()) {
|
|
||||||
bindings.put(service.getName(), service);
|
|
||||||
}
|
|
||||||
|
|
||||||
return bindings;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
public URI getScriptLocation(String input) {
|
public Object runScript(String input, Bindings bindings) throws Throwable {
|
||||||
try {
|
URI resource = scriptProvider.getScriptLocation(input);
|
||||||
return new URL(input).toURI();
|
String script = scriptProvider.fetchScript(resource);
|
||||||
} catch (Exception eu) {
|
return evaluate(script, bindings);
|
||||||
if (input.startsWith("script:")) {
|
|
||||||
try {
|
|
||||||
return new URI("script", input.substring(7), null, null, null);
|
|
||||||
} catch (URISyntaxException e) {
|
|
||||||
throw new IllegalArgumentException(e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
try {
|
|
||||||
File file = new File(input);
|
|
||||||
if (!file.exists()) {
|
|
||||||
throw new FileNotFoundException(file.getPath());
|
|
||||||
}
|
|
||||||
return file.toURI();
|
|
||||||
} catch (Exception e) {
|
|
||||||
throw new IllegalArgumentException(e);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
public Object run(String input, Bindings bindings) throws Throwable {
|
|
||||||
return run(getScriptLocation(input), bindings);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
public Object run(URI scriptLocation, Bindings bindings) throws Throwable {
|
|
||||||
if (scriptLocation.getScheme().equals("file")) {
|
|
||||||
return evalute(new InputStreamReader(new FileInputStream(new File(scriptLocation)), "UTF-8"), bindings);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (scriptLocation.getScheme().equals("system")) {
|
|
||||||
return evalute(new InputStreamReader(System.in), bindings);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (scriptLocation.getScheme().equals("script")) {
|
|
||||||
return evalute(new StringReader(scriptLocation.getAuthority()), bindings);
|
|
||||||
}
|
|
||||||
|
|
||||||
// fetch remote script only if modified
|
|
||||||
CachedResource<String> script = new CachedResource<String>(scriptLocation.toString(), String.class, 0) {
|
|
||||||
|
|
||||||
@Override
|
|
||||||
public String process(ByteBuffer data) {
|
|
||||||
return Charset.forName("UTF-8").decode(data).toString();
|
|
||||||
}
|
|
||||||
};
|
|
||||||
return evaluate(script.get(), bindings);
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
public Object evalute(Reader script, Bindings bindings) throws Throwable {
|
|
||||||
return evaluate(readAll(script), bindings);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
@ -190,6 +102,45 @@ class ScriptShell {
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
protected Bindings initializeBindings(CmdlineInterface cli, ArgumentBean args, Map<String, ?> parameters, AccessControlContext acc) {
|
||||||
|
Bindings bindings = new SimpleBindings();
|
||||||
|
|
||||||
|
// bind external parameters
|
||||||
|
for (Entry<String, ?> it : parameters.entrySet()) {
|
||||||
|
bindings.put(it.getKey(), it.getValue());
|
||||||
|
}
|
||||||
|
|
||||||
|
// bind API objects
|
||||||
|
bindings.put("_cli", PrivilegedInvocation.newProxy(CmdlineInterface.class, cli, acc));
|
||||||
|
bindings.put("_script", new File(args.script));
|
||||||
|
bindings.put("_args", args);
|
||||||
|
bindings.put("_shell", this);
|
||||||
|
|
||||||
|
bindings.put("_types", MediaTypes.getDefault());
|
||||||
|
bindings.put("_log", CLILogger);
|
||||||
|
|
||||||
|
// bind Java properties and environment variables
|
||||||
|
bindings.put("_parameter", new AssociativeScriptObject(parameters));
|
||||||
|
bindings.put("_system", new AssociativeScriptObject(System.getProperties()));
|
||||||
|
bindings.put("_environment", new AssociativeScriptObject(System.getenv()));
|
||||||
|
|
||||||
|
// bind console object
|
||||||
|
bindings.put("console", System.console());
|
||||||
|
|
||||||
|
// bind Episode data providers
|
||||||
|
for (EpisodeListProvider service : WebServices.getEpisodeListProviders()) {
|
||||||
|
bindings.put(service.getName(), service);
|
||||||
|
}
|
||||||
|
|
||||||
|
// bind Movie data providers
|
||||||
|
for (MovieIdentificationService service : WebServices.getMovieIdentificationServices()) {
|
||||||
|
bindings.put(service.getName(), service);
|
||||||
|
}
|
||||||
|
|
||||||
|
return bindings;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
protected AccessControlContext getSandboxAccessControlContext() {
|
protected AccessControlContext getSandboxAccessControlContext() {
|
||||||
Permissions permissions = new Permissions();
|
Permissions permissions = new Permissions();
|
||||||
|
|
||||||
|
|
|
@ -182,6 +182,14 @@ List.metaClass.sortBySimilarity = { prime, Closure toStringFunction = { obj -> o
|
||||||
return delegate.sort{ a, b -> similarity(toStringFunction(b), prime).compareTo(similarity(toStringFunction(a), prime)) }
|
return delegate.sort{ a, b -> similarity(toStringFunction(b), prime).compareTo(similarity(toStringFunction(a), prime)) }
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// call scripts
|
||||||
|
def include(String input, Map parameters = [:], Object... args) {
|
||||||
|
// initialize default parameter
|
||||||
|
parameters.args = (args as List).findResults{ it as File }
|
||||||
|
|
||||||
|
// run given script and catch exceptions
|
||||||
|
_guarded { _shell.runScript(input, new javax.script.SimpleBindings(parameters)) }
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
// CLI bindings
|
// CLI bindings
|
||||||
|
|
|
@ -76,6 +76,12 @@ public abstract class CachedResource<T extends Serializable> {
|
||||||
} catch (Exception e) {
|
} catch (Exception e) {
|
||||||
throw new IOException(e);
|
throw new IOException(e);
|
||||||
}
|
}
|
||||||
|
} else {
|
||||||
|
try {
|
||||||
|
product = type.cast(element.getValue());
|
||||||
|
} catch (Exception e) {
|
||||||
|
Logger.getLogger(getClass().getName()).log(Level.WARNING, e.getMessage());
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
|
|
|
@ -0,0 +1,136 @@
|
||||||
|
// xbmc functions
|
||||||
|
def invokeScanVideoLibrary(host, port = 9090) {
|
||||||
|
try {
|
||||||
|
telnet(host, 9090) { writer, reader ->
|
||||||
|
writer.println('{"id":1,"method":"VideoLibrary.Scan","params":[],"jsonrpc":"2.0"}') // API call for latest XBMC release
|
||||||
|
}
|
||||||
|
return true
|
||||||
|
} catch(e) {
|
||||||
|
println "${e.class.simpleName}: ${e.message}"
|
||||||
|
return false
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
// functions for TheTVDB artwork/nfo
|
||||||
|
def fetchSeriesBanner(outputFile, series, bannerType, bannerType2, season, locale) {
|
||||||
|
// select and fetch banner
|
||||||
|
def banner = [locale, null].findResult { TheTVDB.getBanner(series, [BannerType:bannerType, BannerType2:bannerType2, Season:season, Language:it]) }
|
||||||
|
if (banner == null) {
|
||||||
|
println "Banner not found: $outputFile / $bannerType:$bannerType2"
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
println "Fetching $outputFile => $banner"
|
||||||
|
return banner.url.saveAs(outputFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
def fetchSeriesNfo(outputFile, series, locale) {
|
||||||
|
def info = TheTVDB.getSeriesInfo(series, locale)
|
||||||
|
info.applyXmlTemplate('''<tvshow xmlns:gsp='http://groovy.codehaus.org/2005/gsp'>
|
||||||
|
<title>$name</title>
|
||||||
|
<year>$firstAired.year</year>
|
||||||
|
<top250></top250>
|
||||||
|
<seasons>-1</seasons>
|
||||||
|
<episode></episode>
|
||||||
|
<episodeguideurl></episodeguideurl>
|
||||||
|
<displayseason>-1</displayseason>
|
||||||
|
<displayepisode>-1</displayepisode>
|
||||||
|
<rating>$rating</rating>
|
||||||
|
<votes>$ratingCount</votes>
|
||||||
|
<outline></outline>
|
||||||
|
<plot>$overview</plot>
|
||||||
|
<tagline></tagline>
|
||||||
|
<runtime>$runtime</runtime>
|
||||||
|
<mpaa>$contentRating</mpaa>
|
||||||
|
<playcount></playcount>
|
||||||
|
<lastplayed></lastplayed>
|
||||||
|
<id>$id</id>
|
||||||
|
<episodeguide><url cache="${id}.xml">http://www.thetvdb.com/api/1D62F2F90030C444/series/${id}/all/''' + locale.language + '''.zip</url></episodeguide>
|
||||||
|
<genre>${!genres.empty ? genres[0] : ''}</genre>
|
||||||
|
<set></set>
|
||||||
|
<credits></credits>
|
||||||
|
<director></director>
|
||||||
|
<thumb>$bannerUrl</thumb>
|
||||||
|
<premiered>$firstAired</premiered>
|
||||||
|
<status>$status</status>
|
||||||
|
<studio>$network</studio>
|
||||||
|
<trailer></trailer>
|
||||||
|
<gsp:scriptlet> actors.each { </gsp:scriptlet>
|
||||||
|
<actor>
|
||||||
|
<name>$it</name>
|
||||||
|
<role></role>
|
||||||
|
</actor>
|
||||||
|
<gsp:scriptlet> } </gsp:scriptlet>
|
||||||
|
<artist></artist>
|
||||||
|
</tvshow>
|
||||||
|
''')
|
||||||
|
.replaceAll(/\t|\r|\n/, '') // xbmc can't handle leading/trailing whitespace properly
|
||||||
|
.saveAs(outputFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
def fetchSeriesArtworkAndNfo(seriesDir, seasonDir, series, season, locale = _args.locale) {
|
||||||
|
try {
|
||||||
|
// fetch nfo
|
||||||
|
fetchSeriesNfo(seriesDir['tvshow.nfo'], series, locale)
|
||||||
|
|
||||||
|
// fetch series banner, fanart, posters, etc
|
||||||
|
["680x1000", null].findResult{ fetchSeriesBanner(seriesDir['folder.jpg'], series, "poster", it, null, locale) }
|
||||||
|
["graphical", null].findResult{ fetchSeriesBanner(seriesDir['banner.jpg'], series, "series", it, null, locale) }
|
||||||
|
|
||||||
|
// fetch highest resolution fanart
|
||||||
|
["1920x1080", "1280x720", null].findResult{ fetchSeriesBanner(seriesDir["fanart.jpg"], series, "fanart", it, null, locale) }
|
||||||
|
|
||||||
|
// fetch season banners
|
||||||
|
if (seasonDir != seriesDir) {
|
||||||
|
fetchSeriesBanner(seasonDir["folder.jpg"], series, "season", "season", season, locale)
|
||||||
|
fetchSeriesBanner(seasonDir["banner.jpg"], series, "season", "seasonwide", season, locale)
|
||||||
|
}
|
||||||
|
} catch(e) {
|
||||||
|
println "${e.class.simpleName}: ${e.message}"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
// functions for TheMovieDB artwork/nfo
|
||||||
|
def fetchMovieArtwork(outputFile, movieInfo, artworkType, artworkSize) {
|
||||||
|
// select and fetch artwork
|
||||||
|
def artwork = movieInfo.images.find { it.type == artworkType && it.size == artworkSize }
|
||||||
|
if (artwork == null) {
|
||||||
|
println "Artwork not found: $outputFile"
|
||||||
|
return null
|
||||||
|
}
|
||||||
|
println "Fetching $outputFile => $artwork"
|
||||||
|
return artwork.url.saveAs(outputFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
def fetchMovieNfo(outputFile, movieInfo) {
|
||||||
|
movieInfo.applyXmlTemplate('''<movie>
|
||||||
|
<title>$name</title>
|
||||||
|
<year>$released.year</year>
|
||||||
|
<rating>$rating</rating>
|
||||||
|
<votes>$votes</votes>
|
||||||
|
<plot>$overview</plot>
|
||||||
|
<runtime>$runtime</runtime>
|
||||||
|
<mpaa>$certification</mpaa>
|
||||||
|
<genre>${!genres.empty ? genres[0] : ''}</genre>
|
||||||
|
<id>tt${imdbId.pad(7)}</id>
|
||||||
|
</movie>
|
||||||
|
''')
|
||||||
|
.replaceAll(/\t|\r|\n/, '') // xbmc can't handle leading/trailing whitespace properly
|
||||||
|
.saveAs(outputFile)
|
||||||
|
}
|
||||||
|
|
||||||
|
def fetchMovieArtworkAndNfo(movieDir, movie, locale = _args.locale) {
|
||||||
|
try {
|
||||||
|
def movieInfo = TheMovieDB.getMovieInfo(movie, locale)
|
||||||
|
// fetch nfo
|
||||||
|
fetchMovieNfo(movieDir['movie.nfo'], movieInfo)
|
||||||
|
|
||||||
|
// fetch series banner, fanart, posters, etc
|
||||||
|
fetchMovieArtwork(movieDir['folder.jpg'], movieInfo, 'poster', 'original')
|
||||||
|
fetchMovieArtwork(movieDir['backdrop.jpg'], movieInfo, 'backdrop', 'original')
|
||||||
|
} catch(e) {
|
||||||
|
println "${e.class.simpleName}: ${e.message}"
|
||||||
|
}
|
||||||
|
}
|
|
@ -1,4 +1,4 @@
|
||||||
// filebot -script "http://filebot.sf.net/scripts/utorrent-postprocess.groovy" --output "X:/media" --action copy --conflict override -non-strict -trust-script -Xxbmc=localhost "-Xut_dir=%D" "-Xut_file=%F" "-Xut_label=%L" "-Xut_state=%S" "-Xut_kind=%K"
|
// filebot -script "fn:utorrent-postprocess" --output "X:/media" --action copy --conflict override -non-strict -trust-script -Xxbmc=localhost "-Xut_dir=%D" "-Xut_file=%F" "-Xut_label=%L" "-Xut_state=%S" "-Xut_kind=%K"
|
||||||
def input = []
|
def input = []
|
||||||
|
|
||||||
// print input parameters
|
// print input parameters
|
||||||
|
@ -19,6 +19,9 @@ input = input.findAll{ it.isVideo() || it.isSubtitle() }
|
||||||
// print input fileset
|
// print input fileset
|
||||||
input.each{ println "Input: $it" }
|
input.each{ println "Input: $it" }
|
||||||
|
|
||||||
|
// xbmc artwork/nfo utility
|
||||||
|
include("fn:lib/xbmc")
|
||||||
|
|
||||||
// group episodes/movies and rename according to XBMC standards
|
// group episodes/movies and rename according to XBMC standards
|
||||||
def groups = input.groupBy{
|
def groups = input.groupBy{
|
||||||
def tvs = detectSeriesName(it)
|
def tvs = detectSeriesName(it)
|
||||||
|
@ -39,38 +42,35 @@ def groups = input.groupBy{
|
||||||
}
|
}
|
||||||
|
|
||||||
groups.each{ group, files ->
|
groups.each{ group, files ->
|
||||||
|
// fetch subtitles
|
||||||
|
def subs = getMissingSubtitles(file:files, output:"srt", encoding:"utf-8")
|
||||||
|
if (subs) files += subs
|
||||||
|
|
||||||
// EPISODE MODE
|
// EPISODE MODE
|
||||||
if (group.tvs && !group.mov) {
|
if (group.tvs && !group.mov) {
|
||||||
def output = rename(file:files, format:'TV Shows/{n}{episode.special ? "/Special" : "/Season "+s}/{n} - {episode.special ? "S00E"+special.pad(2) : s00e00} - {t}', db:'TheTVDB')
|
def dest = rename(file:files, format:'TV Shows/{n}/{episode.special ? "Special" : "Season "+s}/{n} - {episode.special ? "S00E"+special.pad(2) : s00e00} - {t}', db:'TheTVDB')
|
||||||
|
|
||||||
output*.dir.unique().each{ dir ->
|
dest.mapByFolder().keySet().each{ dir ->
|
||||||
println "Fetching artwork for $dir from TheTVDB"
|
println "Fetching artwork for $dir from TheTVDB"
|
||||||
def query = group.tvs
|
def query = group.tvs
|
||||||
def sxe = output.findResult{ parseEpisodeNumber(it) }
|
def sxe = dest.findResult{ parseEpisodeNumber(it) }
|
||||||
def options = TheTVDB.search(query, _args.locale)
|
def options = TheTVDB.search(query)
|
||||||
if (options.isEmpty()) {
|
if (options.isEmpty()) {
|
||||||
println "TV Series not found: $query"
|
println "TV Series not found: $query"
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
options = options.sortBySimilarity(query, { it.name })
|
options = options.sortBySimilarity(query, { it.name })
|
||||||
def series = options[0]
|
fetchSeriesArtworkAndNfo(dir.dir, dir, options[0], sxe && sxe.season > 0 ? sxe.season : 1)
|
||||||
def seriesDir = [dir.dir, dir].sortBySimilarity(series.name, { it.name })[0]
|
|
||||||
def season = sxe && sxe.season > 0 ? sxe.season : 1
|
|
||||||
fetchSeriesBannersAndNfo(seriesDir, dir, series, season)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
// MOVIE MODE
|
// MOVIE MODE
|
||||||
if (group.mov && !group.tvs) {
|
if (group.mov && !group.tvs) {
|
||||||
def output = rename(file:files, format:'Movies/{n} ({y})/{n} ({y}){" CD$pi"}', db:'TheMovieDB')
|
def dest = rename(file:files, format:'Movies/{n} ({y})/{n} ({y}){" CD$pi"}', db:'TheMovieDB')
|
||||||
|
|
||||||
output*.dir.unique().each{ dir ->
|
dest.mapByFolder().keySet().each{ dir ->
|
||||||
println "Fetching artwork for $dir from TheMovieDB"
|
println "Fetching artwork for $dir from TheMovieDB"
|
||||||
try {
|
fetchMovieArtworkAndNfo(dir, group.mov)
|
||||||
fetchMovieArtworkAndNfo(dir, group.mov)
|
|
||||||
} catch(e) {
|
|
||||||
println "${e.class.simpleName}: ${e.message}"
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -78,124 +78,7 @@ groups.each{ group, files ->
|
||||||
|
|
||||||
|
|
||||||
// make XBMC scan for new content
|
// make XBMC scan for new content
|
||||||
try {
|
xbmc.split(/[\s,|]+/).each{
|
||||||
xbmc.split(/[\s,|]+/).each{
|
println "Notify XMBC: $it"
|
||||||
println "Notify XBMC: $it"
|
invokeScanVideoLibrary(it)
|
||||||
telnet(it, 9090) { writer, reader ->
|
|
||||||
def msg = '{"id":1,"method":"VideoLibrary.Scan","params":[],"jsonrpc":"2.0"}'
|
|
||||||
writer.println(msg)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
} catch(e) {
|
|
||||||
println "${e.class.simpleName}: ${e.message}"
|
|
||||||
}
|
|
||||||
// END OF SCRIPT
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
// FUNCTIONS for TMDB and TVDB artwork/nfo
|
|
||||||
def fetchBanner(outputFile, series, bannerType, bannerType2 = null, season = null) {
|
|
||||||
// select and fetch banner
|
|
||||||
def banner = [_args.locale.language, null].findResult { TheTVDB.getBanner(series, [BannerType:bannerType, BannerType2:bannerType2, Season:season, Language:it]) }
|
|
||||||
if (banner == null) {
|
|
||||||
println "Banner not found: $outputFile / $bannerType:$bannerType2"
|
|
||||||
return null
|
|
||||||
}
|
|
||||||
println "Fetching $outputFile => $banner"
|
|
||||||
return banner.url.saveAs(outputFile)
|
|
||||||
}
|
|
||||||
def fetchSeriesNfo(outputFile, series) {
|
|
||||||
def info = TheTVDB.getSeriesInfo(series, _args.locale)
|
|
||||||
info.applyXmlTemplate('''<tvshow xmlns:gsp='http://groovy.codehaus.org/2005/gsp'>
|
|
||||||
<title>$name</title>
|
|
||||||
<year>$firstAired.year</year>
|
|
||||||
<top250></top250>
|
|
||||||
<seasons>-1</seasons>
|
|
||||||
<episode></episode>
|
|
||||||
<episodeguideurl></episodeguideurl>
|
|
||||||
<displayseason>-1</displayseason>
|
|
||||||
<displayepisode>-1</displayepisode>
|
|
||||||
<rating>$rating</rating>
|
|
||||||
<votes>$ratingCount</votes>
|
|
||||||
<outline></outline>
|
|
||||||
<plot>$overview</plot>
|
|
||||||
<tagline></tagline>
|
|
||||||
<runtime>$runtime</runtime>
|
|
||||||
<mpaa>$contentRating</mpaa>
|
|
||||||
<playcount></playcount>
|
|
||||||
<lastplayed></lastplayed>
|
|
||||||
<id>$id</id>
|
|
||||||
<episodeguide><url cache="${id}.xml">http://www.thetvdb.com/api/1D62F2F90030C444/series/${id}/all/''' + _args.locale.language + '''.zip</url></episodeguide>
|
|
||||||
<genre>${!genres.empty ? genres[0] : ''}</genre>
|
|
||||||
<set></set>
|
|
||||||
<credits></credits>
|
|
||||||
<director></director>
|
|
||||||
<thumb>$bannerUrl</thumb>
|
|
||||||
<premiered>$firstAired</premiered>
|
|
||||||
<status>$status</status>
|
|
||||||
<studio>$network</studio>
|
|
||||||
<trailer></trailer>
|
|
||||||
<gsp:scriptlet> actors.each { </gsp:scriptlet>
|
|
||||||
<actor>
|
|
||||||
<name>$it</name>
|
|
||||||
<role></role>
|
|
||||||
</actor>
|
|
||||||
<gsp:scriptlet> } </gsp:scriptlet>
|
|
||||||
<artist></artist>
|
|
||||||
</tvshow>
|
|
||||||
''')
|
|
||||||
.replaceAll(/\t|\r|\n/, '') // xbmc can't handle leading/trailing whitespace properly
|
|
||||||
.saveAs(outputFile)
|
|
||||||
}
|
|
||||||
def fetchSeriesBannersAndNfo(seriesDir, seasonDir, series, season) {
|
|
||||||
println "Fetch nfo and banners for $series / Season $season"
|
|
||||||
// fetch nfo
|
|
||||||
fetchSeriesNfo(seriesDir['tvshow.nfo'], series)
|
|
||||||
// fetch series banner, fanart, posters, etc
|
|
||||||
["680x1000", null].findResult{ fetchBanner(seriesDir['folder.jpg'], series, "poster", it) }
|
|
||||||
["graphical", null].findResult{ fetchBanner(seriesDir['banner.jpg'], series, "series", it) }
|
|
||||||
// fetch highest resolution fanart
|
|
||||||
["1920x1080", "1280x720", null].findResult{ fetchBanner(seriesDir["fanart.jpg"], series, "fanart", it) }
|
|
||||||
// fetch season banners
|
|
||||||
if (seasonDir != seriesDir) {
|
|
||||||
fetchBanner(seasonDir["folder.jpg"], series, "season", "season", season)
|
|
||||||
fetchBanner(seasonDir["banner.jpg"], series, "season", "seasonwide", season)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
def fetchArtwork(outputFile, movieInfo, artworkType, artworkSize) {
|
|
||||||
// select and fetch artwork
|
|
||||||
def artwork = movieInfo.images.find { it.type == artworkType && it.size == artworkSize }
|
|
||||||
if (artwork == null) {
|
|
||||||
println "Artwork not found: $outputFile"
|
|
||||||
return null
|
|
||||||
}
|
|
||||||
println "Fetching $outputFile => $artwork"
|
|
||||||
return artwork.url.saveAs(outputFile)
|
|
||||||
}
|
|
||||||
def fetchMovieNfo(outputFile, movieInfo) {
|
|
||||||
movieInfo.applyXmlTemplate('''<movie>
|
|
||||||
<title>$name</title>
|
|
||||||
<year>$released.year</year>
|
|
||||||
<rating>$rating</rating>
|
|
||||||
<votes>$votes</votes>
|
|
||||||
<plot>$overview</plot>
|
|
||||||
<runtime>$runtime</runtime>
|
|
||||||
<mpaa>$certification</mpaa>
|
|
||||||
<genre>${!genres.empty ? genres[0] : ''}</genre>
|
|
||||||
<id>tt${imdbId.pad(7)}</id>
|
|
||||||
</movie>
|
|
||||||
''')
|
|
||||||
.replaceAll(/\t|\r|\n/, '') // xbmc can't handle leading/trailing whitespace properly
|
|
||||||
.saveAs(outputFile)
|
|
||||||
}
|
|
||||||
def fetchMovieArtworkAndNfo(movieDir, movie) {
|
|
||||||
println "Fetch nfo and artwork for $movie"
|
|
||||||
def movieInfo = TheMovieDB.getMovieInfo(movie, Locale.ENGLISH)
|
|
||||||
// fetch nfo
|
|
||||||
fetchMovieNfo(movieDir['movie.nfo'], movieInfo)
|
|
||||||
// fetch series banner, fanart, posters, etc
|
|
||||||
fetchArtwork(movieDir['folder.jpg'], movieInfo, 'poster', 'original')
|
|
||||||
fetchArtwork(movieDir['backdrop.jpg'], movieInfo, 'backdrop', 'original')
|
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue