1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93
use crate::utils::{exec, pckg}; use duckscript::types::command::{Command, CommandResult, Commands}; use duckscript::types::instruction::Instruction; use duckscript::types::runtime::StateValue; use std::collections::HashMap; #[cfg(test)] #[path = "./mod_test.rs"] mod mod_test; #[derive(Clone)] pub(crate) struct CommandImpl { package: String, } impl Command for CommandImpl { fn name(&self) -> String { pckg::concat(&self.package, "Execute") } fn aliases(&self) -> Vec<String> { vec!["exec".to_string()] } fn help(&self) -> String { include_str!("help.md").to_string() } fn clone_and_box(&self) -> Box<dyn Command> { Box::new((*self).clone()) } fn requires_context(&self) -> bool { true } fn run_with_context( &self, arguments: Vec<String>, _state: &mut HashMap<String, StateValue>, variables: &mut HashMap<String, String>, output_variable: Option<String>, _instructions: &Vec<Instruction>, _commands: &mut Commands, _line: usize, ) -> CommandResult { let allow_input = output_variable.is_some(); let print_output = !allow_input; let (start_index, fail_on_error) = if !arguments.is_empty() && arguments[0] == "--fail-on-error" { (1, output_variable.is_none()) } else { (0, false) }; match exec::exec(&arguments, print_output, allow_input, start_index) { Ok((stdout, stderr, exit_code)) => match output_variable { Some(name) => { let mut key = String::from(&name); key.push_str(".stdout"); variables.insert(key.clone(), stdout); key = String::from(&name); key.push_str(".stderr"); variables.insert(key.clone(), stderr); key = String::from(&name); key.push_str(".code"); variables.insert(key.clone(), exit_code.to_string()); CommandResult::Continue(None) } None => { if fail_on_error && exit_code != 0 { CommandResult::Error( format!("Error while executing command, exit code: {}", exit_code) .to_string(), ) } else { CommandResult::Continue(None) } } }, Err(error) => CommandResult::Error(error), } } } pub(crate) fn create(package: &str) -> Box<dyn Command> { Box::new(CommandImpl { package: package.to_string(), }) }