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

use crate::ShellError;
use std::io::{BufRead, BufReader, Read};

pub struct BufferedReader<R: Read> {
    input: BufReader<R>,
    error: bool,
}

impl<R: Read> BufferedReader<R> {
    pub fn new(input: BufReader<R>) -> Self {
        Self {
            input,
            error: false,
        }
    }

    pub fn into_inner(self) -> BufReader<R> {
        self.input
    }
}

impl<R: Read> Iterator for BufferedReader<R> {
    type Item = Result<Vec<u8>, ShellError>;

    fn next(&mut self) -> Option<Self::Item> {
        // Don't try to read more data if an error occurs
        if self.error {
            return None;
        }

        let buffer = self.input.fill_buf();
        match buffer {
            Ok(s) => {
                let result = s.to_vec();

                let buffer_len = s.len();

                if buffer_len == 0 {
                    None
                } else {
                    self.input.consume(buffer_len);

                    Some(Ok(result))
                }
            }
            Err(e) => {
                self.error = true;
                Some(Err(ShellError::IOError { msg: e.to_string() }))
            }
        }
    }
}