Update statement node ASTs

1 files changed, 13 insertions, 14 deletions

diff --git a/lib/handlebars/compiler/visitor.js b/lib/handlebars/compiler/visitor.js
index dc7f527..e48c9bc 100644
--- a/lib/handlebars/compiler/visitor.js
+++ b/lib/handlebars/compiler/visitor.js
@@ -4,7 +4,7 @@ Visitor.prototype = {
   constructor: Visitor,
 
   accept: function(object) {
-    return object && this[object.type] && this[object.type](object);
+    return object && this[object.type](object);
   },
 
   Program: function(program) {
@@ -20,12 +20,21 @@ Visitor.prototype = {
     this.accept(mustache.sexpr);
   },
 
-  block: function(block) {
-    this.accept(block.mustache);
+  BlockStatement: function(block) {
+    this.accept(block.sexpr);
     this.accept(block.program);
     this.accept(block.inverse);
   },
 
+  PartialStatement: function(partial) {
+    this.accept(partial.partialName);
+    this.accept(partial.context);
+    this.accept(partial.hash);
+  },
+
+  ContentStatement: function(content) {},
+  CommentStatement: function(comment) {},
+
   sexpr: function(sexpr) {
     var params = sexpr.params, paramStrings = [], hash;
 
@@ -44,13 +53,6 @@ Visitor.prototype = {
     }
   },
 
-  partial: function(partial) {
-    this.accept(partial.partialName);
-    this.accept(partial.context);
-    this.accept(partial.hash);
-  },
-  PARTIAL_NAME: function(partialName) {},
-
   DATA: function(data) {
     this.accept(data.id);
   },
@@ -58,10 +60,7 @@ Visitor.prototype = {
   STRING: function(string) {},
   NUMBER: function(number) {},
   BOOLEAN: function(bool) {},
-  ID: function(id) {},
-
-  content: function(content) {},
-  comment: function(comment) {}
+  ID: function(id) {}
 };
 
 export default Visitor;