1 files changed, 44 insertions, 32 deletions

diff --git a/tools/hscript-simulate/simulator.cc b/tools/hscript-simulate/simulator.cc
index 1731c62..ade54dc 100644
--- a/tools/hscript-simulate/simulator.cc
+++ b/tools/hscript-simulate/simulator.cc
@@ -14,19 +14,18 @@
 #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;
+    bool strict{}, needs_help{}, disable_pretty{}, version_only{};
     using Horizon::ScriptOptionFlags;
+    using namespace boost::program_options;
 
     /* Default to pretty if we are using a TTY, unless -n specified. */
     if(isatty(1) && isatty(2)) {
@@ -35,42 +34,44 @@ int main(int argc, char *argv[]) {
 
     opts.set(ScriptOptionFlags::Simulate);
 
-    boost::program_options::options_description cli_hidden;
+    options_description cli_hidden;
     cli_hidden.add_options()
         ("installfile", "Installfile to load");
-    boost::program_options::options_description cli_visible("Allowed 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;
+        ("help,h", bool_switch(&needs_help), "Display this message.")
+        ("version,v", bool_switch(&version_only), "Show program version information.")
+        ("no-colour,n", bool_switch(&disable_pretty), "Do not 'prettify' output.")
+        ("strict,s", bool_switch(&strict), "Use strict parsing mode (enable more warnings/errors).");
+    options_description cli;
     cli.add(cli_visible).add(cli_hidden);
-    boost::program_options::positional_options_description cli_pos;
+    positional_options_description cli_pos;
     cli_pos.add("installfile", -1);
-    boost::program_options::variables_map args;
+    variables_map args;
     try {
-        boost::program_options::store(
-            boost::program_options::command_line_parser(argc, argv)
+        store(
+            command_line_parser(argc, argv)
                 .options(cli)
                 .positional(cli_pos)
                 .run(),
             args);
-        boost::program_options::notify(args);
+        notify(args);
     } catch(const boost::program_options::error& cli_err) {
-        std::cout << "An invalid option was entered." << std::endl;
-        return cli_failure(cli_visible);
+        std::cerr << cli_err.what() << std::endl;
+        needs_help = true;
+        result_code = EXIT_FAILURE;
     }
 
-    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(needs_help) {
+        std::cout << cli_visible << std::endl;
+        return result_code;
     }
 
-    if (args.count("no-colour")) {
+    if(disable_pretty) {
         pretty = false;
     }
-    if (args.count("strict")) {
+
+    if(strict) {
         opts.set(ScriptOptionFlags::StrictMode);
     }
 
@@ -86,28 +87,39 @@ int main(int argc, char *argv[]) {
 #ifdef NON_LIBRE_FIRMWARE
     colour_if_pretty(std::cout, "31");
     std::cout << " (supports non-free firmware)";
-    reset_if_pretty(std::cout);
 #endif
+    reset_if_pretty(std::cout);
     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;
+        std::cout << "Copyright (c) 2019-2020 Adélie Linux and contributors."
+                  << std::endl
+                  << "This software is licensed to you under the terms of the "
+                  << std::endl << "AGPL 3.0 license, unless otherwise noted."
+                  << std::endl << std::endl;
     }  /* LCOV_EXCL_STOP */
 
+    if(version_only) {
+        return EXIT_SUCCESS;
+    }
+
+    if(args.count("installfile")) {
+        installfile = args["installfile"].as<std::string>();
+    } else {
+        output_error("<stdin>", "You must specify an installfile");
+        return EXIT_FAILURE;
+    }
+
     my_script = Horizon::Script::load(installfile, opts);
     if(my_script == nullptr) {
-        std::cout << "Could not load the specified script." << std::endl;
+        output_error(installfile, "Could not load the specified script");
         return EXIT_FAILURE;
     }
 
     if(!my_script->execute()) {
-        output_error("internal", "Script failed.  Stop.", "");
+        output_error("internal", "Script failed.  Stop.");
         result_code = EXIT_FAILURE;
     } else {
-        output_info("internal", "Script completed successfully.", "");
+        output_info("internal", "Script completed successfully.");
     }
 
     delete my_script;