Extract stripTestSelectors() function

mainmatter · May 6, 2021 · e5a93fa · e5a93fa
1 parent 9510095
commit e5a93fa
Showing 1 changed file with 15 additions and 11 deletions.
diff --git a/strip-test-selectors.js b/strip-test-selectors.js
@@ -8,6 +8,20 @@ function isTestSelector(attribute) {
   return TEST_SELECTOR_PREFIX.test(attribute);
 }
 
+function stripTestSelectors(node) {
+  if ('sexpr' in node) {
+    node = node.sexpr;
+  }
+
+  node.params = node.params.filter(function(param) {
+    return !isTestSelector(param.original);
+  });
+
+  node.hash.pairs = node.hash.pairs.filter(function(pair) {
+    return !isTestSelector(pair.key);
+  });
+}
+
 function StripTestSelectorsTransform() {
   this.syntax = null;
 }
@@ -21,17 +35,7 @@ StripTestSelectorsTransform.prototype.transform = function(ast) {
         return !isTestSelector(attribute.name);
       });
     } else if (node.type === 'MustacheStatement' || node.type === 'BlockStatement') {
-      if ('sexpr' in node) {
-        node = node.sexpr;
-      }
-
-      node.params = node.params.filter(function(param) {
-        return !isTestSelector(param.original);
-      });
-
-      node.hash.pairs = node.hash.pairs.filter(function(pair) {
-        return !isTestSelector(pair.key);
-      });
+      stripTestSelectors(node);
     }
   });