mirror of
https://github.com/EFForg/rayhunter.git
synced 2026-06-16 01:19:44 -07:00
Carlos Guerra
32 lines
1.0 KiB
Rust
32 lines
1.0 KiB
Rust
use log::info;
|
|
use tokio::sync::mpsc::Receiver;
|
|
use tokio_util::sync::CancellationToken;
|
|
use tokio_util::task::TaskTracker;
|
|
|
|
use crate::config::{self, UiLevel};
|
|
use crate::display::{DisplayState, tplink_framebuffer, tplink_onebit};
|
|
|
|
use std::fs;
|
|
|
|
pub fn update_ui(
|
|
task_tracker: &TaskTracker,
|
|
config: &config::Config,
|
|
shutdown_token: CancellationToken,
|
|
ui_update_rx: Receiver<DisplayState>,
|
|
) {
|
|
let display_level = config.ui_level;
|
|
if display_level == UiLevel::Invisible {
|
|
info!("Invisible mode, not spawning UI.");
|
|
}
|
|
|
|
// Since this is a one-time check at startup, using sync is acceptable
|
|
// The alternative would be to make the entire initialization async
|
|
if fs::exists(tplink_onebit::OLED_PATH).unwrap_or_default() {
|
|
info!("detected one-bit display");
|
|
tplink_onebit::update_ui(task_tracker, config, shutdown_token, ui_update_rx)
|
|
} else {
|
|
info!("fallback to framebuffer");
|
|
tplink_framebuffer::update_ui(task_tracker, config, shutdown_token, ui_update_rx)
|
|
}
|
|
}
|