Adding 'each .. of' syntax

packages/pug-code-gen/index.js

@@ -767,6 +767,52 @@ Compiler.prototype = {
     this.buf.push('  }\n}).call(this);\n');
   },
 
+  visitEachOf: function(each){
+    var indexVarName = each.key || 'pug_index' + this.eachCount;
+    this.eachCount++;
+
+    this.buf.push(''
+      + '// iterate ' + each.obj + '\n'
+      + ';(function(){\n'
+      + '  var $$obj = ' + each.obj + ';\n'
+      + '  if (\'number\' == typeof $$obj.length) {');
+
+    if (each.alternate) {
+      this.buf.push('    if ($$obj.length) {');
+    }
+
+    this.buf.push(''
+      + '      for (var ' + indexVarName + ' = 0, $$l = $$obj.length; ' + indexVarName + ' < $$l; ' + indexVarName + '++) {\n'
+      + '        var ' + each.val + ' = $$obj[' + indexVarName + '];');
+
+    this.visit(each.block, each);
+
+    this.buf.push('      }');
+
+    if (each.alternate) {
+      this.buf.push('    } else {');
+      this.visit(each.alternate, each);
+      this.buf.push('    }');
+    }
+
+    this.buf.push(''
+      + '  } else {\n'
+      + '    var $$l = 0;\n'
+      + '    for (var ' + indexVarName + ' of $$obj) {\n'
+      + '      $$l++;\n'
+      + '      var ' + each.val + ' = ' + indexVarName + ';');
+
+    this.visit(each.block, each);
+
+    this.buf.push('    }');
+    if (each.alternate) {
+      this.buf.push('    if ($$l === 0) {');
+      this.visit(each.alternate, each);
+      this.buf.push('    }');
+    }
+    this.buf.push('  }\n}).call(this);\n');
+  },
+
   /**
    * Visit `attrs`.
    *

packages/pug-lexer/index.js

@@ -971,6 +971,31 @@ Lexer.prototype = {
     }
   },
 
+  /**
+   * EachOf.
+   */
+
+  eachOf: function() {
+    var captures;
+    if (captures = /^(?:each|for) +([a-zA-Z_$][\w$]*)(?: *, *([a-zA-Z_$][\w$]*))? * of *([^\n]+)/.exec(this.input)) {
+      this.consume(captures[0].length);
+      var tok = this.tok('eachOf', captures[1]);
+      tok.key = captures[2] || null;
+      this.incrementColumn(captures[0].length - captures[3].length);
+      this.assertExpression(captures[3])
+      tok.code = captures[3];
+      this.incrementColumn(captures[3].length);
+      this.tokens.push(this.tokEnd(tok));
+      return true;
+    }
+    if (captures = /^- *(?:each|for) +([a-zA-Z_$][\w$]*)(?: *, *([a-zA-Z_$][\w$]*))? +of +([^\n]+)/.exec(this.input)) {
+      this.error(
+        'MALFORMED_EACH',
+        'Pug each and for should no longer be prefixed with a dash ("-"). They are pug keywords and not part of JavaScript.'
+      );
+    }
+  },
+
   /**
    * Code.
    */
@@ -1485,6 +1510,7 @@ Lexer.prototype = {
       || this.callLexerFunction('mixin')
       || this.callLexerFunction('call')
       || this.callLexerFunction('conditional')
+      || this.callLexerFunction('eachOf')
       || this.callLexerFunction('each')
       || this.callLexerFunction('while')
       || this.callLexerFunction('tag')

packages/pug-lexer/test/check-lexer-functions.test.js

@@ -23,6 +23,7 @@ var lexerFunctions = {
   doctype: true,
   dot: true,
   each: true,
+  eachOf: true,
   eos: true,
   endInterpolation: true,
   extends: true,

packages/pug-parser/index.js

@@ -234,6 +234,8 @@ Parser.prototype = {
         return this.parseDot();
       case 'each':
         return this.parseEach();
+      case 'eachOf':
+        return this.parseEachOf();
       case 'code':
         return this.parseCode();
       case 'blockcode':
@@ -761,6 +763,24 @@ loop:
     return node;
   },
 
+  parseEachOf: function(){
+    var tok = this.expect('eachOf');
+    var node = {
+      type: 'EachOf',
+      obj: tok.code,
+      val: tok.val,
+      key: tok.key,
+      block: this.block(),
+      line: tok.loc.start.line,
+      column: tok.loc.start.column,
+      filename: this.filename
+    };
+    if (this.peek().type == 'else') {
+      this.advance();
+      node.alternate = this.block();
+    }
+    return node;
+  },
   /**
    * 'extends' name
    */

packages/pug-walk/index.js

@@ -56,6 +56,14 @@ function walkAST(ast, before, after, options) {
         ast.alternate = walkAST(ast.alternate, before, after, options);
       }
       break;
+    case 'EachOf':
+      if (ast.block) {
+        ast.block = walkAST(ast.block, before, after, options);
+      }
+      if (ast.alternate) {
+        ast.alternate = walkAST(ast.alternate, before, after, options);
+      }
+      break;
     case 'Conditional':
       if (ast.consequent) {
         ast.consequent = walkAST(ast.consequent, before, after, options);