From 0825033850c8d5ef538a87f64786f2a9d6c77aba Mon Sep 17 00:00:00 2001
From: BenyFilho <168232825+BenyFilho@users.noreply.github.com>
Date: Thu, 25 Sep 2025 18:37:32 -0300
Subject: [PATCH] Fix Disconnections - **UPDATED REINSTALL IT** (#3811)

* Update window.injectToFunction

* Fix window.injectToFunction

* window.injectToFunction

* Fix window.injectToFunction

* Fix window.injectToFunction

* Removing override isLidMigrated
---
 src/util/Injected/Store.js | 36 ++++++++++++++++++++++++------------
 1 file changed, 24 insertions(+), 12 deletions(-)

diff --git a/src/util/Injected/Store.js b/src/util/Injected/Store.js
index 05bfa31..07c475d 100644
--- a/src/util/Injected/Store.js
+++ b/src/util/Injected/Store.js
@@ -211,23 +211,35 @@ exports.ExposeStore = () => {
      * @param {Function} callback Modified function
      */
     window.injectToFunction = (target, callback) => {
-        let module = window.require(target.module);
+        try {
+            let module = window.require(target.module);
+            if (!module) return; 
 
-        if (!module) return;
-        
-        const path = target.function.split('.');
-        const funcName = path.pop();
-        for (const key of path) {
-            module = module[key];
+            const path = target.function.split('.');
+            const funcName = path.pop();
+
+            for (const key of path) {
+                if (!module[key]) return;
+                module = module[key];
+            }
+
+            const originalFunction = module[funcName];
+            if (typeof originalFunction !== 'function') return;
+
+            module[funcName] = (...args) => {
+                try {
+                    return callback(originalFunction, ...args);
+                } catch {
+                    return originalFunction(...args);
+                }
+            };
+
+        } catch {
+            return;
         }
-
-        const originalFunction = module[funcName];
-        module[funcName] = (...args) => callback(originalFunction, ...args);
     };
 
     window.injectToFunction({ module: 'WAWebBackendJobsCommon', function: 'mediaTypeFromProtobuf' }, (func, ...args) => { const [proto] = args; return proto.locationMessage ? null : func(...args); });
 
     window.injectToFunction({ module: 'WAWebE2EProtoUtils', function: 'typeAttributeFromProtobuf' }, (func, ...args) => { const [proto] = args; return proto.locationMessage || proto.groupInviteMessage ? 'text' : func(...args); });
-
-    window.injectToFunction({ module: 'WAWebLid1X1MigrationGating', function: 'Lid1X1MigrationUtils.isLidMigrated' }, () => false);
 };