retrieve token balances

crates/torii/grpc/proto/types.proto

@@ -159,7 +159,7 @@ message Token {
     string metadata = 6;
 }
 
-message Balance {
+message TokenBalance {
     string balance = 1;
     string account_address = 2;
     string contract_address = 3;

crates/torii/grpc/proto/world.proto

@@ -45,6 +45,9 @@ service World {
 
     // Retrieve tokens
     rpc RetrieveTokens (RetrieveTokensRequest) returns (RetrieveTokensResponse);
+
+    // Retrieve token balances
+    rpc RetrieveTokenBalances (RetrieveTokenBalancesRequest) returns (RetrieveTokenBalancesResponse);
 }
 
 // A request to retrieve tokens
@@ -58,6 +61,19 @@ message RetrieveTokensResponse {
     repeated types.Token tokens = 1;
 }
 
+// A request to retrieve token balances
+message RetrieveTokenBalancesRequest {
+    // The account addresses to retrieve balances for
+    repeated bytes account_addresses = 1;
+    // The list of token contract addresses to retrieve balances for
+    repeated bytes contract_addresses = 2;
+}
+
+// A response containing token balances
+message RetrieveTokenBalancesResponse {
+    repeated types.TokenBalance balances = 1;
+}
+
 // A request to subscribe to indexer updates.
 message SubscribeIndexerRequest {
     bytes contract_address = 1;

crates/torii/grpc/src/server/mod.rs

@@ -809,6 +809,14 @@ impl DojoWorld {
         Ok(RetrieveTokensResponse { tokens })
     }
 
+    async fn retrieve_balances(
+        &self,
+        account_address: Vec<Felt>,
+        contract_addresses: Vec<Felt>,
+    ) -> Result<RetrieveBalancesResponse, Status> {
+
+    }
+
     async fn subscribe_indexer(
         &self,
         contract_address: Felt,