automod: add user/member to automod context for counters trigger

This commit is contained in:
Dragory 2021-02-13 21:34:47 +02:00
parent e549303b1d
commit 78ff2641d4
No known key found for this signature in database
GPG key ID: 5F387BA66DF8AAC1

View file

@ -1,8 +1,9 @@
import { GuildPluginData } from "knub"; import { GuildPluginData } from "knub";
import { AutomodContext, AutomodPluginType } from "../types"; import { AutomodContext, AutomodPluginType } from "../types";
import { runAutomod } from "../functions/runAutomod"; import { runAutomod } from "../functions/runAutomod";
import { resolveMember, resolveUser, UnknownUser } from "../../../utils";
export function runAutomodOnCounterTrigger( export async function runAutomodOnCounterTrigger(
pluginData: GuildPluginData<AutomodPluginType>, pluginData: GuildPluginData<AutomodPluginType>,
counterName: string, counterName: string,
condition: string, condition: string,
@ -10,6 +11,10 @@ export function runAutomodOnCounterTrigger(
userId: string | null, userId: string | null,
reverse: boolean, reverse: boolean,
) { ) {
const user = userId ? await resolveUser(pluginData.client, userId) : undefined;
const member = (userId && (await resolveMember(pluginData.client, pluginData.guild, userId))) || undefined;
const context: AutomodContext = { const context: AutomodContext = {
timestamp: Date.now(), timestamp: Date.now(),
counterTrigger: { counterTrigger: {
@ -19,6 +24,8 @@ export function runAutomodOnCounterTrigger(
userId, userId,
reverse, reverse,
}, },
user: user instanceof UnknownUser ? undefined : user,
member,
}; };
pluginData.state.queue.add(async () => { pluginData.state.queue.add(async () => {