retriefe tokens

crates/torii/core/src/types.rs

@@ -124,12 +124,13 @@ pub struct Event {
 
 #[derive(FromRow, Deserialize, Debug, Clone)]
 #[serde(rename_all = "camelCase")]
-pub struct ContractMetadata {
+pub struct Token {
     pub id: String,
     pub contract_address: String,
     pub name: String,
     pub symbol: String,
     pub decimals: u8,
+    pub metadata: String,
 }
 
 #[derive(FromRow, Deserialize, Debug, Clone)]

crates/torii/grpc/proto/types.proto

@@ -156,6 +156,7 @@ message Token {
     string name = 3;
     string symbol = 4;
     uint32 decimals = 5;
+    string metadata = 6;
 }
 
 message Balance {

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

@@ -43,6 +43,7 @@ use torii_core::error::{Error, ParseError, QueryError};
 use torii_core::model::{build_sql_query, map_row_to_ty};
 use torii_core::sql::cache::ModelCache;
 use torii_core::sql::utils::sql_string_to_felts;
+use torii_core::types::Token;
 use tower_http::cors::{AllowOrigin, CorsLayer};
 
 use self::subscriptions::entity::EntityManager;
@@ -53,10 +54,7 @@ use crate::proto::types::member_value::ValueType;
 use crate::proto::types::LogicalOperator;
 use crate::proto::world::world_server::WorldServer;
 use crate::proto::world::{
-    RetrieveEntitiesStreamingResponse, RetrieveEventMessagesRequest, SubscribeEntitiesRequest,
-    SubscribeEntityResponse, SubscribeEventMessagesRequest, SubscribeEventsResponse,
-    SubscribeIndexerRequest, SubscribeIndexerResponse, UpdateEventMessagesSubscriptionRequest,
-    WorldMetadataRequest, WorldMetadataResponse,
+    RetrieveEntitiesStreamingResponse, RetrieveEventMessagesRequest, RetrieveTokensRequest, RetrieveTokensResponse, SubscribeEntitiesRequest, SubscribeEntityResponse, SubscribeEventMessagesRequest, SubscribeEventsResponse, SubscribeIndexerRequest, SubscribeIndexerResponse, UpdateEventMessagesSubscriptionRequest, WorldMetadataRequest, WorldMetadataResponse
 };
 use crate::proto::{self};
 use crate::types::schema::SchemaError;
@@ -789,6 +787,28 @@ impl DojoWorld {
         })
     }
 
+    async fn retrieve_tokens(
+        &self,
+        contract_addresses: Vec<Felt>,
+    ) -> Result<RetrieveTokensResponse, Status> {
+        let query = format!(
+            "SELECT * FROM tokens WHERE contract_address IN ({})",
+            contract_addresses
+                .iter()
+                .map(|address| format!("{:#x}", address))
+                .collect::<Vec<_>>()
+                .join(", ")
+        );
+
+        let tokens: Vec<Token> = sqlx::query_as(&query)
+            .fetch_all(&self.pool)
+            .await
+            .map_err(|e| Status::internal(e.to_string()))?;
+
+        let tokens = tokens.iter().map(|token| token.clone().into()).collect();
+        Ok(RetrieveTokensResponse { tokens })
+    }
+
     async fn subscribe_indexer(
         &self,
         contract_address: Felt,
@@ -1165,6 +1185,23 @@ impl proto::world::world_server::World for DojoWorld {
         Ok(Response::new(WorldMetadataResponse { metadata }))
     }
 
+    async fn retrieve_tokens(
+        &self,
+        request: Request<RetrieveTokensRequest>,
+    ) -> Result<Response<RetrieveTokensResponse>, Status> {
+        let RetrieveTokensRequest { contract_addresses } = request.into_inner();
+        let contract_addresses = contract_addresses
+            .iter()
+            .map(|address| Felt::from_bytes_be_slice(address))
+            .collect::<Vec<_>>();
+
+        let tokens = self
+            .retrieve_tokens(contract_addresses)
+            .await
+            .map_err(|e| Status::internal(e.to_string()))?;
+        Ok(Response::new(tokens))
+    }
+
     async fn subscribe_indexer(
         &self,
         request: Request<SubscribeIndexerRequest>,

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

@@ -10,12 +10,25 @@ use starknet::core::types::{
     ContractStorageDiffItem, Felt, FromStrError, StateDiff, StateUpdate, StorageEntry,
 };
 use strum_macros::{AsRefStr, EnumIter, FromRepr};
+use torii_core::types::Token;
 
 use crate::proto::types::member_value;
 use crate::proto::{self};
 
 pub mod schema;
 
+impl From<Token> for proto::types::Token {
+    fn from(value: Token) -> Self {
+        Self {
+            contract_address: value.contract_address,
+            name: value.name,
+            symbol: value.symbol,
+            decimals: value.decimals as u32,
+            metadata: serde_json::to_string(&value.metadata).unwrap(),
+        }
+    }
+}
+
 #[derive(Debug, Serialize, Deserialize, PartialEq, Hash, Eq, Clone)]
 pub struct IndexerUpdate {
     pub head: i64,