tonlib_core_anychain/message/sbt/
ownership_proof.rs

1use num_bigint::BigUint;
2
3use super::SBT_OWNERSHIP_PROOF;
4use crate::cell::{ArcCell, Cell, CellBuilder};
5use crate::message::{HasOpcode, TonMessage, TonMessageError};
6use crate::TonAddress;
7
8/// Creates a body for sbt ownership proof according to TL-B schema:
9///
10/// ```raw
11/// ownership_proof#0524c7ae
12///   query_id:uint64
13///   item_id:uint256
14///   owner:MsgAddress
15///   data:^Cell
16///   revoked_at:uint64
17///   content:(Maybe ^Cell)
18/// = InternalMsgBody;
19/// ```
20#[derive(Clone, Debug, PartialEq)]
21pub struct OwnershipProofMessage {
22    /// arbitrary request number.
23    pub query_id: u64,
24    /// id of NFT.
25    pub item_id: BigUint,
26    /// current owner's address.
27    pub owner: TonAddress,
28    /// data cell passed in prove_ownership.
29    pub data: ArcCell,
30    /// unix time when SBT was revoked, 0 if it was not.
31    pub revoked_at: u64,
32    /// NFT's content, it is passed if with_content was true in prove_ownership.
33    pub content: Option<ArcCell>,
34}
35
36impl OwnershipProofMessage {
37    pub fn new(
38        item_id: BigUint,
39        owner: &TonAddress,
40        data: ArcCell,
41        revoked_at: u64,
42        content: Option<ArcCell>,
43    ) -> Self {
44        OwnershipProofMessage {
45            query_id: 0,
46            item_id,
47            owner: owner.clone(),
48            data,
49            revoked_at,
50            content,
51        }
52    }
53}
54
55impl TonMessage for OwnershipProofMessage {
56    /// ownership_proof#0524c7ae
57    ///   query_id:uint64
58    ///   item_id:uint256
59    ///   owner:MsgAddress
60    ///   data:^Cell
61    ///   revoked_at:uint64
62    ///   content:(Maybe ^Cell)
63    /// = InternalMsgBody;
64    fn build(&self) -> Result<Cell, TonMessageError> {
65        let mut builder = CellBuilder::new();
66        builder.store_u32(32, Self::opcode())?;
67        builder.store_u64(64, self.query_id)?;
68        builder.store_uint(256, &self.item_id)?;
69        builder.store_address(&self.owner)?;
70        builder.store_reference(&self.data)?;
71        builder.store_u64(64, self.revoked_at)?;
72        builder.store_maybe_cell_ref(&self.content)?;
73        Ok(builder.build()?)
74    }
75
76    fn parse(cell: &Cell) -> Result<Self, TonMessageError> {
77        let mut parser = cell.parser();
78
79        let opcode: u32 = parser.load_u32(32)?;
80        let query_id = parser.load_u64(64)?;
81        let item_id = parser.load_uint(256)?;
82        let owner = parser.load_address()?;
83        let data = parser.next_reference()?;
84        let revoked_at = parser.load_u64(64)?;
85        let content = parser.load_maybe_cell_ref()?;
86        parser.ensure_empty()?;
87
88        let result = OwnershipProofMessage {
89            query_id,
90            item_id,
91            owner,
92            data,
93            revoked_at,
94            content,
95        };
96        result.verify_opcode(opcode)?;
97        Ok(result)
98    }
99}
100
101impl HasOpcode for OwnershipProofMessage {
102    fn set_query_id(&mut self, query_id: u64) {
103        self.query_id = query_id;
104    }
105
106    fn query_id(&self) -> u64 {
107        self.query_id
108    }
109
110    fn opcode() -> u32 {
111        SBT_OWNERSHIP_PROOF
112    }
113}
114
115#[cfg(test)]
116mod tests {
117
118    use std::sync::Arc;
119
120    use num_bigint::BigUint;
121
122    use crate::cell::{ArcCell, CellBuilder};
123    use crate::message::{HasOpcode, OwnershipProofMessage, TonMessage};
124    use crate::TonAddress;
125    #[test]
126    fn sbt_owner_info_msg_test() {
127        let query_id = 1234567890;
128        let item_id = BigUint::from(123u64);
129
130        let owner =
131            &TonAddress::from_base64_url("EQAd8QRKoA5sKcug9bwK6vMdmhSAoAxr8vvABvC1TCeTude5")
132                .unwrap();
133
134        let data: ArcCell = Arc::new(
135            CellBuilder::new()
136                .store_u32(12, 123)
137                .unwrap()
138                .build()
139                .unwrap(),
140        );
141        let revoked_at = 123456;
142        let content = Some(Arc::new(
143            CellBuilder::new()
144                .store_u32(12, 456)
145                .unwrap()
146                .build()
147                .unwrap(),
148        ));
149
150        let expected = OwnershipProofMessage {
151            query_id,
152            item_id: item_id.clone(),
153            owner: owner.clone(),
154            data: data.clone(),
155            revoked_at,
156            content: content.clone(),
157        };
158
159        let build_result = OwnershipProofMessage::new(item_id, owner, data, revoked_at, content)
160            .with_query_id(query_id)
161            .build();
162        assert!(build_result.is_ok());
163
164        let cell = build_result.unwrap();
165        let parse_result = OwnershipProofMessage::parse(&cell);
166        assert!(parse_result.is_ok());
167
168        let parsed_msg = parse_result.unwrap();
169        assert_eq!(expected, parsed_msg);
170    }
171}
tonlib_core_anychain/message/sbt/ownership_proof.rs

tonlib_core_anychain/message/sbt/
ownership_proof.rs
