From 48e7488a6363b50c71681b3acc3d00dc9d830a63 Mon Sep 17 00:00:00 2001
From: SpookyDervish <78246495+SpookyDervish@users.noreply.github.com>
Date: Tue, 14 Oct 2025 20:02:22 +1100
Subject: [PATCH] fixed reassignment, starting work on if statements

---
 compiler.py            | 16 +++++++++++++++-
 debug/ir.ll            | 15 ++++++---------
 lexer_token.py         | 17 ++++++++++++++++-
 main.py                |  2 +-
 tests/conditionals.pla |  2 +-
 tests/test.pla         |  8 ++++++--
 6 files changed, 45 insertions(+), 15 deletions(-)

diff --git a/compiler.py b/compiler.py
index 7e391a8..cdfb5e5 100644
--- a/compiler.py
+++ b/compiler.py
@@ -1,7 +1,7 @@
 from llvmlite import ir
 
 from AST import Node, NodeType, Program, Expression
-from AST import ExpressionStatement, AssignmentStatement, BlockStatement, ReturnStatement, FunctionStatement
+from AST import ExpressionStatement, AssignmentStatement, BlockStatement, ReturnStatement, FunctionStatement, ReassignStatement
 from AST import InfixExpression
 from AST import IntegerLiteral, FloatLiteral, IdentifierLiteral
 
@@ -23,6 +23,7 @@ class Compiler:
 		self.module: ir.Module = ir.Module("main")
 		self.builder: ir.IRBuilder = ir.IRBuilder()
 		self.environment: Environment = Environment()
+		self.errors: list[str] = []
 
 	def compile(self, node: Node) -> None:
 		match node.type():
@@ -40,6 +41,8 @@ class Compiler:
 				self.__visit_block_statement(node)
 			case NodeType.ReturnStatement:
 				self.__visit_return_statement(node)
+			case NodeType.ReassignStatement:
+				self.__visit_reassign_statement(node)
 
 			# Expressions
 			case NodeType.InfixExpression:
@@ -115,6 +118,17 @@ class Compiler:
 
 		self.builder = previous_builder
 
+	def __visit_reassign_statement(self, node: ReassignStatement) -> None:
+		name: str = node.ident.value
+		value: Expression = node.right_value
+
+		value, Type = self.__resolve_value(value)
+
+		if self.environment.lookup(name) is None:
+			self.errors.append(f"Identifier {name} has not been declared before it was re-assigned.")
+		else:
+			ptr, _ = self.environment.lookup(name)
+			self.builder.store(value, ptr)
 	# endregion	
 
 	# region Expressions
diff --git a/debug/ir.ll b/debug/ir.ll
index 90aff47..b304946 100644
--- a/debug/ir.ll
+++ b/debug/ir.ll
@@ -2,17 +2,14 @@
 target triple = "x86_64-pc-windows-msvc"
 target datalayout = ""
 
-define i32 @"main"()
-{
-main_entry:
-  %".2" = alloca i32
-  store i32 0, i32* %".2"
-  %".4" = load i32, i32* %".2"
-  ret i32 %".4"
-}
-
 define i32 @"test"()
 {
 test_entry:
+  ret i32 0
+}
+
+define i32 @"main"()
+{
+main_entry:
   ret i32 123
 }
diff --git a/lexer_token.py b/lexer_token.py
index a8d2fd7..e15d6a5 100644
--- a/lexer_token.py
+++ b/lexer_token.py
@@ -23,6 +23,13 @@ class TokenType(Enum):
 	# Assignment symbols
 	EQ = "EQ"
 
+	# Comparison symbols
+	LT = "<"
+	GT = ">"
+	EQ_EQ = "=="
+	LT_EQ = "<="
+	GT_EQ = ">="
+
 	# Symbols
 	LPAREN = "LPAREN"
 	RPAREN = "RPAREN"
@@ -35,6 +42,10 @@ class TokenType(Enum):
 
 	# Keywords
 	RETURN = "RETURN"
+	IF = "IF"
+	UNLESS = "UNLESS"
+	TRUE = "TRUE"
+	FALSE = "FALSE"
 
 	# Typing
 	TYPE = "TYPE"
@@ -54,7 +65,11 @@ class Token:
 	
 
 KEYWORDS: dict[str, TokenType] = {
-	"return": TokenType.RETURN
+	"return": TokenType.RETURN,
+	"if": TokenType.IF,
+	"unless": TokenType.UNLESS,
+	"true": TokenType.TRUE,
+	"false": TokenType.FALSE
 }
 
 ALT_KEYWORDS: dict[str, TokenType] = {
diff --git a/main.py b/main.py
index 4d9a900..ad60a43 100644
--- a/main.py
+++ b/main.py
@@ -12,7 +12,7 @@ from ctypes import CFUNCTYPE, c_int, c_float
 LEXER_DEBUG: bool = False
 PARSER_DEBUG: bool = False
 COMPILER_DEBUG: bool = True
-RUN_CODE: bool = False
+RUN_CODE: bool = True
 
 
 if __name__ == "__main__":
diff --git a/tests/conditionals.pla b/tests/conditionals.pla
index 1dc8d9a..b894b5a 100644
--- a/tests/conditionals.pla
+++ b/tests/conditionals.pla
@@ -1,6 +1,6 @@
 depend "io.pla"
 
-if (1 + 2 == 3) {
+if 1 + 2 == 3 {
 	print("The universe is functional!");
 }
 unless
diff --git a/tests/test.pla b/tests/test.pla
index 71f2359..33bcbb7 100644
--- a/tests/test.pla
+++ b/tests/test.pla
@@ -1,7 +1,11 @@
 main = Func(): Int {
-	x: Int = 0;
+	x: Int = 123;
 
-	x = x * 2;
+	if x == 123 {
+		x = 0;
+	} unless {
+		x = 1;
+	}
 
 	return x;
 }
\ No newline at end of file