From 85eb8cb7bc94570550d88526585d55973610b9bd Mon Sep 17 00:00:00 2001 From: Robin Krahl Date: Tue, 7 Jan 2020 12:41:06 +0000 Subject: Factor out Enum! and Command! macros in arg_util For easier refactoring, we remove the internal enum_int! macro and instead copy its code to the Enum! and Command! macros. --- src/arg_util.rs | 31 ++++++++++++++++++++++++++++--- 1 file changed, 28 insertions(+), 3 deletions(-) diff --git a/src/arg_util.rs b/src/arg_util.rs index 6d1bc50..317d9e3 100644 --- a/src/arg_util.rs +++ b/src/arg_util.rs @@ -33,9 +33,34 @@ macro_rules! Command { )* } - enum_int! {$name, [ - $( $var => $str, )* - ]} + impl ::std::convert::AsRef for $name { + fn as_ref(&self) -> &'static str { + match *self { + $( + $name::$var => $str, + )* + } + } + } + + impl ::std::fmt::Display for $name { + fn fmt(&self, f: &mut ::std::fmt::Formatter<'_>) -> ::std::fmt::Result { + write!(f, "{}", self.as_ref()) + } + } + + impl ::std::str::FromStr for $name { + type Err = &'static str; + + fn from_str(s: &str) -> ::std::result::Result { + match s { + $( + $str => Ok($name::$var), + )* + _ => Err("[error]"), + } + } + } #[allow(unused_qualifications)] impl $name { -- cgit v1.2.3