feat: support extending processor

whxaxes · Jul 4, 2017 · 5feda67 · 5feda67
1 parent 8d41e07
commit 5feda67
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 2 deletions.
diff --git a/lib/compile/ast.js b/lib/compile/ast.js
@@ -16,9 +16,15 @@ class Ast {
     this.blockEnd = options.blockEnd || '%}';
     this.variableStart = options.variableStart || '{{';
     this.variableEnd = options.variableEnd || '}}';
+    this.processor = processor;
     this.commentStart = '{#';
     this.commentEnd = '#}';
 
+    // support extending processor
+    if (options.processor) {
+      this.processor = Object.assign({}, processor, options.processor);
+    }
+
     if (this.blockStart === this.variableStart) {
       throw new Error('blockStart should be different with variableStart!');
     }
@@ -100,7 +106,8 @@ class Ast {
           const matches = expression.match(tagNameRE);
           const tagName = matches[0];
           const isCustom = mus && mus.customTags.has(tagName);
-          const tagHandle = processor[tagName] || (isCustom ? processor.custom : null);
+          const tagHandle = this.processor[tagName]
+            || (isCustom ? this.processor.custom : null);
 
           if (tagHandle) {
             // create ast node
@@ -134,7 +141,7 @@ class Ast {
           collectChars(collector, leftStart + expression + rightEnd);
         } else {
           element = this.genNode(3);
-          processor.variable(element, expression);
+          this.processor.variable(element, expression);
         }
       }
 

diff --git a/test/ast.test.js b/test/ast.test.js
@@ -18,4 +18,15 @@ describe('lib#compile#ast', () => {
     const m = a.genMacro();
     assert(m === a.genMacro());
   });
+
+  it('ast should support extending processor', done => {
+    const str = '{{ test }}';
+    ast(str, {
+      processor: {
+        variable() {
+          done();
+        }
+      }
+    });
+  });
 });