Request Safe info from Safe apps

5afe · Jul 15, 2020 · 4f8be8f · 4f8be8f
1 parent 1aa94ed
commit 4f8be8f
Showing 1 changed file with 16 additions and 7 deletions.
diff --git a/src/routes/safe/components/Apps/index.tsx b/src/routes/safe/components/Apps/index.tsx
@@ -54,8 +54,9 @@ const IframeCoverLoading = styled.div`
   background: white;
 `
 const operations = {
-  SEND_TRANSACTIONS: 'SEND_TRANSACTIONS',
   ON_SAFE_INFO: 'ON_SAFE_INFO',
+  GET_SAFE_INFO: 'GET_SAFE_INFO',
+  SEND_TRANSACTIONS: 'SEND_TRANSACTIONS',
 }
 
 function Apps({ closeModal, closeSnackbar, enqueueSnackbar, openModal }) {
@@ -218,6 +219,11 @@ function Apps({ closeModal, closeSnackbar, enqueueSnackbar, openModal }) {
 
   const getEnabledApps = () => appList.filter((a) => !a.disabled)
 
+  const sendMessageToIframe = useCallback((messageId, data) => {
+    const app = getSelectedApp()
+    iframeEl.contentWindow.postMessage({ messageId, data }, app.url)
+  })
+
   // handle messages from iframe
   useEffect(() => {
     const handleIframeMessage = (data) => {
@@ -230,7 +236,6 @@ function Apps({ closeModal, closeSnackbar, enqueueSnackbar, openModal }) {
         case operations.SEND_TRANSACTIONS: {
           const onConfirm = async () => {
             closeModal()
-
             await sendTransactions(
               dispatch,
               safeAddress,
@@ -252,7 +257,15 @@ function Apps({ closeModal, closeSnackbar, enqueueSnackbar, openModal }) {
             closeModal,
             onConfirm,
           )
+          break
+        }
 
+        case operations.GET_SAFE_INFO: {
+          sendMessageToIframe(operations.ON_SAFE_INFO, {
+            safeAddress,
+            network,
+            ethBalance,
+          })
           break
         }
 
@@ -343,10 +356,6 @@ function Apps({ closeModal, closeSnackbar, enqueueSnackbar, openModal }) {
 
   // on iframe change
   useEffect(() => {
-    const sendMessageToIframe = (messageId, data) => {
-      const app = getSelectedApp()
-      iframeEl.contentWindow.postMessage({ messageId, data }, app.url)
-    }
     const onIframeLoaded = () => {
       setAppIsLoading(false)
       sendMessageToIframe(operations.ON_SAFE_INFO, {
@@ -366,7 +375,7 @@ function Apps({ closeModal, closeSnackbar, enqueueSnackbar, openModal }) {
     return () => {
       iframeEl.removeEventListener('load', onIframeLoaded)
     }
-  }, [ethBalance, getSelectedApp, iframeEl, network, safeAddress, selectedApp])
+  }, [ethBalance, getSelectedApp, iframeEl, network, safeAddress, selectedApp, sendMessageToIframe])
 
   if (loading) {
     return <Loader size="md" />