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

use std::collections::HashMap;

use crate::{
    commands::{
        HelpCmd, HelpManual, NuCmd, PreviewCmd, QuitCmd, SimpleCommand, TryCmd, ViewCommand,
    },
    views::View,
    TableConfig,
};

#[derive(Clone)]
pub enum Command {
    Reactive(Box<dyn SCommand>),
    View {
        cmd: Box<dyn VCommand>,
        is_light: bool,
    },
}

pub struct CommandList {
    commands: HashMap<&'static str, Command>,
    aliases: HashMap<&'static str, &'static str>,
}

macro_rules! cmd_view {
    ($object:expr, $light:expr) => {{
        let object = $object;

        let name = object.name();

        let cmd = Box::new(ViewCmd(object)) as Box<dyn VCommand>;
        let cmd = Command::View {
            cmd,
            is_light: $light,
        };

        (name, cmd)
    }};
    ($object:expr) => {
        cmd_view!($object, false)
    };
}

macro_rules! cmd_react {
    ($object:expr) => {{
        let object = $object;

        let name = object.name();
        let cmd = Command::Reactive(Box::new($object) as Box<dyn SCommand>);

        (name, cmd)
    }};
}

impl CommandList {
    pub fn create_commands(table_cfg: TableConfig) -> Vec<(&'static str, Command)> {
        vec![
            cmd_view!(NuCmd::new(table_cfg)),
            cmd_view!(TryCmd::new(table_cfg), true),
            cmd_view!(PreviewCmd::new(), true),
            cmd_react!(QuitCmd::default()),
        ]
    }

    pub fn create_aliases() -> [(&'static str, &'static str); 3] {
        [
            ("h", HelpCmd::NAME),
            ("q", QuitCmd::NAME),
            ("q!", QuitCmd::NAME),
        ]
    }

    pub fn new(table_cfg: TableConfig) -> Self {
        let mut cmd_list = Self::create_commands(table_cfg);
        let aliases = Self::create_aliases();

        let help_cmd = create_help_command(&cmd_list, &aliases, table_cfg);

        cmd_list.push(cmd_view!(help_cmd, true));

        Self {
            commands: HashMap::from_iter(cmd_list),
            aliases: HashMap::from_iter(aliases),
        }
    }

    pub fn find(&self, args: &str) -> Option<std::io::Result<Command>> {
        let cmd = args.split_once(' ').map_or(args, |(cmd, _)| cmd);
        let args = &args[cmd.len()..];

        let command = self.find_command(cmd);
        parse_command(command, args)
    }

    fn find_command(&self, cmd: &str) -> Option<Command> {
        match self.commands.get(cmd).cloned() {
            None => self
                .aliases
                .get(cmd)
                .and_then(|cmd| self.commands.get(cmd).cloned()),
            cmd => cmd,
        }
    }
}

fn create_help_command(
    commands: &[(&str, Command)],
    aliases: &[(&str, &str)],
    table_cfg: TableConfig,
) -> HelpCmd {
    let help_manuals = create_help_manuals(commands);
    HelpCmd::new(help_manuals, aliases, table_cfg)
}

fn parse_command(command: Option<Command>, args: &str) -> Option<std::io::Result<Command>> {
    match command {
        Some(mut cmd) => {
            let result = match &mut cmd {
                Command::Reactive(cmd) => cmd.parse(args),
                Command::View { cmd, .. } => cmd.parse(args),
            };

            Some(result.map(|_| cmd))
        }
        None => None,
    }
}

fn create_help_manuals(cmd_list: &[(&str, Command)]) -> Vec<HelpManual> {
    let mut help_manuals: Vec<_> = cmd_list
        .iter()
        .map(|(_, cmd)| cmd)
        .map(create_help_manual)
        .collect();

    help_manuals.push(__create_help_manual(
        HelpCmd::default().help(),
        HelpCmd::NAME,
    ));

    help_manuals
}

fn create_help_manual(cmd: &Command) -> HelpManual {
    let name = match cmd {
        Command::Reactive(cmd) => cmd.name(),
        Command::View { cmd, .. } => cmd.name(),
    };

    let manual = match cmd {
        Command::Reactive(cmd) => cmd.help(),
        Command::View { cmd, .. } => cmd.help(),
    };

    __create_help_manual(manual, name)
}

fn __create_help_manual(manual: Option<HelpManual>, name: &'static str) -> HelpManual {
    match manual {
        Some(manual) => manual,
        None => HelpManual {
            name,
            description: "",
            arguments: Vec::new(),
            examples: Vec::new(),
        },
    }
}

// type helper to deal with `Box`es
#[derive(Clone)]
struct ViewCmd<C>(C);

impl<C> ViewCommand for ViewCmd<C>
where
    C: ViewCommand,
    C::View: View + 'static,
{
    type View = Box<dyn View>;

    fn name(&self) -> &'static str {
        self.0.name()
    }

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

    fn help(&self) -> Option<HelpManual> {
        self.0.help()
    }

    fn parse(&mut self, args: &str) -> std::io::Result<()> {
        self.0.parse(args)
    }

    fn spawn(
        &mut self,
        engine_state: &nu_protocol::engine::EngineState,
        stack: &mut nu_protocol::engine::Stack,
        value: Option<nu_protocol::Value>,
    ) -> std::io::Result<Self::View> {
        let view = self.0.spawn(engine_state, stack, value)?;
        Ok(Box::new(view) as Box<dyn View>)
    }
}

pub trait SCommand: SimpleCommand + SCommandClone {}

impl<T> SCommand for T where T: 'static + SimpleCommand + Clone {}

pub trait SCommandClone {
    fn clone_box(&self) -> Box<dyn SCommand>;
}

impl<T> SCommandClone for T
where
    T: 'static + SCommand + Clone,
{
    fn clone_box(&self) -> Box<dyn SCommand> {
        Box::new(self.clone())
    }
}

impl Clone for Box<dyn SCommand> {
    fn clone(&self) -> Box<dyn SCommand> {
        self.clone_box()
    }
}

pub trait VCommand: ViewCommand<View = Box<dyn View>> + VCommandClone {}

impl<T> VCommand for T where T: 'static + ViewCommand<View = Box<dyn View>> + Clone {}

pub trait VCommandClone {
    fn clone_box(&self) -> Box<dyn VCommand>;
}

impl<T> VCommandClone for T
where
    T: 'static + VCommand + Clone,
{
    fn clone_box(&self) -> Box<dyn VCommand> {
        Box::new(self.clone())
    }
}

impl Clone for Box<dyn VCommand> {
    fn clone(&self) -> Box<dyn VCommand> {
        self.clone_box()
    }
}