use crate::result::WebSocketResult;
use crate::stream::sync::AsTcpStream;
pub use crate::stream::sync::Shutdown;
use crate::ws;
use crate::ws::dataframe::DataFrame;
use crate::ws::sender::Sender as SenderTrait;
use std::io::Result as IoResult;
use std::io::Write;
pub struct Writer<W> {
pub stream: W,
pub sender: Sender,
}
impl<W> Writer<W>
where
W: Write,
{
pub fn send_dataframe<D>(&mut self, dataframe: &D) -> WebSocketResult<()>
where
D: DataFrame,
W: Write,
{
self.sender.send_dataframe(&mut self.stream, dataframe)
}
pub fn send_message<M>(&mut self, message: &M) -> WebSocketResult<()>
where
M: ws::Message,
{
self.sender.send_message(&mut self.stream, message)
}
}
impl<S> Writer<S>
where
S: AsTcpStream + Write,
{
pub fn shutdown(&self) -> IoResult<()> {
self.stream.as_tcp().shutdown(Shutdown::Write)
}
pub fn shutdown_all(&self) -> IoResult<()> {
self.stream.as_tcp().shutdown(Shutdown::Both)
}
}
pub struct Sender {
mask: bool,
}
impl Sender {
pub fn new(mask: bool) -> Sender {
Sender { mask }
}
}
impl ws::Sender for Sender {
fn is_masked(&self) -> bool {
self.mask
}
}