diff options
Diffstat (limited to 'mingling')
| -rw-r--r-- | mingling/src/parser/picker.rs | 16 |
1 files changed, 15 insertions, 1 deletions
diff --git a/mingling/src/parser/picker.rs b/mingling/src/parser/picker.rs index c6c2beb..9258002 100644 --- a/mingling/src/parser/picker.rs +++ b/mingling/src/parser/picker.rs @@ -1,7 +1,7 @@ use std::marker::PhantomData; use crate::parser::Argument; -use mingling_core::Flag; +use mingling_core::{EnumTag, Flag}; #[doc(hidden)] pub mod builtin; @@ -316,3 +316,17 @@ impl_pick_structs! { Pick8 Pick9 val_9 T1 val_1, T2 val_2, T3 val_3, T4 val_4, T impl_pick_structs! { Pick9 Pick10 val_10 T1 val_1, T2 val_2, T3 val_3, T4 val_4, T5 val_5, T6 val_6, T7 val_7, T8 val_8, T9 val_9 } impl_pick_structs! { Pick10 Pick11 val_11 T1 val_1, T2 val_2, T3 val_3, T4 val_4, T5 val_5, T6 val_6, T7 val_7, T8 val_8, T9 val_9, T10 val_10 } impl_pick_structs! { Pick11 Pick12 val_12 T1 val_1, T2 val_2, T3 val_3, T4 val_4, T5 val_5, T6 val_6, T7 val_7, T8 val_8, T9 val_9, T10 val_10, T11 val_11 } + +pub trait PickableEnum: EnumTag + Default {} + +impl<T> Pickable for T +where + T: PickableEnum, +{ + type Output = T; + + fn pick(args: &mut Argument, flag: Flag) -> Option<Self::Output> { + let name = args.pick_argument(flag)?; + T::build_enum(name) + } +} |