refactor: extract migration from instrumentation

Author: omermecitoglu

src/database/migrate.ts

@@ -0,0 +1,44 @@
+import getPackageMetadata from "@omer-x/package-metadata";
+import { drizzle } from "drizzle-orm/node-postgres";
+import { migrate } from "drizzle-orm/node-postgres/migrator";
+import { Client } from "pg";
+
+async function migrateAppDatabase() {
+  const packageMetadata = getPackageMetadata();
+
+  const dbCredentials = {
+    host: process.env.POSTGRES_HOST || "localhost",
+    port: parseInt(process.env.POSTGRES_PORT || "5432"),
+    user: process.env.POSTGRES_USER || "postgres",
+    password: process.env.POSTGRES_PASSWORD || "example",
+    database: process.env.POSTGRES_DATABASE || packageMetadata.moduleName,
+  };
+
+  const connect = async () => {
+    const creator = new Client({ ...dbCredentials, database: "postgres" });
+    try {
+      await creator.connect();
+      try {
+        await creator.query(`CREATE DATABASE "${dbCredentials.database}"`);
+      } catch {
+        // do nothing
+      } finally {
+        await creator.end();
+      }
+    } catch {
+      await creator.end();
+      await new Promise(resolve => setTimeout(resolve, 1000));
+      await connect();
+    }
+  };
+  if (dbCredentials.database === "verceldb") return;
+  await connect();
+
+  const client = new Client(dbCredentials);
+  await client.connect();
+  const db = drizzle(client);
+  await migrate(db, { migrationsFolder: "./drizzle" });
+  await client.end();
+}
+
+export default migrateAppDatabase;

src/instrumentation.ts

@@ -1,56 +1,10 @@
 export async function register() {
   if (process.env.NEXT_RUNTIME === "nodejs") {
-    const { drizzle } = await import("drizzle-orm/node-postgres");
-    const { migrate } = await import("drizzle-orm/node-postgres/migrator");
-    const { Client } = await import("pg");
-    const path = await import("path");
-    const fs = await import("fs");
+    process.title = "user-service";
 
-    const getServiceName = () => {
-      try {
-        const filePath = path.resolve(process.cwd(), "package.json");
-        const content = fs.readFileSync(filePath, "utf-8");
-        const data = JSON.parse(content);
-        const packageName = data.name as string;
-        const appName = packageName.split("/").pop();
-        if (!appName) throw new Error(`Invalid package name (${packageName})`);
-        return appName;
-      } catch {
-        return "unknown-service";
-      }
-    };
-
-    const dbCredentials = {
-      host: process.env.DATABASE_HOST || "localhost",
-      port: parseInt(process.env.DATABASE_PORT || "5432"),
-      user: process.env.DATABASE_USER || "postgres",
-      password: process.env.DATABASE_PASS || "example",
-      database: process.env.DATABASE_NAME || getServiceName(),
-    };
-
-    const connect = async () => {
-      const creator = new Client({ ...dbCredentials, database: "postgres" });
-      try {
-        await creator.connect();
-        try {
-          await creator.query(`CREATE DATABASE "${dbCredentials.database}"`);
-        } catch {
-          // do nothing
-        } finally {
-          await creator.end();
-        }
-      } catch {
-        await creator.end();
-        await new Promise(resolve => setTimeout(resolve, 1000));
-        await connect();
-      }
-    };
-    await connect();
-
-    const client = new Client(dbCredentials);
-    await client.connect();
-    const db = drizzle(client);
-    await migrate(db, { migrationsFolder: "./drizzle" });
-    await client.end();
+    if (process.env.VERCEL !== "1") {
+      const { default: migrate } = await import("./database/migrate");
+      await migrate();
+    }
   }
 }