feat: use static object properties as identifiers (#749)

astroturfcss · Mar 17, 2023 · 626c54c · 626c54c
1 parent 35f2e36
commit 626c54c
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/src/utils/getDisplayName.ts b/src/utils/getDisplayName.ts
@@ -1,5 +1,9 @@
 import { NodePath } from '@babel/core';
-import { AssignmentExpression, VariableDeclarator } from '@babel/types';
+import {
+  AssignmentExpression,
+  ObjectProperty,
+  VariableDeclarator,
+} from '@babel/types';
 
 import { getNameFromFile } from './createFilename';
 import getNameFromPath from './getNameFromPath';
@@ -21,6 +25,8 @@ export default function getDisplayName(
       );
     if (path.isExportDefaultDeclaration())
       return getNameFromFile(file.opts.filename);
+    if (path.isObjectProperty())
+      return getNameFromPath((path as NodePath<ObjectProperty>).get('key'));
   }
   return defaultName || null;
 }
diff --git a/test/css-tag.test.js b/test/css-tag.test.js
@@ -143,6 +143,31 @@ describe('css tag', () => {
     },
   );
 
+  testAllRunners(
+    'should use object properties as identifiers',
+    async (runner) => {
+      const [, styles] = await runner(
+        `
+      import { css } from 'astroturf';
+      
+      const styles = {
+        blue: css\`
+          color: blue
+        \`,
+
+        red: css\`
+          color: red;
+        \`,
+      }
+    `,
+        { cssTagName: 'css' },
+      );
+
+      expect(styles[0].identifier).toEqual('blue');
+      expect(styles[1].identifier).toEqual('red');
+    },
+  );
+
   testAllRunners('respects the allowGlobal setting', async (runner) => {
     const [, styles] = await runner(
       `