1 files changed, 67 insertions, 0 deletions
diff --git a/@linaria/packages/babel/src/utils/isStyledOrCss.ts b/@linaria/packages/babel/src/utils/isStyledOrCss.ts
new file mode 100644
index 0000000..04e4789
--- /dev/null
+++ b/@linaria/packages/babel/src/utils/isStyledOrCss.ts
@@ -0,0 +1,67 @@
+import type {
+  CallExpression,
+  Expression,
+  TaggedTemplateExpression,
+} from '@babel/types';
+import type { NodePath } from '@babel/traverse';
+import type { State, TemplateExpression } from '../types';
+import { Core } from '../babel';
+import hasImport from './hasImport';
+
+type Result = NonNullable<TemplateExpression['styled']> | 'css' | null;
+
+const cache = new WeakMap<NodePath<TaggedTemplateExpression>, Result>();
+
+export default function isStyledOrCss(
+  { types: t }: Core,
+  path: NodePath<TaggedTemplateExpression>,
+  state: State
+): Result {
+  if (!cache.has(path)) {
+    const { tag } = path.node;
+
+    const localName = state.file.metadata.localName || 'styled';
+
+    if (
+      t.isCallExpression(tag) &&
+      t.isIdentifier(tag.callee) &&
+      tag.arguments.length === 1 &&
+      tag.callee.name === localName &&
+      hasImport(t, path.scope, state.file.opts.filename, localName, [
+        '@linaria/react',
+        'linaria/react',
+      ])
+    ) {
+      const tagPath = path.get('tag') as NodePath<CallExpression>;
+      cache.set(path, {
+        component: tagPath.get('arguments')[0] as NodePath<Expression>,
+      });
+    } else if (
+      t.isMemberExpression(tag) &&
+      t.isIdentifier(tag.object) &&
+      t.isIdentifier(tag.property) &&
+      tag.object.name === localName &&
+      hasImport(t, path.scope, state.file.opts.filename, localName, [
+        '@linaria/react',
+        'linaria/react',
+      ])
+    ) {
+      cache.set(path, {
+        component: { node: t.stringLiteral(tag.property.name) },
+      });
+    } else if (
+      hasImport(t, path.scope, state.file.opts.filename, 'css', [
+        '@linaria/core',
+        'linaria',
+      ]) &&
+      t.isIdentifier(tag) &&
+      tag.name === 'css'
+    ) {
+      cache.set(path, 'css');
+    } else {
+      cache.set(path, null);
+    }
+  }
+
+  return cache.get(path) ?? null;
+}