datasquirel/package-shared/shell/utils/dbHandler.ts

import fs from "fs";
import path from "path";

import mysql from "serverless-mysql";
import grabDbSSL from "../../utils/backend/grabDbSSL";

let connection = mysql({
    config: {
        host: process.env.DSQL_DB_HOST,
        user: process.env.DSQL_DB_USERNAME,
        password: process.env.DSQL_DB_PASSWORD,
        database: process.env.DSQL_DB_NAME,
        charset: "utf8mb4",
        ssl: grabDbSSL(),
    },
});

type Param = {
    query: string;
    values?: string[] | object;
    database?: string;
};

/**
 * # Main DB Handler Function
 */
export default async function dbHandler({
    query,
    values,
    database,
}: Param): Promise<any[] | object | null> {
    /**
     * Switch Database
     *
     * @description If a database is provided, switch to it
     */
    let isDbCorrect = true;

    if (database) {
        connection = mysql({
            config: {
                host: process.env.DSQL_DB_HOST,
                user: process.env.DSQL_DB_USERNAME,
                password: process.env.DSQL_DB_PASSWORD,
                database: database,
                charset: "utf8mb4",
                ssl: grabDbSSL(),
            },
        });
    }

    if (!isDbCorrect) {
        console.log(
            "Shell Db Handler ERROR in switching Database! Operation Failed!"
        );
        return null;
    }

    /**
     * Declare variables
     *
     * @description Declare "results" variable
     */
    let results;

    /**
     * Fetch from db
     *
     * @description Fetch data from db if no cache
     */
    try {
        if (query && values) {
            results = await connection.query(query, values);
        } else {
            results = await connection.query(query);
        }

        /** ********************* Clean up */
        await connection.end();
    } catch (/** @type {any} */ error: any) {
        if (process.env.FIRST_RUN) {
            return null;
        }

        console.log("ERROR in dbHandler =>", error.message);
        console.log(error);
        console.log(connection.config());

        fs.appendFileSync(
            path.resolve(__dirname, "../.tmp/dbErrorLogs.txt"),
            JSON.stringify(error, null, 4) + "\n" + Date() + "\n\n\n",
            "utf8"
        );
        results = null;
    }

    /**
     * Return results
     *
     * @description Return results add to cache if "req" param is passed
     */
    if (results) {
        return JSON.parse(JSON.stringify(results));
    } else {
        return null;
    }
}