From a0999e30f1a2bc05f35913fe3b31879c682be18f Mon Sep 17 00:00:00 2001
From: Jeff <dev@jeffa.io>
Date: Tue, 21 May 2024 19:48:27 -0400
Subject: [PATCH] Continue implementing as expression

---
 dust-lang/src/abstract_tree/as.rs | 59 ++++++++++++++++++++++---------
 1 file changed, 43 insertions(+), 16 deletions(-)

diff --git a/dust-lang/src/abstract_tree/as.rs b/dust-lang/src/abstract_tree/as.rs
index b6da716..6bc067e 100644
--- a/dust-lang/src/abstract_tree/as.rs
+++ b/dust-lang/src/abstract_tree/as.rs
@@ -1,4 +1,13 @@
-use super::{AbstractNode, Expression, Type, WithPosition};
+use std::borrow::Borrow;
+
+use crate::{
+    context::Context,
+    error::{RuntimeError, ValidationError},
+    value::ValueInner,
+    Value,
+};
+
+use super::{AbstractNode, Action, Expression, Type, WithPosition};
 
 #[derive(Clone, Debug, Eq, PartialEq, PartialOrd, Ord)]
 pub struct As {
@@ -13,26 +22,44 @@ impl As {
 }
 
 impl AbstractNode for As {
-    fn expected_type(
-        &self,
-        context: &mut crate::context::Context,
-    ) -> Result<Type, crate::error::ValidationError> {
-        todo!()
+    fn expected_type(&self, _: &mut Context) -> Result<Type, ValidationError> {
+        Ok(self.r#type.item.clone())
     }
 
     fn validate(
         &self,
-        context: &mut crate::context::Context,
-        manage_memory: bool,
-    ) -> Result<(), crate::error::ValidationError> {
-        todo!()
+        _context: &mut Context,
+        _manage_memory: bool,
+    ) -> Result<(), ValidationError> {
+        match self.r#type.item {
+            Type::Boolean | Type::Float | Type::Integer | Type::String => {}
+            _ => todo!("Create an error for this occurence."),
+        };
+
+        match self.expression.expected_type(_context)? {
+            Type::Boolean | Type::Float | Type::Integer | Type::String => Ok(()),
+            _ => todo!("Create an error for this occurence."),
+        }
     }
 
-    fn run(
-        self,
-        context: &mut crate::context::Context,
-        manage_memory: bool,
-    ) -> Result<super::Action, crate::error::RuntimeError> {
-        todo!()
+    fn run(self, _context: &mut Context, _manage_memory: bool) -> Result<Action, RuntimeError> {
+        let expression_position = self.expression.position();
+        let action = self.expression.run(_context, _manage_memory)?;
+        let value = if let Action::Return(value) = action {
+            value
+        } else {
+            return Err(RuntimeError::ValidationFailure(
+                ValidationError::InterpreterExpectedReturn(expression_position),
+            ));
+        };
+        let (from_value, to_type): (&ValueInner, Type) = (value.inner().borrow(), self.r#type.item);
+
+        let converted = match (from_value, to_type) {
+            (ValueInner::Boolean(boolean), Type::String) => Value::string(boolean.to_string()),
+            (ValueInner::Integer(integer), Type::String) => Value::string(integer.to_string()),
+            _ => todo!("Create an error for this occurence."),
+        };
+
+        Ok(Action::Return(converted))
     }
 }