feat: color! macro support hex

JiatLn · May 15, 2023 · ed78104 · ed78104
1 parent edcdba6
commit ed78104
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/color/color_macro.rs b/src/color/color_macro.rs
@@ -13,6 +13,12 @@ macro_rules! color_args {
 
 #[macro_export]
 macro_rules! color {
+    (#$hex:expr) => {
+        {
+            let hex = format!("#{}", stringify!($hex));
+            $crate::Color::from_hex(&hex).unwrap()
+        }
+    };
     ($color_space:ident, $($args:tt)*) => {
         {
             let args = $crate::color_args!($($args)*);
@@ -64,5 +70,11 @@ mod tests {
 
         let color = color!(hsv, 60.0, 1.0, 1.0);
         assert_eq!(color.hsv(), "hsv(60, 100%, 100%)");
+
+        let color = color!(#f00);
+        assert_eq!(color.hex(), "#ff0000");
+
+        let color = color!(#abcdef);
+        assert_eq!(color.hex(), "#abcdef");
     }
 }
diff --git a/tests/color_macro.rs b/tests/color_macro.rs
@@ -4,4 +4,7 @@ use color_art::color;
 fn test_color_marco() {
     let color = color!(rgb, 255, 255, 0);
     assert_eq!(color.rgb(), "rgb(255, 255, 0)");
+
+    let color = color!(#1890ff);
+    assert_eq!(color.hex(), "#1890ff");
 }