agent-file-tools 0.18.1

use std::sync::atomic::Ordering;
use std::thread;
use std::time::Duration;

use crossbeam_channel::tick;

use super::registry::BgTaskRegistry;
const WATCHDOG_INTERVAL: Duration = Duration::from_millis(500);

pub(crate) fn start(registry: BgTaskRegistry) {
    thread::spawn(move || {
        let ticker = tick(WATCHDOG_INTERVAL);
        while !registry.inner.shutdown.load(Ordering::SeqCst) {
            if ticker.recv().is_err() {
                break;
            }

            let tasks = registry.running_tasks();
            if tasks.is_empty() {
                continue;
            }

            for task in tasks {
                let _ = registry.poll_task(&task);
                if !task.is_running() {
                    continue;
                }

                let timeout_expired = task
                    .state
                    .lock()
                    .ok()
                    .and_then(|state| state.metadata.timeout_ms)
                    .map(|timeout_ms| task.started.elapsed() >= Duration::from_millis(timeout_ms))
                    .unwrap_or(false);
                if timeout_expired {
                    let _ = registry.kill_for_timeout(&task.task_id, &task.session_id);
                    continue;
                }

                registry.reap_child(&task);
            }
        }
    });
}