nushell/src/commands/plugin.rs

use crate::commands::command::SinkCommandArgs;
use crate::errors::ShellError;
use crate::prelude::*;
use serde::{self, Deserialize, Serialize};
use std::io::prelude::*;
use std::io::BufReader;
use std::io::Write;

#[derive(Debug, Serialize, Deserialize)]
pub struct JsonRpc<T> {
    jsonrpc: String,
    pub method: String,
    pub params: T,
}

impl<T> JsonRpc<T> {
    pub fn new<U: Into<String>>(method: U, params: T) -> Self {
        JsonRpc {
            jsonrpc: "2.0".into(),
            method: method.into(),
            params,
        }
    }
}

#[derive(Debug, Serialize, Deserialize)]
#[serde(tag = "method")]
#[allow(non_camel_case_types)]
pub enum NuResult {
    response {
        params: Result<VecDeque<ReturnValue>, ShellError>,
    },
}

pub fn filter_plugin(plugin_name: String, args: CommandArgs) -> Result<OutputStream, ShellError> {
    let mut path = std::path::PathBuf::from(".");
    path.push("target");
    path.push("debug");
    path.push(format!("nu_plugin_{}", plugin_name));

    path = if path.exists() {
        path
    } else {
        std::path::PathBuf::from(format!("nu_plugin_{}", plugin_name))
    };

    let mut child = std::process::Command::new(path)
        .stdin(std::process::Stdio::piped())
        .stdout(std::process::Stdio::piped())
        .spawn()
        .expect("Failed to spawn child process");

    {
        let stdin = child.stdin.as_mut().expect("Failed to open stdin");
        let stdout = child.stdout.as_mut().expect("Failed to open stdout");

        let _reader = BufReader::new(stdout);

        let request = JsonRpc::new("begin_filter", args.args);
        let request_raw = serde_json::to_string(&request).unwrap();
        stdin.write(format!("{}\n", request_raw).as_bytes())?;
    }

    let mut eos = VecDeque::new();
    eos.push_back(Value::Primitive(Primitive::EndOfStream));

    let stream = args
        .input
        .chain(eos)
        .map(move |v| match v {
            Value::Primitive(Primitive::EndOfStream) => {
                let stdin = child.stdin.as_mut().expect("Failed to open stdin");
                let stdout = child.stdout.as_mut().expect("Failed to open stdout");

                let _ = BufReader::new(stdout);
                let request: JsonRpc<std::vec::Vec<Value>> = JsonRpc::new("quit", vec![]);
                let request_raw = serde_json::to_string(&request).unwrap();
                let _ = stdin.write(format!("{}\n", request_raw).as_bytes()); // TODO: Handle error

                VecDeque::new()
            }
            _ => {
                let stdin = child.stdin.as_mut().expect("Failed to open stdin");
                let stdout = child.stdout.as_mut().expect("Failed to open stdout");

                let mut reader = BufReader::new(stdout);

                let request = JsonRpc::new("filter", v);
                let request_raw = serde_json::to_string(&request).unwrap();
                let _ = stdin.write(format!("{}\n", request_raw).as_bytes()); // TODO: Handle error

                let mut input = String::new();
                match reader.read_line(&mut input) {
                    Ok(_) => {
                        let response = serde_json::from_str::<NuResult>(&input);
                        match response {
                            Ok(NuResult::response { params }) => match params {
                                Ok(params) => params,
                                Err(e) => {
                                    let mut result = VecDeque::new();
                                    result.push_back(ReturnValue::Value(Value::Error(Box::new(e))));
                                    result
                                }
                            },
                            Err(e) => {
                                let mut result = VecDeque::new();
                                result.push_back(ReturnValue::Value(Value::Error(Box::new(
                                    ShellError::string(format!(
                                        "Error while processing input: {:?} {}",
                                        e, input
                                    )),
                                ))));
                                result
                            }
                        }
                    }
                    Err(e) => {
                        let mut result = VecDeque::new();
                        result.push_back(ReturnValue::Value(Value::Error(Box::new(
                            ShellError::string(format!("Error while processing input: {:?}", e)),
                        ))));
                        result
                    }
                }
            }
        })
        .flatten();

    Ok(stream.boxed())
}

pub fn sink_plugin(plugin_name: String, args: SinkCommandArgs) -> Result<(), ShellError> {
    let mut path = std::path::PathBuf::from(".");
    path.push("target");
    path.push("debug");
    path.push(format!("nu_plugin_{}", plugin_name));

    path = if path.exists() {
        path
    } else {
        std::path::PathBuf::from(format!("nu_plugin_{}", plugin_name))
    };

    let mut child = std::process::Command::new(path)
        .stdin(std::process::Stdio::piped())
        .spawn()
        .expect("Failed to spawn child process");

    let stdin = child.stdin.as_mut().expect("Failed to open stdin");

    let request = JsonRpc::new("sink", (args.args, args.input));
    let request_raw = serde_json::to_string(&request).unwrap();
    stdin.write(format!("{}\n", request_raw).as_bytes())?;

    let _ = child.wait();

    Ok(())
}