@@ -2,9 +2,7 @@ import { useDojoSDK } from "../hooks";
2
2
import { useState , useEffect , useRef , useCallback } from "react" ;
3
3
import type {
4
4
Token ,
5
- Tokens ,
6
5
TokenBalance ,
7
- TokenBalances ,
8
6
Subscription ,
9
7
} from "@dojoengine/torii-client" ;
10
8
import type {
@@ -16,9 +14,9 @@ import { deepEqual } from "./utils";
16
14
17
15
export function useTokens ( request : GetTokenRequest & GetTokenBalanceRequest ) {
18
16
const { sdk } = useDojoSDK ( ) ;
19
- const [ tokens , setTokens ] = useState < Tokens > ( [ ] ) ;
17
+ const [ tokens , setTokens ] = useState < Token [ ] > ( [ ] ) ;
20
18
const requestRef = useRef < GetTokenRequest | null > ( null ) ;
21
- const [ tokenBalances , setTokenBalances ] = useState < TokenBalances > ( [ ] ) ;
19
+ const [ tokenBalances , setTokenBalances ] = useState < TokenBalance [ ] > ( [ ] ) ;
22
20
const subscriptionRef = useRef < Subscription | null > ( null ) ;
23
21
24
22
const fetchTokens = useCallback ( async ( ) => {
@@ -27,7 +25,7 @@ export function useTokens(request: GetTokenRequest & GetTokenBalanceRequest) {
27
25
tokenIds : request . tokenIds ?? [ ] ,
28
26
} ) ;
29
27
console . log ( "tokens" , tokens ) ;
30
- setTokens ( tokens ) ;
28
+ setTokens ( tokens . items ) ;
31
29
} , [ sdk , request ] ) ;
32
30
33
31
const fetchTokenBalances = useCallback ( async ( ) => {
@@ -48,7 +46,7 @@ export function useTokens(request: GetTokenRequest & GetTokenBalanceRequest) {
48
46
} ) ;
49
47
console . log ( "tokenBalances" , tokenBalances ) ;
50
48
subscriptionRef . current = subscription ;
51
- setTokenBalances ( tokenBalances ) ;
49
+ setTokenBalances ( tokenBalances . items ) ;
52
50
} , [ sdk , request ] ) ;
53
51
54
52
useEffect ( ( ) => {
@@ -95,32 +93,3 @@ function updateTokenBalancesList(
95
93
index === existingBalanceIndex ? newBalance : balance
96
94
) ;
97
95
}
98
-
99
- // /**
100
- // * Subscribe to event changes. This hook fetches initial events from torii and subscribes to new events.
101
- // *
102
- // * @param query ToriiQuery
103
- // */
104
- // export function useTokenBalances(request: GetTokenBalanceRequest) {
105
- // const { sdk, useDojoStore } = useDojoSDK();
106
- // const state = useDojoStore((s) => s);
107
-
108
- // const useEventQueryHook = createSubscriptionHook({
109
- // subscribeMethod: (request) => sdk.onTokenBalanceUpdated(request),
110
- // updateSubscriptionMethod: (subscription, clause) =>
111
- // sdk.updateEventMessageSubscription(subscription, clause, false),
112
- // queryToHashedKeysMethod: (query) =>
113
- // sdk.toriiEventMessagesQueryIntoHashedKeys(query, false),
114
- // processInitialData: (data) => state.mergeEntities(data),
115
- // processUpdateData: (data) => {
116
- // const event = data.pop();
117
- // if (event && event.entityId !== "0x0") {
118
- // state.updateEntity(event);
119
- // }
120
- // },
121
- // getErrorPrefix: () => "Dojo.js - useTokenBalances",
122
- // historical: false,
123
- // });
124
-
125
- // useEventQueryHook(query);
126
- // }
0 commit comments