feat: update error components import statements (#179)

Author: valerybugakov

packages/transforms/src/errorMessageMove/__tests__/errorMessageMove.test.ts

@@ -0,0 +1,19 @@
+import { testCodemod } from '@sourcegraph/codemod-toolkit-ts'
+
+import { errorMessageMove } from '../errorMessageMove'
+
+testCodemod('errorMessageMove', errorMessageMove, [
+    {
+        label: 'default',
+        initialSource: `
+            import { ErrorMessage } from '@sourcegraph/branded/src/components/alerts'
+
+            export const Test = <ErrorMessage className="btn">Hey</ErrorMessage>
+        `,
+        expectedSource: `
+            import { ErrorMessage } from '@sourcegraph/wildcard'
+
+            export const Test = <ErrorMessage className="btn">Hey</ErrorMessage>
+        `,
+    },
+])

packages/transforms/src/errorMessageMove/errorMessageMove.ts

@@ -0,0 +1,43 @@
+import { addOrUpdateSourcegraphWildcardImportIfNeeded } from '@sourcegraph/codemod-toolkit-packages'
+import { getImportDeclarationByModuleSpecifier, JsxTagElement, runTransform } from '@sourcegraph/codemod-toolkit-ts'
+
+const COMPONENTS_TO_MOVE = new Set(['ErrorMessage', 'ErrorAlert'])
+
+/**
+ * Update imports for COMPONENTS_TO_MOVE to `@sourcegraph/wildcard`. See tests for more examples.
+ */
+export const errorMessageMove = runTransform(() => {
+    const jsxTagElementsToUpdate = new Set<string>()
+
+    function processElement(element: JsxTagElement): void {
+        const tagName = element.getTagNameNode().getText()
+
+        if (COMPONENTS_TO_MOVE.has(tagName)) {
+            jsxTagElementsToUpdate.add(tagName)
+        }
+    }
+
+    return {
+        JsxOpeningElement: processElement,
+        JsxSelfClosingElement: processElement,
+        SourceFileExit(sourceFile) {
+            if (jsxTagElementsToUpdate.size === 0) {
+                return
+            }
+            const declaration = getImportDeclarationByModuleSpecifier(
+                sourceFile,
+                '@sourcegraph/branded/src/components/alerts'
+            )
+            declaration?.remove()
+
+            addOrUpdateSourcegraphWildcardImportIfNeeded({
+                sourceFile,
+                importStructure: {
+                    namedImports: [...jsxTagElementsToUpdate],
+                },
+            })
+
+            sourceFile.fixUnusedIdentifiers()
+        },
+    }
+})

packages/transforms/src/errorMessageMove/index.ts

@@ -0,0 +1 @@
+export * from './errorMessageMove'