Yumi/src/utils/DiscordModule.ts

import { Guild, Interaction, Message, GuildMember, CommandInteraction, ButtonInteraction, TextBasedChannel, MessageComponentInteraction, User, SelectMenuInteraction } from "discord.js";

export default class DiscordModule {

    public id?: string;
    public commands?: (string[] | null) = null;
    public commandInteractionName?: (string | null) = null;

    constructor({ id, command, commandInteractionName }: { 
        id?: string,
        command?: (string[] | null),
        commandInteractionName?: (string | null)
    } = {}) {
            this.id = id;
            this.commands = command;
            this.commandInteractionName = commandInteractionName;
    }

    Init(): (void | Promise<void | any>) {}
    Ready(): (void | Promise<void | any>) {}

    GuildOnCommand(command: string, args: any, message: Message): (void | Promise<void | any>) {}
    GuildOnModuleCommand(args: any, message: Message): (void | Promise<void | any>) {}

    GuildInteractionCreate(interaction: Interaction): (void | Promise<void | any>) {}
    GuildModuleInteractionCreate(interaction: Interaction): (void | Promise<void | any>) {}

    GuildCommandInteractionCreate(interaction: CommandInteraction): (void | Promise<void | any>) {}
    GuildModuleCommandInteractionCreate(interaction: CommandInteraction): (void | Promise<void | any>) {}

    GuildSelectMenuInteractionCreate(interaction: SelectMenuInteraction): (void | Promise<void | any>) {}

    GuildButtonInteractionCreate(interaction: ButtonInteraction): (void | Promise<void | any>) {}

    GuildCreate(guild: Guild): (void | Promise<void | any>) {}
    GuildMemberAdd(member: GuildMember): (void | Promise<void | any>) {}
    GuildMessageCreate(message: Message): (void | Promise<void | any>) {}
}

export class HybridInteractionMessage {
    public data: Interaction | Message;
    constructor (data: Interaction | Message) {
        this.data = data;
    }

    public isInteraction(): boolean {
        return this.data instanceof Interaction;
    }

    public isSlashCommand(): boolean {
        return this.data instanceof CommandInteraction && this.data.isCommand();
    }

    public isButton(): boolean {
        return this.data instanceof ButtonInteraction && this.data.isButton();
    }
    
    public isSelectMenu(): boolean {
        return this.data instanceof SelectMenuInteraction && this.data.isSelectMenu();
    }

    public isMessage(): boolean {
        return this.data instanceof Message;
    }

    public getChannel(): (TextBasedChannel | null) {
        return this.data.channel;
    }

    public getGuild(): (Guild | null) {
        return this.data.guild;
    }

    public getUser(): (User | null) {
        return (this.isInteraction()) ? this.getInteraction().user : this.getMessage().author
    }

    public getMember(): (GuildMember | null) {
        return (this.data.member as GuildMember);
    }

    public getInteraction(): Interaction {
        if(this.isMessage())
            throw new Error("Unable to cast interaction to message");

        return this.data as Interaction;
    }

    public getSlashCommand(): CommandInteraction {
        if(this.isInteraction() && !(this.data as CommandInteraction))
            throw new Error("Unable to cast to MessageComponentInteraction");

        return this.data as CommandInteraction;
    }

    public getSelectMenu(): SelectMenuInteraction {
        if(this.isInteraction() && !(this.data as SelectMenuInteraction))
            throw new Error("Unable to cast to SelectMenuInteraction");

        return this.data as SelectMenuInteraction;
    }

    public getMessageComponentInteraction(): MessageComponentInteraction {
        if(this.isInteraction() && !(this.data as MessageComponentInteraction))
            throw new Error("Unable to cast to MessageComponentInteraction");

        return this.data as MessageComponentInteraction;
    }

    public getMessage(): Message {
        if(this.isInteraction())
            throw new Error("Unable to cast message to interaction");

        return this.data as Message;
    }

    public getRaw(): (Interaction | Message) {
        return this.data;
    }
}