ort-openrouter-cli 0.3.4

//! ort: Open Router CLI
//! https://github.com/grahamking/ort
//!
//! MIT License
//! Copyright (c) 2025 Graham King

use core::sync::atomic::{AtomicBool, Ordering};
use core::{mem, ptr};

use crate::libc;

static CANCELLED: AtomicBool = AtomicBool::new(false);
static IS_INIT_DONE: AtomicBool = AtomicBool::new(false);

// A way to stop a running thread
// Loosely inspired by tokio's CancellationToken
#[derive(Clone, Copy)]
pub struct CancelToken(&'static AtomicBool);

impl CancelToken {
    pub fn init() -> Self {
        // If IS_INIT_DONE == false, atomically make it true
        if IS_INIT_DONE
            .compare_exchange(false, true, Ordering::SeqCst, Ordering::SeqCst)
            .is_ok()
        {
            // We changed the flag, so we are the first
            unsafe { install_sigint_handler() };
        }
        CancelToken(&CANCELLED)
    }

    pub fn cancel(&self) {
        self.0.store(true, Ordering::SeqCst);
    }

    pub fn is_cancelled(&self) -> bool {
        self.0.load(Ordering::Relaxed)
    }
}

extern "C" fn handle_sigint(_: i32) {
    CANCELLED.store(true, Ordering::SeqCst);
}

unsafe fn install_sigint_handler() {
    const SIGINT: i32 = 2;
    unsafe {
        let mut sa: libc::sigaction = mem::zeroed();
        sa.sa_flags = 0;
        sa.sa_sigaction = handle_sigint as *const () as usize; // treated as sa_handler when SA_SIGINFO not set
        libc::sigemptyset(&mut sa.sa_mask);
        libc::sigaction(SIGINT, &sa, ptr::null_mut());
    }
}