Fix support for multiple "block lambda" arguments.

[cycript.git] / Driver.hpp

diff --git a/Driver.hpp b/Driver.hpp
index a4458a237a4bca49f5dba063df0007af1fe90ca1..fbc96d931a53a672fae24c799f7fa2397992c519 100644 (file)
--- a/Driver.hpp
+++ b/Driver.hpp
@@ -29,31 +29,43 @@
 #include <vector>
 
 #include "Location.hpp"
-#include "Parser.hpp"
+#include "Options.hpp"
+#include "Pooling.hpp"
+#include "Standard.hpp"
+
+struct CYExpression;
+struct CYScript;
+struct CYWord;
+
+enum CYMark {
+    CYMarkIgnore,
+    CYMarkScript,
+    CYMarkModule,
+};
 
 class _visible CYDriver {
   public:
     CYPool &pool_;
     void *scanner_;
 
+    std::vector<char> buffer_;
+    bool tail_;
+
     std::stack<bool> in_;
+    std::stack<bool> return_;
+    std::stack<bool> template_;
+    std::stack<bool> yield_;
 
     bool newline_;
     bool last_;
-
-    struct {
-        bool AtImplementation;
-        bool Class;
-        bool Function;
-        bool OpenBrace;
-        bool NewLine;
-    } no_;
+    bool next_;
 
     std::istream &data_;
+    CYMark mark_;
 
     int debug_;
     bool strict_;
-    bool commented_;
+    bool highlight_;
 
     enum Condition {
         RegExpCondition,
@@ -109,10 +121,9 @@ class _visible CYDriver {
     CYDriver(CYPool &pool, std::istream &data, const std::string &filename = "");
     ~CYDriver();
 
-    bool Parse();
+    bool Parse(CYMark mark = CYMarkScript);
     void Replace(CYOptions &options);
 
-    Condition GetCondition();
     void SetCondition(Condition condition);
 
     void PushCondition(Condition condition);