pray-cli 1.10.0

Package manager for the language placed before inference — Prayfile CLI (binary: pray)
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

use crate::server::{handle_rpc, ServeAuth};
use pray_core::ssh_rpc::{read_frame, write_frame, RpcRequest, RpcResponse};
use pray_core::{PrayError, PrayResult};
use std::io::{self, BufReader, Write};
use std::path::PathBuf;

pub fn run_stdio_server(root: PathBuf) -> PrayResult<()> {
    std::env::set_var("PRAY_SERVE_STDIO", "1");
    let stdin = io::stdin();
    let mut stdout = io::stdout();
    let mut reader = BufReader::new(stdin.lock());
    loop {
        let frame = match read_frame(&mut reader) {
            Ok(frame) => frame,
            Err(PrayError::Resolution(message)) if message == "rpc stream closed" => return Ok(()),
            Err(error) => return Err(error),
        };
        let request: RpcRequest =
            serde_json::from_slice(&frame).map_err(|error| PrayError::Parse {
                kind: "ssh rpc request",
                message: error.to_string(),
            })?;
        let auth = ServeAuth::stdio();
        let response = match handle_rpc(&root, &auth, &request) {
            Ok(response) => response,
            Err(error) => RpcResponse::error(&request.id, 500, error.to_string()),
        };
        let payload = serde_json::to_vec(&response)
            .map_err(|error| PrayError::Manifest(error.to_string()))?;
        write_frame(&mut stdout, &payload)?;
        stdout.flush()?;
    }
}