saddle-boundary 0.3.34

//! Only raw immutable standard wire is encoded by the dependency driver. Native
//! response storage is prepaid in the original account before copying bytes.
use prost::bytes::{Buf, BufMut};
use saddle_admission::{AdmissionError, ReadOnlyInput, RequestMemory};
use tonic::codec::{Codec, DecodeBuf, Decoder, EncodeBuf, Encoder};

pub(crate) type Response = Result<ReadOnlyInput<Vec<u8>>, AdmissionError>;
pub(crate) struct WireCodec {
    pub memory: RequestMemory,
}
pub(crate) struct WireEncoder;
pub(crate) struct WireDecoder {
    memory: RequestMemory,
}
impl Codec for WireCodec {
    type Encode = ReadOnlyInput<Vec<u8>>;
    type Decode = Response;
    type Encoder = WireEncoder;
    type Decoder = WireDecoder;
    fn encoder(&mut self) -> Self::Encoder {
        WireEncoder
    }
    fn decoder(&mut self) -> Self::Decoder {
        WireDecoder {
            memory: self.memory.clone(),
        }
    }
}
impl Encoder for WireEncoder {
    type Item = ReadOnlyInput<Vec<u8>>;
    type Error = tonic::Status;
    fn encode(&mut self, item: Self::Item, buffer: &mut EncodeBuf<'_>) -> Result<(), Self::Error> {
        buffer.put_slice(item.get());
        // item is destroyed only after the encoder no longer reads its fields.
        Ok(())
    }
}
impl Decoder for WireDecoder {
    type Item = Response;
    type Error = tonic::Status;
    fn decode(&mut self, buffer: &mut DecodeBuf<'_>) -> Result<Option<Self::Item>, Self::Error> {
        let len = buffer.remaining();
        let result = self.memory.framework_output(|builder| {
            builder.write_bytes(len, |target| {
                buffer.copy_to_slice(target);
                Ok(())
            })
        });
        // Resource failure is returned intact to the outer original-source
        // recorder; converting it to tonic::Status would discard its identity.
        Ok(Some(result))
    }
}