hex_patch/app/
text.rs

1use ratatui::text::{Line, Span, Text};
2
3use super::{log::NotificationLevel, settings::color_settings::ColorSettings, App};
4
5impl App {
6    pub(super) fn bytes_to_styled_text(
7        color_settings: &ColorSettings,
8        bytes: &'_ [u8],
9        block_size: usize,
10        blocks_per_row: usize,
11        selected_byte_offset: usize,
12    ) -> Text<'static> {
13        let mut ret = Text::default();
14        ret.lines
15            .reserve(bytes.len() / (block_size * blocks_per_row) + 1);
16        let mut current_line = Line::default();
17        let mut local_block = 0;
18        let mut local_byte = 0;
19        for (byte_index, b) in bytes.iter().enumerate() {
20            let style = if byte_index == selected_byte_offset {
21                color_settings.text_selected
22            } else {
23                Self::get_style_for_byte(color_settings, *b)
24            };
25            let mut next_line = false;
26            let char = Self::u8_to_char(*b);
27            let char_string = char.to_string();
28            let span = Span::styled(char_string, style);
29            current_line.spans.push(span);
30            let mut spacing_string = " ".to_string();
31            local_byte += 1;
32            if local_byte % block_size == 0 {
33                local_byte = 0;
34                spacing_string.push(' ');
35
36                local_block += 1;
37                if local_block % blocks_per_row == 0 {
38                    local_block = 0;
39                    next_line = true;
40                }
41            }
42
43            let span = Span::raw(spacing_string);
44            current_line.spans.push(span);
45
46            if next_line {
47                let new_line = std::mem::take(&mut current_line);
48                ret.lines.push(new_line);
49            }
50        }
51        if !current_line.spans.is_empty() {
52            ret.lines.push(current_line);
53        }
54
55        ret
56    }
57
58    pub(super) fn insert_text(&mut self, text: &str) {
59        self.patch_bytes(text.as_bytes(), false);
60    }
61
62    fn found_text_here(&self, starting_from: usize, text: &str) -> bool {
63        for (i, byte) in text.bytes().enumerate() {
64            if self.data.len() <= starting_from + i || self.data.bytes()[starting_from + i] != byte
65            {
66                return false;
67            }
68        }
69        true
70    }
71
72    pub(super) fn get_text_view(&self, start_row: usize, end_row: usize) -> Text<'static> {
73        let start_byte = start_row * self.blocks_per_row * self.block_size;
74        let end_byte = end_row * self.blocks_per_row * self.block_size;
75        let end_byte = std::cmp::min(end_byte, self.data.len());
76        let bytes = &self.data.bytes()[start_byte..end_byte];
77        let selected_byte_offset = self
78            .get_cursor_position()
79            .global_byte_index
80            .saturating_sub(start_byte);
81        Self::bytes_to_styled_text(
82            &self.settings.color,
83            bytes,
84            self.block_size,
85            self.blocks_per_row,
86            selected_byte_offset,
87        )
88    }
89
90    pub(super) fn find_text(&mut self, text: &str) {
91        if text.is_empty() || self.data.is_empty() {
92            return;
93        }
94        let already_searched = self.text_last_searched_string == text;
95        if !already_searched {
96            self.text_last_searched_string = text.to_string();
97        }
98        let mut search_here = self.get_cursor_position().global_byte_index;
99        // find the next occurrence of the text
100        if already_searched && Self::found_text_here(self, search_here, text) {
101            search_here += text.len();
102        } else {
103            search_here = 0;
104        }
105        let max_search_index = self.data.len() + search_here;
106        while search_here < max_search_index {
107            let actual_search_here = search_here % self.data.len();
108            if Self::found_text_here(self, actual_search_here, text) {
109                self.jump_to(actual_search_here, false);
110                return;
111            }
112            search_here += 1;
113        }
114        self.log(NotificationLevel::Warning, "Text not found");
115    }
116
117    pub(super) fn u8_to_char(input: u8) -> char {
118        match input {
119            0x20..=0x7E => input as char,
120            0x0A => '⏎',
121            0x0C => '↡',
122            0x0D => '↵',
123            0x08 => '⇤',
124            0x09 => '⇥',
125            0x1B => '␛',
126            0x7F => '␡',
127            _ => '.',
128        }
129    }
130}
131
132#[cfg(test)]
133mod tests {
134    use super::*;
135
136    #[test]
137    fn test_bytes_to_styled_text() {
138        let color_settings = ColorSettings::get_default_dark_theme();
139        let bytes = b"CAFEBABE";
140        let block_size = 8;
141        let blocks_per_row = 2;
142        let selected_byte_offset = 0;
143        let text = App::bytes_to_styled_text(
144            &color_settings,
145            bytes,
146            block_size,
147            blocks_per_row,
148            selected_byte_offset,
149        );
150        assert_eq!(text.lines.len(), 1);
151        let mut char_index = 0;
152        for char in text.lines[0]
153            .spans
154            .iter()
155            .flat_map(|span| span.content.chars())
156        {
157            if char.is_alphanumeric() {
158                assert_eq!(char, bytes[char_index] as char);
159                char_index += 1;
160            }
161        }
162        assert_eq!(char_index, bytes.len());
163    }
164}
hex_patch/app/text.rs

hex_patch/app/
text.rs
