import { AllowedMentions, banResponse, BushMessage, dmResponse, permissionsResponse, punishmentEntryRemove, type BanResponse, type BushClient, type BushGuildChannelManager, type BushGuildMember, type BushGuildMemberManager, type BushGuildMemberResolvable, type BushNewsChannel, type BushTextChannel, type BushThreadChannel, type BushUser, type BushUserResolvable, type GuildFeatures, type GuildLogType, type GuildModel } from '#lib'; import { APIMessage } from 'discord-api-types/v10'; import { Collection, Guild, MessageType, PermissionFlagsBits, SnowflakeUtil, ThreadChannel, Webhook, WebhookMessageOptions, type MessageOptions, type MessagePayload, type Snowflake } from 'discord.js'; import type { RawGuildData } from 'discord.js/typings/rawDataTypes'; import _ from 'lodash'; import { Moderation } from '../../common/util/Moderation.js'; import { Guild as GuildDB } from '../../models/instance/Guild.js'; import { ModLogType } from '../../models/instance/ModLog.js'; /** * Represents a guild (or a server) on Discord. * It's recommended to see if a guild is available before performing operations or reading data from it. You can * check this with {@link BushGuild.available}. */ export class BushGuild extends Guild { public declare readonly client: BushClient; public declare members: BushGuildMemberManager; public declare channels: BushGuildChannelManager; public constructor(client: BushClient, data: RawGuildData) { super(client, data); } /** * Checks if the guild has a certain custom feature. * @param feature The feature to check for */ public async hasFeature(feature: GuildFeatures): Promise { const features = await this.getSetting('enabledFeatures'); return features.includes(feature); } /** * Adds a custom feature to the guild. * @param feature The feature to add * @param moderator The moderator responsible for adding a feature */ public async addFeature(feature: GuildFeatures, moderator?: BushGuildMember): Promise { const features = await this.getSetting('enabledFeatures'); const newFeatures = util.addOrRemoveFromArray('add', features, feature); return (await this.setSetting('enabledFeatures', newFeatures, moderator)).enabledFeatures; } /** * Removes a custom feature from the guild. * @param feature The feature to remove * @param moderator The moderator responsible for removing a feature */ public async removeFeature(feature: GuildFeatures, moderator?: BushGuildMember): Promise { const features = await this.getSetting('enabledFeatures'); const newFeatures = util.addOrRemoveFromArray('remove', features, feature); return (await this.setSetting('enabledFeatures', newFeatures, moderator)).enabledFeatures; } /** * Makes a custom feature the opposite of what it was before * @param feature The feature to toggle * @param moderator The moderator responsible for toggling a feature */ public async toggleFeature(feature: GuildFeatures, moderator?: BushGuildMember): Promise { return (await this.hasFeature(feature)) ? await this.removeFeature(feature, moderator) : await this.addFeature(feature, moderator); } /** * Fetches a custom setting for the guild * @param setting The setting to get */ public async getSetting(setting: K): Promise { return ( client.cache.guilds.get(this.id)?.[setting] ?? ((await GuildDB.findByPk(this.id)) ?? GuildDB.build({ id: this.id }))[setting] ); } /** * Sets a custom setting for the guild * @param setting The setting to change * @param value The value to change the setting to * @param moderator The moderator to responsible for changing the setting */ public async setSetting>( setting: K, value: GuildDB[K], moderator?: BushGuildMember ): Promise { const row = (await GuildDB.findByPk(this.id)) ?? GuildDB.build({ id: this.id }); const oldValue = row[setting] as GuildDB[K]; row[setting] = value; client.cache.guilds.set(this.id, row.toJSON() as GuildDB); client.emit('bushUpdateSettings', setting, this, oldValue, row[setting], moderator); return await row.save(); } /** * Get a the log channel configured for a certain log type. * @param logType The type of log channel to get. * @returns Either the log channel or undefined if not configured. */ public async getLogChannel(logType: GuildLogType): Promise { const channelId = (await this.getSetting('logChannels'))[logType]; if (!channelId) return undefined; return ( (this.channels.cache.get(channelId) as BushTextChannel | undefined) ?? ((await this.channels.fetch(channelId)) as BushTextChannel | null) ?? undefined ); } /** * Sends a message to the guild's specified logging channel * @param logType The corresponding channel that the message will be sent to * @param message The parameters for {@link BushTextChannel.send} */ public async sendLogChannel(logType: GuildLogType, message: string | MessagePayload | MessageOptions) { const logChannel = await this.getLogChannel(logType); if (!logChannel || !logChannel.isTextBased()) return; if ( !logChannel .permissionsFor(this.me!.id) ?.has([PermissionFlagsBits.ViewChannel, PermissionFlagsBits.SendMessages, PermissionFlagsBits.EmbedLinks]) ) return; return await logChannel.send(message).catch(() => null); } /** * Sends a formatted error message in a guild's error log channel * @param title The title of the error embed * @param message The description of the error embed */ public async error(title: string, message: string) { void client.console.info(_.camelCase(title), message.replace(/\*\*(.*?)\*\*/g, '<<$1>>')); void this.sendLogChannel('error', { embeds: [{ title: title, description: message, color: util.colors.error }] }); } /** * Bans a user, dms them, creates a mod log entry, and creates a punishment entry. * @param options Options for banning the user. * @returns A string status message of the ban. */ public async bushBan(options: GuildBushBanOptions): Promise { // checks if (!this.me!.permissions.has(PermissionFlagsBits.BanMembers)) return banResponse.MISSING_PERMISSIONS; let caseID: string | undefined = undefined; let dmSuccessEvent: boolean | undefined = undefined; const user = await util.resolveNonCachedUser(options.user); const moderator = client.users.resolve(options.moderator ?? client.user!); if (!user || !moderator) return banResponse.CANNOT_RESOLVE_USER; if ((await this.bans.fetch()).has(user.id)) return banResponse.ALREADY_BANNED; const ret = await (async () => { // add modlog entry const { log: modlog } = await Moderation.createModLogEntry({ type: options.duration ? ModLogType.TEMP_BAN : ModLogType.PERM_BAN, user: user, moderator: moderator.id, reason: options.reason, duration: options.duration, guild: this, evidence: options.evidence }); if (!modlog) return banResponse.MODLOG_ERROR; caseID = modlog.id; // dm user dmSuccessEvent = await Moderation.punishDM({ modlog: modlog.id, guild: this, user: user, punishment: 'banned', duration: options.duration ?? 0, reason: options.reason ?? undefined, sendFooter: true }); // ban const banSuccess = await this.bans .create(user?.id ?? options.user, { reason: `${moderator.tag} | ${options.reason ?? 'No reason provided.'}`, deleteMessageDays: options.deleteDays }) .catch(() => false); if (!banSuccess) return banResponse.ACTION_ERROR; // add punishment entry so they can be unbanned later const punishmentEntrySuccess = await Moderation.createPunishmentEntry({ type: 'ban', user: user, guild: this, duration: options.duration, modlog: modlog.id }); if (!punishmentEntrySuccess) return banResponse.PUNISHMENT_ENTRY_ADD_ERROR; if (!dmSuccessEvent) return banResponse.DM_ERROR; return banResponse.SUCCESS; })(); if (!([banResponse.ACTION_ERROR, banResponse.MODLOG_ERROR, banResponse.PUNISHMENT_ENTRY_ADD_ERROR] as const).includes(ret)) client.emit( 'bushBan', user, moderator, this, options.reason ?? undefined, caseID!, options.duration ?? 0, dmSuccessEvent, options.evidence ); return ret; } /** * {@link bushBan} with less resolving and checks * @param options Options for banning the user. * @returns A string status message of the ban. * **Preconditions:** * - {@link me} has the `BanMembers` permission * **Warning:** * - Doesn't emit bushBan Event */ public async massBanOne(options: GuildMassBanOneOptions): Promise { if (this.bans.cache.has(options.user)) return banResponse.ALREADY_BANNED; const ret = await (async () => { // add modlog entry const { log: modlog } = await Moderation.createModLogEntrySimple({ type: ModLogType.PERM_BAN, user: options.user, moderator: options.moderator, reason: options.reason, duration: 0, guild: this.id }); if (!modlog) return banResponse.MODLOG_ERROR; let dmSuccessEvent: boolean | undefined = undefined; // dm user if (this.members.cache.has(options.user)) { dmSuccessEvent = await Moderation.punishDM({ modlog: modlog.id, guild: this, user: options.user, punishment: 'banned', duration: 0, reason: options.reason ?? undefined, sendFooter: true }); } // ban const banSuccess = await this.bans .create(options.user, { reason: `${options.moderator} | ${options.reason}`, deleteMessageDays: options.deleteDays }) .catch(() => false); if (!banSuccess) return banResponse.ACTION_ERROR; // add punishment entry so they can be unbanned later const punishmentEntrySuccess = await Moderation.createPunishmentEntry({ type: 'ban', user: options.user, guild: this, duration: 0, modlog: modlog.id }); if (!punishmentEntrySuccess) return banResponse.PUNISHMENT_ENTRY_ADD_ERROR; if (!dmSuccessEvent) return banResponse.DM_ERROR; return banResponse.SUCCESS; })(); return ret; } /** * Unbans a user, dms them, creates a mod log entry, and destroys the punishment entry. * @param options Options for unbanning the user. * @returns A status message of the unban. */ public async bushUnban(options: GuildBushUnbanOptions): Promise { // checks if (!this.me!.permissions.has(PermissionFlagsBits.BanMembers)) return unbanResponse.MISSING_PERMISSIONS; let caseID: string | undefined = undefined; let dmSuccessEvent: boolean | undefined = undefined; const user = await util.resolveNonCachedUser(options.user); const moderator = client.users.resolve(options.moderator ?? client.user!); if (!user || !moderator) return unbanResponse.CANNOT_RESOLVE_USER; const ret = await (async () => { const bans = await this.bans.fetch(); let notBanned = false; if (!bans.has(user.id)) notBanned = true; const unbanSuccess = await this.bans .remove(user, `${moderator.tag} | ${options.reason ?? 'No reason provided.'}`) .catch((e) => { if (e?.code === 'UNKNOWN_BAN') { notBanned = true; return true; } else return false; }); if (notBanned) return unbanResponse.NOT_BANNED; if (!unbanSuccess) return unbanResponse.ACTION_ERROR; // add modlog entry const { log: modlog } = await Moderation.createModLogEntry({ type: ModLogType.UNBAN, user: user.id, moderator: moderator.id, reason: options.reason, guild: this, evidence: options.evidence }); if (!modlog) return unbanResponse.MODLOG_ERROR; caseID = modlog.id; // remove punishment entry const removePunishmentEntrySuccess = await Moderation.removePunishmentEntry({ type: 'ban', user: user.id, guild: this }); if (!removePunishmentEntrySuccess) return unbanResponse.PUNISHMENT_ENTRY_REMOVE_ERROR; // dm user dmSuccessEvent = await Moderation.punishDM({ guild: this, user: user, punishment: 'unbanned', reason: options.reason ?? undefined, sendFooter: false }); if (!dmSuccessEvent) return unbanResponse.DM_ERROR; return unbanResponse.SUCCESS; })(); if ( !([unbanResponse.ACTION_ERROR, unbanResponse.MODLOG_ERROR, unbanResponse.PUNISHMENT_ENTRY_REMOVE_ERROR] as const).includes( ret ) ) client.emit('bushUnban', user, moderator, this, options.reason ?? undefined, caseID!, dmSuccessEvent!, options.evidence); return ret; } /** * Denies send permissions in specified channels * @param options The options for locking down the guild */ public async lockdown(options: LockdownOptions): Promise { if (!options.all && !options.channel) return 'all not chosen and no channel specified'; const channelIds = options.all ? await this.getSetting('lockdownChannels') : [options.channel!.id]; if (!channelIds.length) return 'no channels configured'; const mappedChannels = channelIds.map((id) => this.channels.cache.get(id)); const invalidChannels = mappedChannels.filter((c) => c === undefined); if (invalidChannels.length) return `invalid channel configured: ${invalidChannels.join(', ')}`; const moderator = this.members.resolve(options.moderator); if (!moderator) return 'moderator not found'; const errors = new Collection(); const success = new Collection(); const ret = await (async (): Promise => { for (const _channel of mappedChannels) { const channel = _channel!; if (!channel.isText() && !channel.isThread()) { errors.set(channel.id, new Error('wrong channel type')); success.set(channel.id, false); continue; } if (!channel.permissionsFor(this.me!.id)?.has([PermissionFlagsBits.ManageChannels])) { errors.set(channel.id, new Error('client no permission')); success.set(channel.id, false); continue; } else if (!channel.permissionsFor(moderator)?.has([PermissionFlagsBits.ManageChannels])) { errors.set(channel.id, new Error('moderator no permission')); success.set(channel.id, false); continue; } const reason = `[${options.unlock ? 'Unlockdown' : 'Lockdown'}] ${moderator.user.tag} | ${ options.reason ?? 'No reason provided' }`; const permissionOverwrites = channel.isThread() ? channel.parent!.permissionOverwrites : channel.permissionOverwrites; const perms = { SendMessagesInThreads: options.unlock ? null : false, SendMessages: options.unlock ? null : false }; const permsForMe = { [channel.isThread() ? 'SendMessagesInThreads' : 'SendMessages']: options.unlock ? null : true }; // so I can send messages in the channel const changePermSuccess = await permissionOverwrites.edit(this.id, perms, { reason }).catch((e) => e); if (changePermSuccess instanceof Error) { errors.set(channel.id, changePermSuccess); success.set(channel.id, false); } else { success.set(channel.id, true); await permissionOverwrites.edit(this.me!, permsForMe, { reason }); await channel.send({ embeds: [ { author: { name: moderator.user.tag, icon_url: moderator.displayAvatarURL() }, title: `This channel has been ${options.unlock ? 'un' : ''}locked`, description: options.reason ?? 'No reason provided', color: options.unlock ? util.colors.Green : util.colors.Red, timestamp: new Date().toISOString() } ] }); } } if (errors.size) return errors; else return `success: ${success.filter((c) => c === true).size}`; })(); client.emit(options.unlock ? 'bushUnlockdown' : 'bushLockdown', moderator, options.reason, success, options.all); return ret; } public async quote(rawQuote: APIMessage, channel: BushTextChannel | BushNewsChannel | BushThreadChannel) { if (!channel.isTextBased() || channel.isDMBased() || channel.guildId !== this.id || !this.me) return null; if (!channel.permissionsFor(this.me).has('ManageWebhooks')) return null; const quote = new BushMessage(client, rawQuote); const target = channel instanceof ThreadChannel ? channel.parent : channel; if (!target) return null; const webhooks: Collection = await target.fetchWebhooks().catch((e) => e); if (!(webhooks instanceof Collection)) return null; // find a webhook that we can use let webhook = webhooks.find((w) => !!w.token) ?? null; if (!webhook) webhook = await target .createWebhook(`${client.user!.username} Quotes #${target.name}`, { avatar: client.user!.displayAvatarURL({ size: 2048 }), reason: 'Creating a webhook for quoting' }) .catch(() => null); if (!webhook) return null; const sendOptions: Omit = {}; const displayName = quote.member?.displayName ?? quote.author.username; switch (quote.type) { case MessageType.Default: case MessageType.Reply: case MessageType.ChatInputCommand: case MessageType.ContextMenuCommand: case MessageType.ThreadStarterMessage: sendOptions.content = quote.content || undefined; sendOptions.threadId = channel instanceof ThreadChannel ? channel.id : undefined; sendOptions.embeds = quote.embeds.length ? quote.embeds : undefined; sendOptions.attachments = quote.attachments.size ? [...quote.attachments.values()] : undefined; if (quote.stickers.size && !(quote.content || quote.embeds.length || quote.attachments.size)) sendOptions.content = '[[This message has a sticker but not content]]'; break; case MessageType.RecipientAdd: { const recipient = rawQuote.mentions[0]; if (!recipient) { sendOptions.content = `${util.emojis.error} Cannot resolve recipient.`; break; } if (quote.channel.isThread()) { const recipientDisplay = quote.guild?.members.cache.get(recipient.id)?.displayName ?? recipient.username; sendOptions.content = `${util.emojis.join} ${displayName} added ${recipientDisplay} to the thread.`; } else { // this should never happen sendOptions.content = `${util.emojis.join} ${displayName} added ${recipient.username} to the group.`; } break; } case MessageType.RecipientRemove: { const recipient = rawQuote.mentions[0]; if (!recipient) { sendOptions.content = `${util.emojis.error} Cannot resolve recipient.`; break; } if (quote.channel.isThread()) { const recipientDisplay = quote.guild?.members.cache.get(recipient.id)?.displayName ?? recipient.username; sendOptions.content = `${util.emojis.leave} ${displayName} removed ${recipientDisplay} from the thread.`; } else { // this should never happen sendOptions.content = `${util.emojis.leave} ${displayName} removed ${recipient.username} from the group.`; } break; } case MessageType.ChannelNameChange: sendOptions.content = `<:pencil:957988608994861118> ${displayName} changed the channel name: **${quote.content}**`; break; case MessageType.ChannelPinnedMessage: throw new Error('Not implemented yet: MessageType.ChannelPinnedMessage case'); case MessageType.GuildMemberJoin: { const messages = [ '{username} joined the party.', '{username} is here.', 'Welcome, {username}. We hope you brought pizza.', 'A wild {username} appeared.', '{username} just landed.', '{username} just slid into the server.', '{username} just showed up!', 'Welcome {username}. Say hi!', '{username} hopped into the server.', 'Everyone welcome {username}!', "Glad you're here, {username}.", 'Good to see you, {username}.', 'Yay you made it, {username}!' ]; const timestamp = SnowflakeUtil.timestampFrom(quote.id); // this is the same way that the discord client decides what message to use. const message = messages[timestamp % messages.length].replace(/{username}/g, displayName); sendOptions.content = `${util.emojis.join} ${message}`; break; } case MessageType.UserPremiumGuildSubscription: sendOptions.content = `<:NitroBoost:585558042309820447> ${displayName} just boosted the server${ quote.content ? ` **${quote.content}** times` : '' }!`; break; case MessageType.UserPremiumGuildSubscriptionTier1: case MessageType.UserPremiumGuildSubscriptionTier2: case MessageType.UserPremiumGuildSubscriptionTier3: sendOptions.content = `<:NitroBoost:585558042309820447> ${displayName} just boosted the server${ quote.content ? ` **${quote.content}** times` : '' }! ${quote.guild?.name} has achieved **Level ${quote.type - 8}!**`; break; case MessageType.ChannelFollowAdd: sendOptions.content = `${displayName} has added **${quote.content}** to this channel. Its most important updates will show up here.`; break; case MessageType.GuildDiscoveryDisqualified: sendOptions.content = '<:SystemMessageCross:842172192418693173> This server has been removed from Server Discovery because it no longer passes all the requirements. Check Server Settings for more details.'; break; case MessageType.GuildDiscoveryRequalified: sendOptions.content = '<:SystemMessageCheck:842172191801212949> This server is eligible for Server Discovery again and has been automatically relisted!'; break; case MessageType.GuildDiscoveryGracePeriodInitialWarning: sendOptions.content = '<:SystemMessageWarn:842172192401915971> This server has failed Discovery activity requirements for 1 week. If this server fails for 4 weeks in a row, it will be automatically removed from Discovery.'; break; case MessageType.GuildDiscoveryGracePeriodFinalWarning: sendOptions.content = '<:SystemMessageWarn:842172192401915971> This server has failed Discovery activity requirements for 3 weeks in a row. If this server fails for 1 more week, it will be removed from Discovery.'; break; case MessageType.ThreadCreated: { const threadId = rawQuote.message_reference?.channel_id; sendOptions.content = `<:thread:865033845753249813> ${displayName} started a thread: **[${quote.content}](https://discord.com/channels/${quote.guildId}/${threadId} )**. See all threads.`; break; } case MessageType.GuildInviteReminder: sendOptions.content = 'Wondering who to invite? Start by inviting anyone who can help you build the server!'; break; case MessageType.ChannelIconChange: case MessageType.Call: default: sendOptions.content = `${util.emojis.error} I cannot quote **${ MessageType[quote.type] || quote.type }** messages, please report this to my developers.`; break; } sendOptions.allowedMentions = AllowedMentions.none(); sendOptions.username = quote.member?.displayName ?? quote.author.username; sendOptions.avatarURL = quote.member?.displayAvatarURL({ size: 2048 }) ?? quote.author.displayAvatarURL({ size: 2048 }); return await webhook.send(sendOptions); /* .catch((e: any) => e); */ } } export interface BushGuild extends Guild { get me(): BushGuildMember | null; } /** * Options for unbanning a user */ export interface GuildBushUnbanOptions { /** * The user to unban */ user: BushUserResolvable | BushUser; /** * The reason for unbanning the user */ reason?: string | null; /** * The moderator who unbanned the user */ moderator?: BushUserResolvable; /** * The evidence for the unban */ evidence?: string; } export interface GuildMassBanOneOptions { /** * The user to ban */ user: Snowflake; /** * The reason to ban the user */ reason: string; /** * The moderator who banned the user */ moderator: Snowflake; /** * The number of days to delete the user's messages for */ deleteDays?: number; } /** * Options for banning a user */ export interface GuildBushBanOptions { /** * The user to ban */ user: BushUserResolvable; /** * The reason to ban the user */ reason?: string | null; /** * The moderator who banned the user */ moderator?: BushUserResolvable; /** * The duration of the ban */ duration?: number; /** * The number of days to delete the user's messages for */ deleteDays?: number; /** * The evidence for the ban */ evidence?: string; } type ValueOf = T[keyof T]; export const unbanResponse = Object.freeze({ ...dmResponse, ...permissionsResponse, ...punishmentEntryRemove, NOT_BANNED: 'user not banned' } as const); /** * Response returned when unbanning a user */ export type UnbanResponse = ValueOf; /** * Options for locking down channel(s) */ export interface LockdownOptions { /** * The moderator responsible for the lockdown */ moderator: BushGuildMemberResolvable; /** * Whether to lock down all (specified) channels */ all: boolean; /** * Reason for the lockdown */ reason?: string; /** * A specific channel to lockdown */ channel?: BushThreadChannel | BushNewsChannel | BushTextChannel; /** * Whether or not to unlock the channel(s) instead of locking them */ unlock?: boolean; } /** * Response returned when locking down a channel */ export type LockdownResponse = | `success: ${number}` | 'all not chosen and no channel specified' | 'no channels configured' | `invalid channel configured: ${string}` | 'moderator not found' | Collection;