fix: Fixed import not being parsed correctly

alinalihassan · Mar 12, 2022 · d515f7b · d515f7b
1 parent 5d3f973
commit d515f7b
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/src/Backend/Codegen.h b/src/Backend/Codegen.h
@@ -3,7 +3,7 @@
 #include "AST/ExprVisitor.h"
 #include "AST/StmtVisitor.h"
 #include "Frontend/Parser.h"
-#include "LesmaJIT.h"
+#include "JIT/LesmaJIT.h"
 #include "Symbol/SymbolTable.h"
 #include <clang/Driver/Driver.h>
 #include <filesystem>

diff --git a/src/Frontend/Parser.cpp b/src/Frontend/Parser.cpp
@@ -425,9 +425,11 @@ Statement *Parser::ParseImport() {
 
     if (AdvanceIfMatchAny<TokenType::AS>()) {
         auto alias = Consume(TokenType::IDENTIFIER);
+        ConsumeNewline();
         return new Import({loc.Start, alias.getEnd()}, token->lexeme, alias->lexeme);
     }
 
+    ConsumeNewline();
     return new Import({loc.Start, token.getEnd()}, filepath, getBasename(token->lexeme));
 }
 

diff --git a/src/LesmaJIT.h → src/JIT/LesmaJIT.h b/src/LesmaJIT.h → src/JIT/LesmaJIT.h