use std::time::{Duration, Instant};

use serde_json::{json, Value};

use crate::coordinator_session::CoordinatorSession;
use crate::daemon::{Args, RuntimeTask};
use crate::node_identity::signed_node_request_json;

pub(crate) fn poll_task_cancellation(
    session: &mut CoordinatorSession,
    args: &Args,
    task: &RuntimeTask,
    node_private_key: &str,
) -> Result<bool, Box<dyn std::error::Error>> {
    let deadline = Instant::now() + Duration::from_millis(args.control_poll_ms);
    loop {
        let control = session.request(signed_node_request_json(
            args,
            node_private_key,
            "poll_task_control",
            json!({
                "type": "poll_task_control",
                "tenant": &args.tenant,
                "project": &args.project,
                "process": &task.process,
                "node": &args.node,
                "task": &task.task,
            }),
        )?)?;
        if control
            .get("cancel_requested")
            .and_then(Value::as_bool)
            .unwrap_or(false)
            || control
                .get("abort_requested")
                .and_then(Value::as_bool)
                .unwrap_or(false)
        {
            return Ok(true);
        }
        let now = Instant::now();
        if now >= deadline {
            return Ok(false);
        }
        std::thread::sleep((deadline - now).min(Duration::from_millis(100)));
    }
}