Refactor blocks, programs and inverses

1 files changed, 9 insertions, 2 deletions

diff --git a/lib/handlebars/compiler/base.js b/lib/handlebars/compiler/base.js
index 722f09a..7ab1843 100644
--- a/lib/handlebars/compiler/base.js
+++ b/lib/handlebars/compiler/base.js
@@ -1,12 +1,19 @@
 import parser from "./parser";
 import AST from "./ast";
+import { stripFlags, prepareBlock } from "./helpers";
 
 export { parser };
 
 export function parse(input) {
   // Just return if an already-compile AST was passed in.
-  if(input.constructor === AST.ProgramNode) { return input; }
+  if (input.constructor === AST.ProgramNode) { return input; }
+
+  for (var key in AST) {
+    parser.yy[key] = AST[key];
+  }
+
+  parser.yy.stripFlags = stripFlags;
+  parser.yy.prepareBlock = prepareBlock;
 
-  parser.yy = AST;
   return parser.parse(input);
 }