Shasha/resources/structures.js

'use strict';

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

Structures.extend("Guild", g => {
    return class Guild extends g {
        constructor(client, data) {
            super(client, data);
            this.dbLoaded = false;
        }

        async dbLoad() {
            return 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 || {};
                this.quoteOTD = r?.settings?.quoteOTD || {};
                this.eventChannels = r?.settings?.eventChannels || {};

                return this.dbLoaded = true;
            });
        }

        /**
         * 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;
                            }
                        }
                    }
                }
                return found;
            } catch (e) { }
        }

        async setDescription(set) {
            return database.collection("Guild").updateOne({document: this.id}, { $set: { description: set }, $setOnInsert: { document: this.id }}, {upsert: true}, (e, r) => {
                if (e) return errLog(e, null, this.client);
                this.description = set;
                return true;
            });
        }

        async addInfraction(add) {
            try {
                const r = await database.collection("Guild").findOne({ document: this.id });
                this.infractions = r?.moderation?.infractions;
                return database.collection("Guild").updateOne({document: this.id}, {$push:{"moderation.infractions":add}}, (e) => {
                    if (e) return errLog(e, null, this.client);
                    this.infractions.push(add);
                    return true;
                });
            } catch (e) { }
        }

        async setQuoteOTD(set) {
            return database.collection("Guild").updateOne({document: this.id}, {$set: {"settings.quoteOTD": set}, $setOnInsert: { document: this.id }}, {upsert: true}, (e) => {
                if (e) return errLog(e, null, this.client);
                this.quoteOTD = set;
                return true;
            });
        }

        async setEventChannels(set) {
            return database.collection("Guild").updateOne({document: this.id}, {$set: {"settings.eventChannels": set}, $setOnInsert: { document: this.id }}, {upsert: true}, (e) => {
                if (e) return errLog(e, null, this.client);
                this.eventChannels = set;
                return true;
            });
        }

        async setDefaultEmbed(set) {
            return database.collection("Guild").updateOne({document: this.id}, {$set:{"settings.defaultEmbed": set}, $setOnInsert: { document: this.id }}, {upsert: true}, (e) => {
                if (e) return errLog(e, null, this.client);
                this.defaultEmbed = set;
                return true;
            });
        }

        async setModerationSettings(set) {
            return database.collection("Guild").updateOne({document:this.id}, {$set:{"moderation.settings": set}, $setOnInsert: { document: this.id }}, {upsert: true}, (e) => {
                if (e) return errLog(e, null, this.client);
                this.moderation = set;
                return true;
            });
        }
    }
});

Structures.extend("User", u => {
    return class User extends u {
        constructor(client, data) {
            super(client, data);
            this.dbLoaded = false;
            this.cutie = true;
        }

        async dbLoad() {
            return database.collection("User").findOne({document: this.id}).then((r, e) => {
                if (e) return errLog(e, null, this.client);
                this.defaultEmbed = r?.settings?.defaultEmbed || {};
                this.cachedAvatarURL = this.displayAvatarURL({format: "png", size: 4096, dynamic: true});
                this.interactions = r?.interactions || {};
                this.description = r?.description;
                return this.dbLoaded = true;
            });
        }

        async setInteractions(count) {
            return database.collection("User").updateOne({document: this.id}, { $set: { interactions: count }, $setOnInsert: { document: this.id }}, {upsert: true}, (e, r) => {
                if (e) return errLog(e, null, this.client);
                this.interactions = count;
                return true;
            });
        }

        async setDescription(set) {
            return database.collection("User").updateOne({document: this.id}, { $set: { description: set }, $setOnInsert: { document: this.id }}, {upsert: true}, (e, r) => {
                if (e) return errLog(e, null, this.client);
                this.description = set;
                return true;
            });
        }

        async setDefaultEmbed(set) {
            return database.collection("User").updateOne({document: this.id}, {$set:{"settings.defaultEmbed": set}, $setOnInsert: { document: this.id }}, {upsert: true}, (e) => {
                if (e) return errLog(e, null, this.client);
                this.defaultEmbed = set;
                return true;
            });
        }
    }
});