Examples with ScriptingContext org.pepsoft.worldpainter.tools.scripts.ScriptingContext used on opensource projects

Example 1 with ScriptingContext

use of org.pepsoft.worldpainter.tools.scripts.ScriptingContext in project WorldPainter by Captain-Chaos.

the class ScriptingTool method main.

public static void main(String[] args) throws IOException, ClassNotFoundException {
    // Initialise logging
    LoggerContext logContext = (LoggerContext) LoggerFactory.getILoggerFactory();
    try {
        JoranConfigurator configurator = new JoranConfigurator();
        configurator.setContext(logContext);
        logContext.reset();
        configurator.doConfigure(ClassLoader.getSystemResourceAsStream("logback-scriptingtool.xml"));
    } catch (JoranException e) {
    // StatusPrinter will handle this
    }
    StatusPrinter.printInCaseOfErrorsOrWarnings(logContext);
    System.err.println("WorldPainter scripting host version " + Version.VERSION + ".\n" + "Copyright 2011-2018 pepsoft.org, The Netherlands.\n" + "This is free software distributed under the terms of the GPL, version 3, a copy\n" + "of which you can find in the installation directory.\n");
    // Check arguments
    if (args.length < 1) {
        System.err.println("Usage:\n" + "\n" + "    wpscript <scriptfile> [<scriptarg> ...]\n" + "\n" + "Where <scriptfile> is the filename, including extension, of the script to\n" + "execute, and [<scriptarg> ...] an optional list of one or more arguments for\n" + "the script, which will be available to the script in the arguments (from index\n" + "0) or argv (from index 1) array.");
        System.exit(1);
    }
    // Load script
    File scriptFile = new File(args[0]);
    if (!scriptFile.isFile()) {
        System.err.println(args[0] + " does not exist or is not a regular file");
        System.exit(1);
    }
    String scriptFileName = scriptFile.getName();
    int p = scriptFileName.lastIndexOf('.');
    if (p == -1) {
        System.err.println("Script file name " + scriptFileName + " has no extension");
        System.exit(1);
    }
    String extension = scriptFileName.substring(p + 1);
    ScriptEngineManager scriptEngineManager = new ScriptEngineManager();
    ScriptEngine scriptEngine = scriptEngineManager.getEngineByExtension(extension);
    if (scriptEngine == null) {
        System.err.println("Script file language " + extension + " not supported");
        System.exit(1);
    }
    scriptEngine.put(ScriptEngine.FILENAME, scriptFileName);
    // Initialise WorldPainter configuration
    Configuration config = Configuration.load();
    if (config == null) {
        logger.info("Creating new configuration");
        config = new Configuration();
    }
    Configuration.setInstance(config);
    // Load trusted WorldPainter root certificate
    X509Certificate trustedCert = null;
    try {
        CertificateFactory certificateFactory = CertificateFactory.getInstance("X.509");
        trustedCert = (X509Certificate) certificateFactory.generateCertificate(ClassLoader.getSystemResourceAsStream("wproot.pem"));
    } catch (CertificateException e) {
        logger.error("Certificate exception while loading trusted root certificate", e);
    }
    // Load the plugins
    if (trustedCert != null) {
        File pluginsDir = new File(Configuration.getConfigDir(), "plugins");
        if (pluginsDir.isDirectory()) {
            PluginManager.loadPlugins(pluginsDir, trustedCert.getPublicKey());
        }
    } else {
        logger.error("Trusted root certificate not available; not loading plugins");
    }
    WPPluginManager.initialise(config.getUuid());
    if (args.length > 1) {
        System.err.print("Executing script \"" + scriptFileName + "\" with arguments ");
        for (int i = 1; i < args.length; i++) {
            if (i > 1) {
                System.err.print(", ");
            }
            System.err.print("\"" + args[i] + "\"");
        }
        System.err.println("\n");
    } else {
        System.err.println("Executing script \"" + scriptFileName + "\" with no arguments.\n");
    }
    // Parse arguments
    List<String> argList = new ArrayList<>();
    Map<String, String> paramMap = new HashMap<>();
    for (String arg : args) {
        if (arg.startsWith("--") && (arg.length() > 2) && (arg.charAt(2) != '-')) {
            p = arg.indexOf('=');
            if (p != -1) {
                String key = arg.substring(2, p);
                String value = arg.substring(p + 1);
                paramMap.put(key, value);
            } else {
                paramMap.put(arg.substring(2), "true");
            }
        } else if (arg.startsWith("-") && (arg.length() > 1) && (arg.charAt(1) != '-')) {
            for (int i = 1; i < arg.length(); i++) {
                paramMap.put(arg.substring(i, i + 1), "true");
            }
        } else {
            argList.add(arg);
        }
    }
    // Initialise script context
    ScriptingContext context = new ScriptingContext(true);
    Bindings bindings = scriptEngine.getBindings(ScriptContext.ENGINE_SCOPE);
    bindings.put("wp", context);
    String[] argArray = argList.toArray(new String[argList.size()]);
    bindings.put("argc", argArray.length);
    bindings.put("argv", argArray);
    String[] scriptArgs = new String[argArray.length - 1];
    System.arraycopy(argArray, 1, scriptArgs, 0, scriptArgs.length);
    bindings.put("arguments", scriptArgs);
    bindings.put("params", paramMap);
    // Execute script
    try {
        scriptEngine.eval(new FileReader(scriptFile));
        // Check that go() was invoked on the last operation:
        context.checkGoCalled(null);
    } catch (RuntimeException e) {
        logger.error(e.getClass().getSimpleName() + " occurred while executing " + scriptFileName, e);
        System.exit(2);
    } catch (ScriptException e) {
        logger.error("ScriptException occurred while executing " + scriptFileName, e);
        System.exit(2);
    }
}