feat: replace `--pinyin` flag with generic `--post="+pinyin"` option

2 files changed, 43 insertions, 0 deletions

diff --git a/src/post_proc/pinyin.rs b/src/post_proc/pinyin.rs
new file mode 100644
index 0000000..c06b040
--- /dev/null
+++ b/src/post_proc/pinyin.rs
@@ -0,0 +1,31 @@
+use crate::post_proc::TextPostProcesser;
+
+/// Converts Chinese characters to pinyin (romanized phonetic representation).
+///
+/// # Styles
+/// - `plain` — ni hao (default)
+/// - `tone` — nǐ hǎo (with tone marks, requires `with_tone` feature)
+/// - `tone-num` — ni3 hao3 (tone number at end)
+/// - `first-letter` — n h (first letter only)
+pub struct PinyinPostProcesser;
+
+impl TextPostProcesser for PinyinPostProcesser {
+    fn processer_name() -> &'static str {
+        "pinyin"
+    }
+
+    fn process(param: &[&str], text: &str) -> String {
+        let style = param.first().copied().unwrap_or("plain");
+        let result: Vec<&'static str> = match style {
+            "tone" => pinyin::to_pinyin_vec(text, pinyin::Pinyin::with_tone),
+            "tone-num" | "tone_num" => {
+                pinyin::to_pinyin_vec(text, pinyin::Pinyin::with_tone_num_end)
+            }
+            "first-letter" | "first_letter" => {
+                pinyin::to_pinyin_vec(text, pinyin::Pinyin::first_letter)
+            }
+            _ => pinyin::to_pinyin_vec(text, pinyin::Pinyin::plain),
+        };
+        result.join(" ")
+    }
+}
diff --git a/src/post_proc/trait.rs b/src/post_proc/trait.rs
new file mode 100644
index 0000000..7fced5b
--- /dev/null
+++ b/src/post_proc/trait.rs
@@ -0,0 +1,12 @@
+/// A text post-processor that transforms transcribed text.
+pub trait TextPostProcesser {
+    /// Name used in `--post="+name(args)"`.
+    fn processer_name() -> &'static str;
+
+    /// Execute the post-processing.
+    ///
+    /// # Arguments
+    /// * `param` — arguments parsed from `--post="+name(arg1, arg2)"`
+    /// * `text` — the transcribed text to process
+    fn process(param: &[&str], text: &str) -> String;
+}