Jack Wills
96 lines
2.7 KiB
Rust
96 lines
2.7 KiB
Rust
use app_data::AppData;
|
|
use app_error::AppError;
|
|
use bollard::Docker;
|
|
use docker_data::DockerData;
|
|
use parking_lot::Mutex;
|
|
use parse_args::CliArgs;
|
|
use std::sync::{atomic::AtomicBool, Arc};
|
|
use tracing::{info, Level};
|
|
|
|
mod app_data;
|
|
mod app_error;
|
|
mod docker_data;
|
|
mod input_handler;
|
|
mod parse_args;
|
|
mod ui;
|
|
|
|
use ui::{create_ui, GuiState};
|
|
|
|
fn setup_tracing() {
|
|
tracing_subscriber::fmt().with_max_level(Level::INFO).init();
|
|
}
|
|
|
|
#[tokio::main]
|
|
async fn main() {
|
|
setup_tracing();
|
|
let args = CliArgs::new();
|
|
let app_data = Arc::new(Mutex::new(AppData::default(args.clone())));
|
|
let gui_state = Arc::new(Mutex::new(GuiState::default()));
|
|
|
|
let docker_args = args.clone();
|
|
let docker_app_data = Arc::clone(&app_data);
|
|
let docker_gui_state = Arc::clone(&gui_state);
|
|
|
|
let (docker_sx, docker_rx) = tokio::sync::mpsc::channel(16);
|
|
// Create docker daemon handler, and only spawn up the docker data handler if ping returns non-error
|
|
let docker = Arc::new(Docker::connect_with_socket_defaults().unwrap());
|
|
match docker.ping().await {
|
|
Ok(_) => {
|
|
let docker = Arc::clone(&docker);
|
|
tokio::spawn(async move {
|
|
DockerData::init(
|
|
docker_args,
|
|
docker_app_data,
|
|
docker,
|
|
docker_gui_state,
|
|
docker_rx,
|
|
)
|
|
.await;
|
|
});
|
|
}
|
|
Err(_) => app_data.lock().set_error(AppError::DockerConnect),
|
|
}
|
|
|
|
let input_app_data = Arc::clone(&app_data);
|
|
|
|
let (input_sx, input_rx) = tokio::sync::mpsc::channel(16);
|
|
|
|
// let input_docker = Arc::clone(&docker);
|
|
let is_running = Arc::new(AtomicBool::new(true));
|
|
let input_is_running = Arc::clone(&is_running);
|
|
let input_gui_state = Arc::clone(&gui_state);
|
|
let input_docker_sender = docker_sx.clone();
|
|
|
|
// Spawn input handling into own tokio thread
|
|
tokio::spawn(async {
|
|
input_handler::InputHandler::init(
|
|
input_app_data,
|
|
input_rx,
|
|
input_docker_sender,
|
|
input_gui_state,
|
|
input_is_running,
|
|
)
|
|
.await;
|
|
});
|
|
|
|
// Debug mode for testing, mostly pointless, doesn't take terminal nor draw gui
|
|
if !args.gui {
|
|
loop {
|
|
info!("in debug mode");
|
|
tokio::time::sleep(std::time::Duration::from_millis(5000)).await;
|
|
}
|
|
} else {
|
|
let update_duration = std::time::Duration::from_millis(args.docker_interval as u64);
|
|
create_ui(
|
|
app_data,
|
|
input_sx,
|
|
is_running,
|
|
gui_state,
|
|
docker_sx,
|
|
update_duration,
|
|
)
|
|
.await
|
|
.unwrap();
|
|
}
|
|
}
|