From 63d8dfdbd0628083c70effabe1211765f3d9fb86 Mon Sep 17 00:00:00 2001
From: bry <bryanzettler@gmail.com>
Date: Fri, 6 Sep 2024 14:43:49 -0500
Subject: [PATCH 1/2] fixes wallet-adapter not showing up on jup.ag

---
 src/features/browser/BrowserWebViewScreen.tsx | 60 ++++++++++++-------
 src/features/browser/walletStandard.js        | 24 ++++----
 2 files changed, 50 insertions(+), 34 deletions(-)

diff --git a/src/features/browser/BrowserWebViewScreen.tsx b/src/features/browser/BrowserWebViewScreen.tsx
index ab0afd9e3..98d9babf5 100644
--- a/src/features/browser/BrowserWebViewScreen.tsx
+++ b/src/features/browser/BrowserWebViewScreen.tsx
@@ -66,7 +66,7 @@ const BrowserWebViewScreen = () => {
   const { favorites, addFavorite, removeFavorite } = useBrowser()
   const isAndroid = useMemo(() => Platform.OS === 'android', [])
   const spacing = useSpacing()
-  const [injected, setInjected] = useState(false)
+  const [isScriptInjected, setIsScriptInjected] = useState(false)
 
   const isFavorite = useMemo(() => {
     return favorites.some((favorite) => favorite === currentUrl)
@@ -376,6 +376,8 @@ const BrowserWebViewScreen = () => {
   )
 
   const injectedJavascript = useCallback(() => {
+    if (isScriptInjected) return ''
+
     const script = `
     ${injectWalletStandard.toString()}
 
@@ -387,15 +389,43 @@ const BrowserWebViewScreen = () => {
     `
 
     return script
-  }, [accountAddress, isAndroid])
+  }, [accountAddress, isAndroid, isScriptInjected])
 
-  // Inject wallet standard into the webview
   const injectModule = useCallback(() => {
-    if (!webview?.current) return
+    if (!webview?.current || isScriptInjected) return
+    setIsScriptInjected(true)
+
+    const injectionScript = `
+      (function() {
+        function injectWhenReady() {
+          ${injectedJavascript()}
+        }
+
+        if (document.readyState === 'loading') {
+          document.addEventListener('wallet-standard:app-ready', injectWhenReady);
+        } else {
+          injectWhenReady();
+        }
+      })();
+    `
 
-    const script = injectedJavascript()
-    webview?.current?.injectJavaScript(script)
-  }, [injectedJavascript])
+    webview.current.injectJavaScript(injectionScript)
+  }, [injectedJavascript, isScriptInjected])
+
+  const onLoadStart = useCallback(() => {
+    setIsScriptInjected(false)
+  }, [])
+
+  const onLoadEnd = useCallback(() => {
+    if (!isScriptInjected) {
+      injectModule()
+    }
+  }, [isScriptInjected, injectModule])
+
+  const onRefresh = useCallback(() => {
+    setIsScriptInjected(false)
+    webview.current?.reload()
+  }, [])
 
   const onNavigationChange = useCallback((event: WebViewNavigation) => {
     const baseUrl = event.url.replace('https://', '').split('/')
@@ -450,20 +480,6 @@ const BrowserWebViewScreen = () => {
     }
   }, [addFavorite, removeFavorite, isFavorite, currentUrl])
 
-  const onRefresh = useCallback(() => {
-    webview.current?.injectJavaScript('')
-    webview.current?.reload()
-    injectModule()
-  }, [injectModule])
-
-  const onLoadEnd = useCallback(() => {
-    if (!injected) {
-      webview.current?.injectJavaScript('')
-      webview.current?.injectJavaScript(injectedJavascript())
-      setInjected(true)
-    }
-  }, [injectedJavascript, injected, setInjected])
-
   const BrowserFooter = useCallback(() => {
     return (
       <Box padding="m" flexDirection="row" backgroundColor="black900">
@@ -512,7 +528,9 @@ const BrowserWebViewScreen = () => {
             ref={webview}
             originWhitelist={['*']}
             javaScriptEnabled
+            onLoadStart={onLoadStart}
             onLoadEnd={onLoadEnd}
+            injectedJavaScriptBeforeContentLoaded={injectedJavascript()}
             onNavigationStateChange={onNavigationChange}
             onMessage={onMessage}
             source={{ uri }}
diff --git a/src/features/browser/walletStandard.js b/src/features/browser/walletStandard.js
index 4385436f1..53c32862f 100644
--- a/src/features/browser/walletStandard.js
+++ b/src/features/browser/walletStandard.js
@@ -397,7 +397,7 @@ class HeliumWallet {
                         })),
                     }
                 })
-                
+
                 resolve(signedMessages)
               }
             }
@@ -406,26 +406,24 @@ class HeliumWallet {
     }
 }
 
-window.heliumWallet = wallet
 const walletObj = new HeliumWallet(wallet)
 
-const registerEvent = new CustomEvent('wallet-standard:register-wallet', {
-    bubbles: false,
-    cancelable: false,
-    composed: false,
-    detail: ({ register }) => register(walletObj),
-})
-
-window.dispatchEvent(registerEvent)
-
-// Attach the reference to the window, guarding against errors.
 try {
     Object.defineProperty(window, 'heliumWallet', { value: wallet })
 } catch (error) {
     console.error(error)
 }
 
-parent.addEventListener('wallet-standard:app-ready', function (event) {
+// Listen for the 'wallet-standard:app-ready' event
+window.addEventListener('wallet-standard:app-ready', function (event) {
+    const registerEvent = new CustomEvent('wallet-standard:register-wallet', {
+        bubbles: false,
+        cancelable: false,
+        composed: false,
+        detail: ({ register }) => register(walletObj),
+    })
+
+    window.dispatchEvent(registerEvent)
     event.detail.register(walletObj)
 })
 }`

From 4af42a9ba7e9428384cae4b94c17f2f4d77b5f98 Mon Sep 17 00:00:00 2001
From: bry <bryanzettler@gmail.com>
Date: Fri, 6 Sep 2024 14:47:42 -0500
Subject: [PATCH 2/2] Use more generic initial event

---
 src/features/browser/BrowserWebViewScreen.tsx | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/features/browser/BrowserWebViewScreen.tsx b/src/features/browser/BrowserWebViewScreen.tsx
index 98d9babf5..829c9549b 100644
--- a/src/features/browser/BrowserWebViewScreen.tsx
+++ b/src/features/browser/BrowserWebViewScreen.tsx
@@ -402,7 +402,7 @@ const BrowserWebViewScreen = () => {
         }
 
         if (document.readyState === 'loading') {
-          document.addEventListener('wallet-standard:app-ready', injectWhenReady);
+          document.addEventListener('DOMContentLoaded', injectWhenReady);
         } else {
           injectWhenReady();
         }