prns-core 0.3.4

Pure Reticulum engine and wire contract for Personal Reticulum
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

use alloc::vec::Vec;
use core::convert::Infallible;

use rmp::encode::{self, ByteBuf, ValueWriteError};

use super::MessagePackEncodeError;

pub(crate) struct MessagePackEncoder {
    bytes: ByteBuf,
}

impl MessagePackEncoder {
    pub(crate) fn new() -> Self {
        Self {
            bytes: ByteBuf::new(),
        }
    }

    pub(crate) fn finish(self) -> Vec<u8> {
        self.bytes.into_vec()
    }

    pub(crate) fn nil(&mut self) {
        infallible(encode::write_nil(&mut self.bytes));
    }

    pub(crate) fn boolean(&mut self, value: bool) {
        infallible(encode::write_bool(&mut self.bytes, value));
    }

    pub(crate) fn signed(&mut self, value: i64) {
        infallible_value(encode::write_sint(&mut self.bytes, value));
    }

    pub(crate) fn unsigned(&mut self, value: u64) {
        infallible_value(encode::write_uint(&mut self.bytes, value));
    }

    pub(crate) fn float(&mut self, value: f64) {
        infallible_value(encode::write_f64(&mut self.bytes, value));
    }

    pub(crate) fn string(&mut self, value: &str) -> Result<(), MessagePackEncodeError> {
        u32::try_from(value.len()).map_err(|_| MessagePackEncodeError)?;
        infallible_value(encode::write_str(&mut self.bytes, value));
        Ok(())
    }

    pub(crate) fn binary(&mut self, value: &[u8]) -> Result<(), MessagePackEncodeError> {
        u32::try_from(value.len()).map_err(|_| MessagePackEncodeError)?;
        infallible_value(encode::write_bin(&mut self.bytes, value));
        Ok(())
    }

    pub(crate) fn array(&mut self, length: usize) -> Result<(), MessagePackEncodeError> {
        let length = u32::try_from(length).map_err(|_| MessagePackEncodeError)?;
        infallible_value(encode::write_array_len(&mut self.bytes, length));
        Ok(())
    }

    pub(crate) fn map(&mut self, length: usize) -> Result<(), MessagePackEncodeError> {
        let length = u32::try_from(length).map_err(|_| MessagePackEncodeError)?;
        infallible_value(encode::write_map_len(&mut self.bytes, length));
        Ok(())
    }

    #[cfg(feature = "shared-instance-rpc")]
    pub(crate) fn field(&mut self, name: &str) -> Result<(), MessagePackEncodeError> {
        self.string(name)
    }

    #[cfg(feature = "shared-instance-rpc")]
    pub(crate) fn string_field(
        &mut self,
        name: &str,
        value: &str,
    ) -> Result<(), MessagePackEncodeError> {
        self.field(name)?;
        self.string(value)
    }

    #[cfg(feature = "shared-instance-rpc")]
    pub(crate) fn unsigned_field(
        &mut self,
        name: &str,
        value: u64,
    ) -> Result<(), MessagePackEncodeError> {
        self.field(name)?;
        self.unsigned(value);
        Ok(())
    }
}

fn infallible<T>(result: Result<T, Infallible>) -> T {
    match result {
        Ok(value) => value,
        Err(never) => match never {},
    }
}

fn infallible_value<T>(result: Result<T, ValueWriteError<Infallible>>) -> T {
    match result {
        Ok(value) => value,
        Err(ValueWriteError::InvalidMarkerWrite(never))
        | Err(ValueWriteError::InvalidDataWrite(never)) => match never {},
    }
}