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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

use serde::{Deserialize, Serialize};
use serde_cbor::{from_slice, to_vec, Value};
use std::{
    error::Error,
    io::{Read, Write as IoWrite},
    pin::Pin,
};
use tokio::io::{AsyncRead, AsyncReadExt, AsyncWrite, AsyncWriteExt};

use crate::IOPluginError;

const BUF_SIZE: usize = 100;

pub fn io_read<T: for<'a> Deserialize<'a>>(source: &mut (dyn Read + Send)) -> Result<T, Box<dyn Error>> {
    let mut vec = Vec::<u8>::new();
    let mut buf = [0; BUF_SIZE];
    let mut size = BUF_SIZE;
    while size == BUF_SIZE {
        size = match source.read(&mut buf) {
            Ok(0) => Err(IOPluginError::PipeClosed.into()),
            Err(err) => Err(err.into()),
            Ok(read) => Ok::<_, Box<dyn Error>>(read),
        }?;
        vec.extend(&buf[..size]);
    }
    Ok(from_slice(vec.as_slice())?)
}

pub fn io_write<T: Serialize, Write: IoWrite + Send>(sink: &mut Write, message: T) -> Result<(), Box<dyn Error>> {
    let message = to_vec(&message)?;
    sink.write_all(&message)?;
    sink.flush()?;
    Ok(())
}

pub async fn io_read_async<T: for<'a> Deserialize<'a>>(
    mut source: Pin<&mut (dyn AsyncRead + Send)>,
) -> Result<T, Box<dyn Error>> {
    let mut vec = Vec::<u8>::new();
    let mut buf = [0; BUF_SIZE];
    let mut size = BUF_SIZE;
    while size == BUF_SIZE {
        size = match source.read(&mut buf).await {
            Ok(0) => Err(IOPluginError::PipeClosed.into()),
            Err(err) => Err(err.into()),
            Ok(read) => Ok::<_, Box<dyn Error>>(read),
        }?;
        vec.extend(&buf[..size]);
    }
    Ok(from_slice(vec.as_slice())?)
}

pub async fn io_write_async<T: Serialize, Write: AsyncWrite + Send>(
    mut sink: Pin<&mut Write>,
    message: T,
) -> Result<(), Box<dyn Error>> {
    let message = to_vec(&message)?;
    sink.write_all(&message).await?;
    sink.flush().await?;
    Ok(())
}

#[allow(dead_code)]
pub type Generic = Value;