use serde::{Deserialize, Serialize}; use tree_sitter::Node; use crate::{AbstractTree, Error, Map, Result, Type, Value, BUILT_IN_FUNCTIONS}; #[derive(Debug, Clone, Serialize, Deserialize, Eq, PartialEq, PartialOrd, Ord)] pub struct Identifier(String); impl Identifier { pub fn new(inner: String) -> Self { Identifier(inner) } pub fn take_inner(self) -> String { self.0 } pub fn inner(&self) -> &String { &self.0 } } impl AbstractTree for Identifier { fn from_syntax_node(source: &str, node: Node, _context: &Map) -> Result { Error::expect_syntax_node(source, "identifier", node)?; let text = &source[node.byte_range()]; Ok(Identifier(text.to_string())) } fn run(&self, _source: &str, context: &Map) -> Result { if let Some(value) = context.variables()?.get(&self.0) { Ok(value.clone()) } else { Err(Error::VariableIdentifierNotFound(self.inner().clone())) } } fn expected_type(&self, context: &Map) -> Result { if let Some(value) = context.variables()?.get(&self.0) { value.r#type(context) } else { for built_in_function in BUILT_IN_FUNCTIONS { if self.0 == built_in_function.name() { return Ok(built_in_function.r#type()); } } Ok(Type::Any) } } }