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
|
use arg_picker::{IntoPicker, PickerArg, PickerArgs};
use mingling_core::{Program, ProgramCollect, setup::ProgramSetup};
use crate::setups::picker::{
JSON_FLAG, JSON_PRETTY_FLAG, RON_FLAG, RON_PRETTY_FLAG, TOML_FLAG, YAML_FLAG,
};
/// Sets up the structural renderer for the program:
///
/// - Adds a `--renderer` global argument to specify the renderer type
pub struct StructuralRendererSimpleSetup;
impl<C> ProgramSetup<C> for StructuralRendererSimpleSetup
where
C: ProgramCollect<Enum = C>,
{
fn setup(self, program: &mut Program<C>) {
program.global_argument("--renderer", |p, renderer| {
p.structural_renderer_name = renderer.into();
});
}
}
/// Sets up the structural renderer for the program:
///
/// - Adds global flags to specify the renderer type:
/// * `--json` for JSON output
/// * `--json-pretty` for pretty-printed JSON output
/// * `--yaml` for YAML output
/// * `--toml` for TOML output
/// * `--ron` for RON output
/// * `--ron-pretty` for pretty-printed RON output
pub struct StructuralRendererSetup;
impl<C> ProgramSetup<C> for StructuralRendererSetup
where
C: ProgramCollect<Enum = C>,
{
fn setup(self, program: &mut Program<C>) {
let args = program.take_args();
let args = process_renderer_flags(args, program);
program.replace_args(args);
}
}
fn process_renderer_flags<C>(args: Vec<String>, program: &mut Program<C>) -> Vec<String>
where
C: ProgramCollect<Enum = C>,
{
let remains_arg = PickerArg::<PickerArgs>::new(&[], None, true);
let (json, json_pretty, yaml, toml, ron, ron_pretty, remains) = args
.pick(&JSON_FLAG)
.pick(&JSON_PRETTY_FLAG)
.pick(&YAML_FLAG)
.pick(&TOML_FLAG)
.pick(&RON_FLAG)
.pick(&RON_PRETTY_FLAG)
.pick(&remains_arg)
.unwrap();
#[cfg(feature = "json_serde_fmt")]
if *json {
program.structural_renderer_name = crate::StructuralRendererSetting::Json;
}
#[cfg(feature = "json_serde_fmt")]
if *json_pretty {
program.structural_renderer_name = crate::StructuralRendererSetting::JsonPretty;
}
#[cfg(feature = "yaml_serde_fmt")]
if *yaml {
program.structural_renderer_name = crate::StructuralRendererSetting::Yaml;
}
#[cfg(feature = "toml_serde_fmt")]
if *toml {
program.structural_renderer_name = crate::StructuralRendererSetting::Toml;
}
#[cfg(feature = "ron_serde_fmt")]
if *ron {
program.structural_renderer_name = crate::StructuralRendererSetting::Ron;
}
#[cfg(feature = "ron_serde_fmt")]
if *ron_pretty {
program.structural_renderer_name = crate::StructuralRendererSetting::RonPretty;
}
remains.into()
}
|
