use std::task::Poll;
use bytes::Buf;
use crate::container::{Container, Frame};
#[derive(Default)]
pub struct Wire;
impl Container for Wire {
type Error = crate::Error;
fn write(&self, group: &mut moq_net::group::Producer, frames: &[Frame]) -> Result<(), Self::Error> {
for frame in frames {
let hang_frame = hang::container::Frame {
timestamp: frame.timestamp,
payload: frame.payload.clone(),
};
hang_frame.write_to(group)?;
}
Ok(())
}
fn poll_read(
&self,
group: &mut moq_net::group::Consumer,
waiter: &kio::Waiter,
) -> Poll<Result<Option<Vec<Frame>>, Self::Error>> {
use std::task::ready;
loop {
let Some(data) = ready!(group.poll_read_frame(waiter).map_err(hang::Error::from)?) else {
return Poll::Ready(Ok(None));
};
let mut hang_frame = hang::container::Frame::decode(data.payload)?;
if !hang_frame.payload.has_remaining() {
continue;
}
let payload = hang_frame.payload.copy_to_bytes(hang_frame.payload.remaining());
return Poll::Ready(Ok(Some(vec![Frame {
timestamp: hang_frame.timestamp,
payload,
keyframe: false,
duration: None,
}])));
}
}
}