fix: get original at start

replugged-org · Jan 24, 2025 · e8a1e5c · e8a1e5c
1 parent adeccd3
commit e8a1e5c
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/src/renderer/coremods/rdtComponentSourceFix/index.ts b/src/renderer/coremods/rdtComponentSourceFix/index.ts
@@ -1,13 +1,19 @@
+import { AnyFunction } from "src/types";
+
+function getOriginal(fn?: AnyFunction): AnyFunction | undefined {
+  return Function.toString.apply(fn).includes("objInjections.injections.get(funcName)")
+    ? (fn?.prototype?.constructor ?? fn)
+    : fn;
+}
+
 export function start(): void {
-  let $$type = window.$type;
+  let $$type = getOriginal(window.$type);
   Object.defineProperty(window, "$type", {
     get: () => {
       return $$type;
     },
     set: (v) => {
-      $$type = Function.toString.apply(v).includes("objInjections.injections.get(funcName)")
-        ? (v?.prototype?.constructor ?? v)
-        : v;
+      $$type = getOriginal(v);
     },
     configurable: true,
     enumerable: true,