patches: Make $self more robust

1 files changed, 7 insertions, 3 deletions

diff --git a/src/utils/patches.ts b/src/utils/patches.ts
index 8ecd68e..0f83d40 100644
--- a/src/utils/patches.ts
+++ b/src/utils/patches.ts
@@ -27,9 +27,13 @@ export function canonicalizeMatch(match: RegExp | string) {
     return new RegExp(canonSource, match.flags);
 }
 
-export function canonicalizeReplace(replace: string | ReplaceFn, pluginName: string) {
-    if (typeof replace === "function") return replace;
-    return replace.replaceAll("$self", `Vencord.Plugins.plugins.${pluginName}`);
+export function canonicalizeReplace(replace: string | ReplaceFn, pluginName: string): string | ReplaceFn {
+    const self = `Vencord.Plugins.plugins[${JSON.stringify(pluginName)}]`;
+
+    if (typeof replace !== "function")
+        return replace.replaceAll("$self", self);
+
+    return (...args) => replace(...args).replaceAll("$self", self);
 }
 
 export function canonicalizeDescriptor<T>(descriptor: TypedPropertyDescriptor<T>, canonicalize: (value: T) => T) {