Convert class based template compilation plugin to functional style

mainmatter · May 6, 2021 · 84640e4 · 84640e4
1 parent fc27e64
commit 84640e4
Showing 1 changed file with 21 additions and 25 deletions.
diff --git a/strip-test-selectors.js b/strip-test-selectors.js
@@ -18,30 +18,26 @@ function stripTestSelectors(node) {
   });
 }
 
-function StripTestSelectorsTransform() {
-  this.syntax = null;
+function transform() {
+  return {
+    name: 'strip-test-selectors',
+
+    visitor: {
+      ElementNode(node) {
+        node.attributes = node.attributes.filter(function(attribute) {
+          return !isTestSelector(attribute.name);
+        });
+      },
+
+      MustacheStatement(node) {
+        stripTestSelectors(node);
+      },
+
+      BlockStatement(node) {
+        stripTestSelectors(node);
+      },
+    }
+  };
 }
 
-StripTestSelectorsTransform.prototype.transform = function(ast) {
-  let { traverse } = this.syntax;
-
-  traverse(ast, {
-    ElementNode(node) {
-      node.attributes = node.attributes.filter(function(attribute) {
-        return !isTestSelector(attribute.name);
-      });
-    },
-
-    MustacheStatement(node) {
-      stripTestSelectors(node);
-    },
-
-    BlockStatement(node) {
-      stripTestSelectors(node);
-    },
-  });
-
-  return ast;
-};
-
-module.exports = StripTestSelectorsTransform;
+module.exports = transform;