1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24

use crate::error::NbsError;
use byteorder::{LittleEndian, ReadBytesExt, WriteBytesExt};
use std::io;

pub trait ReadStringExt: ReadBytesExt {
    fn read_string(&mut self) -> Result<String, NbsError> {
        let len = self.read_i32::<LittleEndian>()?;
        let mut buffer: Vec<u8> = Vec::with_capacity(len as usize);
        buffer.resize(len as usize, 0u8);
        self.read_exact(&mut buffer)?;
        Ok(String::from_utf8(buffer)?)
    }
}

pub trait WriteStringExt: WriteBytesExt {
    fn write_string(&mut self, s: &str) -> io::Result<()> {
        self.write_i32::<LittleEndian>(s.len() as i32)?;
        self.write_all(s.as_bytes())?;
        Ok(())
    }
}

impl<R> ReadStringExt for R where R: ReadBytesExt {}
impl<W> WriteStringExt for W where W: WriteBytesExt {}