Do not assume lookahead (wish I could force it).

[cycript.git] / Cycript.yy.in

diff --git a/Cycript.yy.in b/Cycript.yy.in
index 1345060c783790e236b99787affc9410cf1cd119..465b32b50f9e198445483dbc8339d1cb532bf51d 100644 (file)
--- a/Cycript.yy.in
+++ b/Cycript.yy.in
@@ -536,11 +536,11 @@ LexNoBrace
     ;
 
 LexNoFunction
-    : { driver.no_.Function = true; }
+    : { if (yychar == yyempty_) driver.no_.Function = true; else if (yychar == token::Function) yychar = token::Function_; }
     ;
 
 LexNoAtImplementation
-    : { driver.no_.AtImplementation = true; }
+    : { if (yychar == yyempty_) driver.no_.AtImplementation = true; else if (yychar == token::AtImplementation) yychar = token::AtImplementation_; }
     ;
 
 LexSetStatement