nushell/crates/nu-parser/src/known_external.rs

use nu_protocol::ast::Expr;
use nu_protocol::engine::{EngineState, Stack, StateWorkingSet};
use nu_protocol::PipelineData;
use nu_protocol::{ast::Call, engine::Command, ShellError, Signature};

#[derive(Clone)]
pub struct KnownExternal {
    pub name: String,
    pub signature: Box<Signature>,
    pub usage: String,
}

impl Command for KnownExternal {
    fn name(&self) -> &str {
        &self.name
    }

    fn signature(&self) -> Signature {
        *self.signature.clone()
    }

    fn usage(&self) -> &str {
        &self.usage
    }

    fn is_known_external(&self) -> bool {
        true
    }

    fn run(
        &self,
        engine_state: &EngineState,
        stack: &mut Stack,
        call: &Call,
        input: PipelineData,
    ) -> Result<PipelineData, ShellError> {
        // FIXME: This is a bit of a hack, and it'd be nice for the parser/AST to be able to handle the original
        // order of the parameters. Until then, we need to recover the original order.
        let call_span = call.span();
        let contents = engine_state.get_span_contents(&call_span);

        let (lexed, _) = crate::lex(contents, call_span.start, &[], &[], true);

        let spans: Vec<_> = lexed.into_iter().map(|x| x.span).collect();
        let mut working_set = StateWorkingSet::new(engine_state);
        let (external_call, _) = crate::parse_external_call(&mut working_set, &spans);

        match external_call.expr {
            Expr::ExternalCall(head, args) => {
                let decl_id = engine_state
                    .find_decl("run-external".as_bytes())
                    .ok_or(ShellError::ExternalNotSupported(head.span))?;

                let command = engine_state.get_decl(decl_id);

                let mut call = Call::new(head.span);

                call.positional.push(*head);

                for arg in args {
                    call.positional.push(arg.clone())
                }

                // if last_expression {
                //     call.named.push((
                //         Spanned {
                //             item: "last-expression".into(),
                //             span: head.span,
                //         },
                //         None,
                //     ))
                // }

                command.run(engine_state, stack, &call, input)
            }
            x => {
                println!("{:?}", x);
                panic!("internal error: known external not actually external")
            }
        }
    }
}