From 9e7c0fd45e169929156bdb317b10d7bb3db65f8b Mon Sep 17 00:00:00 2001
From: 魏曹先生 <1992414357@qq.com>
Date: Sat, 7 Mar 2026 19:37:52 +0800
Subject: Add callback support to chunk_stream_with and implement stream
 writing

---
 src/utils/cmd_macros.rs | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)
 create mode 100644 src/utils/cmd_macros.rs

(limited to 'src/utils/cmd_macros.rs')

diff --git a/src/utils/cmd_macros.rs b/src/utils/cmd_macros.rs
new file mode 100644
index 0000000..11b8da4
--- /dev/null
+++ b/src/utils/cmd_macros.rs
@@ -0,0 +1,47 @@
+#[macro_export]
+macro_rules! special_flag {
+    ($args:expr, $($flag:expr),+) => {{
+        let mut found = false;
+        $(
+            let flag = $flag;
+            if $args.iter().any(|arg| arg == flag) {
+                found = true;
+            }
+            $args.retain(|arg| arg != flag);
+        )+
+        found
+    }};
+}
+
+#[macro_export]
+macro_rules! special_argument {
+    ($args:expr, $($flag:expr),+) => {{
+        let mut value: Option<String> = None;
+        let mut found = false;
+        $(
+            let flag = $flag;
+            if !found {
+                let mut i = 0;
+                while i < $args.len() {
+                    if $args[i] == flag {
+                        if i + 1 < $args.len() {
+                            value = Some($args[i + 1].clone());
+                            $args.remove(i + 1);
+                            $args.remove(i);
+                        } else {
+                            value = None;
+                            $args.remove(i);
+                        }
+                        #[allow(unused_assignments)]
+                        {
+                            found = true;
+                        }
+                        break;
+                    }
+                    i += 1;
+                }
+            }
+        )+
+        value
+    }};
+}
-- 
cgit