aboutsummaryrefslogtreecommitdiff
path: root/mingling/src/setups/basic.rs
blob: 6a6973387dfe07671832ae363c16a7b83310bc2a (plain) (blame)
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
113
114
115
116
117
118
119
use mingling_core::{Flag, Program, ProgramCollect, setup::ProgramSetup};

/// Performs basic program initialization:
///
/// - Collects `--quiet` flag to control message rendering
/// - Collects `--help` flag to enable help mode
/// - Collects `--confirm` flag to skip user confirmation
pub struct BasicProgramSetup;

impl<C> ProgramSetup<C> for BasicProgramSetup
where
    C: ProgramCollect<Enum = C>,
{
    fn setup(self, program: &mut Program<C>) {
        program.with_setup(HelpFlagSetup::new(["-h", "--help"]));
        program.with_setup(QuietFlagSetup::new(["-q", "--quiet"]));
        program.with_setup(ConfirmFlagSetup::new(["-C", "--confirm"]));
    }
}

/// Provides setup for parsing the user help flag
///
/// The default value is `--help / -h`
pub struct HelpFlagSetup {
    flag: Flag,
}

impl HelpFlagSetup {
    /// Creates a new `HelpFlagSetup` with the given flag aliases.
    pub fn new(flag: impl Into<Flag>) -> Self {
        Self { flag: flag.into() }
    }
}

impl<C> ProgramSetup<C> for HelpFlagSetup
where
    C: ProgramCollect<Enum = C>,
{
    fn setup(self, program: &mut Program<C>) {
        program.global_flag(self.flag.clone(), |p| {
            p.user_context.help = true;
        });
    }
}

impl Default for HelpFlagSetup {
    fn default() -> Self {
        Self {
            flag: ["-h", "--help"].into(),
        }
    }
}

/// Provides setup for parsing the quiet flag
///
/// The default value is `--quiet / -q`
pub struct QuietFlagSetup {
    flag: Flag,
}

impl QuietFlagSetup {
    /// Creates a new `QuietFlagSetup` with the given flag aliases.
    pub fn new(flag: impl Into<Flag>) -> Self {
        Self { flag: flag.into() }
    }
}

impl<C> ProgramSetup<C> for QuietFlagSetup
where
    C: ProgramCollect<Enum = C>,
{
    fn setup(self, program: &mut Program<C>) {
        program.global_flag(self.flag.clone(), |p| {
            p.stdout_setting.render_output = false;
            p.stdout_setting.error_output = false;
        });
    }
}

impl Default for QuietFlagSetup {
    fn default() -> Self {
        Self {
            flag: ["-q", "--quiet"].into(),
        }
    }
}

/// Provides setup for parsing the confirm flag
///
/// The default value is `--confirm / -C`
pub struct ConfirmFlagSetup {
    flag: Flag,
}

impl ConfirmFlagSetup {
    /// Creates a new `ConfirmFlagSetup` with the given flag aliases.
    pub fn new(flag: impl Into<Flag>) -> Self {
        Self { flag: flag.into() }
    }
}

impl<C> ProgramSetup<C> for ConfirmFlagSetup
where
    C: ProgramCollect<Enum = C>,
{
    fn setup(self, program: &mut Program<C>) {
        program.global_flag(self.flag.clone(), |p| {
            p.user_context.confirm = true;
        });
    }
}

impl Default for ConfirmFlagSetup {
    fn default() -> Self {
        Self {
            flag: ["-C", "--confirm"].into(),
        }
    }
}