Fix/get balance

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

src/components/Auth/index.tsx

@@ -55,12 +55,10 @@
       setPubKey('')
     }
 
-    await updateBudget(setBudget)
-
     if (isE2E || isDevelopment) {
       setIsAuthenticated(true)
     }
-  }, [setBudget, setPubKey, setIsAuthenticated])
+  }, [setPubKey, setIsAuthenticated])
 
   const handleIsAdmin = useCallback(async () => {
     try {
@@ -112,15 +110,17 @@
 
           await handleAuth()
         } catch (error) {
           console.log(error)
         }
       }
 
+      await updateBudget(setBudget)
+
       await handleIsAdmin()
     }
 
     init()
-  }, [handleAuth, handleIsAdmin])
+  }, [handleAuth, handleIsAdmin, setBudget])
 
   const message = 'This is a private Graph, Contact Admin'
 

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) {