Call StyleSheet / StyleSheetExit in visitors passed to composeVisitors

parcel-bundler · Dec 25, 2024 · 9fa6dc4 · 9fa6dc4
1 parent 7f29035
commit 9fa6dc4
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 0 deletions.
diff --git a/node/composeVisitors.js b/node/composeVisitors.js
@@ -13,6 +13,8 @@ function composeVisitors(visitors) {
 
   /** @type Visitor */
   let res = {};
+  composeSimpleVisitors(res, visitors, 'StyleSheet');
+  composeSimpleVisitors(res, visitors, 'StyleSheetExit');
   composeObjectVisitors(res, visitors, 'Rule', ruleVisitor, wrapUnknownAtRule);
   composeObjectVisitors(res, visitors, 'RuleExit', ruleVisitor, wrapUnknownAtRule);
   composeObjectVisitors(res, visitors, 'Declaration', declarationVisitor, wrapCustomProperty);

diff --git a/node/test/composeVisitors.test.mjs b/node/test/composeVisitors.test.mjs
@@ -686,4 +686,37 @@ test('variables', () => {
   assert.equal(res.code.toString(), 'body{padding:20px;width:600px}');
 });
 
+test('StyleSheet', () => {
+  let styleSheetCalledCount = 0;
+  let styleSheetExitCalledCount = 0;
+  transform({
+    filename: 'test.css',
+    code: Buffer.from(`
+      body {
+        color: blue;
+      }
+    `),
+    visitor: composeVisitors([
+      {
+        StyleSheet() {
+          styleSheetCalledCount++
+        },
+        StyleSheetExit() {
+          styleSheetExitCalledCount++
+        }
+      },
+      {
+        StyleSheet() {
+          styleSheetCalledCount++
+        },
+        StyleSheetExit() {
+          styleSheetExitCalledCount++
+        }
+      }
+    ])
+  });
+  assert.equal(styleSheetCalledCount, 2);
+  assert.equal(styleSheetExitCalledCount, 2);
+});
+
 test.run();