src/types/string_buffer.rs


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

pub struct StringBuffer {
    pub bytes: Vec<u8>,
    pub pointer: usize,
}

impl StringBuffer {
    pub fn new() -> Self {
        Self {
            bytes: Vec::new(),
            pointer: 0,
        }
    }

    pub fn from_str(text: &str) -> Self {
        let mut new = Self::new();
        new.set_str(text);
        new
    }

    pub fn set_str(&mut self, text: &str) {
        self.bytes = text.bytes().collect();
        self.pointer = 0;
    }

    pub fn read(&mut self) -> u8 {
        if let Some(byte) = self.bytes.get(self.pointer) {
            self.pointer += 1;
            *byte
        } else {
            0
        }
    }

    pub fn restart(&mut self) {
        self.pointer = 0;
    }
}