Nucleus/src/utils/database.ts

import Discord from 'discord.js';
import { Collection, MongoClient } from 'mongodb';
import structuredClone from '@ungap/structured-clone';
import config from '../config/main.json' assert {type: 'json'};


const mongoClient = new MongoClient(config.mongoUrl);
await mongoClient.connect()
const database = mongoClient.db("Nucleus");


export const Entry = data => {
    data = data ?? {};
    data.getKey = key => data[key]
    return {
        get(target, prop, receiver) {
            let dataToReturn = data[prop]
            if (dataToReturn === null ) return Reflect.get(target, prop, receiver);
            if (typeof dataToReturn === "object" && !Array.isArray(dataToReturn)) dataToReturn = new Proxy(
                Reflect.get(target, prop, receiver),
                Entry(dataToReturn),
            )
            return dataToReturn ?? Reflect.get(target, prop, receiver);
        }
    }
}


export class Guilds {
    guilds: Collection<GuildConfig>;
    defaultData: GuildConfig;
    async setup() {
        this.guilds = database.collection<GuildConfig>("guilds");
        this.defaultData = (await import("../config/default.json", { assert: { type: "json" }})).default as unknown as GuildConfig;
        return this;
    }

    async read(guild: string) {
        let entry = await this.guilds.findOne({ id: guild });
        return new Proxy(structuredClone(this.defaultData), Entry(entry)) as unknown as GuildConfig
    }

    async write(guild: string, set: object = {}, unset: string[] = []) {
        let uo = {}
        for (let key of unset) {
            uo[key] = "";
        }
        await this.guilds.updateOne({ id: guild }, {
            $unset: uo,
            $set: set
        }, { upsert: true });
    }

    async append(guild: string, key: string, value: any) {
        if (Array.isArray(value)) {
            await this.guilds.updateOne({ id: guild }, {
                $addToSet: { [key]: { $each: value } }
            }, { upsert: true });
        } else {
            await this.guilds.updateOne({ id: guild }, {
                $addToSet: { [key]: value }
            }, { upsert: true });
        }
    }

    async remove(guild: string, key: string, value: any) {
        if (Array.isArray(value)) {
            await this.guilds.updateOne({ id: guild }, {
                $pullAll: { [key]: value }
            }, { upsert: true });
        } else {
            await this.guilds.updateOne({ id: guild }, {
                $pullAll: { [key]: [value] }
            }, { upsert: true });
        }
    }
}


export class History {
    histories: Collection<HistorySchema>;
    defaultData: GuildConfig;

    async setup() {
        this.histories = database.collection<HistorySchema>("history");
        return this;
    }

    async create(type: string, guild: string, user: Discord.User, moderator: Discord.User | null, reason: string | null, before?: null, after?: null, amount?: null) {
        await this.histories.insertOne({
            type: type,
            guild: guild,
            user: user.id,
            moderator: moderator.id,
            reason: reason,
            occurredAt: new Date(),
            before: before,
            after: after,
            amount: amount
        });
    }

    async read(guild: string, user: string, year: number) {
        let entry = (await this.histories.find({
            guild: guild,
            user: user,
            occurredAt: {
                $gte: new Date(year - 1, 11, 31, 23, 59, 59),
                $lt: new Date(year + 1, 0, 1, 0, 0, 0)
            }
        }).toArray()) as HistorySchema[];
        return entry;
    }
}

export class ModNotes {
    modNotes: Collection<ModNoteSchema>;
    defaultData: GuildConfig;

    async setup() {
        this.modNotes = database.collection<ModNoteSchema>("modNotes");
        return this;
    }

    async create(guild: string, user: string, note: string | null) {
        await this.modNotes.updateOne({ guild: guild, user: user }, { $set: { note: note }}, { upsert: true });
    }

    async read(guild: string, user: string) {
        let entry = await this.modNotes.findOne({ guild: guild, user: user });
        return entry?.note ?? null;
    }
}

export class EventSchedulerDatabase {
    events: Collection<EventSchedulerSchema>;
    defaultData: GuildConfig;

    async setup() {
        this.events = database.collection<EventSchedulerSchema>("eventScheduler");
        return this;
    }

    async create(timestamp: Date, data: object) {
        await this.events.insertOne({ timestamp: timestamp, data: data});
    }

    async getNext() {
        let entry = await this.events.findOne({ timestamp: { $lte: new Date() }});
        return entry;
    }

    async remove(timestamp: Date, data: object) {
        await this.events.deleteOne({ timestamp: timestamp, data: data});
    }
}

export interface GuildConfig {
    id: string,
    version: number,
    singleEventNotifications: {
        statsChannelDeleted: boolean
    }
    filters: {
        images: {
            NSFW: boolean,
            size: boolean
        },
        malware: boolean,
        wordFilter: {
            enabled: boolean,
            words: {
                strict: string[],
                loose: string[]
            },
            allowed: {
                users: string[],
                roles: string[],
                channels: string[]
            }
        },
        invite: {
            enabled: boolean,
            allowed: {
                users: string[],
                channels: string[],
                roles: string[]
            }
        },
        pings: {
            mass: number,
            everyone: boolean,
            roles: boolean,
            allowed: {
                roles: string[],
                rolesToMention: string[],
                users: string[],
                channels: string[]
            }
        }
    }
    welcome: {
        enabled: boolean,
        verificationRequired: {
            message: boolean,
            role: string | null
        },
        welcomeRole: string | null,
        channel: string | null,
        message: string | null,
    }
    stats: {
        enabled: boolean,
        channel: string | null,
        text: string | null,
    }[]
    logging: {
        logs: {
            enabled: boolean,
            channel: string | null,
            toLog: string | null,
        },
        staff: {
            channel: string | null,
        }
    }
    verify: {
        enabled: boolean,
        role: string | null,
    }
    tickets: {
        enabled: boolean,
        category: string | null,
        types: string | null,
        customTypes: string[],
        useCustom: boolean,
        supportRole: string | null,
        maxTickets: number
    }
    moderation: {
        mute: {
            timeout: boolean,
            role: string | null,
            text: string | null,
            link: string | null
        },
        kick: {
            text: string | null,
            link: string | null
        },
        ban: {
            text: string | null,
            link: string | null
        },
        softban: {
            text: string | null,
            link: string | null
        },
        warn: {
            text: string | null,
            link: string | null
        },
        role: {
            role: string | null,
        }
    }
    tracks: {
        name: string,
        retainPrevious: boolean,
        nullable: boolean,
        track: string[],
        manageableBy: string[]
    }[]
    roleMenu: {
        enabled: boolean,
        allowWebUI: boolean,
        options: {
            name: string,
            description: string,
            min: number,
            max: number,
            options: {
                name: string,
                description: string | null,
                role: string
            }[]
        }[]
    }
    tags: {}
};

export interface HistorySchema {
    type: string,
    guild: string,
    user: string,
    moderator: string | null,
    reason: string,
    occurredAt: Date,
    before: string | null,
    after: string | null,
    amount: string | null
}

export interface ModNoteSchema {
    guild: string,
    user: string,
    note: string
}

export interface EventSchedulerSchema {
    timestamp: Date,
    data: object
}