kf-protocol-api 2.0.0

use std::fs::File;
use std::io::Cursor;
use std::io::Error as IoError;
use std::io::ErrorKind;
use std::io::Read;
use std::path::Path;

use log::debug;
use log::trace;

use kf_protocol::bytes::Buf;
use kf_protocol::bytes::BufMut;
use kf_protocol::Decoder;
use kf_protocol::Encoder;
use kf_protocol::Version;

use crate::RequestHeader;

#[derive(Debug, Default)]
pub struct ResponseMessage<P> {
    pub correlation_id: i32,
    pub response: P,
}


impl<P> ResponseMessage<P> {


    pub fn from_header(header: &RequestHeader,response: P) -> Self  {
        Self::new(header.correlation_id(),response)
    }

     pub fn new(correlation_id: i32, response: P) -> Self {
        Self {
            correlation_id,
            response,
        }
    }

}



impl<P> ResponseMessage<P>
    where P: Decoder
{
    pub fn decode_from<T>(src: &mut T,version: Version) -> Result<Self,IoError>
        where T:Buf  {

        let mut correlation_id: i32 = 0;
        correlation_id.decode(src,version)?;
        trace!("decoded correlation id: {}",correlation_id);

        let response = P::decode_from(src, version)?;
        Ok(ResponseMessage {
            correlation_id,
            response
        })
        
    }

    
    pub fn decode_from_file<H: AsRef<Path>>(
        file_name: H,
        version: Version
    ) -> Result<Self, IoError> {

        debug!("decoding from file: {:#?}", file_name.as_ref());
        let mut f = File::open(file_name)?;
        let mut buffer: [u8; 1000] = [0; 1000];

        f.read(&mut buffer)?;
        let data = buffer.to_vec();

        let mut src = Cursor::new(&data);

        let mut size: i32 = 0;
        size.decode(&mut src,version)?;
        trace!("decoded response size: {} bytes", size);

        if src.remaining() < size as usize {
                return Err(IoError::new(
                    ErrorKind::UnexpectedEof,
                    "not enought for response",
            ));
        }

        Self::decode_from(&mut src,version)
    }

}

impl<P> Encoder for ResponseMessage<P>
where
    P: Encoder + Default,
{
    fn write_size(&self,version: Version) -> usize {
        self.correlation_id.write_size(version) + self.response.write_size(version)
    }

    fn encode<T>(&self, out: &mut T,version: Version) -> Result<(), IoError>
    where
        T: BufMut,
    {
        let len = self.write_size(version) as i32;
        trace!("encoding kf response: {} version: {}, len: {}", std::any::type_name::<P>(),version, len);
        len.encode(out,version)?;
        self.correlation_id.encode(out,version)?;
        self.response.encode(out,version)?;
        Ok(())
    }
}