use std::collections::BTreeMap;
use serde_json::value::Value as Json;
use crate::error::RenderError;
use crate::local_vars::LocalVars;
#[derive(Clone, Debug)]
pub enum BlockParamHolder {
Path(Vec<String>),
Value(Json),
}
impl BlockParamHolder {
pub fn value(v: Json) -> BlockParamHolder {
BlockParamHolder::Value(v)
}
pub fn path(r: Vec<String>) -> BlockParamHolder {
BlockParamHolder::Path(r)
}
}
#[derive(Clone, Debug, Default)]
pub struct BlockParams<'reg> {
data: BTreeMap<&'reg str, BlockParamHolder>,
}
impl<'reg> BlockParams<'reg> {
pub fn new() -> BlockParams<'reg> {
BlockParams::default()
}
pub fn add_path(&mut self, k: &'reg str, path: Vec<String>) -> Result<(), RenderError> {
self.data.insert(k, BlockParamHolder::path(path));
Ok(())
}
pub fn add_value(&mut self, k: &'reg str, v: Json) -> Result<(), RenderError> {
self.data.insert(k, BlockParamHolder::value(v));
Ok(())
}
pub fn get(&self, k: &str) -> Option<&BlockParamHolder> {
self.data.get(k)
}
}
#[derive(Debug, Clone, Default)]
pub struct BlockContext<'reg> {
base_path: Vec<String>,
base_value: Option<Json>,
block_params: BlockParams<'reg>,
local_variables: LocalVars,
}
impl<'reg> BlockContext<'reg> {
pub fn new() -> BlockContext<'reg> {
BlockContext::default()
}
pub fn set_local_var(&mut self, name: &str, value: Json) {
self.local_variables.put(name, value);
}
pub fn get_local_var(&self, name: &str) -> Option<&Json> {
self.local_variables.get(name)
}
pub fn base_path(&self) -> &Vec<String> {
&self.base_path
}
pub fn base_path_mut(&mut self) -> &mut Vec<String> {
&mut self.base_path
}
pub fn base_value(&self) -> Option<&Json> {
self.base_value.as_ref()
}
pub fn set_base_value(&mut self, value: Json) {
self.base_value = Some(value);
}
pub fn get_block_param(&self, block_param_name: &str) -> Option<&BlockParamHolder> {
self.block_params.get(block_param_name)
}
pub fn set_block_params(&mut self, block_params: BlockParams<'reg>) {
self.block_params = block_params;
}
}