feat: support Shuangpin

Signed-off-by: Chen Linxuan <me@black-desk.cn>

Author: Chen Linxuan

README.md

@@ -30,6 +30,41 @@ set colored-stats on
 - bash-completion
 - rust toolchains
 
+## Configuring Pinyin Schema
+
+`bash-pinyin-completion-rs` supports multiple Pinyin schemes:
+
+- **Quanpin**: Quanpin (full Pinyin) without tone marking - e.g., "zhongguo" for "中国"
+- **ShuangpinAbc**: Shuangpin (double Pinyin, or two-letter Pinyin) - 智能 ABC / Intelligent ABC scheme
+- **ShuangpinJiajia**: Shuangpin (double Pinyin, or two-letter Pinyin) - 拼音加加 / Pinyin Jiajia scheme
+- **ShuangpinMicrosoft**: Shuangpin (double Pinyin, or two-letter Pinyin) - 微软拼音 / MSPY scheme
+- **ShuangpinThunisoft**: Shuangpin (double Pinyin, or two-letter Pinyin) - 紫光拼音 / Thunisoft scheme
+- **ShuangpinXiaohe**: Shuangpin (double Pinyin, or two-letter Pinyin) - 小鹤 / Xiaohe scheme
+- **ShuangpinZrm**: Shuangpin (double Pinyin, or two-letter Pinyin) - 自然码 / Ziranma scheme
+
+You may configure the active scheme/schema with the `PINYIN_COMP_MODE` variable,
+typically set in `.bashrc`. If not set or value is invalid, `bash-pinyin-completion-rs`
+defaults to `Quanpin`.
+
+For example, to enable the 小鹤 / Xiaohe Shuangpin scheme:
+
+```bash
+export PINYIN_COMP_MODE="ShuangpinXiaohe"
+```
+
+To use Quanpin together with Shuangpin (Xiaohe):
+
+```bash
+export PINYIN_COMP_MODE="Quanpin,ShuangpinXiaohe"
+```
+
+### Notes on Completion Modes
+
+- Prefix matching (e.g., "zg" for "中国") is enabled by default with Quanpin,
+  but will be disabled if any Shuangpin schema is enabled.
+- Mixing Shuangpin schemas is not supported -
+  if multiple Shuangpin schemas are enabled, only the first one will take effect.
+
 ## Bug report
 If you encounter any issues, please report them on the GitHub issues page.
 

src/main.rs

@@ -1,6 +1,53 @@
 use ib_pinyin::{matcher::PinyinMatcher, pinyin::PinyinNotation};
+use std::env;
 use std::io::{BufRead, BufReader};
 
+fn parse_pinyin_notation_env() -> PinyinNotation {
+    let env_val = env::var("PINYIN_COMP_MODE").unwrap_or_default();
+    let mut notation = PinyinNotation::empty();
+    let mut shuangpin = Option::<PinyinNotation>::None;
+
+    for mode in env_val.split(',') {
+        let mode = mode.trim();
+        match mode {
+            "Quanpin" => {
+                notation |= PinyinNotation::Ascii;
+            }
+            "ShuangpinAbc" => {
+                shuangpin.get_or_insert(PinyinNotation::DiletterAbc);
+            }
+            "ShuangpinJiajia" => {
+                shuangpin.get_or_insert(PinyinNotation::DiletterJiajia);
+            }
+            "ShuangpinMicrosoft" => {
+                shuangpin.get_or_insert(PinyinNotation::DiletterMicrosoft);
+            }
+            "ShuangpinThunisoft" => {
+                shuangpin.get_or_insert(PinyinNotation::DiletterThunisoft);
+            }
+            "ShuangpinXiaohe" => {
+                shuangpin.get_or_insert(PinyinNotation::DiletterXiaohe);
+            }
+            "ShuangpinZrm" => {
+                shuangpin.get_or_insert(PinyinNotation::DiletterZrm);
+            }
+            _ => {}
+        }
+    }
+
+    notation |= shuangpin.unwrap_or(PinyinNotation::empty());
+
+    if notation.is_empty() {
+        notation = PinyinNotation::Ascii;
+    }
+
+    if notation == PinyinNotation::Ascii {
+        notation |= PinyinNotation::AsciiFirstLetter;
+    }
+
+    notation
+}
+
 fn main() {
     let args: Vec<String> = std::env::args().collect();
     // Print usage
@@ -12,8 +59,9 @@ fn main() {
 
     let input: &str = &args[1];
     if han_re.is_match(input) {return};
+    let notation = parse_pinyin_notation_env();
     let matcher = PinyinMatcher::builder(input)
-        .pinyin_notations(PinyinNotation::Ascii | PinyinNotation::AsciiFirstLetter)
+        .pinyin_notations(notation)
         .build();
 
     let stdin = std::io::stdin();