[main.c] Don't assert if EOF on current_char()...

- [main.c] Add TODO.
- [main.c] Fix bug in consume_string().

diff --git a/main.c b/main.c
index e6d8a25c2f98413528f0f7d357ec2f134e33b134..da504646c33346928d9a895fa372e99937a28e22 100644 (file)
--- a/main.c
+++ b/main.c
@@ -1370,7 +1370,7 @@ int col(Lexer *l) {
 }
 
 char current_char(Lexer *l) {
-    ASSERT(!eof(l), "Trying to get char after EOF");
+    if (eof(l)) return '\0';
     return l->src.data[l->cur];
 }
 
@@ -1429,6 +1429,8 @@ bool consume_string(Lexer *l, String_view *string_sv_out, Location *loc_out, Err
     consume_char(l);
     String_view src_copy = get_src_copy(l);
 
+    // TODO: Here we parse the string until `"` EVEN IF its on multiple lines...
+    // Should we allow this?
     *string_sv_out = sv_lpop_until_char(&src_copy, '"');
 
     loc_out->filename = l->filename;
@@ -1438,7 +1440,7 @@ bool consume_string(Lexer *l, String_view *string_sv_out, Location *loc_out, Err
     // Advance by the len of sv
     l->cur += string_sv_out->count;
 
-    if (eof(l)) {
+    if (current_char(l) != '"') {
         err->loc = *loc_out;
         stbsp_snprintf(err->buf, ERROR_BUF_CAP, "Unterminted String");
         return false;