1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
/*
* simulator.cc - Implementation of the HorizonScript simulation utility
* Project Horizon
*
* Copyright (c) 2019 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)";
#endif
reset_if_pretty(std::cout);
std::cout << std::endl;
if(isatty(1)) { /* LCOV_EXCL_START */
std::cout << "Copyright (c) 2019 Adélie Linux and contributors. AGPL-3.0 license." << std::endl;
std::cout << 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("installfile", "Script failed. Stop.", "");
result_code = EXIT_FAILURE;
} else {
output_info("installfile", "Script completed.", "");
}
delete my_script;
return result_code;
}
|
