Shasha/resources/structures.js

'use strict';

const { Structures } = require("discord.js"),
{ database } = require("../database/mongo");
const { errLog } = require("./functions");

Structures.extend("Guild", g => {
    return class Guild extends g {
        constructor(client, data) {
            super(client, data);
            this.dbLoaded = false;
        }
        async dbLoad() {
            const ret = await database.collection("Guild").findOne({document: this.id}).then((r, j) => {
                if (j) return errLog(j, null, this.client);
                this.infractions = r?.moderation?.infractions;
                this.moderation = r?.moderation?.settings;
                this.defaultEmbed = r?.settings?.defaultEmbed;
                return this.dbLoaded = true;
            });
            return ret;
        }
        /**
         * Get user infractions
         * @param {String} get - User ID
         * @returns {Promise<Object[]>} Infractions object
         */
        async getInfractions(get) {
            try {
                const r = await database.collection("Guild").findOne({ document: this.id });
                this.infractions = r?.moderation?.infractions;
                let found = [];
                if (this.infractions.length > 0) {
                    for (const inf of this.infractions) {
                        for (const user of inf.by) {
                            if (user.id === get) {
                                found.push(inf);
                                break;
                            }
                        }
                    }
                    if (found.length > 0) {
                        return found;
                    }
                }
            } catch (e) { }
        }
        setDefaultEmbed(set) {
            const ret = database.collection("Guild").updateOne({document: this.id}, {$set:{"settings.defaultEmbed": set}}, {upsert: true}, (e) => {
                if (e) return errLog(e, null, this.client);
                this.defaultEmbed = set;
                return true;
            });
            return ret;
        }
        setModerationSettings(set) {
            const ret = database.collection("Guild").updateOne({document:this.id}, {$set:{"moderation.settings": set}}, {upsert: true}, (e) => {
                if (e) return errLog(e, null, this.client);
                this.moderation = set;
                return true;
            });
            return ret;
        }
    }
});

Structures.extend("User", u => {
    return class User extends u {
        constructor(client, data) {
            super(client, data);
            this.dbLoaded = false;
        }
        async dbLoad() {
            const ret = await database.collection("User").findOne({document: this.id}).then((r, j) => {
                if (j) return errLog(j, null, this.client);
                this.defaultEmbed = r?.settings?.defaultEmbed;
                return this.dbLoaded = true;
            });
            return ret;
        }
        setDefaultEmbed(set) {
            const ret = database.collection("User").updateOne({document: this.id}, {$set:{"settings.defaultEmbed": set}}, {upsert: true}, (e) => {
                if (e) return errLog(e, null, this.client);
                this.defaultEmbed = set;
                return true;
            });
            return ret;
        }
    }
});