nushell/src/commands/command.rs

85 lines
1.8 KiB
Rust
Raw Normal View History

2019-05-10 18:59:12 +02:00
use crate::errors::ShellError;
use crate::object::Value;
2019-05-28 08:45:18 +02:00
use crate::parser::CommandConfig;
use crate::prelude::*;
2019-05-11 10:08:21 +02:00
use std::path::PathBuf;
2019-05-10 18:59:12 +02:00
pub struct CommandArgs {
2019-05-24 06:34:43 +02:00
pub host: Arc<Mutex<dyn Host + Send>>,
pub env: Arc<Mutex<Environment>>,
2019-05-16 02:21:46 +02:00
pub args: Vec<Value>,
pub input: InputStream,
2019-05-16 02:21:46 +02:00
}
impl CommandArgs {
2019-05-22 09:12:03 +02:00
crate fn from_context(
ctx: &'caller mut Context,
args: Vec<Value>,
input: InputStream,
) -> CommandArgs {
2019-05-22 09:12:03 +02:00
CommandArgs {
host: ctx.host.clone(),
env: ctx.env.clone(),
2019-05-22 09:12:03 +02:00
args,
input,
}
}
2019-05-11 10:08:21 +02:00
}
2019-05-15 18:12:38 +02:00
#[derive(Debug)]
pub enum CommandAction {
2019-05-11 10:08:21 +02:00
ChangeCwd(PathBuf),
}
2019-05-15 18:12:38 +02:00
#[derive(Debug)]
pub enum ReturnValue {
Value(Value),
Action(CommandAction),
2019-05-11 10:08:21 +02:00
}
impl ReturnValue {
crate fn change_cwd(path: PathBuf) -> ReturnValue {
ReturnValue::Action(CommandAction::ChangeCwd(path))
2019-05-11 10:08:21 +02:00
}
2019-05-10 18:59:12 +02:00
}
pub trait Command {
fn run(&self, args: CommandArgs) -> Result<OutputStream, ShellError>;
2019-05-28 08:45:18 +02:00
fn name(&self) -> &str;
fn config(&self) -> CommandConfig {
CommandConfig {
name: self.name().to_string(),
mandatory_positional: vec![],
optional_positional: vec![],
rest_positional: true,
named: indexmap::IndexMap::new(),
}
}
}
pub struct FnCommand {
name: String,
func: fn(CommandArgs) -> Result<OutputStream, ShellError>,
}
2019-05-22 09:12:03 +02:00
2019-05-28 08:45:18 +02:00
impl Command for FnCommand {
fn run(&self, args: CommandArgs) -> Result<OutputStream, ShellError> {
2019-05-28 08:45:18 +02:00
(self.func)(args)
2019-05-22 09:12:03 +02:00
}
2019-05-28 08:45:18 +02:00
fn name(&self) -> &str {
&self.name
}
}
pub fn command(
name: &str,
func: fn(CommandArgs) -> Result<OutputStream, ShellError>,
) -> Arc<dyn Command> {
Arc::new(FnCommand {
name: name.to_string(),
func,
})
2019-05-22 09:12:03 +02:00
}