音声認識プラグインでSTTリクエスト送信後のレスポンス読み取り処理を追加

Author: sile

src/plugin.rs

@@ -1,6 +1,6 @@
 use std::collections::HashMap;
 use std::fs::File;
-use std::io::{BufWriter, Write};
+use std::io::{BufRead, BufReader, BufWriter, Write};
 use std::path::PathBuf;
 use std::process::{Command, Stdio};
 use std::time::Duration;
@@ -63,6 +63,7 @@ impl InputStream {
 pub struct SpeechToTextPlugin {
     process: std::process::Child,
     stdin: std::process::ChildStdin,
+    stdout: BufReader<std::process::ChildStdout>,
     chunk_duration: Duration,
     input_streams: HashMap<MediaStreamId, InputStream>,
     output_file: BufWriter<File>,
@@ -91,6 +92,7 @@ impl SpeechToTextPlugin {
             .collect();
         Ok(Self {
             stdin: process.stdin.take().or_fail()?,
+            stdout: BufReader::new(process.stdout.take().or_fail()?),
             process,
             chunk_duration: spec.chunk_duration,
             input_streams,
@@ -104,10 +106,11 @@ impl SpeechToTextPlugin {
         input_stream_id: MediaStreamId,
         chunk: Vec<i16>,
     ) -> orfail::Result<()> {
+        let request_id: usize = 0;
         let request = nojson::object(|f| {
             f.member("jsonrpc", "2.0")?;
             f.member("method", "stt")?;
-            f.member("id", 0)?;
+            f.member("id", request_id)?;
             f.member(
                 "params",
                 nojson::object(|f| {
@@ -121,6 +124,23 @@ impl SpeechToTextPlugin {
         writeln!(self.stdin, "{request}").or_fail()?;
         self.stdin.flush().or_fail()?;
 
+        let mut response_line = String::new();
+        self.stdout.read_line(&mut response_line).or_fail()?;
+
+        let json = nojson::RawJson::parse(&response_line).or_fail()?;
+        let obj = JsonObject::new(json.value()).or_fail()?;
+        (obj.get_required::<String>("jsonrpc").or_fail()? == "2.0").or_fail()?;
+        (obj.get_required::<usize>("id").or_fail()? == request_id).or_fail()?;
+
+        // TODO: handle error
+        let result = json
+            .value()
+            .to_member("result")
+            .or_fail()?
+            .required()
+            .or_fail()?;
+        let obj = JsonObject::new(result).or_fail()?;
+        let text: String = obj.get_required("text").or_fail()?;
         todo!()
     }
 }