refactor: ♻️ Rewrite build system to fit the monorepo architecture

packages/kit/tables/db.ts

@@ -0,0 +1,80 @@
+import { join } from "node:path";
+import { config } from "@versia-server/config";
+import { databaseLogger } from "@versia-server/logging";
+import { SQL } from "bun";
+import chalk from "chalk";
+import { type BunSQLDatabase, drizzle } from "drizzle-orm/bun-sql";
+import { withReplicas } from "drizzle-orm/pg-core";
+import { migrate } from "drizzle-orm/postgres-js/migrator";
+import * as schema from "./schema.ts";
+
+const primaryDb = new SQL({
+    host: config.postgres.host,
+    port: config.postgres.port,
+    user: config.postgres.username,
+    password: config.postgres.password,
+    database: config.postgres.database,
+});
+
+const replicas = config.postgres.replicas.map(
+    (replica) =>
+        new SQL({
+            host: replica.host,
+            port: replica.port,
+            user: replica.username,
+            password: replica.password,
+            database: replica.database,
+        }),
+);
+
+export const db =
+    (replicas.length ?? 0) > 0
+        ? withReplicas(
+              drizzle(primaryDb, { schema }),
+              replicas.map((r) => drizzle(r, { schema })) as [
+                  // biome-ignore lint/style/useNamingConvention: Required by drizzle-orm
+                  BunSQLDatabase<typeof schema> & { $client: SQL },
+                  // biome-ignore lint/style/useNamingConvention: Required by drizzle-orm
+                  ...(BunSQLDatabase<typeof schema> & { $client: SQL })[],
+              ],
+          )
+        : drizzle(primaryDb, { schema });
+
+export const setupDatabase = async (info = true): Promise<void> => {
+    for (const dbPool of [primaryDb, ...replicas]) {
+        try {
+            await dbPool.connect();
+        } catch (e) {
+            if (
+                (e as Error).message ===
+                "Client has already been connected. You cannot reuse a client."
+            ) {
+                return;
+            }
+
+            databaseLogger.fatal`Failed to connect to database ${chalk.bold(
+                // Index of the database in the array
+                replicas.indexOf(dbPool) === -1
+                    ? "primary"
+                    : `replica-${replicas.indexOf(dbPool)}`,
+            )}. Please check your configuration.`;
+
+            throw e;
+        }
+    }
+
+    // Migrate the database
+    info && databaseLogger.info`Migrating database...`;
+
+    try {
+        await migrate(db, {
+            migrationsFolder: join(import.meta.dir, "migrations"),
+        });
+    } catch (e) {
+        databaseLogger.fatal`Failed to migrate database. Please check your configuration.`;
+
+        throw e;
+    }
+
+    info && databaseLogger.info`Database migrated`;
+};