256 lines
7.7 KiB
Java
256 lines
7.7 KiB
Java
|
||
package net.sourceforge.filebot.cli;
|
||
|
||
|
||
import static net.sourceforge.filebot.Settings.*;
|
||
import static net.sourceforge.filebot.cli.CLILogging.*;
|
||
import static net.sourceforge.tuned.ExceptionUtilities.*;
|
||
import static net.sourceforge.tuned.FileUtilities.*;
|
||
|
||
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.URL;
|
||
import java.nio.ByteBuffer;
|
||
import java.nio.charset.Charset;
|
||
import java.security.AccessController;
|
||
import java.util.Collection;
|
||
import java.util.LinkedHashSet;
|
||
import java.util.MissingResourceException;
|
||
import java.util.logging.Level;
|
||
import java.util.regex.Pattern;
|
||
|
||
import javax.script.Bindings;
|
||
import javax.script.SimpleBindings;
|
||
|
||
import net.sourceforge.filebot.Analytics;
|
||
import net.sourceforge.filebot.MediaTypes;
|
||
import net.sourceforge.filebot.StandardRenameAction;
|
||
import net.sourceforge.filebot.cli.ScriptShell.Script;
|
||
import net.sourceforge.filebot.cli.ScriptShell.ScriptProvider;
|
||
import net.sourceforge.filebot.web.CachedResource;
|
||
|
||
import org.kohsuke.args4j.CmdLineException;
|
||
import org.kohsuke.args4j.CmdLineParser;
|
||
|
||
|
||
public class ArgumentProcessor {
|
||
|
||
public void printHelp(ArgumentBean argumentBean) {
|
||
new CmdLineParser(argumentBean).printUsage(System.out);
|
||
}
|
||
|
||
|
||
public ArgumentBean parse(String[] args) throws CmdLineException {
|
||
ArgumentBean bean = new ArgumentBean();
|
||
CmdLineParser parser = new CmdLineParser(bean);
|
||
parser.parseArgument(args);
|
||
return bean;
|
||
}
|
||
|
||
|
||
public int process(ArgumentBean args, CmdlineInterface cli) {
|
||
Analytics.trackView(ArgumentProcessor.class, "FileBot CLI");
|
||
CLILogger.setLevel(args.getLogLevel());
|
||
|
||
try {
|
||
// print episode info
|
||
if (args.list) {
|
||
for (String eps : cli.fetchEpisodeList(args.query, args.format, args.db, args.order, args.lang)) {
|
||
System.out.println(eps);
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
// print media info
|
||
if (args.mediaInfo) {
|
||
for (File file : args.getFiles(true)) {
|
||
System.out.println(cli.getMediaInfo(file, args.format));
|
||
}
|
||
return 0;
|
||
}
|
||
|
||
// execute CLI operations
|
||
if (args.script == null) {
|
||
// file operations
|
||
Collection<File> files = new LinkedHashSet<File>(args.getFiles(true));
|
||
|
||
if (args.extract) {
|
||
files.addAll(cli.extract(files, args.output, args.conflict, null, true));
|
||
}
|
||
|
||
if (args.getSubtitles) {
|
||
files.addAll(cli.getSubtitles(files, args.db, args.query, args.lang, args.output, args.encoding, !args.nonStrict));
|
||
} else if (args.getMissingSubtitles) {
|
||
files.addAll(cli.getMissingSubtitles(files, args.db, args.query, args.lang, args.output, args.encoding, !args.nonStrict));
|
||
}
|
||
|
||
if (args.rename) {
|
||
cli.rename(files, StandardRenameAction.forName(args.action), args.conflict, args.output, args.format, args.db, args.query, args.order, args.filter, args.lang, !args.nonStrict);
|
||
}
|
||
|
||
if (args.check) {
|
||
// check verification file
|
||
if (containsOnly(files, MediaTypes.getDefaultFilter("verification"))) {
|
||
if (!cli.check(files)) {
|
||
throw new Exception("Data corruption detected"); // one or more hashes mismatch
|
||
}
|
||
} else {
|
||
cli.compute(files, args.output, args.encoding);
|
||
}
|
||
}
|
||
} else {
|
||
// execute user script
|
||
System.setProperty("grape.root", new File(getApplicationFolder(), "grape").getAbsolutePath());
|
||
|
||
Bindings bindings = new SimpleBindings();
|
||
bindings.put("args", args.getFiles(false));
|
||
|
||
DefaultScriptProvider scriptProvider = new DefaultScriptProvider(args.trustScript);
|
||
URI script = scriptProvider.getScriptLocation(args.script);
|
||
|
||
if (!scriptProvider.isInlineScheme(script.getScheme())) {
|
||
if (scriptProvider.getResourceTemplate(script.getScheme()) != null) {
|
||
scriptProvider.setBaseScheme(new URI(script.getScheme(), "%s", null));
|
||
} else if ("file".equals(script.getScheme())) {
|
||
File base = new File(script).getParentFile();
|
||
File template = new File(base, "%s.groovy");
|
||
scriptProvider.setBaseScheme(template.toURI());
|
||
} else {
|
||
File base = new File(script.getPath()).getParentFile();
|
||
String template = normalizePathSeparators(new File(base, "%s.groovy").getPath());
|
||
scriptProvider.setBaseScheme(new URI(script.getScheme(), script.getHost(), template, script.getQuery(), script.getFragment()));
|
||
}
|
||
}
|
||
|
||
Analytics.trackEvent("CLI", "ExecuteScript", script.getScheme());
|
||
ScriptShell shell = new ScriptShell(cli, args, AccessController.getContext(), scriptProvider);
|
||
shell.runScript(script, bindings);
|
||
}
|
||
|
||
CLILogger.finest("Done ヾ(＠⌒ー⌒＠)ノ");
|
||
return 0;
|
||
} catch (Throwable e) {
|
||
CLILogger.log(Level.SEVERE, String.format("%s: %s", getRootCause(e).getClass().getSimpleName(), getRootCauseMessage(e)), getRootCause(e).getClass() == Exception.class ? null : getRootCause(e));
|
||
CLILogger.finest("Failure (°_°)");
|
||
return -1;
|
||
}
|
||
}
|
||
|
||
|
||
public static class DefaultScriptProvider implements ScriptProvider {
|
||
|
||
private final boolean trustRemoteScript;
|
||
|
||
private URI baseScheme;
|
||
|
||
|
||
public DefaultScriptProvider(boolean trustRemoteScript) {
|
||
this.trustRemoteScript = trustRemoteScript;
|
||
}
|
||
|
||
|
||
public void setBaseScheme(URI baseScheme) {
|
||
this.baseScheme = baseScheme;
|
||
}
|
||
|
||
|
||
public String getResourceTemplate(String scheme) {
|
||
try {
|
||
return getApplicationProperty("script." + scheme);
|
||
} catch (MissingResourceException e) {
|
||
return null;
|
||
}
|
||
}
|
||
|
||
|
||
public boolean isInlineScheme(String scheme) {
|
||
return "g".equals(scheme) || "system".equals(scheme);
|
||
}
|
||
|
||
|
||
@Override
|
||
public URI getScriptLocation(String input) {
|
||
try {
|
||
return new URL(input).toURI();
|
||
} catch (Exception _) {
|
||
try {
|
||
// system:in
|
||
if (input.equals("system:in")) {
|
||
return new URI("system", "in", null);
|
||
}
|
||
|
||
// g:println 'hello world'
|
||
if (input.startsWith("g:")) {
|
||
return new URI("g", input.substring(2), null);
|
||
}
|
||
|
||
// fn:sortivo / svn:sortivo
|
||
if (Pattern.matches("\\w+:.+", input)) {
|
||
String scheme = input.substring(0, input.indexOf(':'));
|
||
if (getResourceTemplate(scheme) != null) {
|
||
return new URI(scheme, input.substring(scheme.length() + 1, input.length()), null);
|
||
}
|
||
}
|
||
|
||
File file = new File(input);
|
||
if (baseScheme != null && !file.isAbsolute()) {
|
||
return new URI(baseScheme.getScheme(), String.format(baseScheme.getSchemeSpecificPart(), input), null);
|
||
}
|
||
|
||
// X:/foo/bar.groovy
|
||
if (!file.isFile()) {
|
||
throw new FileNotFoundException(file.getPath());
|
||
}
|
||
return file.getAbsoluteFile().toURI();
|
||
} catch (Exception e) {
|
||
throw new IllegalArgumentException(e);
|
||
}
|
||
}
|
||
}
|
||
|
||
|
||
@Override
|
||
public Script fetchScript(URI uri) throws IOException {
|
||
if (uri.getScheme().equals("file")) {
|
||
return new Script(readAll(new InputStreamReader(new FileInputStream(new File(uri)), "UTF-8")), true);
|
||
}
|
||
|
||
if (uri.getScheme().equals("system")) {
|
||
return new Script(readAll(new InputStreamReader(System.in)), true);
|
||
}
|
||
|
||
if (uri.getScheme().equals("g")) {
|
||
return new Script(uri.getSchemeSpecificPart(), true);
|
||
}
|
||
|
||
// remote script
|
||
String url;
|
||
boolean trusted;
|
||
|
||
String resolver = getResourceTemplate(uri.getScheme());
|
||
if (resolver != null) {
|
||
url = String.format(resolver, uri.getSchemeSpecificPart());
|
||
trusted = true;
|
||
} else {
|
||
url = uri.toString();
|
||
trusted = trustRemoteScript;
|
||
}
|
||
|
||
// fetch remote script only if modified
|
||
CachedResource<String> script = new CachedResource<String>(url, String.class, 24 * 60 * 60 * 1000) {
|
||
|
||
@Override
|
||
public String process(ByteBuffer data) {
|
||
return Charset.forName("UTF-8").decode(data).toString();
|
||
}
|
||
};
|
||
return new Script(script.get(), trusted);
|
||
}
|
||
}
|
||
|
||
}
|