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 105 106 107 108 109 110 111 112 113 114
use crate::sdk::std::json::OBJECT_VALUE; use crate::utils::pckg; use duckscript::types::command::{Command, CommandResult, Commands}; use duckscript::types::instruction::Instruction; use duckscript::types::runtime::StateValue; use serde_json::{Result, Value}; use std::collections::HashMap; #[cfg(test)] #[path = "./mod_test.rs"] mod mod_test; fn parse_json(data: &str) -> Result<Value> { let value: Value = serde_json::from_str(data)?; Ok(value) } fn create_variables(data: Value, name: &str, variables: &mut HashMap<String, String>) { match data { Value::Null => variables.remove(name), Value::Bool(value) => variables.insert(name.to_string(), value.to_string()), Value::Number(value) => variables.insert(name.to_string(), value.to_string()), Value::String(value) => variables.insert(name.to_string(), value), Value::Array(list) => { let mut index = 0; for item in list { let child_name = format!("{}[{}]", name, index); create_variables(item, &child_name, variables); index = index + 1; } variables.insert(format!("{}.length", name), index.to_string()); None } Value::Object(map) => { variables.insert(name.to_string(), OBJECT_VALUE.to_string()); for (key, value) in map { let child_name = format!("{}.{}", name, key); create_variables(value, &child_name, variables); } None } }; } #[derive(Clone)] pub(crate) struct CommandImpl { package: String, } impl Command for CommandImpl { fn name(&self) -> String { pckg::concat(&self.package, "Parse") } fn aliases(&self) -> Vec<String> { vec!["json_parse".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("No JSON string provided.".to_string()) } else { match parse_json(&arguments[0]) { Ok(data) => { let output = match output_variable { Some(name) => { create_variables(data, &name, variables); match variables.get(&name) { Some(value) => Some(value.to_string()), None => None, } } None => Some("true".to_string()), }; CommandResult::Continue(output) } Err(error) => CommandResult::Error(error.to_string()), } } } } pub(crate) fn create(package: &str) -> Box<dyn Command> { Box::new(CommandImpl { package: package.to_string(), }) }