Remove closures used to simulate block scope in compiled output

pugjs · Sep 24, 2010 · 31d25f1 · 31d25f1
1 parent 483f8d4
commit 31d25f1
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/lib/compiler.js b/lib/compiler.js
@@ -223,17 +223,30 @@ Compiler.prototype = {
      */
 
     visitCode: function(code){
+        var end;
+
+        if (code.block) {
+            // Optimize for if / else
+            if (0 == code.val.trim().indexOf('if') ||
+                0 == code.val.trim().indexOf('else')) {
+                end = true;
+                code.val += ' {';
+            }
+        }
+
+        // Buffer code
         if (code.buffer) {
             var val = code.val.trimLeft();
             if (code.escape) val = 'escape(' + val + ')';
             this.buf.push("buf.push(" + val + ");");
         } else {
             this.buf.push(code.val);
         }
+
+        // Block support
         if (code.block) {
-            this.buf.push('  (function(){');
             this.visit(code.block);
-            this.buf.push('  }).call(this);');
+            if (end) this.buf.push('}');
         }
     },