filebot/source/net/filebot/cli/ArgumentProcessor.java

package net.filebot.cli;

import static java.nio.charset.StandardCharsets.*;
import static net.filebot.ExitCode.*;
import static net.filebot.Logging.*;
import static net.filebot.MediaTypes.*;
import static net.filebot.Settings.*;
import static net.filebot.util.ExceptionUtilities.*;
import static net.filebot.util.FileUtilities.*;

import java.io.File;
import java.util.LinkedHashSet;
import java.util.Set;
import java.util.logging.Level;
import java.util.stream.Stream;

import javax.script.Bindings;
import javax.script.SimpleBindings;

import org.apache.commons.io.IOUtils;

import net.filebot.LicenseError;

public class ArgumentProcessor {

	public int run(ArgumentBean args) {
		try {
			// interactive mode enables basic selection and confirmation dialogs in the CLI
			CmdlineInterface cli = args.isInteractive() ? new CmdlineOperationsTextUI() : new CmdlineOperations();

			// execute simple command
			if (args.script == null) {
				return runCommand(cli, args);
			}

			// execute user script
			runScript(cli, args);

			// script finished successfully
			return finish(SUCCESS);
		} catch (Throwable e) {
			// script failed with exception -> exit with non-zero exit code (and use positive code to avoid issues with launch4j launcher)
			return finish(getExitCode(e));
		}
	}

	private int finish(int code) {
		log.finest(format("%s %s", getErrorMessage(code), getErrorKaomoji(code)));
		return code;
	}

	private int getExitCode(Throwable e) {
		if (findCause(e, LicenseError.class) != null) {
			log.severe(message("License Error", e.getMessage()));
			if (LICENSE.isFile()) {
				printStegosaurus("Please purchase a FileBot License:", getPurchaseURL());

				// on Linux, if there is no interactive console, then we are likely called by a background service running as a different user
				if (isLinuxApp() && System.console() == null) {
					log.severe(format("FileBot requires a valid license. Please run `sudo -H -u %s filebot --license *.psm` to install your FileBot license.", System.getProperty("user.name")));
				} else {
					log.severe(format("FileBot requires a valid license. Please run `filebot --license *.psm` install your FileBot license."));
				}
			}
			return BAD_LICENSE;
		}

		if (findCause(e, CmdlineException.class) != null) {
			log.log(Level.WARNING, findCause(e, CmdlineException.class)::getMessage);
			return FAILURE;
		}

		if (findCause(e, ScriptDeath.class) != null) {
			ScriptDeath d = findCause(e, ScriptDeath.class);
			log.log(Level.WARNING, d::getMessage);
			return d.getExitCode();
		}

		log.log(Level.SEVERE, e, e::getMessage);
		return ERROR;
	}

	public int runCommand(CmdlineInterface cli, ArgumentBean args) throws Exception {
		// print episode info or rename files in linear order
		if (args.list) {
			if (args.rename) {
				// rename files in linear order
				return cli.rename(args.getEpisodeListProvider(), args.getSearchQuery(), args.getExpressionFileFormat(), args.getExpressionFilter(), args.getSortOrder(), args.getLanguage().getLocale(), args.isStrict(), args.getFiles(true), args.getRenameAction(), args.getConflictAction(), args.getOutputPath(), args.getExecCommand()).isEmpty() ? 1 : 0;
			} else {
				// print episode info
				return print(cli.fetchEpisodeList(args.getEpisodeListProvider(), args.getSearchQuery(), args.getExpressionFormat(), args.getExpressionFilter(), args.getSortOrder(), args.getLanguage().getLocale(), args.isStrict()));
			}
		}

		// print media info or execute commands based on media info
		if (args.mediaInfo) {
			ExecCommand exec = args.getExecCommand();
			if (exec != null) {
				// execute command for each file
				return cli.execute(args.getFiles(true), args.getFileFilter(), args.getExpressionFormat(), exec).allMatch(r -> r == SUCCESS) ? SUCCESS : ERROR;
			} else {
				// print media info
				return print(cli.getMediaInfo(args.getFiles(true), args.getFileFilter(), args.getExpressionFormat()));
			}
		}

		// revert files
		if (args.revert) {
			return cli.revert(args.getFiles(false), args.getFileFilter(), args.getRenameAction()).size() > 0 ? SUCCESS : ERROR;
		}

		// file operations
		Set<File> files = new LinkedHashSet<File>(args.getFiles(true));

		if (args.extract) {
			files.addAll(cli.extract(files, args.getOutputPath(), args.getConflictAction(), null, args.isStrict()));
		}

		if (args.getSubtitles) {
			files.addAll(cli.getMissingSubtitles(files, args.getSearchQuery(), args.getLanguage(), args.getSubtitleOutputFormat(), args.getEncoding(), args.getSubtitleNamingFormat(), args.isStrict()));
		}

		if (args.rename) {
			cli.rename(files, args.getRenameAction(), args.getConflictAction(), args.getAbsoluteOutputFolder(), args.getExpressionFileFormat(), args.getDatasource(), args.getSearchQuery(), args.getSortOrder(), args.getExpressionFilter(), args.getLanguage().getLocale(), args.isStrict(), args.getExecCommand());
		}

		if (args.check) {
			// check verification file
			if (containsOnly(files, VERIFICATION_FILES)) {
				if (!cli.check(files)) {
					throw new Exception("Data corruption detected"); // one or more hashes do not match
				}
			} else {
				cli.compute(files, args.getOutputPath(), args.getOutputHashType(), args.getEncoding());
			}
		}

		return SUCCESS;
	}

	private int print(Stream<?> values) {
		return values.peek(System.out::println).mapToInt(v -> 1).sum() > 0 ? SUCCESS : ERROR;
	}

	private void printStegosaurus(String line1, String line2) {
		try {
			log.info(format(IOUtils.toString(getClass().getResource("Stegosaurus.format"), UTF_8), line1, line2));
		} catch (Exception e) {
			debug.log(Level.WARNING, e::toString);
		}
	}

	public void runScript(CmdlineInterface cli, ArgumentBean args) throws Throwable {
		Bindings bindings = new SimpleBindings();
		bindings.put(ScriptShell.SHELL_ARGS_BINDING_NAME, args);
		bindings.put(ScriptShell.ARGV_BINDING_NAME, args.getFiles(false));

		ScriptSource source = ScriptSource.findScriptProvider(args.script);
		ScriptShell shell = new ScriptShell(source.getScriptProvider(args.script), cli, args.defines);
		shell.runScript(source.accept(args.script), bindings);
	}

}