Transform ECMAScript 6 import as CommonJS require.

[cycript.git] / Driver.hpp

diff --git a/Driver.hpp b/Driver.hpp
index 2a0955ac60325b6839440006046575dddb9ed49b..adb03f20f612c9e864e3a677fbb9d78fed1ac4c2 100644 (file)
--- a/Driver.hpp
+++ b/Driver.hpp
@@ -29,7 +29,19 @@
 #include <vector>
 
 #include "Location.hpp"
-#include "Parser.hpp"
+#include "Options.hpp"
+#include "Pooling.hpp"
+#include "Standard.hpp"
+
+struct CYClassTail;
+struct CYExpression;
+struct CYScript;
+struct CYWord;
+
+enum CYMark {
+    CYMarkScript,
+    CYMarkModule,
+};
 
 class _visible CYDriver {
   public:
@@ -40,20 +52,25 @@ class _visible CYDriver {
     bool tail_;
 
     std::stack<bool> in_;
+    std::stack<bool> return_;
+    std::stack<bool> super_;
     std::stack<bool> template_;
+    std::stack<bool> yield_;
+
+    std::stack<CYClassTail *> class_;
 
+    CYMark mark_;
+    int hold_;
     bool newline_;
     bool last_;
-    bool next_;
 
-    std::istream &data_;
+    std::streambuf &data_;
 
     int debug_;
     bool strict_;
     bool highlight_;
 
     enum Condition {
-        RegExpCondition,
         XMLContentCondition,
         XMLTagCondition,
     };
@@ -103,12 +120,13 @@ class _visible CYDriver {
     void ScannerDestroy();
 
   public:
-    CYDriver(CYPool &pool, std::istream &data, const std::string &filename = "");
+    CYDriver(CYPool &pool, std::streambuf &data, const std::string &filename = "");
     ~CYDriver();
 
-    bool Parse();
+    bool Parse(CYMark mark = CYMarkModule);
     void Replace(CYOptions &options);
 
+    void SetRegEx(bool equal);
     void SetCondition(Condition condition);
 
     void PushCondition(Condition condition);