cerberus-cli 0.1.7

Active Aggressive CLI Tools for Search Vulnerability
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

use crossterm::{
    cursor, event::{self, Event, KeyCode, KeyEvent}, execute, terminal::{size, Clear, ClearType, SetSize}
};
use std::{io::{self, Write}, time::Duration};

use crate::pages::{
    geoip::{handle_geoip_events, page::draw_geoip},
    home::draw_home,
    home_events::handle_home_events,
    settings::draw_settings,
    settings_events::handle_settings_events,
    settings_typing::SettingsMenu,
};

pub enum Page {
    Home,
    Settings(SettingsMenu),
    GeoIP,
}

pub enum Transition {
    To(Page),
    Quit,
    Refresh,
}

fn check_terminal_size() -> io::Result<bool> {
    let (width, height) = size()?;
    Ok(width >= 80 && height >= 43)
}


pub fn run_app<W: Write>(stdout: &mut W) -> io::Result<()> {
    let mut current_page = Page::Home;

    loop {
        // Clear the screen
        execute!(stdout, Clear(ClearType::All))?;

        // Handle events
        match current_page {
            Page::Home => match handle_home_events(stdout)? {
                Transition::To(page) => current_page = page,
                Transition::Quit => return Ok(()),
                Transition::Refresh => draw_home(stdout)?,
            },
            Page::Settings(ref mut submenu) => match handle_settings_events(stdout, submenu)? {
                Transition::To(page) => current_page = page,
                Transition::Quit => return Ok(()),
                Transition::Refresh => draw_settings(stdout, submenu)?,
            },
            Page::GeoIP => match handle_geoip_events(stdout)? {
                Transition::To(page) => current_page = page,
                Transition::Quit => return Ok(()),
                Transition::Refresh => draw_geoip(stdout)?,
            },
        }

    }
}