Pass managed user id header when fetching wallet balance

interledger · Oct 3, 2024 · 4c0a5c6 · 4c0a5c6
1 parent 237b91a
commit 4c0a5c6
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 3 deletions.
diff --git a/packages/wallet/backend/src/account/service.ts b/packages/wallet/backend/src/account/service.ts
@@ -162,9 +162,19 @@ export class AccountService implements IAccountService {
   }
 
   async getAccountBalance(account: Account): Promise<number> {
+    const user = await User.query()
+      .findById(account.userId)
+      .select('gateHubUserId')
+
+    if (!user || !user.gateHubUserId) {
+      throw new NotFound()
+    }
+
     const balances = await this.gateHubClient.getWalletBalance(
-      account.gateHubWalletId
+      account.gateHubWalletId,
+      user.gateHubUserId
     )
+
     return Number(
       balances.find((balance) => balance.vault.asset_code === account.assetCode)
         ?.total ?? 0

diff --git a/packages/wallet/backend/src/gatehub/client.ts b/packages/wallet/backend/src/gatehub/client.ts
@@ -267,10 +267,15 @@ export class GateHubClient {
     return response
   }
 
-  async getWalletBalance(walletId: string): Promise<IWalletBalance[]> {
+  async getWalletBalance(
+    walletId: string,
+    managedUserUuid: string
+  ): Promise<IWalletBalance[]> {
     const url = `${this.apiUrl}/core/v1/wallets/${walletId}/balances`
 
-    const response = await this.request<IWalletBalance[]>('GET', url)
+    const response = await this.request<IWalletBalance[]>('GET', url, '', {
+      managedUserUuid
+    })
 
     return response
   }