nushell/crates/nu-command/src/default_context.rs

use std::{cell::RefCell, rc::Rc};

use nu_protocol::{
    engine::{EngineState, StateWorkingSet},
    Signature,
};

use crate::*;

pub fn create_default_context() -> Rc<RefCell<EngineState>> {
    let engine_state = Rc::new(RefCell::new(EngineState::new()));
    let delta = {
        let engine_state = engine_state.borrow();
        let mut working_set = StateWorkingSet::new(&*engine_state);

        working_set.add_decl(Box::new(Alias));
        working_set.add_decl(Box::new(Benchmark));
        working_set.add_decl(Box::new(BuildString));
        working_set.add_decl(Box::new(Cd));
        working_set.add_decl(Box::new(Def));
        working_set.add_decl(Box::new(Do));
        working_set.add_decl(Box::new(Each));
        working_set.add_decl(Box::new(External));
        working_set.add_decl(Box::new(For));
        working_set.add_decl(Box::new(From));
        working_set.add_decl(Box::new(FromJson));
        working_set.add_decl(Box::new(Get));
        working_set.add_decl(Box::new(Help));
        working_set.add_decl(Box::new(If));
        working_set.add_decl(Box::new(Length));
        working_set.add_decl(Box::new(Let));
        working_set.add_decl(Box::new(LetEnv));
        working_set.add_decl(Box::new(Lines));
        working_set.add_decl(Box::new(Ls));
        working_set.add_decl(Box::new(Module));
        working_set.add_decl(Box::new(Ps));
        working_set.add_decl(Box::new(Select));
        working_set.add_decl(Box::new(Sys));
        working_set.add_decl(Box::new(Table));
        working_set.add_decl(Box::new(Use));
        working_set.add_decl(Box::new(Where));
        working_set.add_decl(Box::new(Wrap));

        // This is a WIP proof of concept
        working_set.add_decl(Box::new(ListGitBranches));
        working_set.add_decl(Box::new(Git));
        working_set.add_decl(Box::new(GitCheckout));

        let sig = Signature::build("exit");
        working_set.add_decl(sig.predeclare());
        let sig = Signature::build("vars");
        working_set.add_decl(sig.predeclare());
        let sig = Signature::build("decls");
        working_set.add_decl(sig.predeclare());
        let sig = Signature::build("blocks");
        working_set.add_decl(sig.predeclare());
        let sig = Signature::build("stack");
        working_set.add_decl(sig.predeclare());
        let sig = Signature::build("contents");
        working_set.add_decl(sig.predeclare());

        working_set.render()
    };

    {
        EngineState::merge_delta(&mut *engine_state.borrow_mut(), delta);
    }

    engine_state
}