use std::io::Write;
use std::io::Result as IoResult;
use result::WebSocketResult;
use ws::dataframe::DataFrame;
use stream::sync::AsTcpStream;
use ws;
use ws::sender::Sender as SenderTrait;
pub use stream::sync::Shutdown;
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: mask }
}
}
impl ws::Sender for Sender {
fn is_masked(&self) -> bool {
self.mask
}
}