@mymj/midjourney/libs/command.js

Node.js client for the unofficial MidJourney API.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.Command = exports.Commands = void 0;
exports.Commands = [
    "ask",
    "blend",
    "describe",
    "fast",
    "turbo",
    "help",
    "imagine",
    "info",
    "prefer",
    "private",
    "public",
    "relax",
    "settings",
    "show",
    "stealth",
    "shorten",
    "subscribe",
];
function getCommandName(name) {
    for (const command of exports.Commands) {
        if (command === name) {
            return command;
        }
    }
}
class Command {
    config;
    constructor(config) {
        this.config = config;
    }
    cache = {};
    async cacheCommand(name) {
        if (this.cache[name] !== undefined) {
            return this.cache[name];
        }
        if (this.config.ChannelId) {
            const command = await this.getCommand(name);
            this.cache[name] = command;
        }
        if (this.cache[name] !== undefined) {
            return this.cache[name];
        }
        return await this.getCommand(name);
    }
    async userCommand() {
        let url = `${this.config.DiscordBaseUrl}/api/v9/users/@me/application-command-index`;
        const response = await this.config.fetch(url, {
            headers: { authorization: this.config.SalaiToken },
        });
        if (response.status > 300) {
            console.error(response);
            throw new Error(`Failed to get application_commands of user for command`);
        }
        const data = await response.json();
        if (data?.application_commands) {
            data.application_commands.forEach((command) => {
                const name = getCommandName(command.name);
                if (name) {
                    this.cache[name] = command;
                }
            });
        }
    }
    async allCommand() {
        try {
            let url;
            if (!this.config.ServerId) {
                url = `${this.config.DiscordBaseUrl}/api/v9/channels/${this.config.ChannelId}/application-command-index`;
            }
            else {
                url = `${this.config.DiscordBaseUrl}/api/v9/guilds/${this.config.ServerId}/application-command-index`;
            }
            const response = await this.config.fetch(url, {
                headers: { authorization: this.config.SalaiToken },
            });
            if (response.status > 300) {
                console.error(response);
                throw new Error(`Failed to get application_commands for command`);
            }
            const data = await response.json();
            if (data?.application_commands) {
                data.application_commands.forEach((command) => {
                    const name = getCommandName(command.name);
                    if (name) {
                        this.cache[name] = command;
                    }
                });
            }
        }
        catch (error) {
            console.error('Fail to get command from channel or server');
        }
        if (!this.cache['info']) {
            await this.userCommand();
        }
    }
    async getCommand(name) {
        if (this.cache[name]) {
            return this.cache[name];
        }
        await this.allCommand();
        if (this.cache[name]) {
            return this.cache[name];
        }
        throw new Error(`Missing permission, please make sure you have added MJ Bot.`);
    }
    async imaginePayload(prompt, nonce) {
        const data = await this.commandData("imagine", [
            {
                type: 3,
                name: "prompt",
                value: prompt,
            },
        ]);
        return this.data2Paylod(data, nonce);
    }
    async PreferPayload(nonce) {
        const data = await this.commandData("prefer", [
            {
                type: 1,
                name: "remix",
                options: [],
            },
        ]);
        return this.data2Paylod(data, nonce);
    }
    async shortenPayload(prompt, nonce) {
        const data = await this.commandData("shorten", [
            {
                type: 3,
                name: "prompt",
                value: prompt,
            },
        ]);
        return this.data2Paylod(data, nonce);
    }
    async describePayload(image, nonce) {
        const data = await this.commandData("describe", [
            {
                type: 11,
                name: "image",
                value: image.id,
            },
        ], [
            {
                id: image.id,
                filename: image.filename,
                uploaded_filename: image.upload_filename,
            },
        ]);
        return this.data2Paylod(data, nonce);
    }
    async commandPayload(name, nonce) {
        const data = await this.commandData(name);
        return this.data2Paylod(data, nonce);
    }
    async commandData(name, options = [], attachments = []) {
        const command = await this.getCommand(name);
        const data = {
            version: command.version,
            id: command.id,
            name: command.name,
            type: command.type,
            options,
            application_command: command,
            attachments,
        };
        return data;
    }
    //TODO data type
    data2Paylod(data, nonce) {
        const payload = {
            type: 2,
            application_id: data.application_command.application_id,
            guild_id: this.config.ServerId,
            channel_id: this.config.ChannelId,
            session_id: this.config.SessionId,
            nonce,
            data,
        };
        return payload;
    }
}
exports.Command = Command;
//# sourceMappingURL=command.js.map