zappyzep/backend/src/plugins/Slowmode/util/onMessageCreate.ts
2021-11-02 23:38:02 +02:00

79 lines
3.2 KiB
TypeScript

import { Snowflake, TextChannel } from "discord.js";
import { GuildPluginData } from "knub";
import { SavedMessage } from "../../../data/entities/SavedMessage";
import { hasPermission } from "../../../pluginUtils";
import { resolveMember } from "../../../utils";
import { getMissingChannelPermissions } from "../../../utils/getMissingChannelPermissions";
import { messageLock } from "../../../utils/lockNameHelpers";
import { missingPermissionError } from "../../../utils/missingPermissionError";
import { LogsPlugin } from "../../Logs/LogsPlugin";
import { BOT_SLOWMODE_PERMISSIONS } from "../requiredPermissions";
import { SlowmodePluginType } from "../types";
import { applyBotSlowmodeToUserId } from "./applyBotSlowmodeToUserId";
import { SlowmodeChannel } from "../../../data/entities/SlowmodeChannel";
export async function onMessageCreate(pluginData: GuildPluginData<SlowmodePluginType>, msg: SavedMessage) {
if (msg.is_bot) return;
const channel = pluginData.guild.channels.cache.get(msg.channel_id as Snowflake) as TextChannel;
if (!channel) return;
// Don't apply slowmode if the lock was interrupted earlier (e.g. the message was caught by word filters)
const thisMsgLock = await pluginData.locks.acquire(messageLock(msg));
if (thisMsgLock.interrupted) return;
// Check if this channel even *has* a bot-maintained slowmode
let channelSlowmode: SlowmodeChannel | null;
if (pluginData.state.channelSlowmodeCache.has(channel.id)) {
channelSlowmode = pluginData.state.channelSlowmodeCache.get(channel.id) ?? null;
} else {
channelSlowmode = (await pluginData.state.slowmodes.getChannelSlowmode(channel.id)) ?? null;
pluginData.state.channelSlowmodeCache.set(channel.id, channelSlowmode);
}
if (!channelSlowmode) {
return thisMsgLock.unlock();
}
// Make sure this user is affected by the slowmode
const member = await resolveMember(pluginData.client, pluginData.guild, msg.user_id);
const isAffected = await hasPermission(pluginData, "is_affected", {
channelId: channel.id,
userId: msg.user_id,
member,
});
if (!isAffected) {
return thisMsgLock.unlock();
}
// Make sure we have the appropriate permissions to manage this slowmode
const me = pluginData.guild.members.cache.get(pluginData.client.user!.id)!;
const missingPermissions = getMissingChannelPermissions(me, channel, BOT_SLOWMODE_PERMISSIONS);
if (missingPermissions) {
const logs = pluginData.getPlugin(LogsPlugin);
logs.logBotAlert({
body: `Unable to manage bot slowmode in <#${channel.id}>. ${missingPermissionError(missingPermissions)}`,
});
return;
}
// Delete any extra messages sent after a slowmode was already applied
const userHasSlowmode = await pluginData.state.slowmodes.userHasSlowmode(channel.id, msg.user_id);
if (userHasSlowmode) {
try {
// FIXME: Debug
// tslint:disable-next-line:no-console
console.log(
`[DEBUG] [SLOWMODE] Deleting message ${msg.id} from channel ${channel.id} in guild ${pluginData.guild.id}`,
);
await channel.messages.delete(msg.id);
thisMsgLock.interrupt();
} catch (err) {
thisMsgLock.unlock();
}
return;
}
await applyBotSlowmodeToUserId(pluginData, channel, msg.user_id);
thisMsgLock.unlock();
}