refactor: switch to OnceExit() API.

postcss · Dec 30, 2020 · c75eee9 · c75eee9
1 parent 5b9aa8a
commit c75eee9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 42 deletions.
diff --git a/src/__tests__/integration.js b/src/__tests__/integration.js
diff --git a/src/index.js b/src/index.js
@@ -8,20 +8,25 @@ function pluginCreator(opts) {
     mediaQueries: false,
     selectors: false
   }, opts);
+
   return {
     postcssPlugin: 'postcss-calc',
-    Declaration(node, {result}) {
-      transform(node, "value", options, result);
-    },
-    AtRule(node, {result}) {
-      if (options.mediaQueries) {
-        transform(node, "params", options, result);
-      }
-    },
-    Rule(node, {result}) {
-      if (options.selectors) {
-        transform(node, "selector", options, result);
-      }
+    OnceExit(css, { result }) {
+      css.walk(node => {
+        const { type } = node;
+
+        if (type === 'decl') {
+          transform(node, "value", options, result);
+        }
+
+        if (type === 'atrule' && options.mediaQueries) {
+          transform(node, "params", options, result);
+        }
+
+        if (type === 'rule' && options.selectors) {
+          transform(node, "selector", options, result);
+        }
+      });
     }
   }
 }