System Initiative

use std::io::{ BufRead, Write, }; use object_tree::{ GraphError, NameStr, NodeChild, NodeKind, NodeWithChildren, ReadBytes, WriteBytes, read_key_value_line, write_key_value_line, }; use serde::{ Deserialize, Serialize, }; use super::PkgNode; use crate::{ AttrFuncInputSpec, MapKeyFuncSpec, PropSpec, }; const PROP_CHILD_TYPE_PROPS: &str = "props"; const PROP_CHILD_TYPE_ATTR_FUNC_INPUTS: &str = "attr_func_inputs"; const PROP_CHILD_TYPE_MAP_KEY_FUNCS: &str = "map_key_funcs"; const KEY_KIND_STR: &str = "kind"; #[remain::sorted] #[derive(Debug, Deserialize, Serialize)] #[serde(rename_all = "camelCase")] pub enum PropChild { AttrFuncInputs(Vec<AttrFuncInputSpec>), MapKeyFuncs(Vec<MapKeyFuncSpec>), Props(Vec<PropSpec>), } #[remain::sorted] #[derive(Clone, Copy, Debug, Serialize, PartialEq, Eq)] pub enum PropChildNode { AttrFuncInputs, MapKeyFuncs, Props, } impl PropChildNode { pub fn kind_str(&self) -> &'static str { match self { Self::AttrFuncInputs => PROP_CHILD_TYPE_ATTR_FUNC_INPUTS, Self::Props => PROP_CHILD_TYPE_PROPS, Self::MapKeyFuncs => PROP_CHILD_TYPE_MAP_KEY_FUNCS, } } } impl NameStr for PropChildNode { fn name(&self) -> &str { match self { Self::AttrFuncInputs => PROP_CHILD_TYPE_ATTR_FUNC_INPUTS, Self::MapKeyFuncs => PROP_CHILD_TYPE_MAP_KEY_FUNCS, Self::Props => PROP_CHILD_TYPE_PROPS, } } } impl WriteBytes for PropChildNode { fn write_bytes<W: Write>(&self, writer: &mut W) -> Result<(), GraphError> { write_key_value_line(writer, KEY_KIND_STR, self.kind_str())?; Ok(()) } } impl ReadBytes for PropChildNode { fn read_bytes<R: BufRead>(reader: &mut R) -> Result<Option<Self>, GraphError> where Self: std::marker::Sized, { let kind_str = read_key_value_line(reader, KEY_KIND_STR)?; let node = match kind_str.as_str() { PROP_CHILD_TYPE_ATTR_FUNC_INPUTS => Self::AttrFuncInputs, PROP_CHILD_TYPE_MAP_KEY_FUNCS => Self::MapKeyFuncs, PROP_CHILD_TYPE_PROPS => Self::Props, _ => return Ok(None), }; Ok(Some(node)) } } impl NodeChild for PropChild { type NodeType = PkgNode; fn as_node_with_children(&self) -> NodeWithChildren<Self::NodeType> { match self { Self::AttrFuncInputs(inputs) => NodeWithChildren::new( NodeKind::Tree, Self::NodeType::PropChild(PropChildNode::AttrFuncInputs), inputs .iter() .map(|input| { Box::new(input.clone()) as Box<dyn NodeChild<NodeType = Self::NodeType>> }) .collect(), ), Self::MapKeyFuncs(map_key_funcs) => NodeWithChildren::new( NodeKind::Tree, Self::NodeType::PropChild(PropChildNode::MapKeyFuncs), map_key_funcs .iter() .map(|map_key_func| { Box::new(map_key_func.clone()) as Box<dyn NodeChild<NodeType = Self::NodeType>> }) .collect(), ), Self::Props(props) => NodeWithChildren::new( NodeKind::Tree, Self::NodeType::PropChild(PropChildNode::Props), props .iter() .map(|prop| { Box::new(prop.clone()) as Box<dyn NodeChild<NodeType = Self::NodeType>> }) .collect(), ), } } }