(parser) feat: impl MakeFunction for Variables

Author: t-webber

src/parser/variable/declaration.rs

@@ -9,6 +9,7 @@ use super::{Variable, traits};
 use crate::parser::display::repr_option_vec;
 use crate::parser::keyword::attributes::{AttributeKeyword, UserDefinedTypes};
 use crate::parser::literal::{Attribute, repr_vec_attr};
+use crate::parser::modifiers::functions::{CanMakeFnRes, MakeFunction};
 use crate::parser::modifiers::push::Push;
 use crate::parser::operators::api::OperatorConversions;
 use crate::parser::tree::api::{Ast, CanPush};
@@ -93,6 +94,30 @@ impl AttributeVariable {
     }
 }
 
+impl MakeFunction for AttributeVariable {
+    fn can_make_function(&self) -> CanMakeFnRes {
+        match self.declarations.last()? {
+            Some(declaration) => declaration.value.as_ref()?.can_make_function(),
+            None => CanMakeFnRes::None,
+        }
+    }
+
+    fn make_function(&mut self, depth: u32, arguments: Vec<Ast>) {
+        match self
+            .declarations
+            .last_mut()
+            .expect("checked with can_make_function")
+        {
+            Some(declaration) => declaration
+                .value
+                .as_mut()
+                .expect("checked with can_make_function")
+                .make_function(depth, arguments),
+            None => unreachable!(),
+        }
+    }
+}
+
 impl CanPush for AttributeVariable {
     fn can_push_leaf(&self) -> bool {
         self.declarations.last().is_some_and(|opt| {

src/parser/variable/mod.rs

@@ -109,11 +109,15 @@ impl Variable {
 
 impl MakeFunction for Variable {
     fn can_make_function(&self) -> CanMakeFnRes {
-        todo!()
+        if self.full {
+            CanMakeFnRes::None
+        } else {
+            self.value.can_make_function()
+        }
     }
 
     fn make_function(&mut self, depth: u32, arguments: Vec<Ast>) {
-        todo!()
+        self.value.make_function(depth, arguments);
     }
 }
 

src/parser/variable/value.rs

@@ -8,6 +8,8 @@ use super::name::VariableName;
 use super::traits::{PureType, VariableConversion};
 use crate::parser::keyword::attributes::UserDefinedTypes;
 use crate::parser::literal::Attribute;
+use crate::parser::modifiers::functions::{CanMakeFnRes, MakeFunction};
+use crate::parser::tree::Ast;
 use crate::parser::tree::api::{CanPush, PushAttribute};
 use crate::utils::display;
 
@@ -108,6 +110,22 @@ impl VariableValue {
     }
 }
 
+impl MakeFunction for VariableValue {
+    fn can_make_function(&self) -> CanMakeFnRes {
+        match self {
+            Self::AttributeVariable(var) => var.can_make_function(),
+            Self::VariableName(_) => CanMakeFnRes::None,
+        }
+    }
+
+    fn make_function(&mut self, depth: u32, arguments: Vec<Ast>) {
+        match self {
+            Self::AttributeVariable(var) => var.make_function(depth, arguments),
+            Self::VariableName(_) => unreachable!(),
+        }
+    }
+}
+
 impl CanPush for VariableValue {
     fn can_push_leaf(&self) -> bool {
         match self {