Updates

src/lib/db-handler.ts

@@ -1,9 +1,14 @@
-import type { DBInsertReturn, DBResponseObject } from "../types";
-import MariaDBClient from "./mariadb";
+import grabMariaDBClient from "../functions/grab-mariadb-client";
+import type {
+    BunMariaDBConfig,
+    DBInsertReturn,
+    DBResponseObject,
+} from "../types";
 
 type Param = {
     query: string;
     values?: any[];
+    config?: BunMariaDBConfig;
 };
 
 /**
@@ -12,9 +17,17 @@ type Param = {
  */
 export default async function dbHandler<
     T extends { [k: string]: any } = { [k: string]: any },
->({ query, values }: Param): Promise<DBResponseObject> {
+>({ query, values, config }: Param): Promise<DBResponseObject<T>> {
     try {
-        const res = await MariaDBClient.unsafe(query, values);
+        const CLIENT = config
+            ? grabMariaDBClient({ config })
+            : global.MARIADB_CLIENT;
+
+        if (!CLIENT) {
+            throw new Error(`Couldn't grab MariaDB Client.`);
+        }
+
+        const res = await CLIENT.unsafe(query, values);
 
         const count = res.count;