@pushchain/ui-kit 5.1.1 → 5.2.0-alpha.1

package/src/lib/context/WalletContext.js

@@ -12,6 +12,9 @@ const WalletProviderRegistry_1 = require("../providers/walletProviders/WalletPro
 const PushWalletToast_1 = require("../components/PushWalletToast");
 const LoginModal_1 = require("../components/LoginModal");
 const WalletContextMap_1 = require("./WalletContextMap");
+const useWaapAuth_1 = require("../providers/waap/useWaapAuth");
+const waapProvider_1 = require("../providers/waap/waapProvider");
+const PushChainContext_1 = require("./PushChainContext");
 exports.WalletContext = (0, react_1.createContext)(null);
 const WalletContextProvider = ({ children, config, app, themeMode = constants_1.PushUI.CONSTANTS.THEME.DARK, themeOverrides, }) => {
     const [universalAccount, setUniversalAccount] = (0, react_1.useState)(null);
@@ -23,7 +26,6 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
     const [connectionStatus, setConnectionStatus] = (0, react_1.useState)(types_1.ConnectionStatus.NOT_CONNECTED);
     const [externalWallet, setExternalWallet] = (0, react_1.useState)(null); // to connect with external wallet
     const [progress, setProgress] = (0, react_1.useState)(null);
-    const signatureResolverRef = (0, react_1.useRef)(null);
     const upgradeResolverRef = (0, react_1.useRef)(null);
     const pushChainClientRef = (0, react_1.useRef)(null);
     const setPushChainClient = (client) => {
@@ -31,25 +33,32 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
     };
     const [modalAppData, setModalAppData] = (0, react_1.useState)(app
         ? {
-            title: app === null || app === void 0 ? void 0 : app.title,
-            logoURL: app === null || app === void 0 ? void 0 : app.logoUrl,
-            description: app === null || app === void 0 ? void 0 : app.description,
+            title: app?.title,
+            logoURL: app?.logoUrl,
+            description: app?.description,
         }
         : undefined);
     const [walletAppData, setWalletAppData] = (0, react_1.useState)(app
         ? {
-            title: app === null || app === void 0 ? void 0 : app.title,
-            logoURL: app === null || app === void 0 ? void 0 : app.logoUrl,
-            description: app === null || app === void 0 ? void 0 : app.description,
+            title: app?.title,
+            logoURL: app?.logoUrl,
+            description: app?.description,
         }
         : undefined);
     const [activeTriggerId, setActiveTriggerId] = (0, react_1.useState)();
+    const { loginWithWaapSocial } = (0, useWaapAuth_1.useWaapAuth)();
     const toggleButtonRefs = react_1.default.useRef({});
     const updateModalAppData = (newData) => {
-        setModalAppData((prevData) => (Object.assign(Object.assign({}, prevData), newData)));
+        setModalAppData((prevData) => ({
+            ...prevData,
+            ...newData,
+        }));
     };
     const updateWalletAppData = (newData) => {
-        setWalletAppData((prevData) => (Object.assign(Object.assign({}, prevData), newData)));
+        setWalletAppData((prevData) => ({
+            ...prevData,
+            ...newData,
+        }));
     };
     const handleConnectToPushWallet = () => {
         setWalletVisibility(true);
@@ -65,13 +74,15 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
         };
         sendMessageToPushWallet({
             type: constants_1.APP_TO_WALLET_ACTION.WALLET_CONFIG,
-            data: Object.assign({}, walletConfig),
+            data: {
+                ...walletConfig,
+            },
         });
     };
     const handleUserLogOutEvent = () => {
         if (externalWallet) {
             const providerReceived = WalletProviderRegistry_1.walletRegistry.getProvider(externalWallet.providerName);
-            providerReceived === null || providerReceived === void 0 ? void 0 : providerReceived.disconnect();
+            providerReceived?.disconnect();
         }
         setConnectionStatus(types_1.ConnectionStatus.NOT_CONNECTED);
         setUniversalAccount(null);
@@ -80,13 +91,12 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
         setIframeLoading(true);
         setExternalWallet(null);
         setIsReadOnly(false);
-        localStorage.removeItem(`walletInfo_${(config === null || config === void 0 ? void 0 : config.uid) || 'default'}`);
+        localStorage.removeItem(`walletInfo_${config?.uid || 'default'}`);
         document.body.style.overflow = '';
     };
     // sending events to wallet from dapp
     const sendMessageToPushWallet = (message) => {
-        var _a;
-        if ((_a = iframeRef === null || iframeRef === void 0 ? void 0 : iframeRef.current) === null || _a === void 0 ? void 0 : _a.contentWindow) {
+        if (iframeRef?.current?.contentWindow) {
             try {
                 iframeRef.current.contentWindow.postMessage(message, constants_1.WALLET_CONFIG_URL[config.network]);
             }
@@ -114,9 +124,9 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
         if (response.account) {
             setUniversalAccount(response.account);
         }
-        localStorage.setItem(`walletInfo_${(config === null || config === void 0 ? void 0 : config.uid) || 'default'}`, JSON.stringify({
+        localStorage.setItem(`walletInfo_${config?.uid || 'default'}`, JSON.stringify({
             account: response.account,
-            uid: (config === null || config === void 0 ? void 0 : config.uid) || 'default'
+            uid: config?.uid || 'default'
         }));
     };
     const handleAppConnectionRejection = () => {
@@ -124,13 +134,13 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
         setUniversalAccount(null);
     };
     // Connect external wallet
-    const handleExternalWalletConnection = (data) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    const handleExternalWalletConnection = async (data) => {
         try {
             const providerReceived = WalletProviderRegistry_1.walletRegistry.getProvider(data.provider);
             if (!providerReceived) {
                 return;
             }
-            const walletInfo = yield providerReceived.connect(data.chain);
+            const walletInfo = await providerReceived.connect(data.chain);
             setConnectionStatus(types_1.ConnectionStatus.CONNECTED);
             setMinimiseWallet(true);
             const result = core_1.PushChain.utils.account.fromChainAgnostic(walletInfo.caipAddress);
@@ -140,14 +150,17 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
                 providerName: data.provider,
                 chainType: data.chain,
             };
-            localStorage.setItem(`walletInfo_${(config === null || config === void 0 ? void 0 : config.uid) || 'default'}`, JSON.stringify({
+            localStorage.setItem(`walletInfo_${config?.uid || 'default'}`, JSON.stringify({
                 wallet: connectedWallet,
-                uid: (config === null || config === void 0 ? void 0 : config.uid) || 'default'
+                uid: config?.uid || 'default'
             }));
             setExternalWallet(connectedWallet);
             sendMessageToPushWallet({
                 type: constants_1.APP_TO_WALLET_ACTION.CONNECTION_STATUS,
-                data: Object.assign({ status: 'successful' }, connectedWallet),
+                data: {
+                    status: 'successful',
+                    ...connectedWallet,
+                },
             });
         }
         catch (error) {
@@ -160,9 +173,9 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
             });
             throw new Error('Failed to connect to provider');
         }
-    });
+    };
     // handles external wallet signature request
-    const handleExternalWalletSignRequest = (data) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    const handleExternalWalletSignRequest = async (data) => {
         if (!externalWallet) {
             throw new Error('No External wallet connected');
         }
@@ -171,15 +184,16 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
             if (!providerReceived) {
                 throw new Error('Provider not found');
             }
-            const signature = yield providerReceived.signMessage(data);
+            await providerReceived.switchNetwork?.(externalWallet.chainType);
+            const signature = await providerReceived.signMessage(data);
             return signature;
         }
         catch (error) {
             console.log('Error in generating signature', error);
             throw new Error('Signature request failed');
         }
-    });
-    const handleExternalWalletSignTransactionRequest = (data) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    };
+    const handleExternalWalletSignTransactionRequest = async (data) => {
         if (!externalWallet) {
             throw new Error('No External wallet connected');
         }
@@ -188,15 +202,16 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
             if (!providerReceived) {
                 throw new Error('Provider not found');
             }
-            const signature = yield providerReceived.signAndSendTransaction(data);
+            await providerReceived.switchNetwork?.(externalWallet.chainType);
+            const signature = await providerReceived.signAndSendTransaction(data);
             return signature;
         }
         catch (error) {
             console.log('Error in generating signature', error);
             throw new Error('Signature request failed');
         }
-    });
-    const handleExternalWalletSignTypedDataRequest = (data) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    };
+    const handleExternalWalletSignTypedDataRequest = async (data) => {
         if (!externalWallet) {
             throw new Error('No External wallet connected');
         }
@@ -205,124 +220,118 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
             if (!providerReceived) {
                 throw new Error('Provider not found');
             }
-            const signature = yield providerReceived.signTypedData(data);
+            await providerReceived.switchNetwork?.(externalWallet.chainType);
+            const signature = await providerReceived.signTypedData(data);
             return signature;
         }
         catch (error) {
             console.log('Error in generating signature', error);
             throw new Error('Signature request failed');
         }
-    });
-    // handles Push wallet signature request
-    const handleSendSignRequestToPushWallet = (data) => {
-        return new Promise((resolve, reject) => {
-            if (signatureResolverRef.current) {
-                reject(new Error('Another sign request is already in progress'));
-                return;
-            }
-            setMinimiseWallet(false);
-            signatureResolverRef.current = {
-                success: (response) => {
-                    resolve(response.signature);
-                    signatureResolverRef.current = null; // Clean up
-                    setMinimiseWallet(true);
-                },
-                error: (response) => {
-                    signatureResolverRef.current = null; // Clean up
-                    reject(new Error('Signature request failed'));
-                    setMinimiseWallet(true);
-                },
-            };
-            // Send the sign request to the wallet tab
+    };
+    const handleSocialConnection = async () => {
+        const result = await loginWithWaapSocial();
+        if (!result) {
             sendMessageToPushWallet({
-                type: constants_1.APP_TO_WALLET_ACTION.SIGN_MESSAGE,
-                data,
+                type: constants_1.APP_TO_WALLET_ACTION.SOCIAL_CONNECTION_STATUS,
+                data: {
+                    error: true,
+                },
             });
+            return;
+        }
+        ;
+        const w = await core_1.PushChain.utils.account.convertExecutorToOriginAccount(result.address);
+        if (!w.account)
+            return;
+        sendMessageToPushWallet({
+            type: constants_1.APP_TO_WALLET_ACTION.SOCIAL_CONNECTION_STATUS,
+            data: {
+                account: w.account,
+            },
         });
     };
-    const handleSendSignTransactionRequestToPushWallet = (data) => {
-        return new Promise((resolve, reject) => {
-            if (signatureResolverRef.current) {
-                reject(new Error('Another sign request is already in progress'));
-                return;
-            }
-            setMinimiseWallet(false);
-            signatureResolverRef.current = {
-                success: (response) => {
-                    resolve(response.signature);
-                    signatureResolverRef.current = null; // Clean up
-                    setMinimiseWallet(true);
-                },
-                error: (response) => {
-                    signatureResolverRef.current = null; // Clean up
-                    reject(new Error('Signature request failed'));
-                    setMinimiseWallet(true);
-                },
-            };
-            // Send the sign request to the wallet tab
+    const handleSocialSignAndSendMessage = async (message) => {
+        try {
+            const signature = await (0, waapProvider_1.waapSignMessage)(message);
             sendMessageToPushWallet({
-                type: constants_1.APP_TO_WALLET_ACTION.SIGN_TRANSACTION,
-                data,
+                type: constants_1.APP_TO_WALLET_ACTION.SIGN_MESSAGE,
+                data: { signature },
             });
-        });
-    };
-    const handleSendSignTypedDataRequestToPushWallet = (data) => {
-        return new Promise((resolve, reject) => {
-            if (signatureResolverRef.current) {
-                reject(new Error('Another sign request is already in progress'));
-                return;
-            }
-            setMinimiseWallet(false);
-            signatureResolverRef.current = {
-                success: (response) => {
-                    resolve(response.signature);
-                    signatureResolverRef.current = null; // Clean up
-                    setMinimiseWallet(true);
+        }
+        catch (error) {
+            sendMessageToPushWallet({
+                type: constants_1.APP_TO_WALLET_ACTION.ERROR,
+                data: {
+                    error: error,
                 },
-                error: (response) => {
-                    signatureResolverRef.current = null; // Clean up
-                    reject(new Error('Signature request failed'));
-                    setMinimiseWallet(true);
+            });
+        }
+    };
+    const handleSocialSignAndSendTransaction = async (txn) => {
+        try {
+            const signature = await (0, waapProvider_1.waapSignAndSendTransaction)(txn);
+            sendMessageToPushWallet({
+                type: constants_1.WALLET_TO_APP_ACTION.SIGN_TRANSACTION,
+                data: { signature },
+            });
+        }
+        catch (error) {
+            sendMessageToPushWallet({
+                type: constants_1.WALLET_TO_APP_ACTION.ERROR,
+                data: {
+                    error: error,
                 },
-            };
-            // Send the sign request to the wallet tab
+            });
+        }
+    };
+    const handleSocialSignTypedData = async (typedData) => {
+        try {
+            const signature = await (0, waapProvider_1.waapSignTypedData)(typedData);
             sendMessageToPushWallet({
-                type: constants_1.APP_TO_WALLET_ACTION.SIGN_TYPED_DATA,
-                data,
+                type: constants_1.WALLET_TO_APP_ACTION.SIGN_TYPED_DATA,
+                data: { signature },
             });
-        });
+        }
+        catch (error) {
+            sendMessageToPushWallet({
+                type: constants_1.WALLET_TO_APP_ACTION.ERROR,
+                data: {
+                    error: error,
+                },
+            });
+        }
     };
-    // sending Message sign request to wallet based on which wallet is connected (external or pushwallet)
-    const handleSignMessage = (data) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    const handleSignMessage = async (data) => {
         let signature;
         if (externalWallet) {
-            signature = yield handleExternalWalletSignRequest(data);
+            signature = await handleExternalWalletSignRequest(data);
         }
         else {
-            signature = yield handleSendSignRequestToPushWallet(data);
+            signature = await (0, waapProvider_1.waapSignMessage)(data);
         }
         return signature;
-    });
-    const handleSignAndSendTransaction = (data) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    };
+    const handleSignAndSendTransaction = async (data) => {
         let signature;
         if (externalWallet) {
-            signature = yield handleExternalWalletSignTransactionRequest(data);
+            signature = await handleExternalWalletSignTransactionRequest(data);
         }
         else {
-            signature = yield handleSendSignTransactionRequestToPushWallet(data);
+            signature = await (0, waapProvider_1.waapSignAndSendTransaction)(data);
         }
         return signature;
-    });
-    const handleSignTypedData = (data) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    };
+    const handleSignTypedData = async (data) => {
         let signature;
         if (externalWallet) {
-            signature = yield handleExternalWalletSignTypedDataRequest(data);
+            signature = await handleExternalWalletSignTypedDataRequest(data);
         }
         else {
-            signature = yield handleSendSignTypedDataRequestToPushWallet(data);
+            signature = await (0, waapProvider_1.waapSignTypedData)(data);
         }
         return signature;
-    });
+    };
     const getAuthWindowConfig = () => {
         // Calculate the screen width and height
         const screenWidth = window.screen.width;
@@ -367,13 +376,13 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
             }, 100000);
         });
     };
-    const checkAndShowUpgradeIfNeeded = (pushChainClient) => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
+    const checkAndShowUpgradeIfNeeded = async (pushChainClient) => {
         setPushChainClient(pushChainClient);
         if (!pushChainClient) {
             return false;
         }
         try {
-            yield pushChainClient.accountStatusReady;
+            await pushChainClient.accountStatusReady;
             if (!pushChainClient.isReadMode && pushChainClient.accountStatus.uea.loaded && pushChainClient.accountStatus.uea.requiresUpgrade) {
                 setMinimiseWallet(false);
                 sendMessageToPushWallet({
@@ -403,33 +412,30 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
             console.error('Error checking upgrade status:', error);
             return false;
         }
-    });
+    };
     const handleUpgradeAccountSuccess = () => {
-        var _a;
         setMinimiseWallet(true);
         // Resolve any pending upgrade promise
-        if ((_a = upgradeResolverRef.current) === null || _a === void 0 ? void 0 : _a.success) {
+        if (upgradeResolverRef.current?.success) {
             upgradeResolverRef.current.success();
             upgradeResolverRef.current = null;
         }
     };
     const handleUpgradeAccountError = (error) => {
-        var _a;
         setMinimiseWallet(true);
         // Reject any pending upgrade promise
-        if ((_a = upgradeResolverRef.current) === null || _a === void 0 ? void 0 : _a.error) {
+        if (upgradeResolverRef.current?.error) {
             upgradeResolverRef.current.error(error instanceof Error ? error : new Error('Upgrade failed'));
             upgradeResolverRef.current = null;
         }
     };
-    const handleUpgradeAccount = () => tslib_1.__awaiter(void 0, void 0, void 0, function* () {
-        var _a;
+    const handleUpgradeAccount = async () => {
         const pushChainClient = pushChainClientRef.current;
         if (!pushChainClient) {
             throw new Error('PushChain client not initialized');
         }
         try {
-            yield pushChainClient.upgradeAccount({
+            await pushChainClient.upgradeAccount({
                 progressHook: (progress) => {
                     setProgress(progress);
                     if (progress.id === progress_hook_types_1.PROGRESS_HOOK.UEA_MIG_9901) {
@@ -452,20 +458,19 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
                 data: { success: false, error: error instanceof Error ? error.message : 'Upgrade failed' },
             });
             // Reject the upgrade promise
-            if ((_a = upgradeResolverRef.current) === null || _a === void 0 ? void 0 : _a.error) {
+            if (upgradeResolverRef.current?.error) {
                 upgradeResolverRef.current.error(error instanceof Error ? error : new Error('Upgrade failed'));
             }
         }
-    });
+    };
     (0, react_1.useEffect)(() => {
-        var _a;
-        const walletInfo = localStorage.getItem(`walletInfo_${(config === null || config === void 0 ? void 0 : config.uid) || 'default'}`);
+        const walletInfo = localStorage.getItem(`walletInfo_${config?.uid || 'default'}`);
         const walletData = walletInfo ? JSON.parse(walletInfo) : null;
         if (!walletData)
             return;
-        if (walletData.uid !== (config === null || config === void 0 ? void 0 : config.uid))
+        if (walletData.uid !== config?.uid)
             return;
-        if ((walletData === null || walletData === void 0 ? void 0 : walletData.wallet) && ((_a = walletData.wallet) === null || _a === void 0 ? void 0 : _a.providerName)) {
+        if (walletData?.wallet && walletData.wallet?.providerName) {
             setUniversalAccount(core_1.PushChain.utils.account.fromChainAgnostic(walletData.wallet.address));
             setExternalWallet(walletData.wallet);
         }
@@ -485,25 +490,33 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
         if (externalWallet) {
             sendMessageToPushWallet({
                 type: constants_1.APP_TO_WALLET_ACTION.READ_ONLY_CONNECTION_STATUS,
-                data: Object.assign({ status: 'successful' }, externalWallet),
+                data: {
+                    status: 'successful',
+                    ...externalWallet,
+                },
             });
         }
         else if (universalAccount) {
             sendMessageToPushWallet({
                 type: constants_1.APP_TO_WALLET_ACTION.READ_ONLY_CONNECTION_STATUS,
-                data: Object.assign({ status: 'successful' }, universalAccount),
+                data: {
+                    status: 'successful',
+                    ...universalAccount,
+                },
             });
         }
     }, [isIframeLoading, externalWallet]);
     (0, react_1.useEffect)(() => {
         const messageHandler = (event) => {
-            var _a, _b, _c, _d, _e, _f, _g, _h, _j;
-            if (((_a = iframeRef.current) === null || _a === void 0 ? void 0 : _a.contentWindow) !== event.source)
+            if (iframeRef.current?.contentWindow !== event.source)
                 return;
             switch (event.data.type) {
                 case constants_1.WALLET_TO_APP_ACTION.CONNECT_EXTERNAL_WALLET:
                     handleExternalWalletConnection(event.data.data);
                     break;
+                case constants_1.WALLET_TO_APP_ACTION.CONNECT_SOCIAL:
+                    handleSocialConnection();
+                    break;
                 case constants_1.WALLET_TO_APP_ACTION.IS_LOGGED_IN:
                     handleIsLoggedInAction();
                     break;
@@ -513,27 +526,18 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
                 case constants_1.WALLET_TO_APP_ACTION.APP_CONNECTION_REJECTED:
                     handleAppConnectionRejection();
                     break;
-                case constants_1.WALLET_TO_APP_ACTION.SIGN_MESSAGE:
-                    if (signatureResolverRef.current) {
-                        (_c = (_b = signatureResolverRef === null || signatureResolverRef === void 0 ? void 0 : signatureResolverRef.current) === null || _b === void 0 ? void 0 : _b.success) === null || _c === void 0 ? void 0 : _c.call(_b, event.data.data);
-                    }
+                case constants_1.APP_TO_WALLET_ACTION.SIGN_MESSAGE:
+                    handleSocialSignAndSendMessage(event.data.data);
                     break;
-                case constants_1.WALLET_TO_APP_ACTION.SIGN_TRANSACTION:
-                    if (signatureResolverRef.current) {
-                        (_e = (_d = signatureResolverRef === null || signatureResolverRef === void 0 ? void 0 : signatureResolverRef.current) === null || _d === void 0 ? void 0 : _d.success) === null || _e === void 0 ? void 0 : _e.call(_d, event.data.data);
-                    }
+                case constants_1.APP_TO_WALLET_ACTION.SIGN_TRANSACTION:
+                    handleSocialSignAndSendTransaction(event.data.data);
                     break;
-                case constants_1.WALLET_TO_APP_ACTION.SIGN_TYPED_DATA:
-                    if (signatureResolverRef.current) {
-                        (_g = (_f = signatureResolverRef === null || signatureResolverRef === void 0 ? void 0 : signatureResolverRef.current) === null || _f === void 0 ? void 0 : _f.success) === null || _g === void 0 ? void 0 : _g.call(_f, event.data.data);
-                    }
+                case constants_1.APP_TO_WALLET_ACTION.SIGN_TYPED_DATA:
+                    handleSocialSignTypedData(event.data.data);
                     break;
                 case constants_1.WALLET_TO_APP_ACTION.IS_LOGGED_OUT:
                     handleUserLogOutEvent();
                     break;
-                case constants_1.WALLET_TO_APP_ACTION.ERROR:
-                    (_j = (_h = signatureResolverRef === null || signatureResolverRef === void 0 ? void 0 : signatureResolverRef.current) === null || _h === void 0 ? void 0 : _h.error) === null || _j === void 0 ? void 0 : _j.call(_h, event.data.data);
-                    break;
                 case constants_1.WALLET_TO_APP_ACTION.CLOSE_IFRAME:
                     if (universalAccount)
                         setMinimiseWallet(true);
@@ -556,8 +560,8 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
         window.addEventListener('message', messageHandler);
         return () => window.removeEventListener('message', messageHandler);
     }, [universalAccount]);
-    const WalletContext = (0, WalletContextMap_1.getWalletContext)((config === null || config === void 0 ? void 0 : config.uid) || 'default');
-    return ((0, jsx_runtime_1.jsxs)(WalletContext.Provider, { value: {
+    const WalletContext = (0, WalletContextMap_1.getWalletContext)(config?.uid || 'default');
+    return ((0, jsx_runtime_1.jsx)(WalletContext.Provider, { value: {
             app,
             config,
             connectionStatus,
@@ -586,7 +590,7 @@ const WalletContextProvider = ({ children, config, app, themeMode = constants_1.
             setActiveTriggerId,
             toggleButtonRefs,
             connectionType: externalWallet ? 'external' : 'social',
-        }, children: [(0, jsx_runtime_1.jsx)(LoginModal_1.LoginModal, { iframeRef: iframeRef, themeMode: themeMode, modalAppData: modalAppData, isWalletVisible: isWalletVisible, isIframeLoading: isIframeLoading, setIframeLoading: setIframeLoading, sendWalletConfig: sendWalletConfig, config: config, universalAccount: universalAccount, isWalletMinimised: isWalletMinimised, setMinimiseWallet: setMinimiseWallet, handleUserLogOutEvent: handleUserLogOutEvent, sendMessageToPushWallet: sendMessageToPushWallet, isReadOnly: isReadOnly, toggleButtonRefs: toggleButtonRefs, activeTriggerId: activeTriggerId }), progress && ((0, jsx_runtime_1.jsx)(PushWalletToast_1.PushWalletToast, { progress: progress, setProgress: setProgress })), children] }));
+        }, children: (0, jsx_runtime_1.jsxs)(PushChainContext_1.PushChainProvider, { uid: config?.uid || 'default', universalAccount: universalAccount, handleSignMessage: handleSignMessage, handleSignAndSendTransaction: handleSignAndSendTransaction, handleSignTypedData: handleSignTypedData, handleExternalWalletConnection: handleExternalWalletConnection, requestPushWalletConnection: requestPushWalletConnection, config: config, setProgress: setProgress, isReadOnly: isReadOnly, setIsReadOnly: setIsReadOnly, checkAndShowUpgradeIfNeeded: checkAndShowUpgradeIfNeeded, children: [(0, jsx_runtime_1.jsx)(LoginModal_1.LoginModal, { iframeRef: iframeRef, themeMode: themeMode, modalAppData: modalAppData, isWalletVisible: isWalletVisible, isIframeLoading: isIframeLoading, setIframeLoading: setIframeLoading, sendWalletConfig: sendWalletConfig, config: config, universalAccount: universalAccount, isWalletMinimised: isWalletMinimised, setMinimiseWallet: setMinimiseWallet, handleUserLogOutEvent: handleUserLogOutEvent, sendMessageToPushWallet: sendMessageToPushWallet, isReadOnly: isReadOnly, toggleButtonRefs: toggleButtonRefs, activeTriggerId: activeTriggerId }), progress && ((0, jsx_runtime_1.jsx)(PushWalletToast_1.PushWalletToast, { progress: progress, setProgress: setProgress })), children] }) }));
 };
 exports.WalletContextProvider = WalletContextProvider;
 //# sourceMappingURL=WalletContext.js.map