StyAndCo/zeppelin
3
0
Fork 0
mirror of https://github.com/ZeppelinBot/Zeppelin.git synced 2025-05-22 01:05:02 +00:00
Code Releases Activity

refactor(triggers/mime-type): use attachment data

Browse source
This commit is contained in:
Hiroyuki 2021-08-11 00:18:45 -04:00
parent abb493ab8f
commit 45f9f30aeb
No known key found for this signature in database
GPG key ID: C15AC26538975A24
3 changed files with 5 additions and 210 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -1,7 +1,5 @@
import { automodTrigger } from "../helpers";
import * as t from "io-ts";
import fetch from "node-fetch";
import { fromBuffer } from "file-type";
import { asSingleLine, messageSummary, verboseChannelMention } from "src/utils";
import { Snowflake, TextChannel, Util } from "discord.js";
@ -32,17 +30,16 @@ export const MatchMimeTypeTrigger = automodTrigger<MatchResultType>()({
if (!attachments) return null;
for (const attachment of attachments) {
const res = await fetch(attachment.url);
const mimeType = await fromBuffer(await res.buffer());
const { contentType } = attachment;
const blacklist = trigger.blacklist_enabled
? (trigger.mime_type_blacklist || []).map(_t => _t.toLowerCase())
: null;
if (mimeType && blacklist?.includes(mimeType.mime)) {
if (contentType && blacklist?.includes(contentType)) {
return {
extra: {
matchedType: mimeType.mime,
matchedType: contentType,
mode: "blacklist",
},
};
@ -52,10 +49,10 @@ export const MatchMimeTypeTrigger = automodTrigger<MatchResultType>()({
? (trigger.mime_type_whitelist || []).map(_t => _t.toLowerCase())
: null;
if (whitelist && (!mimeType || !whitelist.includes(mimeType.mime))) {
if (whitelist && (!contentType || !whitelist.includes(contentType))) {
return {
extra: {
matchedType: mimeType?.mime || "unknown",
matchedType: contentType || "unknown",
mode: "whitelist",
},
};