summaryrefslogtreecommitdiff
path: root/src/hex.rs
blob: 6dfbe7f8f1dcc8de274ef088e1f7ce3422beb8a3 (plain)
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
use std::{fs, io};
use std::io::stdout;
use ratatui::{backend::CrosstermBackend, Terminal};
use crossterm::{execute, terminal::{disable_raw_mode, enable_raw_mode, EnterAlternateScreen, LeaveAlternateScreen}};
use crate::{input, render};

pub struct HexEditor {
    filename: String,
    data: Vec<u8>,
    cursor: usize,
    terminal: Terminal<CrosstermBackend<std::io::Stdout>>,
}

impl HexEditor {
    pub fn new(filename: String, data: Vec<u8>) -> io::Result<Self> {
        enable_raw_mode()?;
        execute!(stdout(), EnterAlternateScreen)?;
        let terminal = Terminal::new(CrosstermBackend::new(stdout()))?;
        Ok(Self { filename, data, cursor: 0, terminal })
    }

    pub fn run(&mut self) -> io::Result<()> {
        while input::handle_input(&mut self.cursor, &mut self.data[..])? {
            render::draw(&mut self.terminal, &self.data, self.cursor, &self.filename)?;
        }
        self.cleanup()
    }

    pub fn cleanup(&mut self) -> io::Result<()> {
        disable_raw_mode()?;
        execute!(stdout(), LeaveAlternateScreen)?;
        fs::write(&self.filename, &self.data)?;
        Ok(())
    }
}