use std::path::Path;
use super::helpers::discover_machines;
use crate::core::{state, types};
pub(super) fn parse_rfc3339_to_epoch(s: &str) -> Option<u64> {
if s.len() < 19 {
return None;
}
let year: u64 = s.get(0..4)?.parse().ok()?;
let month: u64 = s.get(5..7)?.parse().ok()?;
let day: u64 = s.get(8..10)?.parse().ok()?;
let hour: u64 = s.get(11..13)?.parse().ok()?;
let min: u64 = s.get(14..16)?.parse().ok()?;
let sec: u64 = s.get(17..19)?.parse().ok()?;
let mut days: u64 = 0;
for y in 1970..year {
days += if (y.is_multiple_of(4) && !y.is_multiple_of(100)) || y.is_multiple_of(400) {
366
} else {
365
};
}
let table = [0, 31, 28, 31, 30, 31, 30, 31, 31, 30, 31, 30];
let leap = (year.is_multiple_of(4) && !year.is_multiple_of(100)) || year.is_multiple_of(400);
let mut md: u64 = 0;
for m in 1..month.min(13) {
md += table[m as usize];
if m == 2 && leap {
md += 1;
}
}
days += md + (day - 1);
Some(days * 86_400 + hour * 3600 + min * 60 + sec)
}
pub(super) fn now_epoch() -> u64 {
std::time::SystemTime::now()
.duration_since(std::time::UNIX_EPOCH)
.map(|d| d.as_secs())
.unwrap_or(0)
}
pub(super) fn filtered_machines(state_dir: &Path, machine: Option<&str>) -> Vec<String> {
let all = discover_machines(state_dir);
match machine {
Some(m) => all.into_iter().filter(|n| n == m).collect(),
None => all,
}
}
pub(super) fn classify_resources(lock: &types::StateLock) -> (usize, usize, usize, usize) {
let mut converged = 0usize;
let mut drifted = 0usize;
let mut failed = 0usize;
let mut unknown = 0usize;
for rl in lock.resources.values() {
match rl.status {
types::ResourceStatus::Converged => converged += 1,
types::ResourceStatus::Drifted => drifted += 1,
types::ResourceStatus::Failed => failed += 1,
types::ResourceStatus::Unknown => unknown += 1,
}
}
(converged, drifted, failed, unknown)
}
pub(crate) fn cmd_status_fleet_apply_cadence(
state_dir: &Path,
machine: Option<&str>,
json: bool,
) -> Result<(), String> {
let machines = filtered_machines(state_dir, machine);
let now = now_epoch();
let mut rows: Vec<(String, String, f64)> = Vec::new();
for m in &machines {
if let Ok(Some(lock)) = state::load_lock(state_dir, m) {
let age_hours = match parse_rfc3339_to_epoch(&lock.generated_at) {
Some(epoch) if now >= epoch => (now - epoch) as f64 / 3600.0,
_ => 0.0,
};
rows.push((m.clone(), lock.generated_at.clone(), age_hours));
}
}
if json {
let entries: Vec<serde_json::Value> = rows.iter().map(|(m, ts, age)| {
serde_json::json!({"machine": m, "last_apply": ts, "age_hours": (*age as u64)})
}).collect();
println!(
"{}",
serde_json::to_string_pretty(
&serde_json::json!({"fleet_apply_cadence":{"machines": entries}})
)
.unwrap_or_default()
);
} else {
println!("=== Fleet Apply Cadence ===");
if rows.is_empty() {
println!(" No machine state found.");
}
for (m, _ts, age) in &rows {
println!(" {}: last apply {}h ago", m, *age as u64);
}
}
Ok(())
}
pub(crate) fn cmd_status_machine_resource_error_classification(
state_dir: &Path,
machine: Option<&str>,
json: bool,
) -> Result<(), String> {
let machines = filtered_machines(state_dir, machine);
let mut rows: Vec<(String, usize, usize, usize, usize)> = Vec::new();
for m in &machines {
if let Ok(Some(lock)) = state::load_lock(state_dir, m) {
let (c, d, f, u) = classify_resources(&lock);
rows.push((m.clone(), c, d, f, u));
}
}
if json {
let entries: Vec<serde_json::Value> = rows.iter().map(|(m, c, d, f, u)| {
serde_json::json!({"machine": m, "converged": c, "drifted": d, "failed": f, "unknown": u})
}).collect();
println!(
"{}",
serde_json::to_string_pretty(
&serde_json::json!({"error_classification":{"machines": entries}})
)
.unwrap_or_default()
);
} else {
println!("=== Resource Error Classification ===");
if rows.is_empty() {
println!(" No machine state found.");
}
for (m, c, d, f, u) in &rows {
println!(" {m}: converged={c}, drifted={d}, failed={f}, unknown={u}");
}
}
Ok(())
}
pub(crate) fn cmd_status_fleet_resource_convergence_summary(
state_dir: &Path,
machine: Option<&str>,
json: bool,
) -> Result<(), String> {
let machines = filtered_machines(state_dir, machine);
let mut total = 0usize;
let mut converged = 0usize;
for m in &machines {
if let Ok(Some(lock)) = state::load_lock(state_dir, m) {
let (c, _d, _f, _u) = classify_resources(&lock);
total += lock.resources.len();
converged += c;
}
}
let pct = if total > 0 {
converged as f64 / total as f64 * 100.0
} else {
0.0
};
if json {
println!("{}", serde_json::to_string_pretty(
&serde_json::json!({"convergence_summary":{"total": total, "converged": converged, "percentage": (pct * 10.0).round() / 10.0}})
).unwrap_or_default());
} else {
println!("=== Fleet Convergence Summary ===");
println!(" Total: {total}, Converged: {converged}, Convergence: {pct:.1}%");
}
Ok(())
}