1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
|
use std::io::{self, BufRead, Write};
use std::path::PathBuf;
use camino::Utf8PathBuf;
use crate::mcp::fault::{FaultKind, FaultRecord, FaultStage};
use crate::mcp::protocol::{WorkerOutcome, WorkerRequest, WorkerResponse};
use crate::mcp::service::WorkerService;
pub(crate) fn serve(project: PathBuf) -> Result<(), fidget_spinner_store_sqlite::StoreError> {
let project = Utf8PathBuf::from(project.to_string_lossy().into_owned());
let mut service = WorkerService::new(&project)?;
let stdin = io::stdin();
let mut stdout = io::stdout().lock();
for line in stdin.lock().lines() {
let line = match line {
Ok(line) => line,
Err(error) => {
eprintln!("worker stdin failure: {error}");
continue;
}
};
if line.trim().is_empty() {
continue;
}
let request = match serde_json::from_str::<WorkerRequest>(&line) {
Ok(request) => request,
Err(error) => {
let response = WorkerResponse {
id: crate::mcp::protocol::HostRequestId(0),
outcome: WorkerOutcome::Fault {
fault: FaultRecord::new(
FaultKind::InvalidInput,
FaultStage::Protocol,
"worker.parse",
format!("invalid worker request: {error}"),
),
},
};
write_message(&mut stdout, &response)?;
continue;
}
};
let WorkerRequest::Execute { id, operation } = request;
let outcome = match service.execute(operation) {
Ok(result) => WorkerOutcome::Success { result },
Err(fault) => WorkerOutcome::Fault { fault },
};
write_message(&mut stdout, &WorkerResponse { id, outcome })?;
}
Ok(())
}
fn write_message(
stdout: &mut impl Write,
response: &WorkerResponse,
) -> Result<(), fidget_spinner_store_sqlite::StoreError> {
serde_json::to_writer(&mut *stdout, response)?;
stdout.write_all(b"\n")?;
stdout.flush()?;
Ok(())
}
|