path: root/tools/hscript-simulate/simulator.cc

/*
 * simulator.cc - Implementation of the HorizonScript simulation utility
 * Project Horizon
 *
 * Copyright (c) 2019-2020 Adélie Linux and contributors.  All rights reserved.
 * This code is licensed under the AGPL 3.0 license, as noted in the
 * LICENSE-code file in the root directory of this repository.
 *
 * SPDX-License-Identifier: AGPL-3.0-only
 */

#include <boost/program_options.hpp>
#include <unistd.h>
#include "hscript/script.hh"
#include "util/output.hh"

bool pretty = false;

static int cli_failure(boost::program_options::options_description cli) {
        std::cout << cli << std::endl;
        return EXIT_FAILURE;
}

int main(int argc, char *argv[]) {
    const Horizon::Script *my_script;
    Horizon::ScriptOptions opts;
    int result_code = EXIT_SUCCESS;
    std::string installfile;
    using Horizon::ScriptOptionFlags;

    /* Default to pretty if we are using a TTY, unless -n specified. */
    if(isatty(1) && isatty(2)) {
        pretty = true;  /* LCOV_EXCL_LINE */
    }

    opts.set(ScriptOptionFlags::Simulate);

    boost::program_options::options_description cli_hidden;
    cli_hidden.add_options()
        ("installfile", "Installfile to load");
    boost::program_options::options_description cli_visible("Allowed options");
    cli_visible.add_options()
        ("no-colour,n", "Do not 'prettify' output")
        ("strict,s", "Strict parsing");
    boost::program_options::options_description cli;
    cli.add(cli_visible).add(cli_hidden);
    boost::program_options::positional_options_description cli_pos;
    cli_pos.add("installfile", -1);
    boost::program_options::variables_map args;
    try {
        boost::program_options::store(
            boost::program_options::command_line_parser(argc, argv)
                .options(cli)
                .positional(cli_pos)
                .run(),
            args);
        boost::program_options::notify(args);
    } catch(const boost::program_options::error& cli_err) {
        std::cout << "An invalid option was entered." << std::endl;
        return cli_failure(cli_visible);
    }

    if (args.count("installfile")) {
        installfile = args["installfile"].as<std::string>();
    } else {
        std::cout << "You must provide an installfile." << std::endl;
        return cli_failure(cli_visible);
    }

    if (args.count("no-colour")) {
        pretty = false;
    }
    if (args.count("strict")) {
        opts.set(ScriptOptionFlags::StrictMode);
    }

    if(!isatty(1)) {
        std::cout << "#!/bin/sh" << std::endl << std::endl;
    }
    bold_if_pretty(std::cout);
    if(!isatty(1)) std::cout << "# Generated by ";
    std::cout << "HorizonScript Simulation Utility version " << VERSTR;
#ifndef HAS_INSTALL_ENV
    std::cout << " (runtime environment only)";
#endif
#ifdef NON_LIBRE_FIRMWARE
    colour_if_pretty(std::cout, "31");
    std::cout << " (supports non-free firmware)";
    reset_if_pretty(std::cout);
#endif
    std::cout << std::endl;
    if(isatty(1)) {  /* LCOV_EXCL_START */
        std::cout << "Copyright (c) 2019-2020 Adélie Linux and contributors.";
        std::cout << std::endl;
        std::cout << "This software is licensed to you under the terms of the "
                  << std::endl << "AGPL 3.0 license, unless otherwise noted.";
        std::cout << std::endl << std::endl;
    }  /* LCOV_EXCL_STOP */

    my_script = Horizon::Script::load(installfile, opts);
    if(my_script == nullptr) {
        std::cout << "Could not load the specified script." << std::endl;
        return EXIT_FAILURE;
    }

    if(!my_script->execute()) {
        output_error("internal", "Script failed.  Stop.", "");
        result_code = EXIT_FAILURE;
    } else {
        output_info("internal", "Script completed successfully.", "");
    }

    delete my_script;
    return result_code;
}