8 Commits

Author SHA1 Message Date
tmerritt 17780c02d5 UI evolution
** Re-evaluating need for users to edit list of hosts at runtime **
2025-05-01 14:54:26 -04:00
tmerritt 141d0ee899 BUGFIX: Delete works for lists of hosts using popups 2025-05-01 09:54:09 -04:00
tmerritt 022359d3fd minor improvements for better 'rustifying' of it 2025-04-30 20:31:19 -04:00
tmerritt 21b41f5593 WIP: working on better ui interaction 2025-04-30 16:09:22 -04:00
tmerritt 9ff76954e0 BUGFIX: removes extra quotes from log entry time entries
FEATURE: Allows delete of target

UPDATE: UI Uses more popups to make the UI more 'layered'
2025-04-30 16:08:56 -04:00
tmerritt 8d0452d752 Fix version tag 2025-04-29 09:29:51 -04:00
tmerritt 49092d9ad7 Updates to use:
- if passed file doesnt exist
 - check for hosts.txt
- if no file passed
 - check for hosts.txt
- if hosts.txt missing
 - default hosts
2025-04-29 09:29:13 -04:00
tmerritt 0261914cda Updates label for VPN endpoints and added more hosts to check 2025-04-28 15:48:54 -04:00
13 changed files with 401 additions and 513 deletions
Generated
+1 -1
View File
@@ -709,7 +709,7 @@ dependencies = [
[[package]] [[package]]
name = "pp" name = "pp"
version = "0.2.0" version = "0.2.2-PREVIEW"
dependencies = [ dependencies = [
"ansi_term", "ansi_term",
"chrono", "chrono",
+1 -1
View File
@@ -1,6 +1,6 @@
[package] [package]
name = "pp" name = "pp"
version = "0.2.0" version = "0.2.2-PREVIEW"
edition = "2024" edition = "2024"
[dependencies] [dependencies]
+8 -4
View File
@@ -1,14 +1,18 @@
address,name address,name
10.3.100.1,Belleville Gateway 10.3.100.1,Belleville Gateway
10.11.31.3,Belleville VPN 11-31 10.11.31.1,Belleville VPN 11-31 (P)
10.12.32.1,Belleville VPN 12-32 10.11.31.3,Belleville VPN 11-31 (B)
10.12.32.1,Belleville VPN 12-32 (P)
10.12.32.3,Belleville VPN 12-32 (B)
192.186.110.6,Belleville Cogeco 192.186.110.6,Belleville Cogeco
129.222.197.36,Belleville Starlink 129.222.197.36,Belleville Starlink
10.2.100.1,Lindsay Gateway 10.2.100.1,Lindsay Gateway
24.143.184.98,Lindsay Cogeco 24.143.184.98,Lindsay Cogeco
192.168.1.50,Lindsay Sign 192.168.1.50,Lindsay Sign
10.11.21.1,Lindsay VPN 11-21 10.11.21.1,Lindsay VPN 11-21 (P)
10.12.22.1,Lindsay VPN 11-22 10.11.21.2,Lindsay VPN 11-21 (L)
10.11.21.1,Lindsay VPN 11-22 (P)
10.12.22.2,Lindsay VPN 11-22 (L)
8.8.8.8,Google DNS 8.8.8.8,Google DNS
1.1.1.1,1111 DNS 1.1.1.1,1111 DNS
192.168.0.1,Peterborough Gateway 192.168.0.1,Peterborough Gateway
+1 -1
View File
@@ -5,6 +5,6 @@ use clap::Parser;
#[command(version, about, long_about = None)] #[command(version, about, long_about = None)]
pub struct AppSettings { pub struct AppSettings {
/// File of list of hosts /// File of list of hosts
#[arg(short, long)] #[arg(short, long, default_value = None)]
pub ping_host_file: Option<PathBuf>, pub ping_host_file: Option<PathBuf>,
} }
+10 -236
View File
@@ -1,240 +1,14 @@
use std::collections::BTreeMap;
use std::fs::File;
use std::io;
use std::io::BufRead;
use std::net::Ipv4Addr;
use std::path::{Path, PathBuf};
use std::str::FromStr;
use std::sync::mpsc;
use std::time::{Duration, SystemTime};
use clap::Parser; use clap::Parser;
use pp::ping_result::PingResult; use ratatui::widgets::TableState;
use pp::ping_request::PingRequest;
use pp::manager::Manager;
use pp::SECONDS_BETWEEN_DISPLAY;
use pp::target_state::TargetState;
use std::{env, error::Error, ffi::OsString, process};
use color_eyre::owo_colors::OwoColorize;
use crossterm::style::Stylize;
use log::debug;
use pp::app_settings::AppSettings; use pp::app_settings::AppSettings;
use pp::tui::ratatui_app::RatatuiApp;
fn main() -> color_eyre::Result<()> {
// find out what file we are using to get our hosts
let settings = AppSettings::parse();
const SECONDS_IN_MINUTE: u32 = 60; color_eyre::install()?;
const SECONDS_IN_HOUR: u32 = SECONDS_IN_MINUTE * 60; let terminal = ratatui::init();
const SECONDS_IN_DAY: u32 = SECONDS_IN_HOUR * 24; let result = RatatuiApp::new(settings.ping_host_file).run(terminal);
ratatui::restore();
pub fn duration_to_string(to_convert: Duration) -> String { result
let mut total_seconds = to_convert.as_secs() as u32;
let mut working_string = String::new();
if total_seconds > 86400 {
// days
let num_days = (total_seconds / SECONDS_IN_DAY) as u32;
working_string = format!("{} days", num_days);
total_seconds = total_seconds - (num_days * SECONDS_IN_DAY);
}
if total_seconds > 3600 {
// hours
let num_hours = (total_seconds / SECONDS_IN_HOUR) as u32;
if num_hours > 0 {
working_string = format!("{} {} hours", working_string, num_hours);
total_seconds = total_seconds - (num_hours * SECONDS_IN_HOUR);
}
}
if total_seconds > 60 {
let num_minutes = (total_seconds / SECONDS_IN_MINUTE) as u32;
if num_minutes > 0 {
working_string = format!("{} {} minutes", working_string, num_minutes);
total_seconds = total_seconds - (num_minutes * SECONDS_IN_MINUTE);
}
// minutes
}
working_string = format!("{} {} seconds", working_string, total_seconds);
working_string
} }
struct PPState {}
impl PPState {
pub fn get_default_targets() -> BTreeMap<String, TargetState> {
let mut working = BTreeMap::new();
working.insert("Localhost".to_string(),
TargetState {
name: "Localhost".to_string(),
target: Ipv4Addr::new(127, 0, 0, 1),
..TargetState::default()
},
);
working.insert("Home Gateway".to_string(),
TargetState {
name: "Home Gateway".to_string(),
target: Ipv4Addr::new(172, 24, 0, 1),
..TargetState::default()
},
);
working.insert("1111 DNS".to_string(),
TargetState {
name: "1111 DNS".to_string(),
target: Ipv4Addr::new(1, 1, 1, 1),
..TargetState::default()
},
);
working.insert("Google DNS".to_string(),
TargetState {
name: "Google DNS".to_string(),
target: Ipv4Addr::new(8, 8, 8, 8),
..TargetState::default()
},
);
working.insert("Site IP 1".to_string(),
TargetState {
name: "Site IP 1".to_string(),
target: Ipv4Addr::new(216, 121, 247, 231),
..TargetState::default()
},
);
working.insert("Site IP 2".to_string(),
TargetState {
name: "Site IP 2".to_string(),
target: Ipv4Addr::new(216, 234, 202, 122),
..TargetState::default()
},
);
working.insert("Site IP 3".to_string(),
TargetState {
name: "Site IP 3".to_string(),
target: Ipv4Addr::new(24, 143, 184, 98),
..TargetState::default()
},
);
working
}
pub fn build_targets_from_file(filename: Option<PathBuf>) -> BTreeMap<String, TargetState> {
// PPState::get_default_targets();
if let Some(file) = filename {
let mut working = BTreeMap::new();
if !&file.exists() {
debug!("Cant load hosts from {:?}. Using default host list.", file.clone().as_os_str());
// use
PPState::get_default_targets()
} else {
debug!("LOADING HOSTS FROM {:?}", file.to_str());
let file = File::open(file);
let mut rdr = csv::Reader::from_reader(file.unwrap());
for result in rdr.records() {
let record = result.unwrap();
working.insert(record[1].to_string(),
TargetState {
name: record[1].to_string(),
target: Ipv4Addr::from_str(&record[0]).unwrap(),
alive: false,
last_alive_change: SystemTime::now(),
last_rtt: 0,
});
}
working
}
} else {
PPState::get_default_targets()
}
}
}
fn ips_from_state(to_read_from: BTreeMap<String, TargetState>) -> Vec<Ipv4Addr> {
let mut working: Vec<Ipv4Addr> = vec![];
for current in to_read_from {
working.push(current.1.target);
}
working
}
/// Simple program to greet a person
fn main() {
// Get App Settings
let settings = AppSettings::parse();
print!("Prep to load targets...");
let file_to_check = match settings.ping_host_file {
None => {
PathBuf::from("./hosts.txt")
}
Some(actual) => {
actual
}
};
let mut targets = PPState::build_targets_from_file(Some(file_to_check));
// channel to send requests to ping
let (ping_response_sender, ping_response_listener) = mpsc::channel::<PingResult>();
println!("Setting up requests for {} hosts.", targets.len());
Manager::spawn_manager_thread(ips_from_state(targets.clone()), ping_response_sender.clone());
let mut display_loop_start = SystemTime::now();
let mut duration_since_last_loop = SystemTime::now().duration_since(display_loop_start).unwrap();
loop {
let now = SystemTime::now();
if let Ok(response) = ping_response_listener.recv_timeout(Duration::from_millis(100)) {
let local_targets = targets.clone();
for (_, (name, current_state)) in local_targets.iter().enumerate() {
if current_state.target == response.target {
let last_alive_change = if response.success == current_state.alive {
current_state.last_alive_change
} else {
SystemTime::now()
};
let new_state = TargetState {
name: current_state.name.clone(),
target: current_state.target,
alive: response.success,
last_rtt: response.rtt,
last_alive_change,
};
targets.insert(name.clone(), new_state);
}
}
}
duration_since_last_loop = now
.duration_since(display_loop_start)
.expect("unable to figure out how long ago we displayed stuff");
if duration_since_last_loop.as_secs() > SECONDS_BETWEEN_DISPLAY as u64 {
println!("DISPLAY LOOP");
println!("Host \t\t\t\t\t | Alive \t | RTT \t\t");
for (name, current_result) in targets.clone() {
let time_since_last_change = now
.duration_since(current_result.last_alive_change)
.unwrap_or(Duration::from_secs(0));
let mut target_string = format!("{} ({})", name, current_result.target);
while target_string.len() < 34 {
target_string.push(' ');
// target_string = format!("{} ", target_string);
}
target_string = if current_result.alive {
target_string.green().to_string()
} else {
target_string.red().to_string()
};
println!("{} \t | {} \t | {}\t | Changed {} ago",
target_string,
current_result.alive,
current_result.last_rtt,
duration_to_string(time_since_last_change)
);
}
display_loop_start = now;
}
}
}
-15
View File
@@ -1,15 +0,0 @@
use clap::Parser;
use ratatui::widgets::TableState;
use pp::app_settings::AppSettings;
use pp::tui::ratatui_app::RatatuiApp;
fn main() -> color_eyre::Result<()> {
// find out what file we are using to get our hosts
let settings = AppSettings::parse();
color_eyre::install()?;
let terminal = ratatui::init();
let result = RatatuiApp::new(settings.ping_host_file).run(terminal);
ratatui::restore();
result
}
-1
View File
@@ -1,5 +1,4 @@
pub mod mode_editing; pub mod mode_editing;
pub mod mode_exiting;
pub mod mode_monitoring; pub mod mode_monitoring;
pub mod ratatui_app; pub mod ratatui_app;
pub mod target_state_widget; pub mod target_state_widget;
-2
View File
@@ -8,7 +8,6 @@ use ratatui::widgets::{Block, Paragraph};
use crate::tui::ratatui_screens::RatatuiScreens::Monitoring; use crate::tui::ratatui_screens::RatatuiScreens::Monitoring;
pub struct RatatuiAddingMode {} pub struct RatatuiAddingMode {}
impl RatatuiAddingMode { impl RatatuiAddingMode {
pub fn render(frame: &mut Frame, state: &mut RatatuiApp) { pub fn render(frame: &mut Frame, state: &mut RatatuiApp) {
frame.render_widget( frame.render_widget(
@@ -17,7 +16,6 @@ impl RatatuiAddingMode {
.centered(), .centered(),
frame.area() frame.area()
); );
} }
pub fn handle_crossterm_events(app: &mut RatatuiApp) -> Result<()> { pub fn handle_crossterm_events(app: &mut RatatuiApp) -> Result<()> {
+15 -2
View File
@@ -14,7 +14,14 @@ pub struct RatatuiDeletingMode {}
impl RatatuiDeletingMode { impl RatatuiDeletingMode {
pub fn render(frame: &mut Frame, state: &mut RatatuiApp) { pub fn render(frame: &mut Frame, state: &mut RatatuiApp) {
let title = Line::from("Delete Host").bold().red().centered(); let title = Line::from("Delete Host").bold().red().centered();
let mut body = format!("Do you really want to delete {} (Y/N)", state.selected_host); let as_list = state.state.clone();
let mut nice_status = None;
for (index, (label, data)) in as_list.iter().enumerate() {
if index == state.selected_host {
nice_status = Some(data);
}
}
let mut body = format!("Do you really want to delete {} (Y/N)", nice_status.unwrap().name);
frame.render_widget( frame.render_widget(
Paragraph::new(body) Paragraph::new(body)
@@ -29,7 +36,13 @@ impl RatatuiDeletingMode {
match event::read()? { match event::read()? {
Event::Key(key) if key.kind == KeyEventKind::Press => match key.code { Event::Key(key) if key.kind == KeyEventKind::Press => match key.code {
KeyCode::Enter | KeyCode::Char('y') | KeyCode::Char('Y') => { KeyCode::Enter | KeyCode::Char('y') | KeyCode::Char('Y') => {
println!("TiME TO DELETE SELECTED"); let as_list = &app.state.clone();
for (index, (label, _)) in as_list.iter().enumerate() {
if index == app.selected_host {
app.state.remove(label).unwrap();
app.set_screen(Monitoring)
}
}
} }
KeyCode::Esc | KeyCode::Char('n') | KeyCode::Char('N') => { KeyCode::Esc | KeyCode::Char('n') | KeyCode::Char('N') => {
app.set_screen(Monitoring) app.set_screen(Monitoring)
-43
View File
@@ -1,43 +0,0 @@
use std::time::Duration;
use crossterm::event;
use crossterm::event::{Event, KeyCode, KeyEventKind};
use ratatui::Frame;
use ratatui::style::Stylize;
use ratatui::text::Line;
use ratatui::widgets::{Block, Paragraph};
use crate::tui::ratatui_app::RatatuiApp;
use color_eyre::Result;
use crate::tui::ratatui_screens::RatatuiScreens::Monitoring;
pub struct RatatuiExitingMode {}
impl RatatuiExitingMode {
pub fn render(frame: &mut Frame) {
let title = Line::from("Exit?").bold().red().centered();
let mut body = "Do you want to exit? (Y/N)";
frame.render_widget(
Paragraph::new(body)
.block(Block::bordered().title(title))
.centered(),
frame.area(),
);
}
pub fn handle_crossterm_events(app: &mut RatatuiApp) -> Result<()>{
if event::poll(Duration::from_millis(100))? {
match event::read()? {
Event::Key(key) if key.kind == KeyEventKind::Press => match key.code {
KeyCode::Enter | KeyCode::Char('y') | KeyCode::Char('Y') => {
app.set_running(false);
}
KeyCode::Char('n') | KeyCode::Char('N') => {
app.set_screen(Monitoring);
}
_ => {}
},
_ => {}
}
}
Ok(())
}
}
+240 -120
View File
@@ -1,58 +1,37 @@
use crate::duration_to_string; use crate::duration_to_string;
use crate::target_state::TargetState;
use crate::tui::ratatui_app::RatatuiApp;
use color_eyre::Result; use color_eyre::Result;
use crossterm::event; use crossterm::event;
use crossterm::event::{Event, KeyCode, KeyEventKind}; use crossterm::event::{Event, KeyCode, KeyEventKind};
use ratatui::prelude::*;
use ratatui::widgets::{Block, Borders, Cell, List, ListItem, ListState, Paragraph, Row, Table, TableState};
use ratatui::Frame; use ratatui::Frame;
use ratatui::layout::Flex;
use ratatui::prelude::*;
use ratatui::widgets::{
Block, Borders, Cell, Clear, List, ListItem, Paragraph, Row, Table, TableState,
};
use std::time::{Duration, SystemTime}; use std::time::{Duration, SystemTime};
use ratatui::layout::Direction::Vertical;
use crate::tui::ratatui_app::RatatuiApp;
use crate::tui::ratatui_screens::RatatuiScreens::{Deleting, Editing, Exiting};
pub struct RatatuiMonitoringMode {} pub struct RatatuiMonitoringMode {}
impl RatatuiMonitoringMode { impl RatatuiMonitoringMode {
pub fn render(frame: &mut Frame, state: &mut RatatuiApp) { fn render_hosts_list(frame: &mut Frame, state: &mut RatatuiApp, area: Rect) {
let layouts = Layout::default() // Headers
.direction(Direction::Vertical) let headers = ["Host", "RTT", "Last Change"]
.constraints([
Constraint::Fill(1),
Constraint::Length(10),
Constraint::Length(3),
])
.split(frame.area());
let body_layout = layouts[0];
let logs_layout = layouts[1];
let footer_layout = layouts[2];
//
// let title = Line::from(format!("PP v{}", env!("CARGO_PKG_VERSION")))
// .bold()
// .blue()
// .centered();
// let table_size = layouts[0].area();
// let columns = Layout::default()
// .direction(Vertical)
// .constraints([Constraint::Min(30),
// Constraint::Min(6),
// Constraint::Min(4),
// Constraint::Min(30)]);
let headers = ["Host", "Alive", "RTT", "Last Change"]
.iter() .iter()
.map(|h| Cell::from(*h)); .map(|h| Cell::from(*h));
let header = Row::new(headers) let header = Row::new(headers)
.style(Style::default().fg(Color::Yellow)) .style(Style::default().fg(Color::Yellow).bold())
.bottom_margin(1); .bottom_margin(1);
// Rows
let mut rows = vec![]; let mut rows = vec![];
for (_, current) in state.state.iter() { for (_, current) in state.state.iter() {
let mut name_field = format!("{} ({})", current.name.clone(), current.target.clone()); let name_field = format!(
"{:<40}",
while name_field.len() < 40 { format!("{} ({})", current.name.clone(), current.target.clone())
name_field.push(' '); );
}
let name_style = if current.alive { let name_style = if current.alive {
Style::default().fg(Color::Green) Style::default().fg(Color::Green)
@@ -60,105 +39,246 @@ impl RatatuiMonitoringMode {
Style::default().fg(Color::Red) Style::default().fg(Color::Red)
}; };
let to_push = vec![ rows.push(Row::new(vec![
Cell::from(name_field).style(name_style), Cell::from(name_field).style(name_style),
Cell::from(current.alive.to_string()),
Cell::from(current.last_rtt.to_string()), Cell::from(current.last_rtt.to_string()),
Cell::from( Cell::from(format!(
format!("{} ago.", "{} ago.",
duration_to_string( duration_to_string(
SystemTime::now() SystemTime::now()
.duration_since(current.last_alive_change) .duration_since(current.last_alive_change)
.unwrap() .unwrap()
)
) )
) )),
]; ]));
rows.push(Row::new(to_push));
} }
let table = Table::new(rows, vec![Constraint::Min(30), Constraint::Min(6), Constraint::Min(5), Constraint::Min(30)]) // Table Builder
.header(header) let table = Table::new(rows, vec![
.block(Block::default() Constraint::Min(30),
.title("Hosts") Constraint::Min(6),
.borders(Borders::ALL)) Constraint::Min(5),
.widths(&[ Constraint::Min(30),
Constraint::Min(30), ])
Constraint::Min(6), .header(header)
Constraint::Min(4), .block(
Constraint::Min(30) Block::default()
]) .title(Line::from(format!("PP v{}", env!("CARGO_PKG_VERSION"))))
.highlight_style( .borders(Borders::ALL),
Style::default() )
.bg(Color::Blue) .widths(&[
.fg(Color::White) Constraint::Fill(2), // Host
.add_modifier(Modifier::BOLD), Constraint::Min(4), // RTT
) Constraint::Min(30), // TIme Since
.highlight_symbol(">> "); ])
.row_highlight_style(
Style::default()
.bg(Color::Blue)
.fg(Color::White)
.add_modifier(Modifier::BOLD),
)
.highlight_symbol(">> ");
// frame.render_widget(table, layouts[0]); // frame.render_widget(table, layouts[0]);
frame.render_stateful_widget(table, layouts[0], &mut make_state(state.selected_host)); frame.render_stateful_widget(table, area, &mut make_hosts_list_state(state.selected_host));
}
let footer_text = "Press <ESC> or q to exit"; fn render_logs(frame: &mut Frame, state: &mut RatatuiApp, logs_layout: Rect) {
// let footer_text = "Press <ESC> or q to exit | Press d to delete host | Press a to add host"; let list_items: Vec<ListItem> = state
.get_log_entries(10)
let mut list_items = vec![]; .iter()
for entry in state.get_log_entries(10) { .map(|entry| ListItem::new(entry.clone()))
list_items.push(ListItem::new(entry)); .collect();
}
let list = List::new(list_items)
.block(Block::default().title("Logs").borders(Borders::ALL))
.highlight_style(
Style::default()
.fg(Color::Yellow)
.add_modifier(Modifier::BOLD),
);
// .highlight_symbol(">> ");
let mut list_state = ListState::default();
list_state.select(Some(0));
frame.render_stateful_widget(list, logs_layout, &mut list_state);
// Log
frame.render_widget( frame.render_widget(
Paragraph::new(footer_text) List::new(list_items)
.block(Block::bordered()) .block(Block::default().title("Logs").borders(Borders::ALL))
.centered(), .highlight_style(
Style::default()
.fg(Color::Yellow)
.add_modifier(Modifier::BOLD),
),
logs_layout,
);
}
fn render_footer_block(frame: &mut Frame, footer_layout: Rect) {
frame.render_widget(
Paragraph::new(
"Press <ESC> or q to exit | Press d to delete host | Press a to add host",
)
.block(Block::bordered())
.centered(),
footer_layout, footer_layout,
); );
} }
pub fn render(frame: &mut Frame, state: &mut RatatuiApp) {
let layouts = Layout::default()
.direction(Direction::Vertical)
.constraints([
Constraint::Fill(1), // Top Bar
Constraint::Length(10), // Hosts
Constraint::Length(3), // Logs
])
.split(frame.area());
Self::render_hosts_list(frame, state, layouts[0]);
Self::render_logs(frame, state, layouts[1]);
Self::render_footer_block(frame, layouts[2]);
Self::add_popup(frame, state);
Self::exit_popup(frame, state);
Self::delete_popup(frame, state);
}
fn add_popup(frame: &mut Frame, state: &RatatuiApp) {
if state.showing_add_popup {
let area = RatatuiMonitoringMode::popup_area(frame.area(), 10, 40);
frame.render_widget(Clear, area);
frame.render_widget(
Paragraph::new(format!("Host Address : {}", state.add_host_name))
.block(Block::bordered().title("Label")),
area,
);
}
}
fn exit_popup(frame: &mut Frame, state: &RatatuiApp) {
if state.trying_to_exit {
let area = RatatuiMonitoringMode::popup_area(frame.area(), 5, 20);
frame.render_widget(Clear, area);
frame.render_widget(
Paragraph::new("Are you sure? (Y/N)").block(Block::bordered().title("Exit")),
area,
);
}
}
fn delete_popup(frame: &mut Frame, state: &RatatuiApp) {
if state.trying_to_delete {
let as_list = state.state.clone();
let mut host_name = String::new();
for (index, (label, _)) in as_list.iter().enumerate() {
if index == state.selected_host {
host_name = label.clone();
}
}
let block = Paragraph::new(format!("Delete {}", host_name))
.centered()
.block(Block::bordered());
frame.render_widget(block, frame.area());
}
}
fn popup_area(area: Rect, percent_x: u16, percent_y: u16) -> Rect {
let vertical = Layout::vertical([Constraint::Percentage(percent_x)]).flex(Flex::Center);
let horizontal = Layout::horizontal([Constraint::Percentage(percent_y)]).flex(Flex::Center);
let [area] = vertical.areas(area);
let [area] = horizontal.areas(area);
area
}
fn handle_add_popup_inputs(app: &mut RatatuiApp, key: KeyCode) {
match key {
KeyCode::Esc => {
app.showing_add_popup = false;
}
KeyCode::Backspace => {
app.add_host_name.remove(app.add_host_name.len() - 1);
}
KeyCode::Enter => {
app.state.insert(app.add_host_name.clone(), TargetState {
name: app.add_host_name.clone(),
..Default::default()
});
}
KeyCode::Down => {
dbg!("Move down to next field");
}
KeyCode::Up => {
dbg!("Move up to previous field");
}
_ => {
app.add_host_name = format!("{}{}", app.add_host_name, key);
//dbg!(format!("ADD_HOST_NAME: {}", app.add_host_name.clone()));
}
}
}
fn handle_exit_popup_inputs(app: &mut RatatuiApp, key: KeyCode) {
match key {
KeyCode::Char('y') | KeyCode::Char('Y') => {
app.set_running(false);
}
KeyCode::Char('n') | KeyCode::Char('N') => {
app.trying_to_exit = false;
}
_ => {}
}
}
fn handle_monitoring_screen_inputs(app: &mut RatatuiApp, key: KeyCode) {
// Default monitoring Screen
match key {
KeyCode::Down => {
if app.selected_host + 1 == app.state.len() {
app.selected_host = 0;
} else {
app.selected_host += 1;
}
}
KeyCode::Up => {
if app.selected_host == 0 {
app.selected_host = app.state.len() - 1;
} else {
app.selected_host -= 1;
}
}
KeyCode::Esc | KeyCode::Char('q') | KeyCode::Char('Q') => {
app.trying_to_exit = true;
}
KeyCode::Char('a') | KeyCode::Char('A') => {
app.showing_add_popup = true;
}
KeyCode::Char('d') | KeyCode::Char('D') => {
app.trying_to_delete = true;
}
_ => {}
}
}
fn handle_delete_popup_inputs(app: &mut RatatuiApp, key: KeyCode) {
match key {
KeyCode::Char('y') | KeyCode::Char('Y') => {
app.remove_selected_host();
app.trying_to_delete = false;
}
KeyCode::Char('n') | KeyCode::Char('N') => {
dbg!("Do not delete this item.");
app.trying_to_delete = false;
}
_ => {}
}
}
pub fn handle_crossterm_events(app: &mut RatatuiApp) -> Result<()> { pub fn handle_crossterm_events(app: &mut RatatuiApp) -> Result<()> {
if event::poll(Duration::from_millis(100))? { if event::poll(Duration::from_millis(100))? {
match event::read()? { match event::read()? {
Event::Key(key) if key.kind == KeyEventKind::Press => { Event::Key(key) if key.kind == KeyEventKind::Press => {
match (key.modifiers, key.code) { // adding mode
(_, KeyCode::Down) => { if app.showing_add_popup {
if app.selected_host + 1 == app.state.len() { Self::handle_add_popup_inputs(app, key.code);
app.selected_host = 0; } else if app.trying_to_exit {
} else { Self::handle_exit_popup_inputs(app, key.code);
app.selected_host += 1; } else if app.trying_to_delete {
} Self::handle_delete_popup_inputs(app, key.code);
} } else {
(_, KeyCode::Up) => { Self::handle_monitoring_screen_inputs(app, key.code);
if app.selected_host == 0 {
app.selected_host = app.state.len() - 1;
} else {
app.selected_host -= 1;
}
}
(_, KeyCode::Esc) | (_, KeyCode::Char('q')) | (_, KeyCode::Char('Q')) => {
app.set_screen(Exiting);
}
// (_, KeyCode::Char('e')) | (_, KeyCode::Char('E')) => {
// app.set_screen(Editing);
// }
// (_, KeyCode::Char('d')) | (_, KeyCode::Char('D')) => {
// app.set_screen(Deleting);
// }
_ => {}
} }
} }
_ => {} _ => {}
@@ -168,7 +288,7 @@ impl RatatuiMonitoringMode {
} }
} }
pub fn make_state(selected: usize) -> TableState { pub fn make_hosts_list_state(selected: usize) -> TableState {
let mut state = TableState::default(); let mut state = TableState::default();
state.select(Some(selected)); state.select(Some(selected));
state state
+123 -84
View File
@@ -1,9 +1,14 @@
use crate::tui::ratatui_screens::RatatuiScreens;
use crate::manager::Manager; use crate::manager::Manager;
use crate::ping_result::PingResult; use crate::ping_result::PingResult;
use crate::target_state::TargetState; use crate::target_state::TargetState;
use crate::tui::mode_adding::RatatuiAddingMode;
use crate::tui::mode_deleting::RatatuiDeletingMode;
use crate::tui::mode_editing::RatatuiEditingMode;
use crate::tui::mode_monitoring::RatatuiMonitoringMode;
use crate::tui::ratatui_screens::RatatuiScreens;
use chrono::{DateTime, Local}; use chrono::{DateTime, Local};
use color_eyre::Result; use color_eyre::Result;
use log::debug;
use ratatui::prelude::*; use ratatui::prelude::*;
use ratatui::{DefaultTerminal, Frame}; use ratatui::{DefaultTerminal, Frame};
use std::collections::BTreeMap; use std::collections::BTreeMap;
@@ -13,14 +18,10 @@ use std::path::PathBuf;
use std::str::FromStr; use std::str::FromStr;
use std::sync::mpsc; use std::sync::mpsc;
use std::sync::mpsc::Receiver; use std::sync::mpsc::Receiver;
use std::thread;
use std::time::{Duration, SystemTime}; use std::time::{Duration, SystemTime};
use crate::tui::mode_adding::RatatuiAddingMode;
use crate::tui::mode_deleting::RatatuiDeletingMode;
use crate::tui::mode_editing::RatatuiEditingMode;
use crate::tui::mode_exiting::RatatuiExitingMode;
use crate::tui::mode_monitoring::RatatuiMonitoringMode;
const LOG_ENTRIES_DEFAULT_COUNT: u32 = 10;
#[derive(Default)] #[derive(Default)]
pub struct RatatuiApp { pub struct RatatuiApp {
@@ -29,11 +30,38 @@ pub struct RatatuiApp {
current_screen: RatatuiScreens, current_screen: RatatuiScreens,
log_entries: Vec<String>, log_entries: Vec<String>,
filename: Option<String>, filename: Option<String>,
pub selected_host: usize pub selected_host: usize,
pub showing_add_popup: bool,
pub trying_to_exit: bool,
pub trying_to_delete: bool,
pub add_host_cursor_position: usize,
pub add_host_name: String,
pub num_log_entries: u32
} }
/// Private Methods /// Private Methods
impl RatatuiApp { impl RatatuiApp {
pub fn remove_selected_host(self: &mut RatatuiApp) {
for (index, (name, data)) in self.state.clone().iter().enumerate() {
if index == self.selected_host {
self.state.remove(name).unwrap();
}
}
}
pub fn set_filename(mut self, new_filename: String) {
self.filename = Some(new_filename)
}
pub fn get_filename(self) -> String {
self.filename.unwrap_or(String::new())
}
pub fn add_new_host(mut self, new_target: TargetState) -> Result<()> {
self.state.insert(new_target.name.clone(), new_target);
Ok(())
}
pub fn run(mut self, mut terminal: DefaultTerminal) -> Result<()> { pub fn run(mut self, mut terminal: DefaultTerminal) -> Result<()> {
self.running = true; self.running = true;
// start the 'manager' thread that spawns its ping threads as needed // start the 'manager' thread that spawns its ping threads as needed
@@ -43,31 +71,8 @@ impl RatatuiApp {
// check for any waiting ping results... // check for any waiting ping results...
self.consume_waiting_results(&receiver); self.consume_waiting_results(&receiver);
match self.current_screen { terminal.draw(|frame| RatatuiMonitoringMode::render(frame, &mut self))?;
RatatuiScreens::Deleting => { RatatuiMonitoringMode::handle_crossterm_events(&mut self)?;
terminal
.draw(|frame| RatatuiDeletingMode::render(frame, &mut self))?;
RatatuiDeletingMode::handle_crossterm_events(&mut self)?
}
RatatuiScreens::Monitoring => {
terminal
.draw(|frame| RatatuiMonitoringMode::render(frame, &mut self))?;
RatatuiMonitoringMode::handle_crossterm_events(&mut self)?;
}
RatatuiScreens::Exiting => {
terminal.draw(|frame| RatatuiExitingMode::render(frame))?;
RatatuiExitingMode::handle_crossterm_events(&mut self)?;
}
RatatuiScreens::Editing => {
terminal.draw(|frame| RatatuiEditingMode::render(frame, &mut self.state))?;
RatatuiEditingMode::handle_crossterm_events(&mut self)?;
}
RatatuiScreens::Adding => {
terminal.draw(|frame| RatatuiAddingMode::render(frame, &mut self))?;
RatatuiAddingMode::handle_crossterm_events(&mut self)?
}
}
} }
Ok(()) Ok(())
} }
@@ -92,6 +97,7 @@ impl RatatuiApp {
alive: new_message.success, alive: new_message.success,
last_rtt: new_message.rtt, last_rtt: new_message.rtt,
last_alive_change, last_alive_change,
..TargetState::default()
}; };
local_state.insert(name.clone(), new_state.clone()); local_state.insert(name.clone(), new_state.clone());
let success_message = if new_state.alive { let success_message = if new_state.alive {
@@ -104,8 +110,8 @@ impl RatatuiApp {
if did_change { if did_change {
self.log_entries.push(format!( self.log_entries.push(format!(
"{:?} {} for {}", "{} {} for {}",
current_time.format("%Y-%m-%d %H:%M:%S: ").to_string(), current_time.format("%Y-%m-%d %H:%M:%S:").to_string(),
success_message, success_message,
new_state.name.clone() new_state.name.clone()
)); ));
@@ -125,13 +131,7 @@ impl RatatuiApp {
} }
fn render(&mut self, frame: &mut Frame) { fn render(&mut self, frame: &mut Frame) {
match self.current_screen { RatatuiMonitoringMode::render(frame, self)
RatatuiScreens::Monitoring => RatatuiMonitoringMode::render(frame, self),
RatatuiScreens::Exiting => RatatuiExitingMode::render(frame),
RatatuiScreens::Editing => RatatuiEditingMode::render(frame, &mut self.state),
RatatuiScreens::Deleting => RatatuiDeletingMode::render(frame,self),
RatatuiScreens::Adding => RatatuiAddingMode::render(frame, self)
}
} }
fn quit(&mut self) { fn quit(&mut self) {
@@ -141,56 +141,95 @@ impl RatatuiApp {
fn get_default_targets() -> BTreeMap<String, TargetState> { fn get_default_targets() -> BTreeMap<String, TargetState> {
let mut working = BTreeMap::new(); let mut working = BTreeMap::new();
working.insert("1111 DNS".to_string(), TargetState { working.insert(
name: "1111 DNS".to_string(), "1111 DNS".to_string(),
target: Ipv4Addr::new(1, 1, 1, 1), TargetState {
..TargetState::default() name: "1111 DNS".to_string(),
}); target: Ipv4Addr::new(1, 1, 1, 1),
working.insert("Google DNS".to_string(), TargetState { ..TargetState::default()
name: "Google DNS".to_string(), },
target: Ipv4Addr::new(8, 8, 8, 8), );
..TargetState::default() working.insert(
}); "Google DNS".to_string(),
working.insert("Test Site 1".to_string(), TargetState { TargetState {
name: "Test Site 1".to_string(), name: "Google DNS".to_string(),
target: Ipv4Addr::new(216, 234, 202, 122), target: Ipv4Addr::new(8, 8, 8, 8),
..TargetState::default() ..TargetState::default()
}); },
);
working.insert(
"Test Site 1".to_string(),
TargetState {
name: "Test Site 1".to_string(),
target: Ipv4Addr::new(216, 234, 202, 122),
..TargetState::default()
},
);
working working
} }
} }
/// Public Methods /// Public Methods
impl RatatuiApp { impl RatatuiApp {
pub fn new(option: Option<PathBuf>) -> Self { fn load_hosts_from_file(file_to_load_from: PathBuf) -> BTreeMap<String, TargetState> {
let mut working = BTreeMap::new();
let the_file = File::open(file_to_load_from.clone());
let mut rdr = csv::Reader::from_reader(the_file.unwrap());
for result in rdr.records() {
let record = result.unwrap();
working.insert(
record[1].to_string(),
TargetState {
name: record[1].to_string(),
target: Ipv4Addr::from_str(&record[0]).unwrap(),
..TargetState::default()
},
);
}
working
}
pub fn new(option: Option<PathBuf>) -> Self {
let mut working = Self::default(); let mut working = Self::default();
let targets = if let Some(file) = option { let targets = if let Some(file) = option {
let mut working_btree = BTreeMap::new(); if file.exists() {
if !&file.exists() {
RatatuiApp::get_default_targets();
working.filename = Some("hosts.txt".to_string());
} else {
let real_file = File::open(file.clone());
working.filename = Some(file.as_os_str().to_string_lossy().parse().unwrap()); working.filename = Some(file.as_os_str().to_string_lossy().parse().unwrap());
let mut rdr = csv::Reader::from_reader(real_file.unwrap()); } else {
for result in rdr.records() { // working.log_event("Passed file doesnt exist looking for hosts.txt".to_string());
let record = result.unwrap(); if PathBuf::from_str("hosts.txt").unwrap().exists() {
working_btree.insert(record[1].to_string(), TargetState { // working.log_event("Found hosts.txt. using it as the default".to_string());
name: record[1].to_string(), working.filename = Some("hosts.txt".to_string());
target: Ipv4Addr::from_str(&record[0]).unwrap(), } else {
alive: false, // working.log_event("Didnt find hosts.txt".to_string());
last_alive_change: SystemTime::now(), working.filename = None;
last_rtt: 0,
});
} }
} }
working_btree
if let Some(file) = working.filename.clone() {
RatatuiApp::load_hosts_from_file(PathBuf::from_str(file.as_str()).unwrap())
} else {
RatatuiApp::get_default_targets()
}
} else { } else {
RatatuiApp::get_default_targets() // none was passed for our parameter.
working.filename = Some("hosts.txt".to_string());
if let Some(ref hosts_file) = working.filename {
if PathBuf::from_str(hosts_file).unwrap().exists() {
RatatuiApp::load_hosts_from_file(
PathBuf::from_str(&*hosts_file.clone()).unwrap(),
)
} else {
working.filename = None;
RatatuiApp::get_default_targets()
}
} else {
RatatuiApp::get_default_targets()
}
}; };
working.state = targets; working.state = targets.clone();
working.num_log_entries = LOG_ENTRIES_DEFAULT_COUNT;
working working
} }
@@ -210,11 +249,11 @@ impl RatatuiApp {
} }
pub fn get_log_entries(&self, how_many: u32) -> Vec<String> { pub fn get_log_entries(&self, how_many: u32) -> Vec<String> {
let mut return_value = vec![]; self.log_entries
for current in self.log_entries.clone().into_iter().rev() { .iter()
return_value.push(current); .rev()
} .take(how_many as usize)
return_value .cloned()
.collect()
} }
} }
-1
View File
@@ -3,7 +3,6 @@
pub enum RatatuiScreens { pub enum RatatuiScreens {
#[default] #[default]
Monitoring, Monitoring,
Exiting,
Editing, Editing,
Deleting, Deleting,
Adding Adding