feat: 新增 client-kws

packages/client-rust/src/bin/client-kws.rs

@@ -0,0 +1,68 @@
+use std::path::Path;
+
+use open_xiaoai::{
+    services::monitor::kws::{KwsMonitor, KwsMonitorEvent, KWS_FILE_PATH},
+    utils::{rand::pick_one, shell::run_shell},
+};
+
+static KWS_REPLY_FILE_PATH: &str = "/data/open-xiaoai/kws/reply.txt";
+
+async fn on_started() {
+    let welcome = std::env::args()
+        .nth(1)
+        .unwrap_or("自定义唤醒词已开启".to_string());
+    let _ = run_shell(&format!("/usr/sbin/tts_play.sh '{}'", welcome)).await;
+}
+
+async fn on_keyword(_keyword: String) {
+    println!("🔥 唤醒词: {}", _keyword);
+
+    let mut wakeup_sounds = vec![
+        "file:///usr/share/sound-vendor/AiNiRobot/wakeup_ei_01.wav".to_string(),
+        "file:///usr/share/sound-vendor/AiNiRobot/wakeup_zai_01.wav".to_string(),
+    ];
+
+    if Path::new(KWS_REPLY_FILE_PATH).exists() {
+        // 播放自定义唤醒提示音
+        let content = std::fs::read_to_string(KWS_REPLY_FILE_PATH).unwrap();
+        let replies = content
+            .split("\n")
+            .filter(|line| !line.trim().is_empty())
+            .collect::<Vec<&str>>();
+        if replies.len() > 0 {
+            wakeup_sounds.clear();
+            wakeup_sounds.extend(replies.iter().map(|s| s.to_string()));
+        }
+    }
+
+    let reply = pick_one(&wakeup_sounds);
+    let script = if reply.contains("://") {
+        format!("miplayer -f '{}'", reply)
+    } else {
+        format!("/usr/sbin/tts_play.sh '{}'", reply)
+    };
+    let _ = run_shell(&script).await;
+
+    // 唤醒
+    let _ = run_shell(r#"ubus call pnshelper event_notify '{"src":1,"event":0}'"#).await;
+}
+
+#[tokio::main]
+async fn main() {
+    if Path::new(KWS_FILE_PATH).exists() {
+        on_started().await;
+    }
+
+    KwsMonitor::start(|event| async move {
+        match event {
+            KwsMonitorEvent::Started => on_started().await,
+            KwsMonitorEvent::Keyword(keyword) => on_keyword(keyword).await,
+        }
+        Ok(())
+    })
+    .await;
+
+    loop {
+        tokio::time::sleep(std::time::Duration::from_secs(1)).await;
+    }
+}

packages/client-rust/src/services/monitor/kws.rs

@@ -16,7 +16,7 @@ pub enum KwsMonitorEvent {
 
 pub struct KwsMonitor;
 
-static KWS_FILE_PATH: &str = "/tmp/open-xiaoai/kws.log";
+pub static KWS_FILE_PATH: &str = "/tmp/open-xiaoai/kws.log";
 
 static LAST_TIMESTAMP: AtomicU64 = AtomicU64::new(0);
 

packages/client-rust/src/utils/mod.rs

@@ -1,3 +1,4 @@
 pub mod event;
 pub mod shell;
 pub mod task;
+pub mod rand;

packages/client-rust/src/utils/rand.rs

@@ -0,0 +1,10 @@
+use rand::seq::IndexedRandom;
+use std::cell::RefCell;
+
+thread_local! {
+    static RNG: RefCell<rand::rngs::ThreadRng> = RefCell::new(rand::rng());
+}
+
+pub fn pick_one<T>(items: &[T]) -> &T {
+    RNG.with(|rng| items.choose(&mut *rng.borrow_mut()).unwrap())
+}