summaryrefslogtreecommitdiff
path: root/mingling/src/parser/picker/builtin.rs
blob: 0b1ce08b3afbe56861fa982839e67214b0a85833 (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
use size::Size;

use crate::parser::Pickable;

impl Pickable for String {
    type Output = String;

    fn pick(args: &mut crate::parser::Argument, flag: mingling_core::Flag) -> Option<Self::Output> {
        args.pick_argument(flag)
    }
}

macro_rules! impl_pickable_for_number {
    ($($t:ty),*) => {
        $(
            impl Pickable for $t {
                type Output = $t;

                fn pick(args: &mut crate::parser::Argument, flag: mingling_core::Flag) -> Option<Self::Output> {
                    let Some(picked) = args.pick_argument(flag) else {
                        return None;
                    };
                    picked.parse().ok()
                }
            }
        )*
    };
}

impl_pickable_for_number!(i8, i16, i32, i64, i128, u8, u16, u32, u64, u128, f32, f64);

impl Pickable for bool {
    type Output = bool;

    fn pick(args: &mut crate::parser::Argument, flag: mingling_core::Flag) -> Option<Self::Output> {
        Some(args.pick_flag(flag))
    }
}

impl Pickable for usize {
    type Output = usize;

    fn pick(args: &mut crate::parser::Argument, flag: mingling_core::Flag) -> Option<Self::Output> {
        let Some(picked) = args.pick_argument(flag) else {
            return None;
        };
        let size_parse = Size::from_str(picked.as_str());
        match size_parse {
            Ok(size) => Some(size.bytes() as usize),
            Err(_) => None,
        }
    }
}