Compiler: add support for importing libraries

mrunix00 · Aug 2, 2024 · be2e6a4 · be2e6a4
1 parent 0ba65e5
commit be2e6a4
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/include/ast.h b/include/ast.h
@@ -166,12 +166,14 @@ struct ImportStatement : public AbstractSyntaxTree {
     std::string path;
     explicit ImportStatement(std::string  path);
     bool operator==(const AbstractSyntaxTree &other) const override;
+    void compile(Program &program, Segment &segment) const override;
 };
 
 struct ExportStatement : public AbstractSyntaxTree {
     AbstractSyntaxTree *stm;
     explicit ExportStatement(AbstractSyntaxTree *stm);
     bool operator==(const AbstractSyntaxTree &other) const override;
+    void compile(Program &program, Segment &segment) const override;
 };
 
 std::vector<AbstractSyntaxTree *> parse(const char *input);

diff --git a/src/ast.cpp b/src/ast.cpp
@@ -1,6 +1,8 @@
 #include "ast.h"
 #include "utils.h"
 
+#include <fstream>
+#include <sstream>
 #include <utility>
 
 Node::Node(Token token) : token(std::move(token)) {
@@ -621,17 +623,36 @@ bool ImportStatement::operator==(const AbstractSyntaxTree &other) const {
     auto &otherImportStatement = dynamic_cast<const ImportStatement &>(other);
     return otherImportStatement.path == path;
 }
+void ImportStatement::compile(Program &program, Segment &segment) const {
+   std::ifstream importedFile(path);
+   if (!importedFile.is_open()) {
+       throw std::runtime_error("Unable to open file: " + path);
+   }
+   std::stringstream fileContent;
+   fileContent << importedFile.rdbuf();
+   auto statements = parse(fileContent.str().c_str());
+   for (auto stm : statements) {
+       if(stm->nodeType == AbstractSyntaxTree::Type::ExportStatement)
+           stm->compile(program, segment);
+       delete stm;
+   }
+}
 
 ExportStatement::ExportStatement(AbstractSyntaxTree *stm)
     : stm(stm) {
     nodeType = AbstractSyntaxTree::Type::ExportStatement;
     typeStr = "ExportStatement";
+    assert(stm->nodeType == AbstractSyntaxTree::Type::Declaration,
+           "[ExportStatement]: Only declarations can be exported!");
 }
 bool ExportStatement::operator==(const AbstractSyntaxTree &other) const {
     if (nodeType != other.nodeType) return false;
     auto &otherExportStatement = dynamic_cast<const ExportStatement &>(other);
     return *otherExportStatement.stm == *stm;
 }
+void ExportStatement::compile(Program &program, Segment &segment) const {
+    stm->compile(program, segment);
+}
 
 void compile(Program &program, const char *input) {
     auto ast = parse(input);