From 26e394571afc131f496927c5672834c5665325d6 Mon Sep 17 00:00:00 2001
From: Kai Vandivier <kpvandivier@gmail.com>
Date: Wed, 2 Oct 2024 09:47:51 +0200
Subject: [PATCH] feat: migrate snap files too

---
 cli/src/commands/migrate/js-to-jsx.js | 35 ++++++++++++++++++++++-----
 1 file changed, 29 insertions(+), 6 deletions(-)

diff --git a/cli/src/commands/migrate/js-to-jsx.js b/cli/src/commands/migrate/js-to-jsx.js
index 74702363..d2924a08 100644
--- a/cli/src/commands/migrate/js-to-jsx.js
+++ b/cli/src/commands/migrate/js-to-jsx.js
@@ -34,8 +34,9 @@ const isJsxInFile = async (filepath) => {
     }
 }
 
+const jsRegex = /\.js(\.snap)?/ // handles .js or .js.snap files
 const renameFile = async (filepath) => {
-    const newPath = filepath.concat('x') // Add 'x' to the end to make it 'jsx'
+    const newPath = filepath.replace(jsRegex, '.jsx$1')
     reporter.debug(`Renaming ${filepath} to ${newPath}`)
     await fs.rename(filepath, newPath)
 }
@@ -168,7 +169,7 @@ const handler = async ({
         globMatches.map(async (matchPath) => {
             const jsxIsInFile = await isJsxInFile(matchPath)
             if (jsxIsInFile) {
-                await renameFile(matchPath, renamedFiles)
+                await renameFile(matchPath)
                 renamedFiles.add(matchPath)
             }
         })
@@ -181,7 +182,7 @@ const handler = async ({
     // a renamed item in the set. If so, rewrite the new extension
     const globMatches2 = await fg.glob(globString + '(|x)', globOptions)
     reporter.info(`Scanning ${globMatches2.length} files to update imports...`)
-    let fileUpdatedCount = 0
+    let filesUpdatedCount = 0
     await Promise.all(
         globMatches2.map(async (matchPath) => {
             const importsAreUpdated = await updateImports({
@@ -190,13 +191,35 @@ const handler = async ({
                 skipUpdatingImportsWithoutExtension,
             })
             if (importsAreUpdated) {
-                fileUpdatedCount++
+                filesUpdatedCount++
             }
         })
     )
-    reporter.print(`Updated imports in ${fileUpdatedCount} file(s)`)
+    reporter.print(`Updated imports in ${filesUpdatedCount} file(s)`)
 
-    // 3. Update d2.config.js
+    // 3. Update snapshot files
+    const snapshotGlobString = globString + '.snap'
+    const snapshotGlobMatches = await fg.glob(snapshotGlobString, globOptions)
+    reporter.info(
+        `Checking ${snapshotGlobMatches.length} snapshots to update filenames...`
+    )
+    let snapshotsUpdatedCount = 0
+    await Promise.all(
+        snapshotGlobMatches.map(async (matchPath) => {
+            // Default snapshot location is './__snapshots__/<baseTestFilename>.snap'
+            const baseTestFilename = matchPath
+                .replace(/\.snap$/, '')
+                .replace('__snapshots__/', '')
+            const baseTestFileIsUpdated = renamedFiles.has(baseTestFilename)
+            if (baseTestFileIsUpdated) {
+                await renameFile(matchPath)
+                snapshotsUpdatedCount++
+            }
+        })
+    )
+    reporter.print(`Updated ${snapshotsUpdatedCount} snapshot filename(s)`)
+
+    // 4. Update d2.config.js
     const d2ConfigPath = path.join(process.cwd(), 'd2.config.js')
     reporter.info('Checking d2.config.js for entry points to update...')
     reporter.debug(`d2 config path: ${d2ConfigPath}`)