diff --git a/lib/ast.js b/lib/ast.js index 8faadae..adc1630 100644 --- a/lib/ast.js +++ b/lib/ast.js @@ -54,6 +54,7 @@ switch (node.type) { case 'DoWhileStatement': case 'ForInStatement': + case 'ForOfStatement': case 'ForStatement': case 'WhileStatement': return true; @@ -72,6 +73,7 @@ case 'EmptyStatement': case 'ExpressionStatement': case 'ForInStatement': + case 'ForOfStatement': case 'ForStatement': case 'IfStatement': case 'LabeledStatement': @@ -102,6 +104,7 @@ case 'LabeledStatement': case 'ForStatement': case 'ForInStatement': + case 'ForOfStatement': case 'WhileStatement': case 'WithStatement': return node.body; diff --git a/lib/keyword.js b/lib/keyword.js index 13c8c6a..d9ca798 100644 --- a/lib/keyword.js +++ b/lib/keyword.js @@ -58,7 +58,7 @@ switch (id.length) { case 2: - return (id === 'if') || (id === 'in') || (id === 'do'); + return (id === 'if') || (id === 'in') || (id === 'of') || (id === 'do'); case 3: return (id === 'var') || (id === 'for') || (id === 'new') || (id === 'try'); case 4: