Eliminate nested block level tracking

See wcjohnson/lightscript#33
wcjohnson · Oct 6, 2017 · cc81f86 · cc81f86
1 parent 87afe91
commit cc81f86
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 13 deletions.
diff --git a/src/parser/statement.js b/src/parser/statement.js
@@ -663,7 +663,6 @@ pp.parseBlockBody = function (node, allowDirectives, topLevel, end) {
 
   const oldInWhiteBlock = this.state.inWhiteBlock;
   const oldWhiteBlockIndentLevel = this.state.whiteBlockIndentLevel;
-  this.state.nestedBlockLevel++;
 
   let isEnd;
   if (this.hasPlugin("lightscript") && typeof end === "number") {
@@ -701,7 +700,6 @@ pp.parseBlockBody = function (node, allowDirectives, topLevel, end) {
     node.body.push(stmt);
   }
 
-  this.state.nestedBlockLevel--;
   if (this.hasPlugin("lightscript")) {
     this.state.inWhiteBlock = oldInWhiteBlock;
     this.state.whiteBlockIndentLevel = oldWhiteBlockIndentLevel;

diff --git a/src/plugins/bangCall.js b/src/plugins/bangCall.js
@@ -40,13 +40,15 @@ export default function(parser) {
       this.unexpected(null, "Whitespace required between `!` and first argument.");
     }
 
-    // Read args
-    let first = true;
+    // Collect state
     const oldBangUnwindLevel = this.state.bangUnwindLevel;
-    const oldBangBlockLevel = this.state.bangBlockLevel;
-    this.state.bangBlockLevel = this.state.nestedBlockLevel;
     this.state.bangUnwindLevel = bangIndentLevel + 1;
 
+    const oldBangWhiteBlockLevel = this.state.bangWhiteBlockLevel;
+    this.state.bangWhiteBlockLevel = this.state.whiteBlockIndentLevel;
+
+    // Read args
+    let first = true;
     while (true) {
       // First argument on a different line from the `!` establishes indent level
       if (this.state.curLine !== bangLine && argIndentLevel === null) {
@@ -83,7 +85,7 @@ export default function(parser) {
     }
 
     this.state.bangUnwindLevel = oldBangUnwindLevel;
-    this.state.bangBlockLevel = oldBangBlockLevel;
+    this.state.bangWhiteBlockLevel = oldBangWhiteBlockLevel;
 
     node = this.finishNode(node, nodeType);
 
@@ -97,7 +99,7 @@ export default function(parser) {
   // Subscripts to a bang call must appear at the arg indent level
   pp.shouldUnwindBangSubscript = function() {
     return this.isLineBreak() &&
-      (this.state.bangBlockLevel == this.state.nestedBlockLevel) &&
+      (this.state.bangWhiteBlockLevel === this.state.whiteBlockIndentLevel) &&
       (this.state.indentLevel <= this.state.bangUnwindLevel);
   };
 }
diff --git a/src/plugins/lightscript.js b/src/plugins/lightscript.js
@@ -140,9 +140,7 @@ pp.parseObjectComprehension = function(node) {
 pp.parseInlineWhiteBlock = function(node) {
   if (this.state.type.startsExpr) return this.parseMaybeAssign();
   // oneline statement case
-  this.state.nestedBlockLevel++;
   node.body = [this.parseStatement(true)];
-  this.state.nestedBlockLevel--;
   node.directives = [];
   this.addExtra(node, "curly", false);
   return this.finishNode(node, "BlockStatement");
@@ -167,9 +165,7 @@ pp.parseWhiteBlock = function (isExpression?) {
       return this.parseInlineWhiteBlock(node);
     }
 
-    this.state.nestedBlockLevel++;
     const stmt = this.parseStatement(false);
-    this.state.nestedBlockLevel--;
     return stmt;
   }
 

diff --git a/src/tokenizer/state.js b/src/tokenizer/state.js
@@ -40,7 +40,6 @@ export default class State {
 
     // for lightscript
     this.indentLevel = 0;
-    this.nestedBlockLevel = 0;
     this.inMatchCaseTest = false;
 
     this.type = tt.eof;