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

use num_bigint::{BigInt, BigUint};
use std::marker::PhantomData;
use ton_lib_core::cell::CellBuilder;
use ton_lib_core::cell::CellParser;
use ton_lib_core::error::TLCoreError;

/// Adapter to write data with fixed length into a cell.
/// use `#[tlb_derive(bits_len={BITS_LEN})]` to apply it using TLBDerive macro
pub struct ConstLen<T> {
    bits_len: usize,
    _phantom: PhantomData<T>,
}

impl<T> ConstLen<T> {
    pub fn new(bits_len: usize) -> Self {
        Self {
            bits_len,
            _phantom: PhantomData,
        }
    }
}

// Num, Option<Num> impls
macro_rules! const_len_num_impl {
    ($src:ty) => {
        impl ConstLen<$src> {
            pub fn read(&self, parser: &mut CellParser) -> Result<$src, TLCoreError> { parser.read_num(self.bits_len) }
            pub fn write(&self, builder: &mut CellBuilder, val: &$src) -> Result<(), TLCoreError> {
                builder.write_num(val, self.bits_len)
            }
        }
        impl ConstLen<Option<$src>> {
            pub fn read(&self, parser: &mut CellParser) -> Result<Option<$src>, TLCoreError> {
                match parser.read_bit()? {
                    true => Ok(Some(parser.read_num(self.bits_len)?)),
                    false => Ok(None),
                }
            }
            pub fn write(&self, builder: &mut CellBuilder, val: &Option<$src>) -> Result<(), TLCoreError> {
                builder.write_bit(val.is_some())?;
                if let Some(val) = val {
                    return builder.write_num(val, self.bits_len);
                };
                Ok(())
            }
        }
    };
}

const_len_num_impl!(i8);
const_len_num_impl!(u8);
const_len_num_impl!(i16);
const_len_num_impl!(u16);
const_len_num_impl!(i32);
const_len_num_impl!(u32);
const_len_num_impl!(i64);
const_len_num_impl!(u64);
const_len_num_impl!(i128);
const_len_num_impl!(u128);
const_len_num_impl!(BigInt);
const_len_num_impl!(BigUint);

impl ConstLen<Vec<u8>> {
    pub fn read(&self, parser: &mut CellParser) -> Result<Vec<u8>, TLCoreError> { parser.read_bits(self.bits_len) }
    pub fn write(&self, builder: &mut CellBuilder, val: &Vec<u8>) -> Result<(), TLCoreError> {
        builder.write_bits(val, self.bits_len)
    }
}

impl ConstLen<Option<Vec<u8>>> {
    pub fn read(&self, parser: &mut CellParser) -> Result<Option<Vec<u8>>, TLCoreError> {
        match parser.read_bit()? {
            true => Ok(Some(parser.read_bits(self.bits_len)?)),
            false => Ok(None),
        }
    }
    pub fn write(&self, builder: &mut CellBuilder, val: &Option<Vec<u8>>) -> Result<(), TLCoreError> {
        match val {
            Some(val) => {
                builder.write_bit(true)?;
                builder.write_bits(val, self.bits_len)
            }
            None => builder.write_bit(false),
        }
    }
}

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

    #[test]
    fn test_const_len() -> anyhow::Result<()> {
        let mut builder = TonCell::builder();
        ConstLen::<u32>::new(24).write(&mut builder, &1u32)?;
        let cell = builder.build()?;
        assert_eq!(&cell.data, &[0, 0, 1]);
        let parsed = ConstLen::<u32>::new(24).read(&mut cell.parser())?;
        assert_eq!(parsed, 1u32);
        Ok(())
    }

    #[derive(TLBDerive)]
    struct TestType {
        #[tlb_derive(bits_len = 4)]
        a: u32,
    }

    #[test]
    fn test_cont_len_bits_len() -> anyhow::Result<()> {
        let mut builder = TonCell::builder();
        TestType { a: 1 }.write(&mut builder)?;
        let cell = builder.build()?;
        assert_eq!(&cell.data, &[0b00010000]);
        let parsed = TestType::read(&mut cell.parser())?;
        assert_eq!(parsed.a, 1u32);
        Ok(())
    }
}