BetterFrame/kiosk/src/ui.rs

661 lines
26 KiB
Rust
Raw Normal View History

use std::cell::{Cell, RefCell};
use std::sync::mpsc;
use std::time::{Duration, Instant};
use url::Url;
use gtk4::prelude::*;
use gtk4::{self as gtk, Application, ApplicationWindow, Box as GtkBox, Grid, Label, Orientation, Picture};
use tracing::{info, warn};
use crate::bundle::KioskBundle;
use crate::cec;
use crate::hwmon;
use crate::pipeline;
use crate::server;
use crate::ws_client;
use crate::ServerMsg;
thread_local! {
/// camera_id → (pipeline, paintable, badge). Pipelines stay warm across
/// layout swaps for cameras still referenced or in preload_camera_ids.
/// badge is 'M' / 'S' / ' ' indicating which stream is active.
static WARM_CAMERAS: RefCell<std::collections::HashMap<u32, (gstreamer::Pipeline, gtk::gdk::Paintable, char)>>
= RefCell::new(std::collections::HashMap::new());
/// Most recently rendered bundle. Used for layout-switch + idle revert.
static CURRENT_BUNDLE: RefCell<Option<KioskBundle>> = const { RefCell::new(None) };
/// Server URL + kiosk key for re-rendering on layout-switch.
static CURRENT_AUTH: RefCell<Option<(String, String)>> = const { RefCell::new(None) };
/// Layout id currently on screen, if any.
static CURRENT_LAYOUT_ID: Cell<Option<u32>> = const { Cell::new(None) };
/// Timestamp of the last "activity" event (render, switch, wake).
static LAST_ACTIVITY: RefCell<Instant> = RefCell::new(Instant::now());
/// True after we've fired CEC standby due to sleep timeout.
static IS_ASLEEP: Cell<bool> = const { Cell::new(false) };
/// Has the idle-watchdog already been installed on the main loop?
static WATCHDOG_INSTALLED: Cell<bool> = const { Cell::new(false) };
}
const APP_ID: &str = "dev.betterframe.kiosk";
const BETTERFRAME_LOGO_SVG: &str = include_str!("../../server/src/web-static/betterframe-logo.svg");
const BETTERFRAME_MARK_SVG: &str = include_str!("../../server/src/web-static/betterframe-mark.svg");
pub fn build_app() -> Application {
let app = Application::builder().application_id(APP_ID).build();
app.connect_activate(activate);
app
}
fn activate(app: &Application) {
let window = ApplicationWindow::builder()
.application(app)
.title("BetterFrame")
.fullscreened(true)
.build();
let provider = gtk::CssProvider::new();
2026-05-11 09:50:45 +00:00
provider.load_from_string("window { background-color: #000000; }");
gtk::style_context_add_provider_for_display(
&WidgetExt::display(&window),
&provider,
gtk::STYLE_PROVIDER_PRIORITY_APPLICATION,
);
show_logo(&window);
window.present();
let (tx, rx) = mpsc::channel::<WorkerMsg>();
let server_url = std::env::var("BETTERFRAME_SERVER").ok()
.or_else(|| std::env::args().nth(1));
std::thread::spawn(move || {
let server = server::discover_server(server_url.as_deref());
info!("server: {server}");
let key = if server::is_paired() {
info!("already paired");
server::load_key()
} else {
let (code, expires) = server::initiate_pairing(&server);
info!("pairing code: {code} (expires {expires})");
let _ = tx.send(WorkerMsg::ShowPairingCode(code.clone()));
let (name, key) = server::poll_claim(&server, &code);
info!("paired as: {name}");
key
};
// Try fetching live bundle. If server unreachable, fall back to
// cached on-disk bundle and keep retrying every 30s in the background.
let initial = match server::fetch_bundle(&server, &key) {
Some(b) => {
info!("bundle: {} cameras, {} layouts", b.cameras.len(), b.layouts.len());
Some(b)
}
None => {
if let Some(cached) = server::load_cached_bundle() {
warn!("offline mode: rendering cached bundle");
Some(cached)
} else {
warn!("no bundle available (server unreachable, no cache)");
None
}
}
};
if let Some(bundle) = initial {
let _ = tx.send(WorkerMsg::RenderBundle(bundle, server.clone(), key.clone()));
}
// Spawn WS client in a separate thread for live updates
let server_ws = server.clone();
let key_ws = key.clone();
let (ws_tx, ws_rx) = mpsc::channel::<ServerMsg>();
let tx_for_reload = tx.clone();
let server_for_reload = server.clone();
let key_for_reload = key.clone();
std::thread::spawn(move || {
ws_client::run(&server_ws, &key_ws, ws_tx);
});
// Background retry thread: if we couldn't fetch a live bundle on boot,
// try again every 30s until we get one. Once fetched, send a render.
let retry_tx = tx.clone();
let retry_server = server.clone();
let retry_key = key.clone();
std::thread::spawn(move || {
// Only loop while we have no live bundle yet — best-effort heuristic:
// we attempt once, then sleep. If server unreachable each time we
// keep waiting; once a fetch succeeds we push a fresh render.
// After first success we exit; subsequent updates flow via WS.
loop {
std::thread::sleep(Duration::from_secs(30));
if let Some(b) = server::fetch_bundle(&retry_server, &retry_key) {
info!("offline-retry: fresh bundle fetched, rendering");
let _ = retry_tx.send(WorkerMsg::RenderBundle(
b,
retry_server.clone(),
retry_key.clone(),
));
return;
}
}
});
// Listen for WS messages and dispatch
std::thread::spawn(move || {
for msg in ws_rx {
match msg {
ServerMsg::ReloadBundle => {
info!("reloading bundle");
match server::fetch_bundle(&server_for_reload, &key_for_reload) {
Some(bundle) => {
let _ = tx_for_reload.send(WorkerMsg::RenderBundle(
bundle,
server_for_reload.clone(),
key_for_reload.clone(),
));
}
None => warn!("reload-bundle: fetch failed, keeping current render"),
}
}
ServerMsg::Standby => cec::standby(),
ServerMsg::Wake => {
let _ = tx_for_reload.send(WorkerMsg::Wake);
}
ServerMsg::Fan(pwm) => { hwmon::set_fan(pwm); }
ServerMsg::SwitchLayout(id) => {
let _ = tx_for_reload.send(WorkerMsg::SwitchLayout(id));
}
}
}
});
// Heartbeat loop — reports display geometry + hwmon
loop {
std::thread::sleep(std::time::Duration::from_secs(60));
let displays = query_displays();
let hw = hwmon::read();
server::heartbeat(&server, &key, &displays, &hw);
}
});
// Poll channel from UI thread via timeout
let window_clone = window.clone();
gtk::glib::timeout_add_local(std::time::Duration::from_millis(100), move || {
while let Ok(msg) = rx.try_recv() {
match msg {
WorkerMsg::ShowPairingCode(code) => show_pairing_code(&window_clone, &code),
WorkerMsg::RenderBundle(bundle, server, key) => {
render_bundle(&window_clone, bundle, &server, &key);
install_idle_watchdog(&window_clone);
}
WorkerMsg::SwitchLayout(id) => {
render_layout(&window_clone, id);
}
WorkerMsg::Wake => {
cec::wake();
IS_ASLEEP.with(|c| c.set(false));
mark_activity();
}
}
}
gtk::glib::ControlFlow::Continue
});
}
enum WorkerMsg {
ShowPairingCode(String),
RenderBundle(KioskBundle, String, String),
SwitchLayout(u32),
Wake,
}
/// Reset activity timer. If we were asleep, wake the display first.
fn mark_activity() {
LAST_ACTIVITY.with(|t| *t.borrow_mut() = Instant::now());
if IS_ASLEEP.with(|c| c.get()) {
info!("activity while asleep → waking display");
cec::wake();
IS_ASLEEP.with(|c| c.set(false));
}
}
/// Install the once-per-second watchdog that enforces idle/sleep timeouts.
/// Safe to call multiple times — installs at most once.
fn install_idle_watchdog(window: &ApplicationWindow) {
if WATCHDOG_INSTALLED.with(|c| c.get()) { return; }
WATCHDOG_INSTALLED.with(|c| c.set(true));
let window = window.clone();
gtk::glib::timeout_add_local(Duration::from_secs(1), move || {
let elapsed = LAST_ACTIVITY.with(|t| t.borrow().elapsed());
// Need the bundle to read display timeouts + default layout.
let (idle_to, sleep_to, default_id) = CURRENT_BUNDLE.with(|b| {
match b.borrow().as_ref() {
Some(bundle) => (
bundle.display.idle_timeout_seconds as u64,
bundle.display.sleep_timeout_seconds as u64,
bundle.display.default_layout_id,
),
None => (0, 0, None),
}
});
// Idle revert: if elapsed >= idle timeout AND current layout is not
// default AND current layout doesn't itself reset the idle timer.
if idle_to > 0 && elapsed >= Duration::from_secs(idle_to) {
let cur = CURRENT_LAYOUT_ID.with(|c| c.get());
let cur_resets_idle = CURRENT_BUNDLE.with(|b| {
let bundle = b.borrow();
let Some(bundle) = bundle.as_ref() else { return false };
let Some(cur_id) = cur else { return false };
bundle.layouts.iter().find(|l| l.id == cur_id)
.map(|l| l.resets_idle_timer)
.unwrap_or(false)
});
if let (Some(cur_id), Some(def_id)) = (cur, default_id) {
if cur_id != def_id && cur_resets_idle {
info!("idle timeout reached → reverting to default layout");
render_layout(&window, def_id);
}
}
}
// Sleep: fire CEC standby once, mark asleep.
if sleep_to > 0
&& elapsed >= Duration::from_secs(sleep_to)
&& !IS_ASLEEP.with(|c| c.get())
{
info!("sleep timeout reached → CEC standby");
cec::standby();
IS_ASLEEP.with(|c| c.set(true));
}
gtk::glib::ControlFlow::Continue
});
}
/// Query connected HDMI displays from sysfs. Returns (name, width, height).
/// Reads /sys/class/drm/*/status and /sys/class/drm/*/modes.
fn query_displays() -> Vec<(String, u32, u32)> {
let mut out = Vec::new();
let Ok(entries) = std::fs::read_dir("/sys/class/drm") else { return out };
for entry in entries.flatten() {
let name = entry.file_name().to_string_lossy().to_string();
// Skip non-HDMI connectors and the "card" parents
if !name.contains("-HDMI-") && !name.contains("-DP-") { continue; }
let path = entry.path();
let status = std::fs::read_to_string(path.join("status")).unwrap_or_default();
if status.trim() != "connected" { continue; }
let modes = std::fs::read_to_string(path.join("modes")).unwrap_or_default();
// First line = preferred mode
let mode = modes.lines().next().unwrap_or("");
let parts: Vec<&str> = mode.split('x').collect();
if parts.len() != 2 { continue; }
let w: u32 = parts[0].parse().unwrap_or(0);
let h: u32 = parts[1].trim().parse().unwrap_or(0);
if w == 0 || h == 0 { continue; }
// Strip "cardN-" prefix for cleaner name
let clean_name = name.split_once('-').map(|(_, rest)| rest.to_string()).unwrap_or(name);
out.push((clean_name, w, h));
}
out
}
fn show_pairing_code(window: &ApplicationWindow, code: &str) {
let vbox = GtkBox::new(Orientation::Vertical, 20);
vbox.set_valign(gtk::Align::Center);
vbox.set_halign(gtk::Align::Center);
vbox.set_vexpand(true);
let title = logo_picture(BETTERFRAME_LOGO_SVG, 360, 88, "pairing-logo");
let code_label = Label::new(Some(code));
add_css(&code_label, ".code { font-size: 72px; color: #fff; font-weight: 700; letter-spacing: 12px; font-family: monospace; }");
code_label.add_css_class("code");
let hint = Label::new(Some("Enter this code in BetterFrame admin to pair"));
add_css(&hint, ".hint { font-size: 14px; color: #666; }");
hint.add_css_class("hint");
vbox.append(&title);
vbox.append(&code_label);
vbox.append(&hint);
window.set_child(Some(&vbox));
}
fn render_bundle(window: &ApplicationWindow, bundle: KioskBundle, server_url: &str, kiosk_key: &str) {
// Cache the bundle + auth so layout-switch and idle-revert can re-render
// without needing a full reload.
CURRENT_BUNDLE.with(|b| *b.borrow_mut() = Some(bundle.clone()));
CURRENT_AUTH.with(|a| *a.borrow_mut() = Some((server_url.to_string(), kiosk_key.to_string())));
mark_activity();
let target_layout_id = bundle.display.default_layout_id
.or_else(|| bundle.layouts.iter().find(|l| l.is_default).map(|l| l.id));
let Some(target_layout_id) = target_layout_id else {
warn!("display has no default layout");
clear_warm_cameras();
CURRENT_LAYOUT_ID.with(|c| c.set(None));
show_logo(window);
return;
};
render_layout(window, target_layout_id);
}
/// Render a specific layout id from the cached bundle. If not found, fall back
/// to the display's default layout. If neither exists, show the logo.
fn render_layout(window: &ApplicationWindow, layout_id: u32) {
mark_activity();
// Snapshot what we need out of the cached bundle.
let snapshot: Option<(KioskBundle, String, String)> = CURRENT_BUNDLE.with(|b| {
let bundle = b.borrow();
let bundle = bundle.as_ref()?.clone();
let auth = CURRENT_AUTH.with(|a| a.borrow().clone());
let (server_url, kiosk_key) = auth?;
Some((bundle, server_url, kiosk_key))
});
let Some((bundle, server_url, kiosk_key)) = snapshot else {
warn!("render_layout: no cached bundle yet");
show_logo(window);
return;
};
let layout = bundle.layouts.iter().find(|l| l.id == layout_id)
.or_else(|| {
warn!("render_layout: layout {layout_id} not found, falling back to default");
bundle.display.default_layout_id
.and_then(|did| bundle.layouts.iter().find(|l| l.id == did))
.or_else(|| bundle.layouts.iter().find(|l| l.is_default))
});
let Some(layout) = layout else {
warn!("render_layout: no usable layout");
clear_warm_cameras();
CURRENT_LAYOUT_ID.with(|c| c.set(None));
show_logo(window);
return;
};
if layout.cells.is_empty() {
warn!("layout has no cells");
clear_warm_cameras();
CURRENT_LAYOUT_ID.with(|c| c.set(Some(layout.id)));
show_logo(window);
return;
}
CURRENT_LAYOUT_ID.with(|c| c.set(Some(layout.id)));
info!("rendering layout '{}' (id {}) with {}x{} grid, {} cells",
layout.name, layout.id, layout.grid_cols, layout.grid_rows, layout.cells.len());
// Compute which cameras are needed: cells with content_type=camera + preload_camera_ids
let mut needed: std::collections::HashSet<u32> = std::collections::HashSet::new();
for cell in &layout.cells {
if cell.content_type == "camera" {
if let Some(id) = cell.camera_id { needed.insert(id); }
}
}
for id in &layout.preload_camera_ids { needed.insert(*id); }
// Stop pipelines for cameras no longer needed
WARM_CAMERAS.with(|w| {
let mut warm = w.borrow_mut();
let stale: Vec<u32> = warm.keys().filter(|id| !needed.contains(id)).copied().collect();
for id in stale {
if let Some((pipe, _, _)) = warm.remove(&id) {
info!("stopping pipeline for camera {id} (no longer needed)");
pipeline::stop(&pipe);
}
}
});
let server_url = server_url.as_str();
let kiosk_key = kiosk_key.as_str();
let grid = Grid::new();
grid.set_row_homogeneous(true);
grid.set_column_homogeneous(true);
grid.set_vexpand(true);
grid.set_hexpand(true);
let cam_map: std::collections::HashMap<u32, &crate::bundle::BundleCamera> =
bundle.cameras.iter().map(|c| (c.id, c)).collect();
// Total grid area for the heuristic
let total_area = (layout.grid_cols.max(1) * layout.grid_rows.max(1)) as f32;
// Ensure preloaded cameras have pipelines even if not visible (use sub for warmth)
for cam_id in &layout.preload_camera_ids {
if let Some(cam) = cam_map.get(cam_id) {
ensure_warm(*cam_id, cam, None, 0.0);
}
}
for cell in &layout.cells {
let widget: gtk::Widget = match cell.content_type.as_str() {
"camera" => {
if let Some(cam_id) = cell.camera_id {
if let Some(cam) = cam_map.get(&cam_id) {
let area = (cell.col_span * cell.row_span) as f32 / total_area;
if let Some((paintable, badge)) = ensure_warm(cam_id, cam, cell.stream_selector.as_deref(), area) {
let picture = Picture::for_paintable(&paintable);
picture.set_content_fit(match cell.fit.as_str() {
"contain" => gtk::ContentFit::Contain,
"fill" => gtk::ContentFit::Fill,
_ => gtk::ContentFit::Cover,
});
picture.set_vexpand(true);
picture.set_hexpand(true);
// Wrap in Overlay so we can stack a stream-role badge on top
let overlay = gtk::Overlay::new();
overlay.set_child(Some(&picture));
overlay.set_vexpand(true);
overlay.set_hexpand(true);
if badge == 'M' || badge == 'S' {
let label = Label::new(Some(&badge.to_string()));
label.set_halign(gtk::Align::Start);
label.set_valign(gtk::Align::Start);
label.set_margin_start(4);
label.set_margin_top(4);
add_css(&label, "label { background: rgba(0,0,0,0.6); color: #fff; font-size: 11px; font-weight: 600; padding: 2px 6px; border-radius: 4px; min-width: 14px; }");
overlay.add_overlay(&label);
}
overlay.upcast()
} else {
placeholder(Some(&format!("{} (no stream)", cam.name)))
}
} else {
placeholder(Some("Unknown camera"))
}
} else {
none_cell()
}
}
"html" => {
let html = cell.html_content.as_deref().unwrap_or("");
if html.trim().is_empty() {
none_cell()
} else {
let webview = webkit6::WebView::new();
webkit6::prelude::WebViewExt::load_html(&webview, html, None);
webview.set_vexpand(true);
webview.set_hexpand(true);
webview.upcast()
}
}
"web" => {
let url = cell.web_url.as_deref().unwrap_or("").trim();
if url.is_empty() {
none_cell()
} else {
let webview = webkit6::WebView::new();
load_webview_url(&webview, url, server_url, kiosk_key);
webview.set_vexpand(true);
webview.set_hexpand(true);
webview.upcast()
}
}
"none" => none_cell(),
_ => placeholder(Some("Unknown content")),
};
grid.attach(
&widget,
cell.col as i32,
cell.row as i32,
cell.col_span as i32,
cell.row_span as i32,
);
}
window.set_child(Some(&grid));
}
fn clear_warm_cameras() {
WARM_CAMERAS.with(|w| {
for (_, (pipe, _, _)) in w.borrow().iter() { pipeline::stop(pipe); }
w.borrow_mut().clear();
});
}
fn load_webview_url(webview: &webkit6::WebView, url: &str, server_url: &str, kiosk_key: &str) {
if should_attach_kiosk_auth(url, server_url) {
let request = webkit6::URIRequest::new(url);
if let Some(headers) = request.http_headers() {
headers.append("Authorization", &format!("Bearer {kiosk_key}"));
}
webkit6::prelude::WebViewExt::load_request(webview, &request);
return;
}
webkit6::prelude::WebViewExt::load_uri(webview, url);
}
fn should_attach_kiosk_auth(url: &str, server_url: &str) -> bool {
let Ok(target) = Url::parse(url) else { return false };
let Ok(server) = Url::parse(server_url) else { return false };
if target.scheme() != server.scheme()
|| target.host_str() != server.host_str()
|| target.port_or_known_default() != server.port_or_known_default()
{
return false;
}
let path = target.path();
path.starts_with("/dash/") || path.starts_with("/in/kiosk/")
}
/// Returns (paintable, badge_char) for a camera, creating a warm pipeline if missing.
/// If cached pipeline's stream differs from what the cell needs (M↔S swap due
/// to layout change), tear down old and spin up new.
fn ensure_warm(
cam_id: u32,
cam: &crate::bundle::BundleCamera,
selector: Option<&str>,
area_fraction: f32,
) -> Option<(gtk::gdk::Paintable, char)> {
let (uri, desired_badge) = cam.pick_stream(selector, area_fraction)?;
// Check cached: if badge matches desired, reuse. Else swap.
let cached = WARM_CAMERAS.with(|w| {
w.borrow().get(&cam_id).map(|(p, paint, b)| (p.clone(), paint.clone(), *b))
});
if let Some((pipe, paintable, badge)) = cached {
if badge == desired_badge {
return Some((paintable, badge));
}
info!("camera {cam_id}: stream change {badge} → {desired_badge}, swapping");
pipeline::stop(&pipe);
WARM_CAMERAS.with(|w| { w.borrow_mut().remove(&cam_id); });
}
let (pipe, sink) = pipeline::create_camera_pipeline(&cam.name, &uri)?;
let paintable = sink.property::<gtk::gdk::Paintable>("paintable");
pipeline::play(&pipe);
WARM_CAMERAS.with(|w| {
w.borrow_mut().insert(cam_id, (pipe, paintable.clone(), desired_badge));
});
info!("warmed pipeline for camera {cam_id} (stream: {desired_badge})");
Some((paintable, desired_badge))
}
fn show_logo(window: &ApplicationWindow) {
let vbox = GtkBox::new(Orientation::Vertical, 0);
vbox.set_valign(gtk::Align::Center);
vbox.set_halign(gtk::Align::Center);
vbox.set_vexpand(true);
vbox.set_hexpand(true);
vbox.append(&logo_picture(BETTERFRAME_LOGO_SVG, 480, 118, "idle-logo"));
window.set_child(Some(&vbox));
}
fn none_cell() -> gtk::Widget {
placeholder(None)
}
fn placeholder(text: Option<&str>) -> gtk::Widget {
let vbox = GtkBox::new(Orientation::Vertical, 8);
add_css(
&vbox,
".bf-placeholder { background-color: #111; } .bf-placeholder-text { color: #666; font-size: 14px; }",
);
vbox.add_css_class("bf-placeholder");
vbox.set_valign(gtk::Align::Center);
vbox.set_halign(gtk::Align::Center);
vbox.set_vexpand(true);
vbox.set_hexpand(true);
vbox.append(&logo_picture(BETTERFRAME_MARK_SVG, 56, 56, "cell-logo"));
if let Some(text) = text {
let label = Label::new(Some(text));
label.add_css_class("bf-placeholder-text");
vbox.append(&label);
}
vbox.upcast()
}
fn logo_picture(svg: &'static str, width: i32, height: i32, css_class: &str) -> gtk::Widget {
let bytes = gtk::glib::Bytes::from_static(svg.as_bytes());
match gtk::gdk::Texture::from_bytes(&bytes) {
Ok(texture) => {
let picture = Picture::for_paintable(&texture);
picture.add_css_class(css_class);
picture.set_content_fit(gtk::ContentFit::Contain);
picture.set_can_shrink(true);
picture.set_size_request(width, height);
picture.set_valign(gtk::Align::Center);
picture.set_halign(gtk::Align::Center);
picture.upcast()
}
Err(err) => {
warn!("failed to load embedded logo: {err}");
let label = Label::new(Some("BetterFrame"));
label.set_size_request(width, height);
label.set_valign(gtk::Align::Center);
label.set_halign(gtk::Align::Center);
label.upcast()
}
}
}
fn add_css(widget: &impl IsA<gtk::Widget>, css: &str) {
let provider = gtk::CssProvider::new();
provider.load_from_string(css);
gtk::style_context_add_provider_for_display(
&widget.display(),
&provider,
gtk::STYLE_PROVIDER_PRIORITY_APPLICATION,
);
}