Added a whitelist (always ping)

This commit is contained in:
AntonMacG 2023-11-13 07:21:20 +00:00
parent b789094256
commit e5ffb349f7

View file

@ -7,7 +7,7 @@
import { definePluginSettings } from "@api/Settings"; import { definePluginSettings } from "@api/Settings";
import { Devs } from "@utils/constants"; import { Devs } from "@utils/constants";
import definePlugin, { OptionType } from "@utils/types"; import definePlugin, { OptionType } from "@utils/types";
import { MessageStore, UserStore } from "@webpack/common"; import { MessageStore, showToast, UserStore } from "@webpack/common";
import { MessageJSON } from "discord-types/general"; import { MessageJSON } from "discord-types/general";
export const settings = definePluginSettings({ export const settings = definePluginSettings({
@ -15,12 +15,18 @@ export const settings = definePluginSettings({
type: OptionType.BOOLEAN, type: OptionType.BOOLEAN,
description: "Always get pinged when someone replies to your messages", description: "Always get pinged when someone replies to your messages",
default: false, default: false,
},
replyPingWhitelist: {
type: OptionType.STRING,
description: "Comma-separated list of User IDs to always receive reply pings from",
default: "",
disabled: () => settings.store.alwaysPingOnReply,
} }
}); });
export default definePlugin({ export default definePlugin({
name: "ReplyPingControl", name: "ReplyPingControl",
description: "Control whether to always or never get pinged on message replies", description: "Control whether to always or never get pinged on message replies, with a whitelist feature",
authors: [Devs.ant0n, Devs.MrDiamond], authors: [Devs.ant0n, Devs.MrDiamond],
settings, settings,
@ -36,13 +42,20 @@ export default definePlugin({
const user = UserStore.getCurrentUser(); const user = UserStore.getCurrentUser();
if (message.author.id === user.id) if (message.author.id === user.id)
return; return;
if (this.getRepliedMessage(message)?.author.id !== user.id)
const repliedMessage = this.getRepliedMessage(message);
if (!repliedMessage || repliedMessage.author.id !== user.id)
return; return;
if (!settings.store.alwaysPingOnReply) const whitelist = parseWhitelist(settings.store.replyPingWhitelist);
const isWhitelisted = whitelist.includes(message.author.id);
if (isWhitelisted || settings.store.alwaysPingOnReply) {
if (!message.mentions.some(mention => mention.id === user.id))
message.mentions.push(user as any);
} else {
message.mentions = message.mentions.filter(mention => mention.id !== user.id); message.mentions = message.mentions.filter(mention => mention.id !== user.id);
else if (!message.mentions.some(mention => mention.id === user.id)) }
message.mentions.push(user as any);
}, },
getRepliedMessage(message: MessageJSON) { getRepliedMessage(message: MessageJSON) {
@ -50,3 +63,24 @@ export default definePlugin({
return ref && MessageStore.getMessage(ref.channel_id, ref.message_id); return ref && MessageStore.getMessage(ref.channel_id, ref.message_id);
}, },
}); });
function validateWhitelist(value: string) {
const whitelist = parseWhitelist(value);
if (whitelist.some(id => !isValidUserId(id))) {
showToast("Invalid User ID: One or more User IDs in the whitelist are invalid. Please check your input.");
return false;
}
showToast("Whitelist Updated: Reply ping whitelist has been successfully updated.");
return true;
}
function parseWhitelist(value: string) {
return value.split(",")
.map(id => id.trim())
.filter(id => id !== "");
}
function isValidUserId(id: string) {
return /^\d+$/.test(id); // Validates if the string is a number (basic validation for Discord User IDs)
}