Merge pull request #2187 from stakwork/fix/get-balance

Fix/get balance
stakwork · Oct 17, 2024 · 8b1d62a · 8b1d62a
2 parents 75b81df + 445afa5
commit 8b1d62a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 6 deletions.
diff --git a/src/components/AddContentModal/index.tsx b/src/components/AddContentModal/index.tsx
@@ -113,6 +113,8 @@ const handleSubmitForm = async (
       Authorization: lsatToken,
     })
 
+    await updateBudget(setBudget)
+
     if (res.data.project_id) {
       setRunningProjectId(res.data.project_id)
     }
@@ -127,7 +129,7 @@ const handleSubmitForm = async (
   } catch (err: any) {
     if (err.status === 402) {
       await payLsat(setBudget)
-      await updateBudget(setBudget)
+
       await handleSubmitForm(data, sourceType, setBudget, setRunningProjectId)
     } else {
       let errorMessage = NODE_ADD_ERROR

diff --git a/src/components/Auth/index.tsx b/src/components/Auth/index.tsx
@@ -50,12 +50,10 @@ export const AuthGuard = ({ children }: PropsWithChildren) => {
       setPubKey('')
     }
 
-    await updateBudget(setBudget)
-
     if (isE2E || isDevelopment) {
       setIsAuthenticated(true)
     }
-  }, [setBudget, setPubKey, setIsAuthenticated])
+  }, [setPubKey, setIsAuthenticated])
 
   const handleIsAdmin = useCallback(async () => {
     try {
@@ -113,11 +111,13 @@ export const AuthGuard = ({ children }: PropsWithChildren) => {
         }
       }
 
+      await updateBudget(setBudget)
+
       await handleIsAdmin()
     }
 
     init()
-  }, [handleAuth, handleIsAdmin])
+  }, [handleAuth, handleIsAdmin, setBudget])
 
   const message = 'This is a private Graph, Contact Admin'
 

diff --git a/src/network/fetchGraphData/index.ts b/src/network/fetchGraphData/index.ts
@@ -1,7 +1,7 @@
 import { isDevelopment, isE2E } from '~/constants'
 import { api } from '~/network/api'
 import { FetchDataResponse } from '~/types'
-import { payLsat } from '~/utils'
+import { payLsat, updateBudget } from '~/utils'
 import { getLSat } from '~/utils/getLSat'
 
 // Main function to fetch graph data
@@ -30,6 +30,8 @@ const fetchNodes = async (
     try {
       const response = await api.get<FetchDataResponse>(url, { Authorization: lsatToken }, signal)
 
+      updateBudget(setBudget)
+
       return response
       // eslint-disable-next-line @typescript-eslint/no-explicit-any
     } catch (error: any) {