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,
        }
    }
}