chore: add back --db-dir for do-sqlite

Author: raykyri

packages/kettle/services/cli.ts

@@ -62,6 +62,10 @@ async function main() {
             type: "boolean",
             description: "Include verbose logging",
           })
+          .option("db-dir", {
+            type: "string",
+            description: "Directory for SQLite storage (default: random temp directory)",
+          })
       },
       launcherCommand,
     )
@@ -81,6 +85,10 @@ async function main() {
             description: "Port for the worker HTTP server",
             default: 3001,
           })
+          .option("db-dir", {
+            type: "string",
+            description: "Directory for SQLite storage (default: random temp directory)",
+          })
       },
       startWorkerCommand,
     )

packages/kettle/services/launcher.ts

@@ -187,6 +187,7 @@ export interface LauncherArgs {
   manifest: string
   port?: number
   verbose?: boolean
+  dbDir?: string
 }
 
 export async function launcherCommand(argv: LauncherArgs) {
@@ -250,6 +251,7 @@ export async function launcherCommand(argv: LauncherArgs) {
     workerPort: argv.port ?? 3001,
     quoteServicePort: await findFreePort(),
     bundleDir: bundleDir,
+    dbDir: argv.dbDir,
   })
 
   // Handle graceful shutdown

packages/kettle/services/startWorker.ts

@@ -22,6 +22,7 @@ export interface WorkerConfig {
   workerPort: number
   quoteServicePort: number
   bundleDir: string
+  dbDir?: string
 }
 
 export interface WorkerResult {
@@ -34,7 +35,7 @@ export interface WorkerResult {
 export async function startWorker(
   options: WorkerConfig,
 ): Promise<WorkerResult> {
-  const { workerPort, quoteServicePort, bundleDir } = options
+  const { workerPort, quoteServicePort, bundleDir, dbDir } = options
   const quoteServiceUrl = `http://127.0.0.1:${quoteServicePort}`
 
   const WORKER_JS = "worker.js"
@@ -54,8 +55,16 @@ export async function startWorker(
     mkdirSync(staticDir, { recursive: true })
   }
 
-  // Create temp directory for Durable Object SQLite storage
-  const doStorageDir = mkdtempSync(join(tmpdir(), "kettle-do-storage-"))
+  // Use provided db directory or create temp directory for Durable Object SQLite storage
+  let doStorageDir: string
+  if (dbDir) {
+    doStorageDir = dbDir
+    if (!existsSync(doStorageDir)) {
+      mkdirSync(doStorageDir, { recursive: true })
+    }
+  } else {
+    doStorageDir = mkdtempSync(join(tmpdir(), "kettle-do-storage-"))
+  }
 
   // Start quote service
   const quoteService = startQuoteService(quoteServicePort)
@@ -297,6 +306,7 @@ const config :Workerd.Config = (
 export interface StartWorkerArgs {
   file?: string
   port?: number
+  dbDir?: string
 }
 
 export async function startWorkerCommand(argv: StartWorkerArgs) {
@@ -323,6 +333,7 @@ export async function startWorkerCommand(argv: StartWorkerArgs) {
     workerPort: port,
     quoteServicePort: await findFreePort(),
     bundleDir: join(projectDir, "dist"),
+    dbDir: argv.dbDir,
   })
 
   process.on("SIGINT", () => stop())