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
|
use mingling::{Program, hook::ProgramHook, macros::program_setup};
use shared_functions::info;
use crate::{
ThisProgram,
output::{
ansi_control::disable_ansi,
env_logger::{EnvLogger, init_envlogger},
},
};
#[program_setup]
pub fn color_output_setup(program: &mut Program<ThisProgram>) {
program.global_flag("--no-color", |_| {
disable_ansi();
});
}
#[program_setup]
pub fn env_logger_setup(program: &mut Program<ThisProgram>) {
program.stdout_setting.verbose = program.pick_global_flag(["-V", "--verbose"]);
let log_show_time = program.pick_global_flag("--log-time");
let log_level = program
.pick_global_argument("--log-level")
.unwrap_or("info".to_string());
if program.stdout_setting.verbose {
init_envlogger(EnvLogger {
show_time: log_show_time,
show_level: log_level.as_str() != "disable",
level: match log_level.as_str() {
"trace" => log::Level::Trace,
"debug" => log::Level::Debug,
"warn" => log::Level::Warn,
"error" => log::Level::Error,
_ => log::Level::Info,
},
});
// Add Hook
program.with_hook(
ProgramHook::<ThisProgram>::empty()
.on_begin::<_, ()>(|_| println!("[DEBUG] Program is begin"))
.on_pre_dispatch(|info| {
println!("[DEBUG] Pre dispatch: {}", info.arguments.join(" "))
})
.on_post_dispatch(|info| println!("[DEBUG] Post dispatch: {}", info.entry))
.on_pre_chain(|info| {
println!("[DEBUG] Pre chain: {}", info.input);
})
.on_post_chain(|info| println!("[DEBUG] Post chain: {}", info.output.member_id))
.on_finish(|_| {
println!("[DEBUG] Loop end");
})
.on_pre_render(|info| println!("[DEBUG] Pre render: {}", info.input))
.on_post_render(|_| println!("[DEBUG] Post render")),
);
}
info!("Verbose mode enabled!");
}
|
