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

//! Ockam pipe protocol structures

pub mod internal;

use crate::Message;
use ockam_core::compat::vec::Vec;
use ockam_core::{Decodable, Encodable, Result, TransportMessage, Uint};
use serde::{Deserialize, Serialize};

/// An indexed message for pipes
#[derive(Serialize, Deserialize, Message)]
pub struct PipeMessage {
    /// Pipe message index
    pub index: Uint,
    /// Pipe message raw data
    pub data: Vec<u8>,
}

/// We need to manually implement clone because serde_bare::Uint
/// doesn't, so we can't derive it
impl Clone for PipeMessage {
    fn clone(&self) -> Self {
        Self {
            index: Uint::from(self.index.u64()),
            data: self.data.clone(),
        }
    }
}

impl PipeMessage {
    pub(crate) fn from_transport(index: u64, msg: TransportMessage) -> Result<Self> {
        let data = msg.encode()?;
        Ok(Self {
            index: index.into(),
            data,
        })
    }

    pub(crate) fn to_transport(&self) -> Result<TransportMessage> {
        TransportMessage::decode(&self.data)
    }
}