2019-05-22 09:12:03 +02:00
|
|
|
use crate::prelude::*;
|
|
|
|
use std::sync::Arc;
|
|
|
|
use subprocess::Exec;
|
|
|
|
|
|
|
|
crate enum ClassifiedCommand {
|
|
|
|
Internal(InternalCommand),
|
|
|
|
External(ExternalCommand),
|
|
|
|
}
|
|
|
|
|
|
|
|
impl ClassifiedCommand {
|
2019-05-23 09:23:06 +02:00
|
|
|
crate async fn run(
|
2019-05-22 09:12:03 +02:00
|
|
|
self,
|
2019-05-23 09:23:06 +02:00
|
|
|
input: InputStream,
|
2019-05-23 06:30:43 +02:00
|
|
|
context: &mut Context,
|
2019-05-23 09:23:06 +02:00
|
|
|
) -> Result<InputStream, ShellError> {
|
2019-05-22 09:12:03 +02:00
|
|
|
match self {
|
|
|
|
ClassifiedCommand::Internal(internal) => {
|
|
|
|
let result = context.run_command(internal.command, internal.args, input)?;
|
2019-05-23 09:23:06 +02:00
|
|
|
let env = context.env.clone();
|
2019-05-22 09:12:03 +02:00
|
|
|
|
2019-05-23 09:23:06 +02:00
|
|
|
let stream = result.filter_map(move |v| match v {
|
|
|
|
ReturnValue::Action(action) => match action {
|
|
|
|
CommandAction::ChangeCwd(cwd) => {
|
|
|
|
env.lock().unwrap().cwd = cwd;
|
|
|
|
futures::future::ready(None)
|
|
|
|
}
|
|
|
|
},
|
2019-05-22 09:12:03 +02:00
|
|
|
|
2019-05-23 09:23:06 +02:00
|
|
|
ReturnValue::Value(v) => futures::future::ready(Some(v)),
|
|
|
|
});
|
2019-05-22 09:12:03 +02:00
|
|
|
|
2019-05-23 09:23:06 +02:00
|
|
|
Ok(stream.boxed() as InputStream)
|
2019-05-22 09:12:03 +02:00
|
|
|
}
|
|
|
|
|
|
|
|
ClassifiedCommand::External(external) => {
|
|
|
|
Exec::shell(&external.name)
|
|
|
|
.args(&external.args)
|
2019-05-23 09:23:06 +02:00
|
|
|
.cwd(context.env.lock().unwrap().cwd())
|
2019-05-22 09:12:03 +02:00
|
|
|
.join()
|
|
|
|
.unwrap();
|
2019-05-23 09:23:06 +02:00
|
|
|
|
|
|
|
Ok(VecDeque::new().boxed() as InputStream)
|
2019-05-22 09:12:03 +02:00
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
crate struct InternalCommand {
|
|
|
|
crate command: Arc<dyn Command>,
|
|
|
|
crate args: Vec<Value>,
|
|
|
|
}
|
|
|
|
|
|
|
|
crate struct ExternalCommand {
|
|
|
|
crate name: String,
|
|
|
|
crate args: Vec<String>,
|
|
|
|
}
|