init: first version

Signed-off-by: wxiwnd <wxiwnd@outlook.com>

Author: wxiwnd

.gitignore

@@ -0,0 +1 @@
+/target

Cargo.lock

@@ -0,0 +1,7 @@
+[package]
+name = "bash-pinyin-completion-rs"
+version = "0.1.0"
+edition = "2021"
+
+[dependencies]
+ib-pinyin = "0.2.5"

Cargo.toml

@@ -0,0 +1,30 @@
+#!/usr/bin/env bash
+
+# Load bash-completion if not loaded
+# [ -f /usr/share/bash-completion/bash_completion ] && \
+#     . /usr/share/bash-completion/bash_completion
+
+# Detect bash-completion
+if ! declare -F _comp_compgen_filedir &>/dev/null; then
+    echo "No function _comp_compgen_filedir found. Please install bash-completion first."
+    return
+fi
+
+eval "function __bak_comp_compgen_filedir() { $(declare -f _comp_compgen_filedir | tail -n +2) }"
+
+# replace _comp_compgen_filedir
+_comp_compgen_filedir() {
+    __bak_comp_compgen_filedir "$@"
+
+    local cur="${COMP_WORDS[COMP_CWORD]}"
+
+    # ignore empty
+    [ -z "$cur" ] && return
+
+    local -a pinyin_matches=()
+    while IFS= read -r line; do
+        pinyin_matches+=( "$line" )
+    done < <( <script_path> "$cur" 2>/dev/null )
+
+    COMPREPLY+=( "${pinyin_matches[@]}" )
+}

scripts/bash_pinyin_completion

@@ -0,0 +1,33 @@
+use ib_pinyin::{matcher::PinyinMatcher, pinyin::PinyinNotation};
+
+fn main() {
+    let args: Vec<String> = std::env::args().collect();
+    // Print usage
+    if args.len() < 2 {
+        eprintln!("Usage: {} <pinyin>", args[0]);
+        std::process::exit(1);
+    }
+
+    let input: &str = &args[1];
+    let matcher = PinyinMatcher::builder(input)
+    .pinyin_notations(PinyinNotation::Ascii | PinyinNotation::AsciiFirstLetter)
+    .build();
+
+    let mut results: Vec<String> = Vec::new();
+    // Read current location
+    if let Ok(entries) = std::fs::read_dir(".") {
+        for entry in entries.flatten() {
+                let path = entry.path();
+                if let Some(file_name) = path.file_name() {
+                    if let Some(file_name_str) = file_name.to_str() {
+                        if matcher.is_match(file_name_str) {
+                            results.push(file_name_str.to_string());
+                        }
+                    }
+                }
+            }
+        }
+        for result in results {
+            println!("{}", result);
+        }
+    }