@sanity/ui-codemod 1.0.0-alpha.4 → 1.0.0-alpha.6

package/dist/transforms/latest/text/text.js

@@ -1,6 +1,6 @@
 import "jscodeshift";
 import { addAttribute } from "../../../_chunks-es/addAttribute.js";
-import { transformComponent, getComponentLocalNames, shouldTransformComponent, transformImport, transformAttributes } from "../../../_chunks-es/transformComponent.js";
+import { transformComponent, getComponentLocalNames, getStyledComponentAliases, shouldTransformComponent, transformImport, transformAttributes, transformStyledComponents } from "../../../_chunks-es/transformStyledComponents.js";
 const TEXT_MODS = {
   accent: {
     type: "rename-mapped",
@@ -73,13 +73,24 @@ const TEXT_MODS = {
 function transform(fileInfo, api, options) {
   const { fromPackage, toPackage } = options || {};
   return transformComponent(fileInfo, api, ({ j, root, markChanged }) => {
-    const localNames = getComponentLocalNames(j, root, "Text", options);
-    shouldTransformComponent(j, root, "Text", localNames, options) && (transformImport(j, root, "Text", fromPackage, toPackage) && markChanged(), root.find(j.JSXOpeningElement, {
-      name: { type: "JSXIdentifier", name: "Text" }
-    }).forEach((path) => {
+    const localNames = getComponentLocalNames(j, root, "Text", options), styledAliases = getStyledComponentAliases(
+      j,
+      root,
+      "Text",
+      fileInfo.path,
+      localNames,
+      options
+    );
+    shouldTransformComponent(j, root, "Text", localNames, options, styledAliases) && (transformImport(j, root, "Text", fromPackage, toPackage) && markChanged(), root.find(j.JSXOpeningElement).forEach((path) => {
       let changed = !1;
-      transformAttributes(j, path, TEXT_MODS, TODO_WARNING) && (changed = !0), addAttribute(j, path.node, "as", "div") && (changed = !0), addAttribute(j, path.node, "trim", !0) && (changed = !0), changed && markChanged();
-    }));
+      const name = path.node.name;
+      name.type !== "JSXIdentifier" || !localNames.has(name.name) || (transformAttributes(j, path, TEXT_MODS, TODO_WARNING) && (changed = !0), addAttribute(j, path.node, "as", "div") && (changed = !0), addAttribute(j, path.node, "trim", !0) && (changed = !0), changed && markChanged());
+    }), transformStyledComponents(j, root, styledAliases, () => !0, {
+      callback: (path) => {
+        let changed = !1;
+        return transformAttributes(j, path, TEXT_MODS, TODO_WARNING) && (changed = !0), addAttribute(j, path.node, "as", "div") && (changed = !0), addAttribute(j, path.node, "trim", !0) && (changed = !0), changed && markChanged(), changed;
+      }
+    }) && markChanged());
   });
 }
 export {

package/dist/transforms/latest/text/text.js.map

@@ -1 +1 @@
-{"version":3,"file":"text.js","sources":["../../../../src/transforms/latest/text/text.mods.ts","../../../../src/transforms/latest/text/text.ts"],"sourcesContent":["import type {AttributeMods} from '../../../types/AttributeMods'\n\n/** @internal */\nexport const TEXT_MODS: AttributeMods = {\n  accent: {\n    type: 'rename-mapped',\n    name: 'tone',\n    mapping: {\n      true: 'suggest',\n    },\n  },\n  flex: {\n    type: 'style-mapped',\n    style: 'flex',\n    mapping: {\n      none: '0 0 auto',\n      auto: '1 1 auto',\n      initial: '0 1 auto',\n      1: '1',\n      2: '2',\n      3: '3',\n      4: '4',\n      5: '5',\n      6: '6',\n      7: '7',\n      8: '8',\n      9: '9',\n      10: '10',\n      11: '11',\n      12: '12',\n    },\n  },\n  maxWidth: {\n    type: 'style-mapped',\n    style: 'maxWidth',\n    mapping: {\n      auto: 'none',\n      fill: '100%',\n      0: '20rem',\n      1: '40rem',\n      2: '60rem',\n      3: '80rem',\n      4: '100rem',\n      5: '120rem',\n    },\n  },\n  textOverflow: {\n    type: 'rename-mapped',\n    name: 'truncate',\n    mapping: {\n      ellipsis: 1,\n      clip: 1,\n    },\n  },\n  width: {\n    type: 'style-mapped',\n    style: 'width',\n    mapping: {\n      auto: 'auto',\n      fill: '100%',\n      stretch: 'stretch',\n      min: 'min-content',\n      max: 'max-content',\n      0: '20rem',\n      1: '40rem',\n      2: '60rem',\n      3: '80rem',\n      4: '100rem',\n      5: '120rem',\n    },\n  },\n}\n","import {type API, type FileInfo} from 'jscodeshift'\n\nimport type {BaseOptions} from '../../../types/BaseOptions'\nimport {addAttribute} from '../../../utils/addAttribute'\nimport {getComponentLocalNames} from '../../../utils/getComponentLocalNames'\nimport {shouldTransformComponent} from '../../../utils/shouldTransformComponent'\nimport {transformAttributes} from '../../../utils/transformAttributes'\nimport {transformComponent} from '../../../utils/transformComponent'\nimport {transformImport} from '../../../utils/transformImport'\nimport {TEXT_MODS} from './text.mods'\n\nconst TODO_WARNING = 'Please double check the Text migration below'\n\n/** @internal */\nexport default function transform(\n  fileInfo: FileInfo,\n  api: API,\n  options: BaseOptions,\n): string | undefined {\n  const {fromPackage, toPackage} = options || {}\n\n  return transformComponent(fileInfo, api, ({j, root, markChanged}) => {\n    const localNames = getComponentLocalNames(j, root, 'Text', options)\n\n    if (!shouldTransformComponent(j, root, 'Text', localNames, options)) {\n      return\n    }\n\n    if (transformImport(j, root, 'Text', fromPackage, toPackage)) {\n      markChanged()\n    }\n\n    root\n      .find(j.JSXOpeningElement, {\n        name: {type: 'JSXIdentifier', name: 'Text'},\n      })\n      .forEach((path) => {\n        let changed = false\n\n        if (transformAttributes(j, path, TEXT_MODS, TODO_WARNING)) {\n          changed = true\n        }\n\n        if (addAttribute(j, path.node, 'as', 'div')) {\n          changed = true\n        }\n\n        if (addAttribute(j, path.node, 'trim', true)) {\n          changed = true\n        }\n\n        if (changed) {\n          markChanged()\n        }\n      })\n  })\n}\n"],"names":[],"mappings":";;;AAGO,MAAM,YAA2B;AAAA,EACtC,QAAQ;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,SAAS;AAAA,MACP,MAAM;AAAA,IAAA;AAAA,EACR;AAAA,EAEF,MAAM;AAAA,IACJ,MAAM;AAAA,IACN,OAAO;AAAA,IACP,SAAS;AAAA,MACP,MAAM;AAAA,MACN,MAAM;AAAA,MACN,SAAS;AAAA,MACT,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,IAAI;AAAA,MACJ,IAAI;AAAA,MACJ,IAAI;AAAA,IAAA;AAAA,EACN;AAAA,EAEF,UAAU;AAAA,IACR,MAAM;AAAA,IACN,OAAO;AAAA,IACP,SAAS;AAAA,MACP,MAAM;AAAA,MACN,MAAM;AAAA,MACN,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,IAAA;AAAA,EACL;AAAA,EAEF,cAAc;AAAA,IACZ,MAAM;AAAA,IACN,MAAM;AAAA,IACN,SAAS;AAAA,MACP,UAAU;AAAA,MACV,MAAM;AAAA,IAAA;AAAA,EACR;AAAA,EAEF,OAAO;AAAA,IACL,MAAM;AAAA,IACN,OAAO;AAAA,IACP,SAAS;AAAA,MACP,MAAM;AAAA,MACN,MAAM;AAAA,MACN,SAAS;AAAA,MACT,KAAK;AAAA,MACL,KAAK;AAAA,MACL,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,IAAA;AAAA,EACL;AAEJ,GC5DM,eAAe;AAGrB,SAAwB,UACtB,UACA,KACA,SACoB;AACpB,QAAM,EAAC,aAAa,UAAA,IAAa,WAAW,CAAA;AAE5C,SAAO,mBAAmB,UAAU,KAAK,CAAC,EAAC,GAAG,MAAM,kBAAiB;AACnE,UAAM,aAAa,uBAAuB,GAAG,MAAM,QAAQ,OAAO;AAE7D,6BAAyB,GAAG,MAAM,QAAQ,YAAY,OAAO,MAI9D,gBAAgB,GAAG,MAAM,QAAQ,aAAa,SAAS,KACzD,YAAA,GAGF,KACG,KAAK,EAAE,mBAAmB;AAAA,MACzB,MAAM,EAAC,MAAM,iBAAiB,MAAM,OAAA;AAAA,IAAM,CAC3C,EACA,QAAQ,CAAC,SAAS;AACjB,UAAI,UAAU;AAEV,0BAAoB,GAAG,MAAM,WAAW,YAAY,MACtD,UAAU,KAGR,aAAa,GAAG,KAAK,MAAM,MAAM,KAAK,MACxC,UAAU,KAGR,aAAa,GAAG,KAAK,MAAM,QAAQ,EAAI,MACzC,UAAU,KAGR,WACF,YAAA;AAAA,IAEJ,CAAC;AAAA,EACL,CAAC;AACH;"}
+{"version":3,"file":"text.js","sources":["../../../../src/transforms/latest/text/text.mods.ts","../../../../src/transforms/latest/text/text.ts"],"sourcesContent":["import type {AttributeMods} from '../../../types/AttributeMods'\n\n/** @internal */\nexport const TEXT_MODS: AttributeMods = {\n  accent: {\n    type: 'rename-mapped',\n    name: 'tone',\n    mapping: {\n      true: 'suggest',\n    },\n  },\n  flex: {\n    type: 'style-mapped',\n    style: 'flex',\n    mapping: {\n      none: '0 0 auto',\n      auto: '1 1 auto',\n      initial: '0 1 auto',\n      1: '1',\n      2: '2',\n      3: '3',\n      4: '4',\n      5: '5',\n      6: '6',\n      7: '7',\n      8: '8',\n      9: '9',\n      10: '10',\n      11: '11',\n      12: '12',\n    },\n  },\n  maxWidth: {\n    type: 'style-mapped',\n    style: 'maxWidth',\n    mapping: {\n      auto: 'none',\n      fill: '100%',\n      0: '20rem',\n      1: '40rem',\n      2: '60rem',\n      3: '80rem',\n      4: '100rem',\n      5: '120rem',\n    },\n  },\n  textOverflow: {\n    type: 'rename-mapped',\n    name: 'truncate',\n    mapping: {\n      ellipsis: 1,\n      clip: 1,\n    },\n  },\n  width: {\n    type: 'style-mapped',\n    style: 'width',\n    mapping: {\n      auto: 'auto',\n      fill: '100%',\n      stretch: 'stretch',\n      min: 'min-content',\n      max: 'max-content',\n      0: '20rem',\n      1: '40rem',\n      2: '60rem',\n      3: '80rem',\n      4: '100rem',\n      5: '120rem',\n    },\n  },\n}\n","import {type API, type FileInfo} from 'jscodeshift'\n\nimport type {BaseOptions} from '../../../types/BaseOptions'\nimport {addAttribute} from '../../../utils/addAttribute'\nimport {getComponentLocalNames} from '../../../utils/getComponentLocalNames'\nimport {getStyledComponentAliases} from '../../../utils/getStyledComponentAliases'\nimport {shouldTransformComponent} from '../../../utils/shouldTransformComponent'\nimport {transformAttributes} from '../../../utils/transformAttributes'\nimport {transformComponent} from '../../../utils/transformComponent'\nimport {transformImport} from '../../../utils/transformImport'\nimport {transformStyledComponents} from '../../../utils/transformStyledComponents'\nimport {TEXT_MODS} from './text.mods'\n\nconst TODO_WARNING = 'Please double check the Text migration below'\n\n/** @internal */\nexport default function transform(\n  fileInfo: FileInfo,\n  api: API,\n  options: BaseOptions,\n): string | undefined {\n  const {fromPackage, toPackage} = options || {}\n\n  return transformComponent(fileInfo, api, ({j, root, markChanged}) => {\n    const localNames = getComponentLocalNames(j, root, 'Text', options)\n    const styledAliases = getStyledComponentAliases(\n      j,\n      root,\n      'Text',\n      fileInfo.path,\n      localNames,\n      options,\n    )\n\n    if (!shouldTransformComponent(j, root, 'Text', localNames, options, styledAliases)) {\n      return\n    }\n\n    if (transformImport(j, root, 'Text', fromPackage, toPackage)) {\n      markChanged()\n    }\n\n    root.find(j.JSXOpeningElement).forEach((path) => {\n      let changed = false\n      const name = path.node.name\n\n      if (name.type !== 'JSXIdentifier' || !localNames.has(name.name)) {\n        return\n      }\n\n      if (transformAttributes(j, path, TEXT_MODS, TODO_WARNING)) {\n        changed = true\n      }\n\n      if (addAttribute(j, path.node, 'as', 'div')) {\n        changed = true\n      }\n\n      if (addAttribute(j, path.node, 'trim', true)) {\n        changed = true\n      }\n\n      if (changed) {\n        markChanged()\n      }\n    })\n\n    if (\n      transformStyledComponents(j, root, styledAliases, () => true, {\n        callback: (path) => {\n          let changed = false\n\n          if (transformAttributes(j, path, TEXT_MODS, TODO_WARNING)) {\n            changed = true\n          }\n\n          if (addAttribute(j, path.node, 'as', 'div')) {\n            changed = true\n          }\n\n          if (addAttribute(j, path.node, 'trim', true)) {\n            changed = true\n          }\n\n          if (changed) {\n            markChanged()\n          }\n\n          return changed\n        },\n      })\n    ) {\n      markChanged()\n    }\n  })\n}\n"],"names":[],"mappings":";;;AAGO,MAAM,YAA2B;AAAA,EACtC,QAAQ;AAAA,IACN,MAAM;AAAA,IACN,MAAM;AAAA,IACN,SAAS;AAAA,MACP,MAAM;AAAA,IAAA;AAAA,EACR;AAAA,EAEF,MAAM;AAAA,IACJ,MAAM;AAAA,IACN,OAAO;AAAA,IACP,SAAS;AAAA,MACP,MAAM;AAAA,MACN,MAAM;AAAA,MACN,SAAS;AAAA,MACT,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,IAAI;AAAA,MACJ,IAAI;AAAA,MACJ,IAAI;AAAA,IAAA;AAAA,EACN;AAAA,EAEF,UAAU;AAAA,IACR,MAAM;AAAA,IACN,OAAO;AAAA,IACP,SAAS;AAAA,MACP,MAAM;AAAA,MACN,MAAM;AAAA,MACN,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,IAAA;AAAA,EACL;AAAA,EAEF,cAAc;AAAA,IACZ,MAAM;AAAA,IACN,MAAM;AAAA,IACN,SAAS;AAAA,MACP,UAAU;AAAA,MACV,MAAM;AAAA,IAAA;AAAA,EACR;AAAA,EAEF,OAAO;AAAA,IACL,MAAM;AAAA,IACN,OAAO;AAAA,IACP,SAAS;AAAA,MACP,MAAM;AAAA,MACN,MAAM;AAAA,MACN,SAAS;AAAA,MACT,KAAK;AAAA,MACL,KAAK;AAAA,MACL,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,MACH,GAAG;AAAA,IAAA;AAAA,EACL;AAEJ,GC1DM,eAAe;AAGrB,SAAwB,UACtB,UACA,KACA,SACoB;AACpB,QAAM,EAAC,aAAa,UAAA,IAAa,WAAW,CAAA;AAE5C,SAAO,mBAAmB,UAAU,KAAK,CAAC,EAAC,GAAG,MAAM,kBAAiB;AACnE,UAAM,aAAa,uBAAuB,GAAG,MAAM,QAAQ,OAAO,GAC5D,gBAAgB;AAAA,MACpB;AAAA,MACA;AAAA,MACA;AAAA,MACA,SAAS;AAAA,MACT;AAAA,MACA;AAAA,IAAA;AAGG,6BAAyB,GAAG,MAAM,QAAQ,YAAY,SAAS,aAAa,MAI7E,gBAAgB,GAAG,MAAM,QAAQ,aAAa,SAAS,KACzD,YAAA,GAGF,KAAK,KAAK,EAAE,iBAAiB,EAAE,QAAQ,CAAC,SAAS;AAC/C,UAAI,UAAU;AACd,YAAM,OAAO,KAAK,KAAK;AAEnB,WAAK,SAAS,mBAAmB,CAAC,WAAW,IAAI,KAAK,IAAI,MAI1D,oBAAoB,GAAG,MAAM,WAAW,YAAY,MACtD,UAAU,KAGR,aAAa,GAAG,KAAK,MAAM,MAAM,KAAK,MACxC,UAAU,KAGR,aAAa,GAAG,KAAK,MAAM,QAAQ,EAAI,MACzC,UAAU,KAGR,WACF;IAEJ,CAAC,GAGC,0BAA0B,GAAG,MAAM,eAAe,MAAM,IAAM;AAAA,MAC5D,UAAU,CAAC,SAAS;AAClB,YAAI,UAAU;AAEd,eAAI,oBAAoB,GAAG,MAAM,WAAW,YAAY,MACtD,UAAU,KAGR,aAAa,GAAG,KAAK,MAAM,MAAM,KAAK,MACxC,UAAU,KAGR,aAAa,GAAG,KAAK,MAAM,QAAQ,EAAI,MACzC,UAAU,KAGR,WACF,YAAA,GAGK;AAAA,MACT;AAAA,IAAA,CACD,KAED,YAAA;AAAA,EAEJ,CAAC;AACH;"}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@sanity/ui-codemod",
-  "version": "1.0.0-alpha.4",
+  "version": "1.0.0-alpha.6",
   "description": "",
   "keywords": [],
   "license": "ISC",

package/src/constants/latest/layout-mods.ts

@@ -411,6 +411,10 @@ export const LAYOUT_MODS: AttributeMods = {
       border: 'border-box',
     },
   },
+  space: {
+    type: 'rename-only',
+    name: 'gap',
+  },
   textAlign: {
     type: 'style-only',
     style: 'textAlign',

package/src/transforms/latest/box/box.test.ts

@@ -1,13 +1,6 @@
-import {mkdtempSync, readFileSync, rmSync, writeFileSync} from 'node:fs'
-import {tmpdir} from 'node:os'
-import {join} from 'node:path'
+import {expect} from 'vitest'
 
-import {afterEach, describe, expect, it} from 'vitest'
-
-const applyTransform = require('jscodeshift/dist/testUtils').applyTransform
-
-import {clearModuleParseCache} from '../../../utils/parseModule'
-import {defineInlineTest} from '../../../utils/testUtils'
+import {defineCrossFileTest, defineInlineTest} from '../../../utils/testUtils'
 import transform from './box'
 
 defineInlineTest(
@@ -246,157 +239,99 @@ defineInlineTest(
   'warns and does not transform attributes if styled Box should be replaced',
 )
 
-const tempDirs: string[] = []
-
-afterEach(() => {
-  clearModuleParseCache()
-
-  for (const dir of tempDirs.splice(0)) {
-    rmSync(dir, {recursive: true, force: true})
-  }
-})
-
-describe('cross-file styled aliases', () => {
-  it('transforms attributes on imported styled Box wrappers', () => {
-    const dir = mkdtempSync(join(tmpdir(), 'ui-codemod-box-crossfile-'))
-
-    tempDirs.push(dir)
-
-    writeFileSync(
-      join(dir, 'Component.styled.tsx'),
-      `
-        import {Box} from '@sanity/ui'
-
-        export const RootBox = styled(Box)(({theme}) => ({}))
-      `,
-    )
-
-    writeFileSync(
-      join(dir, 'Component.tsx'),
-      `
-        import {RootBox} from './Component.styled'
-
-        export function Component() {
-          return <RootBox alignItems="center" />
-        }
-      `,
-    )
-
-    const importerPath = join(dir, 'Component.tsx')
-    const source = readFileSync(importerPath, 'utf8')
-    const output = applyTransform(transform, {}, {source, path: importerPath}, {parser: 'tsx'})
-
-    expect(output).toContain('alignItems: "center"')
-    expect(output).not.toContain('<RootBox alignItems="center" />')
-  })
-
-  it('adds todo warning when imported styled Box wrapper should be replaced', () => {
-    const dir = mkdtempSync(join(tmpdir(), 'ui-codemod-box-crossfile-todo-'))
-
-    tempDirs.push(dir)
-
-    writeFileSync(
-      join(dir, 'Component.styled.tsx'),
-      `
-        import {Box} from '@sanity/ui'
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Box} from '@sanity/ui'
 
-        export const RootBox = styled(Box)(({theme}) => ({}))
-      `,
-    )
+    export const RootBox = styled(Box)(({theme}) => ({}))
+  `,
+  `
+    import {RootBox} from './Component.styled'
 
-    writeFileSync(
-      join(dir, 'Component.tsx'),
-      `
-        import {RootBox} from './Component.styled'
+    export function Component() {
+      return <RootBox alignItems="center" />
+    }
+  `,
+  (output) => {
+    expect(output.replace(/\s+/g, ' ')).toContain('<RootBox style={{ alignItems: "center" }} />')
+  },
+  'transforms attributes on imported styled Box wrappers',
+)
 
-        export function Component() {
-          return <RootBox display="flex" alignItems="center" />
-        }
-      `,
-    )
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Box} from '@sanity/ui'
 
-    const importerPath = join(dir, 'Component.tsx')
-    const source = readFileSync(importerPath, 'utf8')
-    const output = applyTransform(transform, {}, {source, path: importerPath}, {parser: 'tsx'})
+    export const RootBox = styled(Box)(({theme}) => ({}))
+  `,
+  `
+    import {RootBox} from './Component.styled'
 
+    export function Component() {
+      return <RootBox display="flex" alignItems="center" />
+    }
+  `,
+  (output) => {
     expect(output).toContain('UI-CODEMOD TODO: Please double check styled(Box) migration below')
     expect(output).toContain('<RootBox display="flex" alignItems="center" />')
-    expect(output).not.toContain('const RootBox = styled(Box)')
-  })
-
-  it('does not rewrite unrelated Box from another package', () => {
-    const dir = mkdtempSync(join(tmpdir(), 'ui-codemod-box-crossfile-unrelated-'))
-
-    tempDirs.push(dir)
-
-    writeFileSync(
-      join(dir, 'Component.styled.tsx'),
-      `
-        import {Box} from '@sanity/ui'
-
-        export const RootBox = styled(Box)(({theme}) => ({}))
-      `,
-    )
-
-    writeFileSync(
-      join(dir, 'Component.tsx'),
-      `
-        import {Box} from 'another-package'
-        import {RootBox} from './Component.styled'
-
-        export function Component() {
-          return (
-            <>
-              <RootBox alignItems="center" />
-              <Box display="flex" />
-            </>
-          )
-        }
-      `,
-    )
-
-    const importerPath = join(dir, 'Component.tsx')
-    const source = readFileSync(importerPath, 'utf8')
-    const output = applyTransform(transform, {}, {source, path: importerPath}, {parser: 'tsx'})
-
-    expect(output).toContain('alignItems: "center"')
-    expect(output).not.toContain('<RootBox alignItems="center" />')
-    expect(output).toContain('<Box display="flex" />')
-    expect(output).not.toContain('<Flex')
-  })
-
-  it('transforms styled Box wrappers imported through barrel re-exports', () => {
-    const dir = mkdtempSync(join(tmpdir(), 'ui-codemod-box-crossfile-barrel-'))
-
-    tempDirs.push(dir)
-
-    writeFileSync(
-      join(dir, 'Component.styled.tsx'),
-      `
-        import {Box} from '@sanity/ui'
-
-        export const RootBox = styled(Box)(({theme}) => ({}))
-      `,
-    )
+  },
+  'adds todo warning when imported styled Box wrapper should be replaced',
+)
 
-    writeFileSync(join(dir, 'index.ts'), `export {RootBox} from './Component.styled'`)
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Box} from '@sanity/ui'
 
-    writeFileSync(
-      join(dir, 'Component.tsx'),
-      `
-        import {RootBox} from './index'
+    export const RootBox = styled(Box)(({theme}) => ({}))
+  `,
+  `
+    import {Box} from 'another-package'
+    import {RootBox} from './Component.styled'
+
+    export function Component() {
+      return (
+        <>
+          <RootBox alignItems="center" />
+          <Box display="flex" />
+        </>
+      )
+    }
+  `,
+  (output) => {
+    expect(output.replace(/\s+/g, ' ')).toContain('<RootBox style={{ alignItems: "center" }} />')
+    expect(output).toContain('<Box display="flex" />')
+  },
+  'does not rewrite unrelated Box from another package',
+)
 
-        export function Component() {
-          return <RootBox alignItems="center" />
-        }
-      `,
-    )
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Box} from '@sanity/ui'
 
-    const importerPath = join(dir, 'Component.tsx')
-    const source = readFileSync(importerPath, 'utf8')
-    const output = applyTransform(transform, {}, {source, path: importerPath}, {parser: 'tsx'})
+    export const RootBox = styled(Box)(({theme}) => ({}))
+  `,
+  `
+    import {RootBox} from './index'
 
-    expect(output).toContain('alignItems: "center"')
-    expect(output).not.toContain('<RootBox alignItems="center" />')
-  })
-})
+    export function Component() {
+      return <RootBox alignItems="center" />
+    }
+  `,
+  (output) => {
+    expect(output.replace(/\s+/g, ' ')).toContain('<RootBox style={{ alignItems: "center" }} />')
+  },
+  'transforms styled Box wrappers imported through barrel re-exports',
+  {
+    extraFiles: {
+      'index.ts': `export {RootBox} from './Component.styled'`,
+    },
+  },
+)

package/src/transforms/latest/card/card.test.ts

@@ -1,4 +1,6 @@
-import {defineInlineTest} from '../../../utils/testUtils'
+import {expect} from 'vitest'
+
+import {defineCrossFileTest, defineInlineTest} from '../../../utils/testUtils'
 import transform from './card'
 
 defineInlineTest(
@@ -205,3 +207,100 @@ defineInlineTest(
   `,
   'warns if Card includes muted prop',
 )
+
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Card} from '@sanity/ui'
+
+    export const RootCard = styled(Card)(({theme}) => ({}))
+  `,
+  `
+    import {RootCard} from './Component.styled'
+
+    export function Component() {
+      return <RootCard padding={4} radius={3} border />
+    }
+  `,
+  (output) => {
+    expect(output).toContain('<RootCard density="regular" />')
+  },
+  'transforms attributes on imported styled Card wrappers',
+)
+
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Card} from '@sanity/ui'
+
+    export const RootCard = styled(Card)(({theme}) => ({}))
+  `,
+  `
+    import {RootCard} from './Component.styled'
+
+    export function Component() {
+      return <RootCard padding={1} />
+    }
+  `,
+  (output) => {
+    expect(output).toContain('UI-CODEMOD TODO: Please double check styled(Card) migration below')
+    expect(output).toContain('<RootCard padding={1} />')
+  },
+  'adds todo warning when imported styled Card wrapper should be replaced',
+)
+
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Card} from '@sanity/ui'
+
+    export const RootCard = styled(Card)(({theme}) => ({}))
+  `,
+  `
+    import {Card} from 'another-package'
+    import {RootCard} from './Component.styled'
+
+    export function Component() {
+      return (
+        <>
+          <RootCard padding={4} radius={3} border />
+          <Card padding={4} radius={3} border />
+        </>
+      )
+    }
+  `,
+  (output) => {
+    expect(output).toContain('<RootCard density="regular" />')
+    expect(output).toContain('<Card padding={4} radius={3} border />')
+  },
+  'does not rewrite unrelated Card from another package',
+)
+
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Card} from '@sanity/ui'
+
+    export const RootCard = styled(Card)(({theme}) => ({}))
+  `,
+  `
+    import {RootCard} from './index'
+
+    export function Component() {
+      return <RootCard padding={4} radius={3} border />
+    }
+  `,
+  (output) => {
+    expect(output).toContain('<RootCard density="regular" />')
+  },
+  'transforms styled Card wrappers imported through barrel re-exports',
+  {
+    extraFiles: {
+      'index.ts': `export {RootCard} from './Component.styled'`,
+    },
+  },
+)

package/src/transforms/latest/card/card.ts

@@ -1,19 +1,22 @@
-import {type API, type FileInfo} from 'jscodeshift'
+import {type API, type FileInfo, type JSXAttribute, type JSXSpreadAttribute} from 'jscodeshift'
 
 import type {BaseOptions} from '../../../types/BaseOptions'
 import {getAttribute} from '../../../utils/getAttribute'
 import {getComponentLocalNames} from '../../../utils/getComponentLocalNames'
 import {getStaticAttributeExpression} from '../../../utils/getStaticAttributeExpression'
+import {getStyledComponentAliases} from '../../../utils/getStyledComponentAliases'
 import {replaceElement} from '../../../utils/replaceElement'
 import {shouldTransformComponent} from '../../../utils/shouldTransformComponent'
 import {transformAttributes} from '../../../utils/transformAttributes'
 import {transformComponent} from '../../../utils/transformComponent'
 import {transformImport} from '../../../utils/transformImport'
+import {transformStyledComponents} from '../../../utils/transformStyledComponents'
 import {BOX_MODS} from '../box/box.mods'
 import {CARD_MODS} from './card.mods'
 
 const CARD_TODO_WARNING = 'Please double check the Card migration below'
 const BOX_TODO_WARNING = 'Please double check the Box migration below'
+const STYLED_TODO_WARNING = 'Please double check styled(Card) migration below'
 
 /** @internal */
 export default function transform(
@@ -25,8 +28,16 @@ export default function transform(
 
   return transformComponent(fileInfo, api, ({j, root, markChanged}) => {
     const localNames = getComponentLocalNames(j, root, 'Card', options)
+    const styledAliases = getStyledComponentAliases(
+      j,
+      root,
+      'Card',
+      fileInfo.path,
+      localNames,
+      options,
+    )
 
-    if (!shouldTransformComponent(j, root, 'Card', localNames, options)) {
+    if (!shouldTransformComponent(j, root, 'Card', localNames, options, styledAliases)) {
       return
     }
 
@@ -34,29 +45,32 @@ export default function transform(
       markChanged()
     }
 
+    const replaceWithBox = (attrs: (JSXAttribute | JSXSpreadAttribute)[]) => {
+      const density = getAttribute(attrs, 'density')
+      const muted = getAttribute(attrs, 'muted')
+      const scheme = getAttribute(attrs, 'scheme')
+      const border = getAttribute(attrs, 'border')
+      const padding = getStaticAttributeExpression(j, attrs, 'padding')
+      const radius = getStaticAttributeExpression(j, attrs, 'radius')
+
+      return !(
+        density ||
+        muted ||
+        scheme ||
+        (border && padding === 3 && radius == 2) ||
+        (border && padding === 4 && radius == 3) ||
+        (border && padding === 5 && radius == 4)
+      )
+    }
+
     if (
       replaceElement(
         j,
         root,
-        (attrs) => {
-          const density = getAttribute(attrs, 'density')
-          const muted = getAttribute(attrs, 'muted')
-          const scheme = getAttribute(attrs, 'scheme')
-          const border = getAttribute(attrs, 'border')
-          const padding = getStaticAttributeExpression(j, attrs, 'padding')
-          const radius = getStaticAttributeExpression(j, attrs, 'radius')
-
-          return !(
-            density ||
-            muted ||
-            scheme ||
-            (border && padding === 3 && radius == 2) ||
-            (border && padding === 4 && radius == 3) ||
-            (border && padding === 5 && radius == 4)
-          )
-        },
+        (attrs) => replaceWithBox(attrs),
         {
           element: 'Card',
+          localNames,
           callback: (path) => transformAttributes(j, path, CARD_MODS, CARD_TODO_WARNING),
         },
         {
@@ -67,5 +81,14 @@ export default function transform(
     ) {
       markChanged()
     }
+
+    if (
+      transformStyledComponents(j, root, styledAliases, (attrs) => !replaceWithBox(attrs), {
+        warning: STYLED_TODO_WARNING,
+        callback: (path) => transformAttributes(j, path, CARD_MODS, CARD_TODO_WARNING),
+      })
+    ) {
+      markChanged()
+    }
   })
 }

package/src/transforms/latest/code/code.test.ts

@@ -1,4 +1,6 @@
-import {defineInlineTest} from '../../../utils/testUtils'
+import {expect} from 'vitest'
+
+import {defineCrossFileTest, defineInlineTest} from '../../../utils/testUtils'
 import transform from './code'
 
 defineInlineTest(
@@ -59,3 +61,84 @@ defineInlineTest(
   `,
   'moves width props to style and uppdates mapped values',
 )
+
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Code} from '@sanity/ui'
+
+    export const RootCode = styled(Code)(({theme}) => ({}))
+  `,
+  `
+    import {RootCode} from './Component.styled'
+
+    export function RootCode() {
+      return <RootCode flex="auto" />
+    }
+  `,
+  (output) => {
+    expect(output.replace(/\s+/g, ' ')).toContain(
+      '<RootCode style={{ flex: "1 1 auto" }} trim={true} />',
+    )
+  },
+  'transforms attributes on imported styled Code wrappers',
+)
+
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Code} from '@sanity/ui'
+
+    export const RootCode = styled(Code)(({theme}) => ({}))
+  `,
+  `
+    import {Code} from 'another-package'
+    import {RootCode} from './Component.styled'
+
+    export function Component() {
+      return (
+        <>
+          <RootCode flex="auto" />
+          <Code flex="auto" />
+        </>
+      )
+    }
+  `,
+  (output) => {
+    expect(output.replace(/\s+/g, ' ')).toContain(
+      '<RootCode style={{ flex: "1 1 auto" }} trim={true} />',
+    )
+    expect(output).toContain('<Code flex="auto" />')
+  },
+  'does not transform attributes on unrelated Code from another package',
+)
+
+defineCrossFileTest(
+  transform,
+  {},
+  `
+    import {Code} from '@sanity/ui'
+
+    export const RootCode = styled(Code)(({theme}) => ({}))
+  `,
+  `
+    import {RootCode} from './index'
+
+    export function Component() {
+      return <RootCode flex="auto" />
+    }
+  `,
+  (output) => {
+    expect(output.replace(/\s+/g, ' ')).toContain(
+      '<RootCode style={{ flex: "1 1 auto" }} trim={true} />',
+    )
+  },
+  'transforms styled Code wrappers imported through barrel re-exports',
+  {
+    extraFiles: {
+      'index.ts': `export {RootCode} from './Component.styled'`,
+    },
+  },
+)