FINISHED

src/db/index.ts

@@ -2,6 +2,7 @@ import { drizzle } from "drizzle-orm/libsql";
 import { createClient } from "@libsql/client";
 import * as schema from "./schema";
 import path from "path";
+import fs from "fs";
 
 // Define the database type based on the schema
 type Database = ReturnType<typeof drizzle<typeof schema>>;
@@ -10,17 +11,16 @@ let _db: Database | null = null;
 
 function initDb(): Database {
   if (!_db) {
-    let url = process.env.DATABASE_URL;
-    if (!url) {
-      url = `file:${path.resolve(process.cwd(), "chronus.db")}`;
-    } else if (
-      !url.startsWith("file:") &&
-      !url.startsWith("libsql:") &&
-      !url.startsWith("http:") &&
-      !url.startsWith("https:")
-    ) {
-      url = `file:${url}`;
+    const envRootDir = process.env.ROOT_DIR
+      ? process.env.ROOT_DIR
+      : import.meta.env.ROOT_DIR;
+    const rootDir = envRootDir || process.cwd();
+
+    if (envRootDir && !fs.existsSync(rootDir)) {
+      fs.mkdirSync(rootDir, { recursive: true });
     }
+
+    const url = `file:${path.join(rootDir, "chronus.db")}`;
     const authToken = process.env.DATABASE_AUTH_TOKEN;
 
     const client = createClient({