From 8f55e2fa7059ef215a1b8369300f0a85103b079b Mon Sep 17 00:00:00 2001
From: Nic Gaffney <gaffney_nic@protonmail.com>
Date: Thu, 18 Jul 2024 19:26:06 -0500
Subject: ASt progress

---
 src/main.zig | 12 +++---------
 1 file changed, 3 insertions(+), 9 deletions(-)

(limited to 'src/main.zig')

diff --git a/src/main.zig b/src/main.zig
index 4a0bbca..2ca0b68 100644
--- a/src/main.zig
+++ b/src/main.zig
@@ -45,21 +45,15 @@ pub fn main() !void {
         switch (t) {
             .ret => {
                 const num = tokIter.next();
-                switch (num.?) {
-                    .intLit => {},
-                    else => break,
-                }
-                switch (tokIter.next().?) {
-                    .semiCol => {},
-                    else => break,
-                }
+                if (!tok.checkType(num.?, tok.TokenType.intLit)) return error.SyntaxError;
+
+                if (!tok.checkType(tokIter.next().?, tok.TokenType.semiCol)) return error.SyntaxError;
                 try outWriter.print(
                     \\  mov rax, 60
                     \\  mov rdi, {}
                     \\  syscall
                     \\
                 , .{num.?.intLit});
-                gpa.allocator().free(t.ret);
             },
             // No other commands
             else => {},
-- 
cgit v1.2.3