ton_lib 0.0.38

A collection of types and utilities for interacting with the TON network
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133

use crate::tlb_adapters::DictKeyAdapterInto;
use crate::tlb_adapters::DictValAdapterTLB;
use crate::tlb_adapters::TLBHashMapE;
use num_bigint::BigUint;
use num_traits::ToPrimitive;
use std::collections::HashMap;
use std::ops::{Deref, DerefMut};
use std::str::FromStr;
use ton_lib_core::error::TLCoreError;
use ton_lib_core::types::tlb_core::VarLenBytes;
use ton_lib_core::TLBDerive;

/// https://github.com/ton-blockchain/ton/blob/050a984163a53df16fb03f66cc445c34bfed48ed/crypto/block/block.tlb#L116
#[derive(Clone, Debug, PartialEq, Eq, TLBDerive)]
pub struct Coins(VarLenBytes<u128, 4>);

/// https://github.com/ton-blockchain/ton/blob/050a984163a53df16fb03f66cc445c34bfed48ed/crypto/block/block.tlb#L124
#[derive(Default, Clone, Debug, PartialEq, TLBDerive)]
pub struct CurrencyCollection {
    pub grams: Coins,
    #[tlb_derive(adapter = "TLBHashMapE::<DictKeyAdapterInto, DictValAdapterTLB, _, _>::new(32)")]
    pub other: HashMap<u32, VarLenBytes<BigUint, 5>>,
}

impl Coins {
    pub const ZERO: Coins = Coins(VarLenBytes {
        data: 0u128,
        bits_len: 0,
    });

    pub fn new<T: Into<u128>>(amount: T) -> Self {
        let amount = amount.into();
        let bits_len = (128 - amount.leading_zeros()).div_ceil(8) * 8;
        Self(VarLenBytes::new(amount, bits_len as usize))
    }
    pub fn from_signed<T: Into<i128>>(amount: T) -> Result<Self, TLCoreError> {
        let amount = amount.into();
        if amount < 0 {
            return Err(TLCoreError::UnexpectedValue {
                expected: "positive int".to_string(),
                actual: format!("{amount}"),
            });
        }
        Ok(Self::new(amount as u128))
    }

    pub fn to_u32(&self) -> Result<u32, TLCoreError> {
        self.0.to_u32().ok_or(TLCoreError::UnexpectedValue {
            expected: "u32".to_string(),
            actual: format!("{}", self.0.data),
        })
    }

    pub fn to_u64(&self) -> Result<u64, TLCoreError> {
        self.0.to_u64().ok_or(TLCoreError::UnexpectedValue {
            expected: "u64".to_string(),
            actual: format!("{}", self.0.data),
        })
    }

    pub fn to_u128(&self) -> u128 { self.0.data }
}

impl CurrencyCollection {
    pub fn new<T: Into<u128>>(grams: T) -> Self {
        Self {
            grams: Coins::new(grams),
            other: Default::default(),
        }
    }
}

mod traits_impl {
    use super::*;

    impl Copy for Coins {}

    impl FromStr for CurrencyCollection {
        type Err = TLCoreError;
        fn from_str(grams: &str) -> Result<Self, Self::Err> { Ok(Self::new(u128::from_str(grams)?)) }
    }

    impl Deref for Coins {
        type Target = u128;
        fn deref(&self) -> &Self::Target { &self.0 }
    }

    impl DerefMut for Coins {
        fn deref_mut(&mut self) -> &mut Self::Target { &mut self.0 }
    }

    impl<T: Into<u128>> From<T> for Coins {
        fn from(value: T) -> Self { Coins::new(value) }
    }

    impl FromStr for Coins {
        type Err = TLCoreError;
        fn from_str(grams: &str) -> Result<Self, Self::Err> { Ok(Self::new(u128::from_str(grams)?)) }
    }

    impl Default for Coins {
        fn default() -> Self { Coins::ZERO }
    }
}

#[cfg(test)]
mod tests {
    use super::*;
    use ton_lib_core::traits::tlb::TLB;

    #[test]
    fn test_currency_collection() -> anyhow::Result<()> {
        let parsed = CurrencyCollection::from_boc_hex("b5ee9c720101010100070000094c143b1d14")?;
        assert_eq!(parsed.grams, 3242439121u32.into());

        let cell_serial = parsed.to_cell()?;
        let parsed_back = CurrencyCollection::from_cell(&cell_serial)?;
        assert_eq!(parsed, parsed_back);
        Ok(())
    }

    #[test]
    fn test_currency_collection_zero_grams() -> anyhow::Result<()> {
        let currency = CurrencyCollection::new(0u32);
        let cell = currency.to_cell()?;
        let parsed = CurrencyCollection::from_cell(&cell)?;
        assert_eq!(parsed.grams, 0u32.into());

        let cell_serial = parsed.to_cell()?;
        assert_eq!(cell_serial, cell);
        Ok(())
    }
}