From c6c396ad67ab2413d068ffb25d96a5b0a0ba17ef Mon Sep 17 00:00:00 2001
From: c+1
Date: Mon, 15 May 2023 12:53:25 -0400
Subject: things

---
 src/lexer.c | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'src/lexer.c')

diff --git a/src/lexer.c b/src/lexer.c
index b61f750..74cc298 100644
--- a/src/lexer.c
+++ b/src/lexer.c
@@ -180,6 +180,20 @@ token_t* lexer_get_def_const(lexer_t* lexer) {
       lexer_next(lexer);
       return token_init(TOKEN_DEFINE_CONST, "=>");
    } else {
+      fputs("Unknown variable state.", stderr);
+      exit(1);
+   }
+}
+
+token_t* lexer_get_def_var(lexer_t* lexer) {
+   lexer_pass(lexer);
+
+   if (lexer_next(lexer), lexer->c == '=') {
+      return lexer_get_def_const(lexer);
+   } else if (lexer_next(lexer), lexer->c == '-') {
+      return lexer_get_def_mut(lexer); 
+   } else {
+      fputs("Unknown variable state.", stderr);
       exit(1);
    }
 }
@@ -191,6 +205,7 @@ token_t* lexer_get_def_mut(lexer_t* lexer) {
       lexer_next(lexer);
       return token_init(TOKEN_DEFINE_MUT, "->");
    } else {
+      fputs("Unknown variable state.", stderr);
       exit(1);
    }
 }
-- 
cgit v1.2.3