Compare commits
3 Commits
f5fa9877b3
...
8d0452d752
| Author | SHA1 | Date | |
|---|---|---|---|
| 8d0452d752 | |||
| 49092d9ad7 | |||
| 0261914cda |
2
Cargo.lock
generated
2
Cargo.lock
generated
@ -709,7 +709,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "pp"
|
name = "pp"
|
||||||
version = "0.2.0"
|
version = "0.2.1"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"ansi_term",
|
"ansi_term",
|
||||||
"chrono",
|
"chrono",
|
||||||
|
|||||||
@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "pp"
|
name = "pp"
|
||||||
version = "0.2.0"
|
version = "0.2.1"
|
||||||
edition = "2024"
|
edition = "2024"
|
||||||
|
|
||||||
[dependencies]
|
[dependencies]
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
@ -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>,
|
||||||
}
|
}
|
||||||
246
src/bin/pp.rs
246
src/bin/pp.rs
@ -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;
|
||||||
const SECONDS_IN_MINUTE: u32 = 60;
|
fn main() -> color_eyre::Result<()> {
|
||||||
const SECONDS_IN_HOUR: u32 = SECONDS_IN_MINUTE * 60;
|
// find out what file we are using to get our hosts
|
||||||
const SECONDS_IN_DAY: u32 = SECONDS_IN_HOUR * 24;
|
|
||||||
|
|
||||||
pub fn duration_to_string(to_convert: Duration) -> String {
|
|
||||||
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();
|
let settings = AppSettings::parse();
|
||||||
|
|
||||||
print!("Prep to load targets...");
|
color_eyre::install()?;
|
||||||
let file_to_check = match settings.ping_host_file {
|
let terminal = ratatui::init();
|
||||||
None => {
|
let result = RatatuiApp::new(settings.ping_host_file).run(terminal);
|
||||||
PathBuf::from("./hosts.txt")
|
ratatui::restore();
|
||||||
}
|
result
|
||||||
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;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
@ -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
|
|
||||||
}
|
|
||||||
|
|
||||||
@ -29,7 +29,7 @@ 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");
|
|
||||||
}
|
}
|
||||||
KeyCode::Esc | KeyCode::Char('n') | KeyCode::Char('N') => {
|
KeyCode::Esc | KeyCode::Char('n') | KeyCode::Char('N') => {
|
||||||
app.set_screen(Monitoring)
|
app.set_screen(Monitoring)
|
||||||
|
|||||||
@ -6,7 +6,6 @@ use ratatui::prelude::*;
|
|||||||
use ratatui::widgets::{Block, Borders, Cell, List, ListItem, ListState, Paragraph, Row, Table, TableState};
|
use ratatui::widgets::{Block, Borders, Cell, List, ListItem, ListState, Paragraph, Row, Table, TableState};
|
||||||
use ratatui::Frame;
|
use ratatui::Frame;
|
||||||
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_app::RatatuiApp;
|
||||||
use crate::tui::ratatui_screens::RatatuiScreens::{Deleting, Editing, Exiting};
|
use crate::tui::ratatui_screens::RatatuiScreens::{Deleting, Editing, Exiting};
|
||||||
|
|
||||||
@ -81,7 +80,7 @@ impl RatatuiMonitoringMode {
|
|||||||
let table = Table::new(rows, vec![Constraint::Min(30), Constraint::Min(6), Constraint::Min(5), Constraint::Min(30)])
|
let table = Table::new(rows, vec![Constraint::Min(30), Constraint::Min(6), Constraint::Min(5), Constraint::Min(30)])
|
||||||
.header(header)
|
.header(header)
|
||||||
.block(Block::default()
|
.block(Block::default()
|
||||||
.title("Hosts")
|
.title(Line::from(format!("PP v{}", env!("CARGO_PKG_VERSION"))))
|
||||||
.borders(Borders::ALL))
|
.borders(Borders::ALL))
|
||||||
.widths(&[
|
.widths(&[
|
||||||
Constraint::Min(30),
|
Constraint::Min(30),
|
||||||
@ -100,8 +99,8 @@ impl RatatuiMonitoringMode {
|
|||||||
// 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, layouts[0], &mut make_state(state.selected_host));
|
||||||
|
|
||||||
let footer_text = "Press <ESC> or q to exit";
|
// let footer_text = "Press <ESC> or q to exit";
|
||||||
// let footer_text = "Press <ESC> or q to exit | Press d to delete host | Press a to add host";
|
let footer_text = format!("Press <ESC> or q to exit | Press d to delete host | Press a to add host - ({:?})", state.filename);
|
||||||
|
|
||||||
let mut list_items = vec![];
|
let mut list_items = vec![];
|
||||||
for entry in state.get_log_entries(10) {
|
for entry in state.get_log_entries(10) {
|
||||||
@ -152,12 +151,12 @@ impl RatatuiMonitoringMode {
|
|||||||
(_, KeyCode::Esc) | (_, KeyCode::Char('q')) | (_, KeyCode::Char('Q')) => {
|
(_, KeyCode::Esc) | (_, KeyCode::Char('q')) | (_, KeyCode::Char('Q')) => {
|
||||||
app.set_screen(Exiting);
|
app.set_screen(Exiting);
|
||||||
}
|
}
|
||||||
// (_, KeyCode::Char('e')) | (_, KeyCode::Char('E')) => {
|
(_, KeyCode::Char('e')) | (_, KeyCode::Char('E')) => {
|
||||||
// app.set_screen(Editing);
|
app.set_screen(Editing);
|
||||||
// }
|
}
|
||||||
// (_, KeyCode::Char('d')) | (_, KeyCode::Char('D')) => {
|
(_, KeyCode::Char('d')) | (_, KeyCode::Char('D')) => {
|
||||||
// app.set_screen(Deleting);
|
app.set_screen(Deleting);
|
||||||
// }
|
}
|
||||||
_ => {}
|
_ => {}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@ -13,7 +13,9 @@ 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 log::debug;
|
||||||
use crate::tui::mode_adding::RatatuiAddingMode;
|
use crate::tui::mode_adding::RatatuiAddingMode;
|
||||||
use crate::tui::mode_deleting::RatatuiDeletingMode;
|
use crate::tui::mode_deleting::RatatuiDeletingMode;
|
||||||
use crate::tui::mode_editing::RatatuiEditingMode;
|
use crate::tui::mode_editing::RatatuiEditingMode;
|
||||||
@ -28,7 +30,7 @@ pub struct RatatuiApp {
|
|||||||
pub state: BTreeMap<String, TargetState>,
|
pub state: BTreeMap<String, TargetState>,
|
||||||
current_screen: RatatuiScreens,
|
current_screen: RatatuiScreens,
|
||||||
log_entries: Vec<String>,
|
log_entries: Vec<String>,
|
||||||
filename: Option<String>,
|
pub(crate) filename: Option<String>,
|
||||||
pub selected_host: usize
|
pub selected_host: usize
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -163,34 +165,62 @@ impl RatatuiApp {
|
|||||||
|
|
||||||
/// Public Methods
|
/// Public Methods
|
||||||
impl RatatuiApp {
|
impl RatatuiApp {
|
||||||
pub fn new(option: Option<PathBuf>) -> Self {
|
|
||||||
|
|
||||||
let mut working = Self::default();
|
fn load_hosts_from_file(file_to_load_from: PathBuf) -> BTreeMap<String, TargetState> {
|
||||||
let targets = if let Some(file) = option {
|
let mut working = BTreeMap::new();
|
||||||
let mut working_btree = BTreeMap::new();
|
|
||||||
if !&file.exists() {
|
let the_file = File::open(file_to_load_from.clone());
|
||||||
RatatuiApp::get_default_targets();
|
let mut rdr = csv::Reader::from_reader(the_file.unwrap());
|
||||||
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());
|
|
||||||
let mut rdr = csv::Reader::from_reader(real_file.unwrap());
|
|
||||||
for result in rdr.records() {
|
for result in rdr.records() {
|
||||||
let record = result.unwrap();
|
let record = result.unwrap();
|
||||||
working_btree.insert(record[1].to_string(), TargetState {
|
working.insert(record[1].to_string(), TargetState {
|
||||||
name: record[1].to_string(),
|
name: record[1].to_string(),
|
||||||
target: Ipv4Addr::from_str(&record[0]).unwrap(),
|
target: Ipv4Addr::from_str(&record[0]).unwrap(),
|
||||||
alive: false,
|
alive: false,
|
||||||
last_alive_change: SystemTime::now(),
|
last_alive_change: SystemTime::now(),
|
||||||
last_rtt: 0,
|
last_rtt: 0
|
||||||
});
|
});
|
||||||
}
|
}
|
||||||
|
|
||||||
|
working
|
||||||
}
|
}
|
||||||
working_btree
|
|
||||||
|
pub fn new(option: Option<PathBuf>) -> Self {
|
||||||
|
let mut working = Self::default();
|
||||||
|
let targets = if let Some(file) = option {
|
||||||
|
if file.exists() {
|
||||||
|
working.filename = Some(file.as_os_str().to_string_lossy().parse().unwrap());
|
||||||
|
} else {
|
||||||
|
// working.log_event("Passed file doesnt exist looking for hosts.txt".to_string());
|
||||||
|
if !&PathBuf::from_str("hosts.txt").unwrap().exists() {
|
||||||
|
// working.log_event("Didnt find hosts.txt".to_string());
|
||||||
|
working.filename = None;
|
||||||
|
} else {
|
||||||
|
// working.log_event("Found hosts.txt. using it as the default".to_string());
|
||||||
|
working.filename = Some("hosts.txt".to_string());
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if let Some(file) = working.filename.clone() {
|
||||||
|
RatatuiApp::load_hosts_from_file(PathBuf::from_str(&*working.filename.clone().unwrap()).unwrap())
|
||||||
} else {
|
} else {
|
||||||
RatatuiApp::get_default_targets()
|
RatatuiApp::get_default_targets()
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
// 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
|
working
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user