feat!: reconstruct prisma commands for litestream

packages/wb/src/commands/prisma.ts

@@ -19,11 +19,12 @@ export const prismaCommand: CommandModule = {
   describe: 'Run prisma commands',
   builder: (yargs) => {
     return yargs
+      .command(cleanUpLitestreamCommand)
+      .command(createLitestreamConfigCommand)
       .command(deployCommand)
       .command(deployForceCommand)
-      .command(createLitestreamConfigCommand)
       .command(listBackupsCommand)
-      .command(litestreamCommand)
+      .command(setUpDBForLitestreamCommand)
       .command(migrateCommand)
       .command(migrateDevCommand)
       .command(resetCommand)
@@ -39,6 +40,30 @@ export const prismaCommand: CommandModule = {
   },
 };
 
+const cleanUpLitestreamCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>> = {
+  command: 'cleanup-litestream',
+  describe: 'Clean up temporal Litestream files',
+  builder,
+  async handler(argv) {
+    const allProjects = await findPrismaProjects(argv);
+    for (const project of prepareForRunningCommand('prisma cleanup-litestream', allProjects)) {
+      await runWithSpawn(prismaScripts.cleanUpLitestream(project), project, argv);
+    }
+  },
+};
+
+const createLitestreamConfigCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>> = {
+  command: 'create-litestream-config',
+  describe: 'Create Litestream configuration file',
+  builder,
+  async handler(argv) {
+    const allProjects = await findPrismaProjects(argv);
+    for (const project of prepareForRunningCommand('prisma create-litestream-config', allProjects)) {
+      createLitestreamConfig(project);
+    }
+  },
+};
+
 const deployCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>> = {
   command: 'deploy',
   describe: 'Apply migration to DB without initializing it',
@@ -64,18 +89,6 @@ const deployForceCommand: CommandModule<unknown, InferredOptionTypes<typeof buil
   },
 };
 
-const createLitestreamConfigCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>> = {
-  command: 'create-litestream-config',
-  describe: 'Create Litestream configuration file',
-  builder,
-  async handler(argv) {
-    const allProjects = await findPrismaProjects(argv);
-    for (const project of prepareForRunningCommand('prisma create-litestream-config', allProjects)) {
-      createLitestreamConfig(project);
-    }
-  },
-};
-
 const listBackupsCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>> = {
   command: 'list-backups',
   describe: 'List Litestream backups',
@@ -88,18 +101,6 @@ const listBackupsCommand: CommandModule<unknown, InferredOptionTypes<typeof buil
   },
 };
 
-const litestreamCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>> = {
-  command: 'litestream',
-  describe: 'Setup DB for Litestream',
-  builder,
-  async handler(argv) {
-    const allProjects = await findPrismaProjects(argv);
-    for (const project of prepareForRunningCommand('prisma litestream', allProjects)) {
-      await runWithSpawn(prismaScripts.litestream(project), project, argv);
-    }
-  },
-};
-
 const migrateCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>> = {
   command: 'migrate',
   describe: 'Apply migration to DB with initializing it',
@@ -182,6 +183,18 @@ const seedCommand: CommandModule<unknown, InferredOptionTypes<typeof seedBuilder
   },
 };
 
+const setUpDBForLitestreamCommand: CommandModule<unknown, InferredOptionTypes<typeof builder>> = {
+  command: 'setup-db-for-litestream',
+  describe: 'Setup DB for Litestream',
+  builder,
+  async handler(argv) {
+    const allProjects = await findPrismaProjects(argv);
+    for (const project of prepareForRunningCommand('prisma setup-db-for-litestream', allProjects)) {
+      await runWithSpawn(prismaScripts.setUpDBForLitestream(project), project, argv);
+    }
+  },
+};
+
 const studioBuilder = {
   ...builder,
   'db-url-or-path': {

packages/wb/src/scripts/prismaScripts.ts

@@ -10,6 +10,12 @@ import { runtimeWithArgs } from '../utils/runtime.js';
  * and `YARN zzz` is replaced with `yarn zzz` or `node_modules/.bin/zzz`.
  */
 class PrismaScripts {
+  cleanUpLitestream(project: Project): string {
+    const dirPath = getDatabaseDirPath(project);
+    // Cleanup existing artifacts to avoid issues with Litestream replication at first.
+    return `rm -Rf ${dirPath}/prod.sqlite3-*; rm -Rf ${dirPath}/prod.sqlite3.*; rm -Rf ${dirPath}/.prod.sqlite3*`;
+  }
+
   deploy(_: Project, additionalOptions = ''): string {
     return `PRISMA migrate deploy ${additionalOptions}`;
   }
@@ -26,29 +32,6 @@ class PrismaScripts {
     return `litestream ltx -config litestream.yml ${dirPath}/prod.sqlite3`;
   }
 
-  litestream(project: Project): string {
-    const dirPath = getDatabaseDirPath(project);
-    // Cleanup existing artifacts to avoid issues with Litestream replication at first.
-    // cf. https://litestream.io/tips/
-    return `rm -Rf ${dirPath}/prod.sqlite3-*; rm -Rf ${dirPath}/prod.sqlite3.*; rm -Rf ${dirPath}/.prod.sqlite3*; ${runtimeWithArgs} -e '
-const { PrismaClient } = require("@prisma/client");
-const prisma = new PrismaClient();
-(async () => {
-  try {
-    await prisma.$queryRawUnsafe("PRAGMA busy_timeout = 5000");
-    await prisma.$queryRawUnsafe("PRAGMA journal_mode = WAL");
-    await prisma.$queryRawUnsafe("PRAGMA synchronous = NORMAL");
-    await prisma.$queryRawUnsafe("PRAGMA wal_autocheckpoint = 0");
-  } catch (error) {
-    console.error("Failed due to:", error);
-    process.exit(1);
-  } finally {
-    await prisma.$disconnect();
-  }
-})();
-'`;
-  }
-
   migrate(project: Project, additionalOptions = ''): string {
     return `PRISMA migrate deploy ${additionalOptions} && PRISMA generate && ${this.seed(project)}`;
   }
@@ -80,6 +63,27 @@ const prisma = new PrismaClient();
     return `if [ -e "prisma/seeds.ts" ]; then BUN build-ts run prisma/seeds.ts; fi`;
   }
 
+  setUpDBForLitestream(_: Project): string {
+    // cf. https://litestream.io/tips/
+    return `${runtimeWithArgs} -e '
+const { PrismaClient } = require("@prisma/client");
+const prisma = new PrismaClient();
+(async () => {
+  try {
+    await prisma.$queryRawUnsafe("PRAGMA busy_timeout = 5000");
+    await prisma.$queryRawUnsafe("PRAGMA journal_mode = WAL");
+    await prisma.$queryRawUnsafe("PRAGMA synchronous = NORMAL");
+    await prisma.$queryRawUnsafe("PRAGMA wal_autocheckpoint = 0");
+  } catch (error) {
+    console.error("Failed due to:", error);
+    process.exit(1);
+  } finally {
+    await prisma.$disconnect();
+  }
+})();
+'`;
+  }
+
   studio(project: Project, dbUrlOrPath?: string, additionalOptions = ''): string {
     const FILE_SCHEMA = 'file:';
     let prefix = '';