fix: mysql transactions do not work with run property of sqlite (#1974)

* fix: mysql transactions do not work with run property of sqlite

* fix: ci issues

packages/db/driver.ts

@@ -2,6 +2,7 @@ import Database from "better-sqlite3";
 import type { Logger } from "drizzle-orm";
 import type { BetterSQLite3Database } from "drizzle-orm/better-sqlite3";
 import { drizzle as drizzleSqlite } from "drizzle-orm/better-sqlite3";
+import type { MySql2Database } from "drizzle-orm/mysql2";
 import { drizzle as drizzleMysql } from "drizzle-orm/mysql2";
 import mysql from "mysql2";
 
@@ -11,7 +12,8 @@ import { env } from "./env";
 import * as mysqlSchema from "./schema/mysql";
 import * as sqliteSchema from "./schema/sqlite";
 
-type HomarrDatabase = BetterSQLite3Database<typeof sqliteSchema>;
+export type HomarrDatabase = BetterSQLite3Database<typeof sqliteSchema>;
+export type HomarrDatabaseMysql = MySql2Database<typeof mysqlSchema>;
 
 const init = () => {
   // eslint-disable-next-line @typescript-eslint/no-unnecessary-condition

packages/db/index.ts

@@ -7,5 +7,7 @@ export * from "drizzle-orm";
 export const db = database;
 
 export type Database = typeof db;
+export type { HomarrDatabaseMysql } from "./driver";
 
 export { createId } from "@paralleldrive/cuid2";
+export { handleDiffrentDbDriverOperationsAsync as handleTransactionsAsync } from "./transactions";

packages/db/transactions.ts

@@ -0,0 +1,24 @@
+import type { HomarrDatabase, HomarrDatabaseMysql } from "./driver";
+import { env } from "./env";
+import * as mysqlSchema from "./schema/mysql";
+
+type MysqlSchema = typeof mysqlSchema;
+
+interface HandleTransactionInput {
+  handleAsync: (db: HomarrDatabaseMysql, schema: MysqlSchema) => Promise<void>;
+  handleSync: (db: HomarrDatabase) => void;
+}
+
+/**
+ * The below method is mostly used to handle transactions in different database drivers.
+ * As better-sqlite3 transactions have to be synchronous, we have to implement them in a different way.
+ * But it can also generally be used when dealing with different database drivers.
+ */
+export const handleDiffrentDbDriverOperationsAsync = async (db: HomarrDatabase, input: HandleTransactionInput) => {
+  if (env.DB_DRIVER !== "mysql2") {
+    input.handleSync(db);
+    return;
+  }
+
+  await input.handleAsync(db as unknown as HomarrDatabaseMysql, mysqlSchema);
+};