geam-time 0.2.1

use super::{Component, GleamTimeHostProfile, TimeSource};
use crate::{HostProviderModule, HostRegistrationError};
use geam_core::provider::{Call, HostResult};
use num_bigint::BigInt;

#[geam_macros::module(
    path = "gleam/time/calendar",
    crate_path = geam_core,
    profile = crate::GleamTimeHostProfile,
    component = crate::Component<Profile::Source>,
)]
mod provider {
    use super::{BigInt, Call, HostResult, TimeSource};

    #[geam_macros::function(profile = Profile)]
    fn local_time_offset_seconds(
        #[geam_macros::call] call: &mut Call<Profile::Source>,
    ) -> HostResult<BigInt> {
        let seconds = call.state_mut().local_offset_seconds()?;
        Ok(BigInt::from(seconds))
    }
}

pub(super) fn host_provider<Profile>() -> Result<HostProviderModule<Profile>, HostRegistrationError>
where
    Profile: GleamTimeHostProfile,
{
    provider::__geam_module::<Profile>()
}

#[cfg(test)]
mod tests {
    use super::host_provider;
    use crate::test_support::{CALENDAR_SOURCE, ScriptedSource, execution};
    use crate::{ExecutionError, HostError, HostFailure, InvariantError, ValueType};
    use crate::{GleamTimeProfile, GleamTimeRunState};
    use geam_stdlib::GleamStdlibRunState;

    #[test]
    fn registers_the_exact_calendar_provider() {
        let provider = host_provider::<GleamTimeProfile>()
            .expect("official Time calendar provider should register");
        let functions = provider.functions().collect::<Vec<_>>();

        assert_eq!(provider.package(), "gleam_time");
        assert_eq!(provider.module(), "gleam/time/calendar");
        assert_eq!(provider.external_types().count(), 0);
        assert_eq!(functions.len(), 1);
        assert_eq!(functions[0].name(), "local_time_offset_seconds");
        assert!(functions[0].scheme().parameters().is_empty());
        assert!(functions[0].type_().argument_types().is_empty());
        assert_eq!(functions[0].type_().return_(), &ValueType::Int);
    }

    #[test]
    fn preserves_local_offset_source_failures() {
        let source = format!("{CALENDAR_SOURCE}\npub fn main() {{\n  current_offset()\n}}\n",);
        let execution = execution::<ScriptedSource>(&source, "gleam/time/calendar");
        let mut state = GleamTimeRunState::new(
            GleamStdlibRunState::from_seed([4; 32]),
            ScriptedSource {
                times: Default::default(),
                offsets: [Err(HostFailure::new("offset unavailable"))].into(),
            },
        );
        let error = execution
            .run_main(&mut state, &mut Vec::new())
            .expect_err("scripted offset failure should remain an execution error");
        let error = expect_offset_host_error(error);

        assert_eq!(error.package(), "gleam_time");
        assert_eq!(error.module(), "gleam/time/calendar");
        assert_eq!(error.function(), "local_time_offset_seconds");
        assert_eq!(error.failure().message(), "offset unavailable");
        assert_eq!(
            error
                .location()
                .path()
                .expect("synthetic offset failure should retain its source path")
                .as_str(),
            "src/gleam/time/calendar.gleam",
        );
        assert_eq!(error.location().line(), Some(7));
    }

    #[test]
    #[should_panic(expected = "scripted offset failure should remain a host error")]
    fn offset_failure_assertion_rejects_other_execution_errors() {
        let _ = expect_offset_host_error(ExecutionError::Invariant(
            InvariantError::ListIndexOutOfBounds {
                item_type: ValueType::Int,
                index: 1,
                length: 0,
            },
        ));
    }

    fn expect_offset_host_error(error: ExecutionError) -> Box<HostError> {
        let ExecutionError::Host(error) = error else {
            panic!("scripted offset failure should remain a host error");
        };
        error
    }
}