bugfix & test

1 files changed, 7 insertions, 2 deletions

diff --git a/Driver.cc b/Driver.cc
index 6d557db..a3db940 100644
--- a/Driver.cc
+++ b/Driver.cc
@@ -7,13 +7,18 @@ Driver::Driver(const char* name)
 
 Driver::~Driver() {}
 
-void Driver::Parse(std::string& filename) {
+std::shared_ptr<ProgramNode> Driver::Parse(std::string& filename) {
     location_.initialize(&filename);
     if (StartLexer(filename) != 0) {
-        return;
+        return nullptr;
     }
     Parser parser(*this, &context_);
     parser.parse();
     StopLexer();
+    return root_;
 }
+
+void Driver::SetRoot(std::shared_ptr<ProgramNode> root) { root_ = root; }
+
+Context* Driver::GetContext() { return &context_; }
 } // namespace pcc