m10_sdk_protos/
metadata.rs

1use prost::Message;
2use prost_types::Any;
3
4use crate::sdk::metadata::*;
5use crate::sdk::transaction::{CreateTransfer, TransferStep};
6use crate::sdk::FinalizedTransfer;
7
8pub fn memo(memo: &str) -> Any {
9    Memo {
10        plaintext: memo.to_string(),
11    }
12    .any()
13}
14
15pub trait Metadata: Message + Sized + Default {
16    const TYPE_URL: &'static str;
17
18    fn any(&self) -> Any {
19        Any {
20            type_url: Self::TYPE_URL.to_string(),
21            value: self.encode_to_vec(),
22        }
23    }
24}
25
26pub trait MetadataExt {
27    fn metadata<M: Metadata>(&self) -> Option<M>;
28    fn memo(&self) -> String {
29        self.metadata::<Memo>().unwrap_or_default().plaintext
30    }
31}
32
33impl MetadataExt for TransferStep {
34    fn metadata<M: Metadata>(&self) -> Option<M> {
35        self.metadata
36            .iter()
37            .filter(|a| a.type_url == M::TYPE_URL)
38            .find_map(|a| M::decode(a.value.as_slice()).ok())
39    }
40}
41
42impl MetadataExt for CreateTransfer {
43    fn metadata<M: Metadata>(&self) -> Option<M> {
44        self.transfer_steps.iter().find_map(|step| step.metadata())
45    }
46}
47
48impl MetadataExt for FinalizedTransfer {
49    fn metadata<M: Metadata>(&self) -> Option<M> {
50        self.transfer_steps.iter().find_map(|step| step.metadata())
51    }
52}
53
54impl Metadata for Attachment {
55    const TYPE_URL: &'static str = "m10.sdk.metadata.Attachment";
56}
57
58impl Metadata for Memo {
59    const TYPE_URL: &'static str = "m10.sdk.metadata.Memo";
60}
61
62impl Metadata for Fee {
63    const TYPE_URL: &'static str = "m10.sdk.metadata.Fee";
64}
65
66impl Metadata for Withdraw {
67    const TYPE_URL: &'static str = "m10.sdk.metadata.Withdraw";
68}
69
70impl Metadata for Deposit {
71    const TYPE_URL: &'static str = "m10.sdk.metadata.Deposit";
72}
73
74impl Metadata for Contract {
75    const TYPE_URL: &'static str = "m10.sdk.metadata.Contract";
76}