data_stream/
text.rs

1use crate::{FromStream, ToStream, collections::SizeSettings};
2
3use std::io::{Error, ErrorKind, Read, Result, Write};
4
5impl<S> ToStream<S> for char {
6    fn to_stream<W: Write>(&self, stream: &mut W) -> Result<()> {
7        let mut buf = [0u8; 4];
8        let encoded = self.encode_utf8(&mut buf);
9        let len = encoded.len() as u8;
10        stream.write_all(&[len])?;
11        stream.write_all(&buf[..len as usize])
12    }
13}
14
15impl<S> FromStream<S> for char {
16    fn from_stream<R: Read>(stream: &mut R) -> Result<Self> {
17        let mut len_buf = [0u8; 1];
18        stream.read_exact(&mut len_buf)?;
19        let len = len_buf[0] as usize;
20
21        if len == 0 || len > 4 {
22            return Err(Error::new(
23                ErrorKind::InvalidData,
24                "Invalid UTF-8 char length",
25            ));
26        }
27
28        let mut buf = [0u8; 4];
29        stream.read_exact(&mut buf[..len])?;
30
31        std::str::from_utf8(&buf[..len])
32            .ok()
33            .and_then(|s| s.chars().next())
34            .ok_or_else(|| Error::new(ErrorKind::InvalidData, "Invalid UTF-8 char"))
35    }
36}
37
38impl<S: SizeSettings> ToStream<S> for String {
39    fn to_stream<W: Write>(&self, stream: &mut W) -> Result<()> {
40        let bytes = self.as_bytes();
41        S::size_to_stream(bytes.len(), stream)?;
42        stream.write_all(bytes)
43    }
44}
45
46impl<S: SizeSettings> FromStream<S> for String {
47    fn from_stream<R: Read>(stream: &mut R) -> Result<Self> {
48        let len = S::size_from_stream(stream)?;
49        let mut buf = vec![0u8; len];
50        stream.read_exact(&mut buf)?;
51        Self::from_utf8(buf).map_err(|e| Error::new(ErrorKind::InvalidData, e))
52    }
53}
54
55impl<S: SizeSettings> ToStream<S> for str {
56    fn to_stream<W: Write>(&self, stream: &mut W) -> Result<()> {
57        let bytes = self.as_bytes();
58        S::size_to_stream(bytes.len(), stream)?;
59        stream.write_all(bytes)
60    }
61}
data_stream/text.rs

data_stream/
text.rs
