datasquirel/package-shared/functions/backend/addUsersTableToDb.js

// @ts-check

const fs = require("fs");
const path = require("path");
const { execSync } = require("child_process");
const serverError = require("./serverError");
const DB_HANDLER = require("../../utils/backend/global-db/DB_HANDLER");
const { default: grabUserSchemaData } = require("./grabUserSchemaData");
const { default: setUserSchemaData } = require("./setUserSchemaData");
const addDbEntry = require("./db/addDbEntry");
const createDbFromSchema = require("../../shell/createDbFromSchema");
const LOCAL_DB_HANDLER = require("../../utils/backend/global-db/LOCAL_DB_HANDLER");

/**
 * # Add User Table to Database
 *
 * @param {object} params
 * @param {number} params.userId - user id
 * @param {string} params.database
 * @param {boolean} [params.useLocal]
 *
 * @returns {Promise<any>} new user auth object payload
 */
module.exports = async function addUsersTableToDb({
    userId,
    database,
    useLocal,
}) {
    /**
     * Initialize
     *
     * @description Initialize
     */
    const dbFullName = `datasquirel_user_${userId}_${database}`;
    /** @type {import("../../types").DSQL_TableSchemaType} */
    const userPreset = require("../../data/presets/users.json");

    try {
        /**
         * Fetch user
         *
         * @description Fetch user from db
         */
        const userSchemaData = grabUserSchemaData({ userId });
        if (!userSchemaData) throw new Error("User schema data not found!");

        let targetDatabase = userSchemaData.filter(
            (db) => db.dbSlug === database
        )[0];

        let existingTableIndex;
        // @ts-ignore
        let existingTable = targetDatabase.tables.filter((table, index) => {
            if (table.tableName === "users") {
                existingTableIndex = index;
                return true;
            }
        });

        if (existingTable && existingTable[0] && existingTableIndex) {
            targetDatabase.tables[existingTableIndex] = userPreset;
        } else {
            targetDatabase.tables.push(userPreset);
        }

        setUserSchemaData({ schemaData: userSchemaData, userId });

        const targetDb = useLocal
            ? await LOCAL_DB_HANDLER(
                  `SELECT id FROM user_databases WHERE user_id=? AND db_slug=?`,
                  [userId, database]
              )
            : await DB_HANDLER(
                  `SELECT id FROM user_databases WHERE user_id=? AND db_slug=?`,
                  [userId, database]
              );

        if (targetDb && targetDb[0]) {
            const newTableEntry = await addDbEntry({
                dbFullName: "datasquirel",
                tableName: "user_database_tables",
                data: {
                    user_id: userId,
                    db_id: targetDb[0].id,
                    db_slug: database,
                    table_name: "Users",
                    table_slug: "users",
                },
                useLocal,
            });
        }

        const dbShellUpdate = await createDbFromSchema({
            userId,
            targetDatabase: dbFullName,
        });

        return `Done!`;
    } catch (/** @type {any} */ error) {
        serverError({
            component: "addUsersTableToDb",
            message: error.message,
            user: { id: userId },
        });
        return error.message;
    }
};