fix: improve prisma scripts (#563)

Author: exKAZUu

packages/wb/src/commands/prisma.ts

@@ -137,6 +137,13 @@ const resetCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>>
     for (const project of prepareForRunningCommand('prisma reset', allProjects)) {
       await runWithSpawn(prismaScripts.reset(project, unknownOptions), project, argv);
     }
+    // Force to reset test database
+    if (process.env.WB_ENV !== 'test') {
+      process.env.WB_ENV = 'test';
+      for (const project of prepareForRunningCommand('WB_ENV=test prisma reset', await findPrismaProjects(argv))) {
+        await runWithSpawn(prismaScripts.reset(project, unknownOptions), project, argv);
+      }
+    }
   },
 };
 

packages/wb/src/scripts/prismaScripts.ts

@@ -23,8 +23,9 @@ class PrismaScripts {
 
   deployForce(project: Project): string {
     const dirPath = getDatabaseDirPath(project);
+    // `prisma migrate reset` sometimes fails if the existing database schema, so we remove it first.
     // Don't skip "migrate deploy" because restored database may be older than the current schema.
-    return `PRISMA migrate reset --force --skip-seed && rm -Rf ${dirPath}/prod.sqlite3*
+    return `rm -Rf ${dirPath}/prod.sqlite3*; PRISMA migrate reset --force --skip-seed && rm -Rf ${dirPath}/prod.sqlite3*
       && litestream restore -config litestream.yml -o ${dirPath}/prod.sqlite3 ${dirPath}/prod.sqlite3 && ls -ahl ${dirPath}/prod.sqlite3 && ALLOW_TO_SKIP_SEED=0 PRISMA migrate deploy`;
   }
 
@@ -43,18 +44,16 @@ class PrismaScripts {
 
   reset(project: Project, additionalOptions = ''): string {
     // cf. https://www.prisma.io/docs/guides/database/seed-database#integrated-seeding-with-prisma-migrate
-    const resetOptions = additionalOptions.trim();
-    const baseReset = `PRISMA migrate reset --force ${resetOptions}`;
-    const resetCommand = project.packageJson.dependencies?.blitz ? `${baseReset} && ${this.seed(project)}` : baseReset;
-    const resetCommandForTest = project.packageJson.dependencies?.blitz
-      ? String.raw`find db \( -name "test.db*" -o -name "test.sqlite*" \) -delete`
-      : String.raw`find prisma \( -name "test.db*" -o -name "test.sqlite*" \) -delete`;
-    return `${resetCommand} && ${resetCommandForTest}`;
+    if (project.packageJson.dependencies?.blitz) {
+      // Blitz does not trigger seed automatically, so we need to run it manually.
+      return `PRISMA migrate reset --force ${additionalOptions} && ${this.seed(project)}`;
+    }
+    return `PRISMA migrate reset --force ${additionalOptions}`;
   }
 
   restore(project: Project, outputPath: string): string {
     const dirPath = getDatabaseDirPath(project);
-    return `${this.removeSqliteArtifacts(outputPath)}; litestream restore -config litestream.yml -o ${outputPath} ${dirPath}/prod.sqlite3`;
+    return `rm -Rf ${outputPath}*; litestream restore -config litestream.yml -o ${outputPath} ${dirPath}/prod.sqlite3`;
   }
 
   seed(project: Project, scriptPath?: string): string {
@@ -117,11 +116,6 @@ const prisma = new PrismaClient();
     }
     return `${prefix}PRISMA studio ${additionalOptions}`;
   }
-
-  private removeSqliteArtifacts(sqlitePath: string): string {
-    // Litestream requires removing WAL/SHM and Litestream sidecar files when recreating databases.
-    return `rm -Rf ${sqlitePath} ${sqlitePath}-shm ${sqlitePath}-wal ${sqlitePath}-litestream`;
-  }
 }
 
 function getDatabaseDirPath(project: Project): string {