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

use framing::{FrameType, Frame, FramePayload, MethodFrame};
use error::Result;

#[derive(Debug, PartialEq, Eq, Clone)]
pub struct EncodedMethod(Vec<u8>);

impl EncodedMethod {
    pub fn new(data: Vec<u8>) -> Self {
        EncodedMethod(data)
    }

    pub fn into_inner(self) -> Vec<u8> {
        self.0
    }

    pub fn inner(&self) -> &[u8] {
        &self.0
    }
}

pub trait Method {
    fn decode(method_frame: MethodFrame) -> Result<Self> where Self: Sized;
    fn encode(&self) -> Result<EncodedMethod>;
    fn name(&self) -> &'static str;
    fn id(&self) -> u16;
    fn class_id(&self) -> u16;

    fn encode_method_frame(&self) -> Result<FramePayload> {
        let frame = MethodFrame { class_id: self.class_id(), method_id: self.id(), arguments: self.encode()? };
        frame.encode()
    }

    fn to_frame(&self, channel: u16) -> Result<Frame> {
        Ok(Frame {
            frame_type: FrameType::METHOD,
            channel: channel,
            payload: self.encode_method_frame()?,
        })
    }
}