forked!!

src/equicordplugins/replyPingControl/index.ts

@@ -0,0 +1,90 @@
+/*
+ * Vencord, a Discord client mod
+ * Copyright (c) 2023 Vendicated and contributors
+ * SPDX-License-Identifier: GPL-3.0-or-later
+ */
+
+import { definePluginSettings } from "@api/Settings";
+import { Devs } from "@utils/constants";
+import definePlugin, { OptionType } from "@utils/types";
+import { MessageStore, showToast, UserStore } from "@webpack/common";
+import { MessageJSON } from "discord-types/general";
+
+let cachedWhitelist: string[] = [];
+
+export const settings = definePluginSettings({
+    alwaysPingOnReply: {
+        type: OptionType.BOOLEAN,
+        description: "Always get pinged when someone replies to your messages",
+        default: false,
+    },
+    replyPingWhitelist: {
+        type: OptionType.STRING,
+        description: "Comma-separated list of User IDs to always receive reply pings from",
+        default: "",
+        disabled: () => settings.store.alwaysPingOnReply,
+        onChange: newValue => {
+            const originalIDs = newValue.split(",")
+                .map(id => id.trim())
+                .filter(id => id !== "");
+
+            const isInvalid = originalIDs.some(id => !isValidUserId(id));
+
+            if (isInvalid) {
+                showToast("Invalid User ID: One or more User IDs in the whitelist are invalid. Please check your input.");
+            } else {
+                cachedWhitelist = originalIDs;
+                showToast("Whitelist Updated: Reply ping whitelist has been successfully updated.");
+            }
+        }
+    }
+});
+
+export default definePlugin({
+    name: "ReplyPingControl",
+    description: "Control whether to always or never get pinged on message replies, with a whitelist feature",
+    authors: [Devs.ant0n, Devs.MrDiamond],
+    settings,
+
+    patches: [{
+        find: "_channelMessages",
+        replacement: {
+            match: /receiveMessage\((\i)\)\{/,
+            replace: "$&$self.modifyMentions($1);"
+        }
+    }],
+
+    modifyMentions(message: MessageJSON) {
+        const user = UserStore.getCurrentUser();
+        if (message.author.id === user.id)
+            return;
+
+        const repliedMessage = this.getRepliedMessage(message);
+        if (!repliedMessage || repliedMessage.author.id !== user.id)
+            return;
+
+        const isWhitelisted = cachedWhitelist.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);
+        }
+    },
+
+    getRepliedMessage(message: MessageJSON) {
+        const ref = message.message_reference;
+        return ref && MessageStore.getMessage(ref.channel_id, ref.message_id);
+    },
+});
+
+function parseWhitelist(value: string) {
+    return value.split(",")
+        .map(id => id.trim())
+        .filter(id => id !== "");
+}
+
+function isValidUserId(id: string) {
+    return /^\d+$/.test(id);
+}