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 94 95 96 97 98 99 100 101 102 103 104
use crate::utils::pckg; use crate::utils::state::{get_as_string, get_handles_sub_state}; use duckscript::types::command::{Command, CommandResult, Commands}; use duckscript::types::instruction::Instruction; use duckscript::types::runtime::StateValue; use java_properties::write; use std::collections::HashMap; use std::str; #[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, "MapToProperties") } fn aliases(&self) -> Vec<String> { vec!["map_to_properties".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 { if arguments.is_empty() { CommandResult::Error("Map handle not provided.".to_string()) } else { let (prefix, key) = if arguments.len() >= 3 && arguments[0] == "--prefix" { (arguments[1].to_string(), arguments[2].to_string()) } else { ("".to_string(), arguments[0].to_string()) }; let state = get_handles_sub_state(state); match state.get(&key) { Some(state_value) => match state_value { StateValue::SubState(ref map) => { let mut properties = HashMap::new(); for (property_key, property_value) in map { let mut var_key = property_key.to_string(); if !prefix.is_empty() { var_key.insert(0, '.'); var_key.insert_str(0, &prefix); } let string_value = match get_as_string(property_value) { Ok(value) => value, Err(error) => return CommandResult::Error(error), }; properties.insert(var_key, string_value); } let mut buffer: Vec<u8> = vec![]; match write(&mut buffer, &properties) { Ok(_) => match str::from_utf8(&buffer) { Ok(text) => CommandResult::Continue(Some(text.trim().to_string())), Err(error) => CommandResult::Error(error.to_string()), }, Err(error) => CommandResult::Error(error.to_string()), } } _ => CommandResult::Error("Invalid handle provided.".to_string()), }, None => { CommandResult::Error(format!("Map for handle: {} not found.", key).to_string()) } } } } } pub(crate) fn create(package: &str) -> Box<dyn Command> { Box::new(CommandImpl { package: package.to_string(), }) }