STTプラグインにJSONRPC通信処理とプロセス終了処理を追加

Author: sile

src/plugin.rs

@@ -1,6 +1,6 @@
 use std::collections::HashMap;
 use std::fs::File;
-use std::io::BufWriter;
+use std::io::{BufWriter, Write};
 use std::path::PathBuf;
 use std::process::{Command, Stdio};
 use std::time::Duration;
@@ -98,11 +98,29 @@ impl SpeechToTextPlugin {
         })
     }
 
+    // TODO: ブロックする可能性があるので専用のスレッドで実行する
     fn execute_stt(
         &mut self,
         input_stream_id: MediaStreamId,
         chunk: Vec<i16>,
     ) -> orfail::Result<()> {
+        let request = nojson::object(|f| {
+            f.member("jsonrpc", "2.0")?;
+            f.member("method", "stt")?;
+            f.member("id", 0)?;
+            f.member(
+                "params",
+                nojson::object(|f| {
+                    f.member("input_stream_id", input_stream_id.get())?;
+                    f.member("audio_data", &chunk)?;
+                    Ok(())
+                }),
+            )?;
+            Ok(())
+        });
+        writeln!(self.stdin, "{request}").or_fail()?;
+        self.stdin.flush().or_fail()?;
+
         todo!()
     }
 }
@@ -118,7 +136,7 @@ impl MediaProcessor for SpeechToTextPlugin {
 
     fn process_input(&mut self, input: MediaProcessorInput) -> orfail::Result<()> {
         if let Some(sample) = input.sample {
-            let mut input_stream = self.input_streams.get_mut(&input.stream_id).or_fail()?;
+            let input_stream = self.input_streams.get_mut(&input.stream_id).or_fail()?;
             let data = sample.expect_audio_data().or_fail()?;
             input_stream
                 .chunk
@@ -144,4 +162,9 @@ impl MediaProcessor for SpeechToTextPlugin {
     }
 }
 
-// TODO: kill when drop
+impl Drop for SpeechToTextPlugin {
+    fn drop(&mut self) {
+        let _ = self.process.kill();
+        let _ = self.process.wait();
+    }
+}