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
|
use std::{env::current_dir, process::exit};
use colored::Colorize;
use mingling::{
Program,
hook::ProgramHook,
macros::program_setup,
setup::{ExitCodeSetup, GeneralRendererSetup, HelpFlagSetup, QuietFlagSetup},
};
use mingling_cli::{ThisProgram, display::markdown, res::ResCurrentDir};
fn main() {
#[cfg(windows)]
colored::control::set_virtual_terminal(true).unwrap();
// Preprocess args to handle cargo-mling invocations
let mut args: Vec<String> = std::env::args().collect();
if args.first().map_or(false, |a| a.contains("cargo-mling")) {
args[0] = "cargo-mling".to_string();
}
if args.get(1).map_or(false, |a| a == "mling") {
args.remove(1);
}
// Build program with preprocessed args
let mut program = Program::<ThisProgram>::new_with_args(args);
// Intercept Version
program.global_flag(["-V", "--version"], |_| {
eprintln!(include_str!("../helps/version.txt"));
exit(0)
});
// Intercept Help
program.with_hook(ProgramHook::empty().on_post_dispatch(|c| match c {
// When dispatcher is not found
ThisProgram::ErrorDispatcherNotFound => {
// And user requests Help
if ThisProgram::this().user_context.help {
// Print help
eprintln!("{}", markdown(include_str!("../helps/mling_help.txt")));
exit(0)
}
}
_ => {}
}));
// Setups
program.with_setup(HelpFlagSetup::new(["-h", "--help"]));
program.with_setup(GeneralRendererSetup);
program.with_setup(StandardOutputSetup);
program.with_setup(ExitCodeSetup::default());
// Resources
program.with_resource(ResCurrentDir {
path: current_dir().unwrap(),
});
// Execute
let quiet = program.stdout_setting.quiet;
let error_output = program.stdout_setting.error_output && !quiet;
let render_output = program.stdout_setting.render_output && !quiet;
let result = program.exec_without_render().unwrap();
if !result.is_empty() {
if result.exit_code == 0 && render_output {
println!("{}", result.trim());
} else if error_output {
eprintln!("{}", result.trim().bright_red());
}
}
exit(result.exit_code);
}
#[program_setup]
fn standard_output_setup(program: &mut Program<ThisProgram>) {
program.with_setup(QuietFlagSetup::new("--silence"));
program.global_flag(["--no-error"], |program| {
program.stdout_setting.error_output = false;
});
program.global_flag(["--no-result"], |program| {
program.stdout_setting.render_output = false;
});
program.global_flag(["--silence", "--quiet"], |program| {
program.stdout_setting.quiet = true;
});
}
|
