fix: traverse exported declarations

Blazity · Aug 4, 2023 · b0b5def · b0b5def
1 parent 1338ada
commit b0b5def
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/src/ASTtoCSS.ts b/src/ASTtoCSS.ts
@@ -154,6 +154,15 @@ export function convertASTtoCSS(ast: t.File): Array<{
           styledComponents.push(component)
         }
       }
+    } else if (t.isExportNamedDeclaration(node) || t.isExportDefaultDeclaration(node)) {
+      if (node.declaration && t.isVariableDeclaration(node.declaration)) {
+        for (const declaration of node.declaration.declarations) {
+          if (t.isVariableDeclarator(declaration) && t.isIdentifier(declaration.id)) {
+            const component = traverseComponent(declaration)
+            styledComponents.push(component)
+          }
+        }
+      }
     }
   }