87 lines
2.8 KiB
TypeScript
87 lines
2.8 KiB
TypeScript
import escapeStringRegexp from "escape-string-regexp";
|
|
import z from "zod";
|
|
import { normalizeText } from "../../../utils/normalizeText";
|
|
import { stripMarkdown } from "../../../utils/stripMarkdown";
|
|
import { getTextMatchPartialSummary } from "../functions/getTextMatchPartialSummary";
|
|
import { MatchableTextType, matchMultipleTextTypesOnMessage } from "../functions/matchMultipleTextTypesOnMessage";
|
|
import { automodTrigger } from "../helpers";
|
|
|
|
interface MatchResultType {
|
|
word: string;
|
|
type: MatchableTextType;
|
|
}
|
|
|
|
const regexCache = new WeakMap<any, RegExp[]>();
|
|
|
|
const configSchema = z.strictObject({
|
|
words: z.array(z.string().max(2000)).max(512),
|
|
case_sensitive: z.boolean().default(false),
|
|
only_full_words: z.boolean().default(true),
|
|
normalize: z.boolean().default(false),
|
|
loose_matching: z.boolean().default(false),
|
|
loose_matching_threshold: z.number().int().default(4),
|
|
strip_markdown: z.boolean().default(false),
|
|
match_messages: z.boolean().default(true),
|
|
match_embeds: z.boolean().default(false),
|
|
match_visible_names: z.boolean().default(false),
|
|
match_usernames: z.boolean().default(false),
|
|
match_nicknames: z.boolean().default(false),
|
|
match_custom_status: z.boolean().default(false),
|
|
});
|
|
|
|
export const MatchWordsTrigger = automodTrigger<MatchResultType>()({
|
|
configSchema,
|
|
|
|
async match({ pluginData, context, triggerConfig: trigger }) {
|
|
if (!context.message) {
|
|
return;
|
|
}
|
|
|
|
if (!regexCache.has(trigger)) {
|
|
const looseMatchingThreshold = Math.min(Math.max(trigger.loose_matching_threshold, 1), 64);
|
|
const patterns = trigger.words.map((word) => {
|
|
let pattern = trigger.loose_matching
|
|
? [...word].map((c) => escapeStringRegexp(c)).join(`(?:\\s*|.{0,${looseMatchingThreshold})`)
|
|
: escapeStringRegexp(word);
|
|
|
|
if (trigger.only_full_words) {
|
|
pattern = `\\b${pattern}\\b`;
|
|
}
|
|
|
|
return pattern;
|
|
});
|
|
|
|
const mergedRegex = new RegExp(patterns.map((p) => `(?:${p})`).join("|"), trigger.case_sensitive ? "" : "i");
|
|
regexCache.set(trigger, [mergedRegex]);
|
|
}
|
|
const regexes = regexCache.get(trigger)!;
|
|
|
|
for await (let [type, str] of matchMultipleTextTypesOnMessage(pluginData, trigger, context.message)) {
|
|
if (trigger.strip_markdown) {
|
|
str = stripMarkdown(str);
|
|
}
|
|
|
|
if (trigger.normalize) {
|
|
str = normalizeText(str);
|
|
}
|
|
|
|
for (const regex of regexes) {
|
|
if (regex.test(str)) {
|
|
return {
|
|
extra: {
|
|
type,
|
|
word: "",
|
|
},
|
|
};
|
|
}
|
|
}
|
|
}
|
|
|
|
return null;
|
|
},
|
|
|
|
renderMatchInformation({ pluginData, contexts, matchResult }) {
|
|
const partialSummary = getTextMatchPartialSummary(pluginData, matchResult.extra.type, contexts[0]);
|
|
return `Matched word in ${partialSummary}`;
|
|
},
|
|
});
|