3 files changed, 78 insertions, 2 deletions

diff --git a/src/parser/ast.c b/src/parser/ast.c
index 97cdb6d..9bdf4d1 100644
--- a/src/parser/ast.c
+++ b/src/parser/ast.c
@@ -25,8 +25,20 @@ void free_expr(struct Expr* expr)
 
 static void free_statement(struct Statement* statement)
 {
-	if (statement->kind == STATEMENT_ASSIGN)
-		free_expr(statement->assign.value);
+	switch (statement->kind)
+	{
+		case STATEMENT_ASSIGN:
+			free_expr(statement->assign.value);
+			break;
+		case STATEMENT_IF:
+			free_expr(statement->branch.left);
+			free_expr(statement->branch.right);
+			free_statement(statement->branch.body);
+			free(statement->branch.body);
+			break;
+		default:
+			break;
+	}
 }
 
 void free_proc(struct ProcDecl* proc)
diff --git a/src/parser/ast.h b/src/parser/ast.h
index 199d802..e85e72f 100644
--- a/src/parser/ast.h
+++ b/src/parser/ast.h
@@ -65,6 +65,7 @@ enum StatementKind
 	STATEMENT_LABEL,
 	STATEMENT_GOTO,
 	STATEMENT_SYSCALL,
+	STATEMENT_IF,
 };
 
 struct AssignStatement
@@ -85,6 +86,14 @@ struct GotoStatement
 	struct Token label;
 };
 
+struct IfStatement
+{
+	struct Expr* left;
+	struct Token comparison;
+	struct Expr* right;
+	struct Statement* body;
+};
+
 struct Statement
 {
 	enum StatementKind kind;
@@ -93,6 +102,7 @@ struct Statement
 		struct AssignStatement assign;
 		struct LabelStatement label;
 		struct GotoStatement jump;
+		struct IfStatement branch;
 	};
 };
 
diff --git a/src/parser/parser.c b/src/parser/parser.c
index 5431bfd..2ae52f2 100644
--- a/src/parser/parser.c
+++ b/src/parser/parser.c
@@ -206,8 +206,62 @@ static struct Expr* parse_expression(struct Parser* parser)
 	return parse_binary(parser, parse_multiplicative, TOKEN_PLUS, TOKEN_MINUS);
 }
 
+static bool is_compare_op(enum TokenType type)
+{
+	return type == TOKEN_EQUAL_EQUAL
+		|| type == TOKEN_BANG_EQUAL
+		|| type == TOKEN_LESS
+		|| type == TOKEN_LESS_EQUAL
+		|| type == TOKEN_GREATER
+		|| type == TOKEN_GREATER_EQUAL;
+}
+
+static bool parse_statement(struct Parser* parser, struct Statement* out);
+
+static bool parse_if(struct Parser* parser, struct Statement* out)
+{
+	struct Expr* left = parse_expression(parser);
+	if (left == NULL)
+		return false;
+
+	if (!is_compare_op(parser->current.type))
+	{
+		error_at(parser, parser->current, "expected a comparison operator");
+		free_expr(left);
+		return false;
+	}
+	advance_parser(parser);
+	struct Token comparison = parser->previous;
+
+	struct Expr* right = parse_expression(parser);
+	if (right == NULL)
+	{
+		free_expr(left);
+		return false;
+	}
+
+	struct Statement* body = malloc(sizeof(struct Statement));
+	if (!parse_statement(parser, body))
+	{
+		free(body);
+		free_expr(left);
+		free_expr(right);
+		return false;
+	}
+
+	out->kind = STATEMENT_IF;
+	out->branch.left = left;
+	out->branch.comparison = comparison;
+	out->branch.right = right;
+	out->branch.body = body;
+	return true;
+}
+
 static bool parse_statement(struct Parser* parser, struct Statement* out)
 {
+	if (match_token(parser, TOKEN_IF))
+		return parse_if(parser, out);
+
 	if (match_token(parser, TOKEN_SYSCALL))
 	{
 		out->kind = STATEMENT_SYSCALL;