entui 0.1.1

A file entropy visualization in the terminal with a nice interface
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
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91

use ratatui::{
    Frame,
    layout::{Constraint, Direction, Layout},
    style::{Color, Modifier, Style},
    symbols,
    text::Span,
    widgets::{Axis, Block, Borders, Chart, Dataset, GraphType},
};

use crate::app::App;

pub fn draw(f: &mut Frame, app: &App) {
    let chunks = Layout::default()
        .direction(Direction::Vertical)
        .constraints([Constraint::Min(0), Constraint::Length(1)].as_ref())
        .split(f.area());

    let datasets = vec![
        Dataset::default()
            .name("EntUI")
            .marker(symbols::Marker::Braille)
            .graph_type(GraphType::Line)
            .style(Style::default().fg(Color::Cyan))
            .data(&app.entropy_data),
    ];

    let x_labels = if app.hex_mode {
        vec![
            Span::raw(format!("0x{:08x}", app.window_start as u64)),
            Span::raw(format!(
                "0x{:08x}",
                app.window_start as u64 + app.window_width as u64 / 2
            )),
            Span::raw(format!(
                "0x{:08x}",
                app.window_start as u64 + app.window_width as u64
            )),
        ]
    } else {
        vec![
            Span::raw(format!("{}", app.window_start as u64)),
            Span::raw(format!(
                "{}",
                app.window_start as u64 + app.window_width as u64 / 2
            )),
            Span::raw(format!(
                "{}",
                app.window_start as u64 + app.window_width as u64
            )),
        ]
    };

    let chart = Chart::new(datasets)
        .block(
            Block::default()
                .title(Span::styled(
                    "Shannon Entropy",
                    Style::default()
                        .fg(Color::Cyan)
                        .add_modifier(Modifier::BOLD),
                ))
                .borders(Borders::ALL),
        )
        .x_axis(
            Axis::default()
                .title("File Offset")
                .style(Style::default().fg(Color::Gray))
                .bounds([app.window_start, app.window_start + app.window_width])
                .labels(x_labels),
        )
        .y_axis(
            Axis::default()
                .title("Entropy")
                .style(Style::default().fg(Color::Gray))
                .bounds([0.0, 8.0])
                .labels(vec![
                    Span::raw("       0 "),
                    Span::raw("       4 "),
                    Span::raw("       8 "),
                ]),
        );

    f.render_widget(chart, chunks[0]);

    let instructions =
        Span::raw("Commands: [-/+] Zoom | [Arrows] Scroll | [h] Toggle Hex Offsets | [q] Quit");
    f.render_widget(Block::default().borders(Borders::NONE).title(""), chunks[1]);
    use ratatui::widgets::Paragraph;
    let p = Paragraph::new(instructions).style(Style::default().fg(Color::White));
    f.render_widget(p, chunks[1]);
}