StyAndCo/zeppelin
3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-05-21 08:45:03 +00:00
Code Releases Activity

fix(automod/mime-type-trigger): types, logic

Browse source
This commit is contained in:
Hiroyuki 2021-08-04 02:20:33 -04:00
parent 01d576b697
commit abb493ab8f
No known key found for this signature in database
GPG key ID: C15AC26538975A24
1 changed files with 18 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

38
backend/src/plugins/Automod/triggers/matchMimeType.ts
View file

@ -2,7 +2,8 @@ import { automodTrigger } from "../helpers";
import * as t from "io-ts"; import * as t from "io-ts";
import fetch from "node-fetch"; import fetch from "node-fetch";
import { fromBuffer } from "file-type"; import { fromBuffer } from "file-type";
import { asSingleLine, disableInlineCode, messageSummary, verboseChannelMention } from "src/utils"; import { asSingleLine, messageSummary, verboseChannelMention } from "src/utils";
import { Snowflake, TextChannel, Util } from "discord.js";
interface MatchResultType { interface MatchResultType {
matchedType: string; matchedType: string;
@ -11,16 +12,16 @@ interface MatchResultType {
export const MatchMimeTypeTrigger = automodTrigger<MatchResultType>()({ export const MatchMimeTypeTrigger = automodTrigger<MatchResultType>()({
configType: t.type({ configType: t.type({
mimetype_blacklist: t.array(t.string), mime_type_blacklist: t.array(t.string),
blacklist_enabled: t.boolean, blacklist_enabled: t.boolean,
mimetype_whitelist: t.array(t.string), mime_type_whitelist: t.array(t.string),
whitelist_enabled: t.boolean, whitelist_enabled: t.boolean,
}), }),
defaultConfig: { defaultConfig: {
mimetype_blacklist: [], mime_type_blacklist: [],
blacklist_enabled: false, blacklist_enabled: false,
mimetype_whitelist: [], mime_type_whitelist: [],
whitelist_enabled: false, whitelist_enabled: false,
}, },
@ -31,15 +32,14 @@ export const MatchMimeTypeTrigger = automodTrigger<MatchResultType>()({
if (!attachments) return null; if (!attachments) return null;
for (const attachment of attachments) { for (const attachment of attachments) {
const res = await fetch(attachment.proxyURL); const res = await fetch(attachment.url);
const mimeType = await fromBuffer(await res.buffer()); const mimeType = await fromBuffer(await res.buffer());
if (!mimeType) return;
const blacklist = trigger.blacklist_enabled const blacklist = trigger.blacklist_enabled
? (trigger.mimetype_blacklist || []).map(_t => _t.toLowerCase()) ? (trigger.mime_type_blacklist || []).map(_t => _t.toLowerCase())
: null; : null;
if (blacklist?.includes(mimeType.mime)) { if (mimeType && blacklist?.includes(mimeType.mime)) {
return { return {
extra: { extra: {
matchedType: mimeType.mime, matchedType: mimeType.mime,
@ -49,13 +49,13 @@ export const MatchMimeTypeTrigger = automodTrigger<MatchResultType>()({
} }
const whitelist = trigger.whitelist_enabled const whitelist = trigger.whitelist_enabled
? (trigger.mimetype_whitelist || []).map(_t => _t.toLowerCase()) ? (trigger.mime_type_whitelist || []).map(_t => _t.toLowerCase())
: null; : null;
if (whitelist?.includes(mimeType.mime)) { if (whitelist && (!mimeType || !whitelist.includes(mimeType.mime))) {
return { return {
extra: { extra: {
matchedType: mimeType.mime, matchedType: mimeType?.mime || "unknown",
mode: "whitelist", mode: "whitelist",
}, },
}; };
@ -66,19 +66,17 @@ export const MatchMimeTypeTrigger = automodTrigger<MatchResultType>()({
}, },
renderMatchInformation({ pluginData, contexts, matchResult }) { renderMatchInformation({ pluginData, contexts, matchResult }) {
const [context] = contexts; const { message } = contexts[0];
const { message } = context; const channel = pluginData.guild.channels.cache.get(message!.channel_id as Snowflake) as TextChannel;
if (!message) return;
const channel = pluginData.guild.channels.cache.get(message.channel_id);
const prettyChannel = verboseChannelMention(channel); const prettyChannel = verboseChannelMention(channel);
const { matchedType, mode } = matchResult.extra; const { matchedType, mode } = matchResult.extra;
return ( return (
asSingleLine(` asSingleLine(`
Matched MIME type \`${disableInlineCode(matchedType)}\` Matched MIME type \`${Util.escapeInlineCode(matchedType)}\`
(${mode === "blacklist" ? "(blacklisted)" : "(not in whitelist)"}) (${mode === "blacklist" ? "blacklisted" : "not in whitelist"})
in message (\`${message.id}\`) in ${prettyChannel} in message (\`${message!.id}\`) in ${prettyChannel}
`) + messageSummary(message) `) + messageSummary(message!)
); );
}, },
}); });