fix: update dependencies to use workspace references; enhance type safety in useTokenBalances function

Author: shunkakinoki

extras/demo-anypay/package.json

@@ -18,16 +18,16 @@
     "clean": "rimraf dist"
   },
   "dependencies": {
-    "@0xsequence/anypay-sdk": "github:0xsequence/sequence.js#dists/wallet/anypay-sdk",
+    "@0xsequence/anypay-sdk": "workspace:*",
     "@0xsequence/api": "0.0.0-anypay-20250527101311",
     "@0xsequence/design-system": "^2.1.6",
     "@0xsequence/hooks": "^5.1.0",
     "@0xsequence/indexer": "^2.3.9",
     "@0xsequence/network": "^2.3.9",
     "@0xsequence/wagmi-connector": "^4.0.1",
-    "@0xsequence/wallet-core": "github:0xsequence/sequence.js#dists/wallet/core",
-    "@0xsequence/wallet-primitives": "github:0xsequence/sequence.js#dists/wallet/primitives",
-    "@0xsequence/wallet-wdk": "github:0xsequence/sequence.js#dists/wallet/wdk",
+    "@0xsequence/wallet-core": "workspace:*",
+    "@0xsequence/wallet-primitives": "workspace:*",
+    "@0xsequence/wallet-wdk": "workspace:*",
     "@radix-ui/react-popover": "^1.1.6",
     "@repo/eslint-config": "workspace:*",
     "@repo/typescript-config": "workspace:*",

packages/wallet/anypay-sdk/src/tokenBalances.ts

@@ -17,18 +17,18 @@ export { type NativeTokenBalance, type TokenBalance }
 const defaultPage = { page: 1, pageSize: 10, more: false }
 
 // Type guard for native token balance
-function isNativeToken(token: TokenBalance | NativeTokenBalance): boolean {
+function isNativeToken(token: TokenBalance | NativeTokenBalance): token is NativeTokenBalance {
   if ('contractAddress' in token) {
     return false
   }
   return true
 }
 
 export function useTokenBalances(address: Address.Address): {
-  tokenBalancesData: any // TODO: Add proper type
+  tokenBalancesData: GetTokenBalancesSummaryReturn | undefined
   isLoadingBalances: boolean
   balanceError: Error | null
-  sortedTokens: any[] // TODO: Add proper type
+  sortedTokens: (TokenBalance | NativeTokenBalance)[]
 } {
   const indexerClient = useIndexerGatewayClient()
 
@@ -58,11 +58,10 @@ export function useTokenBalances(address: Address.Address): {
           },
         })
 
-        // Transform the raw gateway response to match GetTokenBalancesSummaryReturn
         return {
           page: summaryFromGateway.page,
-          balances: summaryFromGateway.balances.flatMap((b: any) => b.results),
-          nativeBalances: summaryFromGateway.nativeBalances.flatMap((b: any) => b.results),
+          balances: summaryFromGateway.balances.flatMap((b) => b.results),
+          nativeBalances: summaryFromGateway.nativeBalances.flatMap((b) => b.results),
         }
       } catch (error) {
         console.error('Failed to fetch token balances:', error)
@@ -79,16 +78,11 @@ export function useTokenBalances(address: Address.Address): {
   })
 
   const sortedTokens = useMemo(() => {
-    if (!tokenBalancesData?.balances) {
+    if (!tokenBalancesData) {
       return []
     }
 
-    // Flatten both native and token balances
-    const nativeBalances = tokenBalancesData.nativeBalances.flatMap(
-      (b: any) => b.results, // TODO: Add proper type
-    )
-    const tokenBalances = tokenBalancesData.balances.flatMap((b: any) => b.results) // TODO: Add proper type
-    const balances = [...nativeBalances, ...tokenBalances]
+    const balances = [...tokenBalancesData.nativeBalances, ...tokenBalancesData.balances]
 
     return [...balances]
       .filter((token) => {