use crate::env_source::{EnvSource, ProcessEnvSource};
use chrono::{DateTime, Utc};
pub fn resolve_now_with_env<E: EnvSource + ?Sized>(env: &E) -> DateTime<Utc> {
env.var("SOURCE_DATE_EPOCH")
.and_then(|s| s.parse::<i64>().ok())
.and_then(|secs| DateTime::<Utc>::from_timestamp(secs, 0))
.unwrap_or_else(Utc::now)
}
pub fn resolve_now() -> DateTime<Utc> {
resolve_now_with_env(&ProcessEnvSource)
}
pub fn source_date_epoch_with_env<E: EnvSource + ?Sized>(env: &E) -> Option<DateTime<Utc>> {
env.var("SOURCE_DATE_EPOCH")
.and_then(|s| s.parse::<i64>().ok())
.and_then(|secs| DateTime::<Utc>::from_timestamp(secs, 0))
}
pub fn source_date_epoch() -> Option<DateTime<Utc>> {
source_date_epoch_with_env(&ProcessEnvSource)
}
#[cfg(test)]
mod tests {
use super::*;
use crate::env_source::MapEnvSource;
#[test]
fn resolve_now_honors_source_date_epoch() {
let env = MapEnvSource::new().with("SOURCE_DATE_EPOCH", "1715000000");
let now = resolve_now_with_env(&env);
assert_eq!(now.timestamp(), 1715000000);
}
#[test]
fn resolve_now_ignores_malformed_sde() {
let env = MapEnvSource::new().with("SOURCE_DATE_EPOCH", "not-a-number");
let now = resolve_now_with_env(&env);
assert!(now.timestamp() > 1_577_836_800);
}
#[test]
fn resolve_now_falls_back_when_unset() {
let env = MapEnvSource::new();
let now = resolve_now_with_env(&env);
assert!(now.timestamp() > 1_577_836_800);
}
#[test]
fn source_date_epoch_returns_some_when_set() {
let env = MapEnvSource::new().with("SOURCE_DATE_EPOCH", "1715000000");
let dt = source_date_epoch_with_env(&env);
assert_eq!(dt.map(|d| d.timestamp()), Some(1715000000));
}
#[test]
fn source_date_epoch_returns_none_when_unset() {
let env = MapEnvSource::new();
assert!(source_date_epoch_with_env(&env).is_none());
}
#[test]
fn source_date_epoch_returns_none_when_malformed() {
let env = MapEnvSource::new().with("SOURCE_DATE_EPOCH", "not-a-number");
assert!(source_date_epoch_with_env(&env).is_none());
}
}