naia-socket-shared 0.25.0

use std::cell::Cell;
use std::time::Duration;

// Thread-local simulated clock — each test thread gets its own independent
// clock so parallel tests never interfere with each other.  The test executor
// is single-threaded per scenario, so no cross-thread clock reads are needed.
thread_local! {
    static SIMULATED_CLOCK: Cell<u64> = const { Cell::new(u64::MAX) }; // MAX = uninitialized sentinel
}

pub struct TestClock;

impl TestClock {
    /// Initialize the simulated clock with a starting time.
    pub fn init(initial_ms: u64) {
        SIMULATED_CLOCK.with(|c| c.set(initial_ms));
    }

    /// Advance the simulated clock by the specified number of milliseconds.
    /// Lazy-inits to 0 if no explicit `init()` has been called on this thread.
    pub fn advance(delta_ms: u64) {
        SIMULATED_CLOCK.with(|c| {
            let current = c.get();
            let base = if current == u64::MAX { 0 } else { current };
            c.set(base + delta_ms);
        });
    }

    /// Reset the simulated clock (for cleanup between tests).
    pub fn reset() {
        SIMULATED_CLOCK.with(|c| c.set(u64::MAX));
    }

    /// Get the current simulated time in milliseconds.
    /// Lazy-inits to 0 on first read if no explicit `init()` has been called
    /// on this thread — every test thread gets a sane starting clock without
    /// boilerplate. Tests that want a specific start time still call
    /// `TestClock::init(N)` explicitly.
    pub fn current_time_ms() -> u64 {
        SIMULATED_CLOCK.with(|c| {
            let millis = c.get();
            if millis == u64::MAX {
                c.set(0);
                0
            } else {
                millis
            }
        })
    }
}

/// Represents a specific moment in simulated test time
#[derive(Clone, Eq, PartialEq, Ord, PartialOrd, Debug)]
pub struct Instant {
    millis_since_start: u64,
}

impl Instant {
    /// Creates an Instant from the current simulated time
    pub fn now() -> Self {
        Self {
            millis_since_start: TestClock::current_time_ms(),
        }
    }

    /// Returns time elapsed since the Instant
    pub fn elapsed(&self, now: &Self) -> Duration {
        if now.millis_since_start >= self.millis_since_start {
            Duration::from_millis(now.millis_since_start - self.millis_since_start)
        } else {
            Duration::ZERO
        }
    }

    /// Returns time until the Instant occurs
    pub fn until(&self, now: &Self) -> Duration {
        if self.millis_since_start >= now.millis_since_start {
            Duration::from_millis(self.millis_since_start - now.millis_since_start)
        } else {
            Duration::ZERO
        }
    }

    pub fn is_after(&self, other: &Self) -> bool {
        self.millis_since_start > other.millis_since_start
    }

    /// Adds a given number of milliseconds to the Instant
    pub fn add_millis(&mut self, millis: u32) {
        self.millis_since_start = self.millis_since_start.saturating_add(millis as u64);
    }

    /// Subtracts a given number of milliseconds from the Instant
    pub fn subtract_millis(&mut self, millis: u32) {
        self.millis_since_start = self.millis_since_start.saturating_sub(millis as u64);
    }

    /// Returns inner Instant implementation (not available in test backend)
    pub fn inner(&self) -> std::time::Instant {
        panic!("inner() is not available in test backend. Use the Instant API directly.");
    }
}