iscp-rs 1.1.2

iSCPv2 Client Library
Documentation 
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

use bytes::BufMut;

use crate::encoding;
use crate::transport::TransportError;

/// エンコード方法
#[derive(Clone, Debug, serde::Serialize)]
#[non_exhaustive]
pub enum EncodingName {
    #[serde(rename = "proto")]
    Protobuf,
}

impl From<encoding::Name> for EncodingName {
    fn from(name: encoding::Name) -> Self {
        match name {
            encoding::Name::Protobuf => EncodingName::Protobuf,
        }
    }
}

/// 圧縮タイプ
#[derive(Clone, Debug, serde::Serialize)]
pub enum CompressionType {
    #[serde(rename = "per-message")]
    PerMessage,
    #[serde(rename = "context-takeover")]
    ContextTakeover,
}

/// 接続開始時のパラメータ
#[derive(Clone, Debug, serde::Serialize)]
#[non_exhaustive]
pub struct NegotiationParams {
    #[serde(rename = "enc")]
    pub encoding_name: EncodingName,
    #[serde(rename = "comp", skip_serializing_if = "Option::is_none")]
    pub compression_type: Option<CompressionType>,
    #[serde(rename = "clevel", skip_serializing_if = "Option::is_none")]
    pub compression_level: Option<i8>,
    #[serde(rename = "cwinbits", skip_serializing_if = "Option::is_none")]
    pub compression_window_bits: Option<u8>,
}

impl NegotiationParams {
    pub(crate) fn to_uri_query_string(&self) -> Result<String, TransportError> {
        serde_qs::to_string(&self).map_err(TransportError::new)
    }

    pub(crate) fn to_bytes(&self) -> Result<Vec<u8>, TransportError> {
        use serde_json::Value;

        let mut buf = Vec::new();

        let value = serde_json::to_value(self).unwrap();
        let map = value.as_object().unwrap();
        for (k, v) in map.iter() {
            let v = match v {
                Value::Null => continue,
                Value::String(s) => s.clone(),
                _ => v.to_string(),
            };

            buf.put_u16(
                k.len()
                    .try_into()
                    .map_err(|_| TransportError::from_msg("too long key for negotiation"))?,
            );
            buf.put(k.as_bytes());

            buf.put_u16(
                v.len()
                    .try_into()
                    .map_err(|_| TransportError::from_msg("too long value for negotiation"))?,
            );
            buf.put(v.as_bytes());
        }

        Ok(buf)
    }
}