use chrono::Utc;
use super::parse_rfc3339;
#[derive(Debug, Clone, Copy, PartialEq, Eq)]
pub enum FreshnessKind {
Index,
RecalledMemory,
StateLikeClaim,
}
pub fn memory_age_days(updated_at: &str) -> Option<i64> {
let updated_at = parse_rfc3339(updated_at)?;
let now = Utc::now();
let clamped = if updated_at > now { now } else { updated_at };
Some((now - clamped).num_days())
}
pub fn memory_age_label(updated_at: &str) -> Option<String> {
let days = memory_age_days(updated_at)?;
Some(match days {
0 => "today".to_string(),
1 => "1 day old".to_string(),
_ => format!("{days} days old"),
})
}
pub fn memory_freshness_text(updated_at: &str, kind: FreshnessKind) -> Option<String> {
let days = memory_age_days(updated_at)?;
if days <= 1 {
return None;
}
let age_label = memory_age_label(updated_at)?;
let text = match kind {
FreshnessKind::Index => {
if days <= 7 {
format!(
"Historical memory index entry ({age_label}); verify against current tools/files before treating it as live project state."
)
} else {
format!(
"Older memory index entry ({age_label}); verify against current tools/files before treating it as current project truth."
)
}
}
FreshnessKind::RecalledMemory => {
if days <= 7 {
format!(
"Historical memory ({age_label}); verify against current task context before treating it as still current."
)
} else {
format!(
"Older historical memory ({age_label}); verify against current tools/files before relying on it."
)
}
}
FreshnessKind::StateLikeClaim => {
if days <= 7 {
format!(
"Historical state-like memory ({age_label}); verify against current code/config before asserting it as fact."
)
} else {
format!(
"Older state-like memory ({age_label}); verify against current code/config before asserting it as fact."
)
}
}
};
Some(text)
}
pub fn render_memory_freshness_note(updated_at: &str, kind: FreshnessKind) -> Option<String> {
memory_freshness_text(updated_at, kind).map(|text| format!("Note: {text}"))
}
#[cfg(test)]
mod tests {
use chrono::{Duration, Utc};
use super::*;
#[test]
fn same_day_timestamp_has_no_warning() {
let timestamp = Utc::now().to_rfc3339();
assert_eq!(memory_age_days(×tamp), Some(0));
assert_eq!(memory_age_label(×tamp).as_deref(), Some("today"));
assert!(memory_freshness_text(×tamp, FreshnessKind::Index).is_none());
}
#[test]
fn multi_day_timestamp_gets_warning_and_age_label() {
let timestamp = (Utc::now() - Duration::days(3)).to_rfc3339();
assert_eq!(memory_age_days(×tamp), Some(3));
assert_eq!(memory_age_label(×tamp).as_deref(), Some("3 days old"));
let text = memory_freshness_text(×tamp, FreshnessKind::RecalledMemory)
.expect("warning expected");
assert!(text.contains("3 days old"));
assert!(text.contains("Historical memory"));
}
#[test]
fn invalid_timestamp_degrades_safely() {
assert!(memory_age_days("not-a-timestamp").is_none());
assert!(memory_age_label("not-a-timestamp").is_none());
assert!(memory_freshness_text("not-a-timestamp", FreshnessKind::Index).is_none());
assert!(render_memory_freshness_note("not-a-timestamp", FreshnessKind::Index).is_none());
}
#[test]
fn future_timestamp_is_clamped_to_zero_age() {
let timestamp = (Utc::now() + Duration::days(5)).to_rfc3339();
assert_eq!(memory_age_days(×tamp), Some(0));
assert_eq!(memory_age_label(×tamp).as_deref(), Some("today"));
assert!(memory_freshness_text(×tamp, FreshnessKind::Index).is_none());
}
#[test]
fn state_like_claim_uses_stronger_wording_than_index() {
let timestamp = (Utc::now() - Duration::days(10)).to_rfc3339();
let index_text = memory_freshness_text(×tamp, FreshnessKind::Index)
.expect("index warning expected");
let state_text = memory_freshness_text(×tamp, FreshnessKind::StateLikeClaim)
.expect("state-like warning expected");
assert!(index_text.contains("current project truth"));
assert!(state_text.contains("current code/config"));
assert_ne!(index_text, state_text);
}
}