hiero_sdk/token/
token_info_query.rs

1// SPDX-License-Identifier: Apache-2.0
2
3use hiero_sdk_proto::services;
4use hiero_sdk_proto::services::token_service_client::TokenServiceClient;
5use tonic::transport::Channel;
6
7use crate::ledger_id::RefLedgerId;
8use crate::query::{
9    AnyQueryData,
10    QueryExecute,
11    ToQueryProtobuf,
12};
13use crate::token::token_info::TokenInfo;
14use crate::{
15    BoxGrpcFuture,
16    Error,
17    Query,
18    ToProtobuf,
19    TokenId,
20    ValidateChecksums,
21};
22
23/// Gets information about Token instance.
24///
25pub type TokenInfoQuery = Query<TokenInfoQueryData>;
26
27#[derive(Default, Clone, Debug)]
28pub struct TokenInfoQueryData {
29    token_id: Option<TokenId>,
30}
31
32impl From<TokenInfoQueryData> for AnyQueryData {
33    #[inline]
34    fn from(data: TokenInfoQueryData) -> Self {
35        Self::TokenInfo(data)
36    }
37}
38
39impl TokenInfoQuery {
40    /// Returns the token ID for which information is requested.
41    #[must_use]
42    pub fn get_token_id(&self) -> Option<TokenId> {
43        self.data.token_id
44    }
45
46    /// Sets the token ID for which information is requested.
47    pub fn token_id(&mut self, id: impl Into<TokenId>) -> &mut Self {
48        self.data.token_id = Some(id.into());
49        self
50    }
51}
52
53impl ToQueryProtobuf for TokenInfoQueryData {
54    fn to_query_protobuf(&self, header: services::QueryHeader) -> services::Query {
55        let token_id = self.token_id.to_protobuf();
56
57        services::Query {
58            query: Some(services::query::Query::TokenGetInfo(services::TokenGetInfoQuery {
59                header: Some(header),
60                token: token_id,
61            })),
62        }
63    }
64}
65
66impl QueryExecute for TokenInfoQueryData {
67    type Response = TokenInfo;
68
69    fn execute(
70        &self,
71        channel: Channel,
72        request: services::Query,
73    ) -> BoxGrpcFuture<'_, services::Response> {
74        Box::pin(async { TokenServiceClient::new(channel).get_token_info(request).await })
75    }
76}
77
78impl ValidateChecksums for TokenInfoQueryData {
79    fn validate_checksums(&self, ledger_id: &RefLedgerId) -> Result<(), Error> {
80        self.token_id.validate_checksums(ledger_id)
81    }
82}
83
84#[cfg(test)]
85mod tests {
86    use expect_test::expect;
87
88    use crate::query::ToQueryProtobuf;
89    use crate::{
90        Hbar,
91        TokenId,
92        TokenInfoQuery,
93    };
94
95    #[test]
96    fn serialize() {
97        expect![[r#"
98            Query {
99                query: Some(
100                    TokenGetInfo(
101                        TokenGetInfoQuery {
102                            header: Some(
103                                QueryHeader {
104                                    payment: None,
105                                    response_type: AnswerOnly,
106                                },
107                            ),
108                            token: Some(
109                                TokenId {
110                                    shard_num: 0,
111                                    realm_num: 0,
112                                    token_num: 5005,
113                                },
114                            ),
115                        },
116                    ),
117                ),
118            }
119        "#]]
120        .assert_debug_eq(
121            &TokenInfoQuery::new()
122                .token_id(TokenId::new(0, 0, 5005))
123                .max_payment_amount(Hbar::from_tinybars(100_000))
124                .data
125                .to_query_protobuf(Default::default()),
126        )
127    }
128
129    #[test]
130    fn get_set_token_id() {
131        let mut query = TokenInfoQuery::new();
132        query.token_id(TokenId::new(0, 0, 5005));
133
134        assert_eq!(query.get_token_id(), Some(TokenId::new(0, 0, 5005)));
135    }
136}
hiero_sdk/token/token_info_query.rs

hiero_sdk/token/
token_info_query.rs
