1 files changed, 137 insertions, 0 deletions
diff --git a/tiger-compiler/src/parse/libparse.cc b/tiger-compiler/src/parse/libparse.cc
new file mode 100644
index 0000000..7c963b6
--- /dev/null
+++ b/tiger-compiler/src/parse/libparse.cc
@@ -0,0 +1,137 @@
+/**
+ ** \file parse/libparse.cc
+ ** \brief Functions and variables exported by the parse module.
+ */
+
+#include <ast/chunk-interface.hh>
+#include <ast/chunk-list.hh>
+#include <misc/file-library.hh>
+#include <misc/indent.hh>
+#include <misc/symbol.hh>
+#include <parse/libparse.hh>
+#include <parse/location.hh>
+#include <parse/tasks.hh>
+#include <parse/tiger-driver.hh>
+#include <parse/tweast.hh>
+
+// Define exported parse functions.
+namespace parse
+{
+  ast::ChunkList* parse_prelude()
+  {
+    if (tasks::prelude.empty())
+      {
+        return nullptr;
+      }
+
+    TigerDriver td;
+
+    return std::get<ast::ChunkList*>(td.parse(td.prelude()));
+  }
+
+  // Parse a Tiger file, return the corresponding abstract syntax.
+  std::pair<ast::ChunkList*, misc::error> parse(const std::string& prelude,
+                                                const std::string& fname,
+                                                misc::file_library& library,
+                                                bool scan_trace_p,
+                                                bool parse_trace_p,
+                                                bool enable_object_extensions_p,
+                                                bool enable_assert_extensions_p)
+  {
+    // Current directory must be that of the file currently processed.
+    library.push_current_directory(misc::path(fname).parent_path());
+
+    TigerDriver td(library);
+    td.scan_trace(scan_trace_p).parse_trace(parse_trace_p);
+    td.enable_object_extensions(enable_object_extensions_p);
+    td.enable_assert_extensions(enable_assert_extensions_p);
+
+    ast::ChunkList* res = nullptr;
+
+    ast_type tree = td.parse_file(fname);
+
+    ast::Exp** exp = std::get_if<ast::Exp*>(&tree);
+    ast::ChunkList** chunks = std::get_if<ast::ChunkList*>(&tree);
+
+    // Try to parse the program as an expression, and check that the
+    // parsing did not fail in that case.
+    if (exp && *exp)
+      {
+        Tweast in;
+
+        if (!prelude.empty())
+          {
+            ast::ChunkList* prelude_chunks =
+              (prelude == "builtin"
+                 ? std::get<ast::ChunkList*>(td.parse(td.prelude()))
+                 : td.parse_import(prelude, location()));
+            if (prelude_chunks)
+              in << prelude_chunks;
+          }
+        in << "function _main() = (" << *exp << "; ())";
+        res = td.parse(in);
+      }
+    // Try to parse the program as a list of declarations, and check
+    // that the parsing did not fail in that case.
+    else if (chunks && *chunks)
+      {
+        // Try to parse the program as a list of declarations.
+        res = *chunks;
+      }
+    // Otherwise, the parsing failed, and an error will be returned as
+    // second member of the return value.
+
+    // Ensure that directory stack is not modified by parse.
+    library.pop_current_directory();
+
+    return std::pair(res, td.error_get());
+  }
+
+  ast_type parse(Tweast& input)
+  {
+    TigerDriver td;
+    td.enable_extensions();
+    td.enable_object_extensions();
+    ast_type res = td.parse(input);
+    if (td.error_get())
+      {
+        misc::error e;
+        e << "Failed to resolve Tweast:" << misc::incendl << input;
+        e << "Got: " << td.error_get();
+        e.ice_here();
+      }
+    return res;
+  }
+
+  ast::Exp* parse(const std::string& str, bool enable_object_extensions_p)
+  {
+    TigerDriver td;
+    td.enable_object_extensions(enable_object_extensions_p);
+    ast::Exp* res = td.parse(str);
+    td.error_get().ice_on_error_here();
+    return res;
+  }
+
+  ast::ChunkList* parse_unit(const std::string& str,
+                             bool enable_object_extensions_p)
+  {
+    TigerDriver td;
+    td.enable_object_extensions(enable_object_extensions_p);
+    std::string rewrite = "function _main() = (" + str + "; ())";
+    ast::ChunkList* res = td.parse(rewrite);
+    td.error_get().ice_on_error_here();
+    return res;
+  }
+
+  // Parse a set of declarations.
+  ast::ChunkInterface* parse_chunks(Tweast& in)
+  {
+    ast::ChunkList* dl = parse(in);
+    assertion(dl->chunks_get().size() == 1);
+    ast::ChunkInterface* res = dl->chunks_get().front();
+    dl->chunks_get().pop_front();
+    delete dl;
+    return res;
+  }
+
+} // namespace parse