From 2dcceb1d5889f6eac6262f38506c58066a898f76 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. --- nitrocli/src/arg_util.rs | 31 ++++++++++++++++++++++++++++--- 1 file changed, 28 insertions(+), 3 deletions(-) diff --git a/nitrocli/src/arg_util.rs b/nitrocli/src/arg_util.rs index fe755f3..68bd23f 100644 --- a/nitrocli/src/arg_util.rs +++ b/nitrocli/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.1