Support Setting Arrays/Fields

Signed-off-by: Will Hawkins <hawkinsw@obs.cr>
2026-03-20 02:32:15 -04:00
parent d7696e822c
commit 04627f20e2
9 changed files with 326 additions and 74 deletions
@@ -246,3 +246,54 @@ extension ExpressionStatement: CompilableStatement {
    return Result.Ok((ExpressionStatement(), context))
  }
 }
+
+extension ParserAssignmentStatement: CompilableStatement {
+  public static func Compile(
+    node: Node, withContext context: CompilerContext
+  ) -> Result<(EvaluatableStatement, CompilerContext)> {
+
+    #RequireNodeType<Node, (EvaluatableStatement, CompilerContext)>(
+      node: node, type: "assignmentStatement", nice_type_name: "assignment statement")
+
+    guard let lvalue_node = node.child(at: 0),
+      lvalue_node.nodeType == "expression"
+    else {
+      return Result.Error(
+        ErrorOnNode(node: node, withError: "Missing lvalue in assignment statement"))
+    }
+
+    guard let rvalue_node = node.child(at: 2),
+      rvalue_node.nodeType == "expression"
+    else {
+      return Result.Error(
+        ErrorOnNode(node: node, withError: "Missing rvalue in assignment statement"))
+    }
+
+    let maybe_parsed_rvalue = Expression.Compile(
+      node: rvalue_node, withContext: context)
+    guard case Result.Ok(let rvalue) = maybe_parsed_rvalue else {
+      return Result.Error(maybe_parsed_rvalue.error()!)
+    }
+
+    let maybe_parsed_lvalue = LValue.Compile(node: lvalue_node, withContext: context)
+    guard case .Ok(let lvalue_identifier) = maybe_parsed_lvalue else {
+      return Result.Error(maybe_parsed_lvalue.error()!)
+    }
+
+    let check_result = lvalue_identifier.check(to: rvalue, inScopes: context.names)
+    guard case .Ok(_) = check_result else {
+      return Result.Error(
+        ErrorOnNode(
+          node: lvalue_node,
+          withError: "\(check_result.error()!)"))
+    }
+
+    return Result.Ok(
+      (
+        ParserAssignmentStatement(
+          withLValue: lvalue_identifier,
+          withValue: rvalue
+        ), context
+      ))
+  }
+}