rust/lib-hedgewars-engine/src/ipc.rs
author unC0Rr
Wed, 21 Nov 2018 15:49:16 +0100
changeset 14260 f0c0d2d217c3
parent 14255 a239e4243cf9
child 14261 1b8c889027a3
permissions -rw-r--r--
IPC implementation

use hedgewars_engine_messages::{messages::*, parser::extract_message};
use netbuf::*;
use std::io::*;

pub struct IPC {
    in_buffer: Buf,
    out_buffer: Buf,
}

impl IPC {
    pub fn new() -> Self {
        Self {
            in_buffer: Buf::new(),
            out_buffer: Buf::new(),
        }
    }

    pub fn send_message(&mut self, message: &EngineMessage) {
        self.out_buffer.write(&message.to_bytes()).unwrap();
    }
}

impl Write for IPC {
    fn write(&mut self, buf: &[u8]) -> Result<usize> {
        self.in_buffer.write(buf)
    }

    fn flush(&mut self) -> Result<()> {
        self.in_buffer.flush()
    }
}

impl Read for IPC {
    fn read(&mut self, buf: &mut [u8]) -> Result<usize> {
        let result = self.out_buffer.as_ref().read(buf);

        if let Ok(read_bytes) = result {
            self.out_buffer.consume(read_bytes);
        }

        result
    }
}

impl Iterator for IPC {
    type Item = EngineMessage;

    fn next(&mut self) -> Option<Self::Item> {
        let (consumed, message) = extract_message(&self.in_buffer[..])?;

        self.in_buffer.consume(consumed);

        Some(message)
    }
}