1 files changed, 48 insertions, 36 deletions

diff --git a/tools/hscript-validate/validator.cc b/tools/hscript-validate/validator.cc
index 77641c7..8277bea 100644
--- a/tools/hscript-validate/validator.cc
+++ b/tools/hscript-validate/validator.cc
@@ -14,69 +14,70 @@
 #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 install{}, keep_going{}, strict{}, needs_help{}, version_only{},
+         dont_pretty{};
     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)) {
         pretty = true;  /* LCOV_EXCL_LINE */
     }
 
-    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()
-        ("install,i", "Set Installation Environment flag (DANGEROUS)")
-        ("keep-going,k", "Continue parsing after errors")
-        ("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.")
+        ("install,i", bool_switch(&install), "Set Installation Environment flag. (DANGEROUS!)")
+        ("keep-going,k", bool_switch(&keep_going), "Continue parsing after errors.")
+        ("no-colour,n", bool_switch(&dont_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);
+        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("install")) {
+    if(install) {
         opts.set(ScriptOptionFlags::InstallEnvironment);
     }
-    if (args.count("keep-going")) {
+    if(keep_going) {
         opts.set(ScriptOptionFlags::KeepGoing);
     }
-    if (args.count("no-colour")) {
+    if(dont_pretty) {
         pretty = false;
     }
-    if (args.count("strict")) {
+    if(strict) {
         opts.set(ScriptOptionFlags::StrictMode);
     }
 
@@ -88,26 +89,37 @@ 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
-    std::cout << std::endl;
-    std::cout << "Copyright (c) 2019-2020 Adélie Linux and contributors."
-              << 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;
+    reset_if_pretty(std::cout);
+    std::cout << std::endl
+              << "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;
+
+    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->validate()) {
-        output_error("internal", "Script failed validation.  Stop.", "");
+        output_error("internal", "Script failed validation.  Stop.");
         result_code = EXIT_FAILURE;
     } else {
-        output_info("internal", "Script passed validation.", "");
+        output_info("internal", "Script passed validation.");
     }
 
     delete my_script;