fix: Fix lexer bug on file ending with a directive

keyvank · Nov 29, 2024 · 2575b92 · 2575b92
1 parent acba071
commit 2575b92
Showing 1 changed file with 8 additions and 7 deletions.
diff --git a/lexer.c b/lexer.c
@@ -183,17 +183,18 @@ typed_token *next_op(char **inp_ptr, int is_newline)
                 }
                 continue;
             }
-            if (*inp == '\n' || *inp == '\0')
-            {
-                line[sz] = '\0';
-                *inp_ptr = inp;
-                typed_token *dir_tkns = tokenize(line);
-                return new_tkn(TKN_DIRECTIVE, dir_tkns, directive_tkn_debug);
-            }
+
+            if (*inp == '\n')
+                break;
+
             line[sz] = *inp;
             sz++;
             inp++;
         }
+        line[sz] = '\0';
+        *inp_ptr = inp;
+        typed_token *dir_tkns = tokenize(line);
+        return new_tkn(TKN_DIRECTIVE, dir_tkns, directive_tkn_debug);
     }
     if (*inp == '?')
     {