repartee 0.4.0

A modern terminal IRC client built with Ratatui and Tokio
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use std::io::{self, Write};
use tokio::sync::mpsc;

use super::protocol::MainMessage;

/// A `Write` implementation that buffers bytes and sends them as
/// `MainMessage::Output` chunks through an mpsc channel on `flush()`.
///
/// A spawned tokio task drains the receiver and writes framed messages
/// to the actual `UnixStream`.
pub struct SocketWriter {
    buffer: Vec<u8>,
    tx: mpsc::UnboundedSender<MainMessage>,
}

impl SocketWriter {
    pub fn new(tx: mpsc::UnboundedSender<MainMessage>) -> Self {
        Self {
            buffer: Vec::with_capacity(8192),
            tx,
        }
    }
}

impl Write for SocketWriter {
    fn write(&mut self, buf: &[u8]) -> io::Result<usize> {
        self.buffer.extend_from_slice(buf);
        Ok(buf.len())
    }

    fn flush(&mut self) -> io::Result<()> {
        if !self.buffer.is_empty() {
            let data = std::mem::replace(&mut self.buffer, Vec::with_capacity(8192));
            self.tx.send(MainMessage::Output(data)).map_err(|_| {
                io::Error::new(io::ErrorKind::BrokenPipe, "socket output channel closed")
            })?;
        }
        Ok(())
    }
}