Replace code with macro

src/main.rs

@@ -35,23 +35,27 @@ mod timer;
 mod runner;
 mod choose;
 
+macro_rules! options {
+    ($name:ident { $($opt:ident,)* }) => {
         #[derive(Copy, Clone, ArgEnum)]
-enum AnimationType {
+        enum $name {
+            $($opt,)*
+        }
+        impl Options for $name {
+            fn all() -> Vec<Self> {
+                vec![$($name::$opt,)*]
+            }
+        }
+    }
+}
+
+options!(AnimationType {
     Circle,
     Rhombus,
-    Rotation
+    Rotation,
-}
+});
 
-impl Options for AnimationType {
+options!(ColorType {
-    fn all() -> Vec<Self> where Self: Sized {
-        use AnimationType::*;
-
-        vec![Circle, Rhombus, Rotation]
-    }
-}
-
-#[derive(Copy, Clone, ArgEnum)]
-enum ColorType {
     Red,
     Green,
     Blue,
@@ -60,30 +64,13 @@ enum ColorType {
     LightBlue,
     Grey,
     Rainbow,
-}
+});
 
-impl Options for ColorType {
+options!(FillModeType {
-    fn all() -> Vec<Self> where Self: Sized {
-        use ColorType::*;
-
-        vec![Red, Green, Blue, LightRed, LightGreen, LightBlue, Grey, Rainbow]
-    }
-}
-
-#[derive(Copy, Clone, ArgEnum)]
-enum FillModeType {
     Circle,
     Level,
-    Stripes
+    Stripes,
-}
+});
-
-impl Options for FillModeType {
-    fn all() -> Vec<Self> where Self: Sized {
-        use FillModeType::*;
-
-        vec![Circle, Level]
-    }
-}
 
 #[derive(Parser)]
 #[clap(author = env ! ("CARGO_PKG_AUTHORS"), version = env ! ("CARGO_PKG_VERSION"), about = env ! ("CARGO_PKG_DESCRIPTION"))]