From 8acb35719d2a06fcf3294820dbd7724aa4d872e1 Mon Sep 17 00:00:00 2001
From: cryptooda <crypto.oda@gmail.com>
Date: Tue, 6 Jan 2026 20:15:53 +0700
Subject: [PATCH] Add PrivyBalanceResponse type for improved type safety in
 wallet balance retrieval

- Introduced a new type definition for Privy balance response to enhance type safety and clarity in the getWalletBalanceImpl function.
- Updated the makePrivyRequest call to utilize the new PrivyBalanceResponse type, streamlining the handling of balance data.
- This change aims to improve code maintainability and reduce potential errors during balance retrieval processes.
---
 .../src/plugins/custom/privy.ts               | 36 ++++++++++++-------
 1 file changed, 24 insertions(+), 12 deletions(-)

diff --git a/src/Managing.Web3Proxy/src/plugins/custom/privy.ts b/src/Managing.Web3Proxy/src/plugins/custom/privy.ts
index 77dc3810..70ae74a7 100644
--- a/src/Managing.Web3Proxy/src/plugins/custom/privy.ts
+++ b/src/Managing.Web3Proxy/src/plugins/custom/privy.ts
@@ -1095,6 +1095,22 @@ export const sendTokenImpl = async (
   }
 };
 
+/**
+ * Type for Privy balance response
+ */
+type PrivyBalanceResponse = {
+  balances: Array<{
+    chain: string;
+    asset: string;
+    raw_value: string;
+    raw_value_decimals: number;
+    display_values: {
+      [key: string]: string;
+      usd: string;
+    };
+  }>;
+};
+
 /**
  * Fallback function to get wallet balances using direct RPC calls
  * Used when Privy API is unavailable (503, timeout, etc.)
@@ -1312,18 +1328,14 @@ export const getWalletBalanceImpl = async (
     const timeoutId = setTimeout(() => controller.abort(), 25000);
     
     try {
-      const balanceResponse = await makePrivyRequest<{
-        balances: Array<{
-          chain: string;
-          asset: string;
-          raw_value: string;
-          raw_value_decimals: number;
-          display_values: {
-            [key: string]: string;
-            usd: string;
-          };
-        };
-      }>(balanceUrl, {}, true, 'GET', fastify, controller.signal);
+      const balanceResponse = await makePrivyRequest<PrivyBalanceResponse>(
+        balanceUrl, 
+        {}, 
+        true, 
+        'GET', 
+        fastify, 
+        controller.signal
+      );
       
       clearTimeout(timeoutId);