ebman/

state.rs

use std::{
    collections::{BTreeMap, BTreeSet},
    path::PathBuf,
};

use crate::util::{config_file, parse_bool, write_atomic};

#[derive(Debug, Default, Clone)]
pub struct PersistedState {
    pub profile: Option<String>,
    pub region: Option<String>,
    pub filter: Option<String>,
    pub sort: Option<String>, // e.g. "app:asc", "health:desc"
    pub grouped: Option<bool>,
    pub redact: Option<bool>,
    pub events_visible: Option<bool>,
    /// Event-timestamp display mode for the Events panel + Detail/Events
    /// tab. `None` means "never set" — the app falls back to the
    /// `EventTimeFormat` default (UTC). Stored as `"utc"|"local"|"age"`.
    pub event_time_format: Option<crate::app::EventTimeFormat>,
    pub selected_env: Option<String>,
    pub pinned: BTreeSet<String>,
    pub pinned_apps: BTreeSet<String>,
    /// Cost Explorer column toggle. Defaults to `None` (off) so the
    /// COST column doesn't render until the operator opts in via
    /// `:cost on`. Persists across sessions because Cost Explorer
    /// access is account-level and the operator's intent is durable.
    pub cost_enabled: Option<bool>,
    pub aliases: BTreeMap<String, String>,
    pub saved_views: BTreeMap<String, String>,
    /// Pre-deploy snapshots keyed by env name. Persists the
    /// `previous_version_label` + `taken_at` captured by every `:deploy`
    /// so a cross-session `:rollback` still has a target (without
    /// falling back to the event-history scan, which has a 100-event
    /// window cap). Stored as `"label|RFC3339-timestamp"` per env.
    pub deploy_snapshots: BTreeMap<String, String>,
    pub hidden_cols: BTreeSet<String>,
    /// User-defined extra metric charts for the Metrics tab. Keyed by the
    /// operator-chosen display label; value is `"namespace|name|stat"`.
    pub custom_metrics: BTreeMap<String, CustomMetricSpec>,
}

/// Parsed shape of a user-defined Metrics-tab chart. Stored line-oriented
/// in state.toml as `metric.LABEL = "namespace|name|stat[|dim=val;dim=val]"`.
/// The fourth pipe-separated field is optional; when absent the app
/// defaults to the env-scoped `EnvironmentName=<env>` dimension.
#[derive(Debug, Clone, PartialEq, Eq)]
pub struct CustomMetricSpec {
    pub namespace: String,
    pub name: String,
    pub stat: String,
    pub dimensions: Vec<(String, String)>,
}

impl CustomMetricSpec {
    /// Parse the `"namespace|name|stat[|k=v;k=v]"` form. Returns None on
    /// malformed input (wrong field count or empty mandatory parts) so the
    /// loader silently drops bad lines instead of aborting startup. A
    /// missing 4th field means "use the env-scoped default dimension at
    /// fetch time".
    pub fn parse(raw: &str) -> Option<Self> {
        let parts: Vec<&str> = raw.split('|').collect();
        if !matches!(parts.len(), 3 | 4) {
            return None;
        }
        let ns = parts[0].trim();
        let name = parts[1].trim();
        let stat = parts[2].trim();
        if ns.is_empty() || name.is_empty() || stat.is_empty() {
            return None;
        }
        let dimensions = if parts.len() == 4 {
            parts[3]
                .split(';')
                .filter_map(|kv| {
                    let (k, v) = kv.split_once('=')?;
                    let k = k.trim();
                    let v = v.trim();
                    if k.is_empty() || v.is_empty() {
                        return None;
                    }
                    Some((k.to_string(), v.to_string()))
                })
                .collect()
        } else {
            Vec::new()
        };
        Some(Self {
            namespace: ns.into(),
            name: name.into(),
            stat: stat.into(),
            dimensions,
        })
    }

    pub fn serialize(&self) -> String {
        if self.dimensions.is_empty() {
            return format!("{}|{}|{}", self.namespace, self.name, self.stat);
        }
        let dims = self
            .dimensions
            .iter()
            .map(|(k, v)| format!("{k}={v}"))
            .collect::<Vec<_>>()
            .join(";");
        format!("{}|{}|{}|{dims}", self.namespace, self.name, self.stat)
    }
}

pub fn load() -> PersistedState {
    let path = state_path();
    let Ok(text) = std::fs::read_to_string(&path) else {
        return PersistedState::default();
    };
    parse(&text)
}

/// True when no `state.toml` exists on disk yet. Used by the
/// first-run nudge to decide whether to surface the "press ? for
/// help" hint at boot. Distinct from "state.toml exists but is
/// empty" — the latter means the operator has run ebman before
/// (we wrote the file) but everything got cleared.
pub fn file_exists() -> bool {
    state_path().exists()
}

pub fn parse(text: &str) -> PersistedState {
    let mut state = PersistedState::default();
    for line in text.lines() {
        let line = line.trim();
        if line.is_empty() || line.starts_with('#') {
            continue;
        }
        let Some((key, raw_val)) = line.split_once('=') else {
            continue;
        };
        let value = raw_val.trim().trim_matches('"').to_string();
        if value.is_empty() {
            continue;
        }
        let k = key.trim();
        match k {
            "profile" => state.profile = Some(value),
            "region" => state.region = Some(value),
            "filter" => state.filter = Some(value),
            "sort" => state.sort = Some(value),
            "grouped" => state.grouped = parse_bool(&value),
            "redact" => state.redact = parse_bool(&value),
            "events_visible" => state.events_visible = parse_bool(&value),
            "event_time_format" => {
                state.event_time_format = crate::app::EventTimeFormat::parse(&value)
            }
            "selected_env" => state.selected_env = Some(value),
            _ if k.starts_with("filter.") => {
                // Legacy named-filter entries from ebman ≤ 0.11. Promote
                // them into the unified `saved_views` store using the
                // filter-only encoding so the operator's existing
                // shortcuts keep working — `]` / `[` cycle picks them
                // up alongside any full views. If the same name also
                // exists as `view.NAME`, the explicit `view.*` wins
                // (preserves operator intent on the off chance both
                // are present). First serialize-after-load drops the
                // `filter.*` lines and writes only `view.*` going
                // forward.
                let name = k.trim_start_matches("filter.").trim().to_string();
                if !name.is_empty() && !state.saved_views.contains_key(&name) {
                    state
                        .saved_views
                        .insert(name, crate::app::encode_filter_only_view(&value));
                }
            }
            "pinned" => {
                state.pinned = value
                    .split(',')
                    .map(|s| s.trim().to_string())
                    .filter(|s| !s.is_empty())
                    .collect();
            }
            "pinned_apps" => {
                state.pinned_apps = value
                    .split(',')
                    .map(|s| s.trim().to_string())
                    .filter(|s| !s.is_empty())
                    .collect();
            }
            "cost_enabled" => state.cost_enabled = parse_bool(&value),
            _ if k.starts_with("alias.") => {
                let name = k.trim_start_matches("alias.").trim().to_string();
                if !name.is_empty() {
                    state.aliases.insert(name, value);
                }
            }
            _ if k.starts_with("view.") => {
                let name = k.trim_start_matches("view.").trim().to_string();
                if !name.is_empty() {
                    state.saved_views.insert(name, value);
                }
            }
            _ if k.starts_with("deploy_snapshot.") => {
                let name = k.trim_start_matches("deploy_snapshot.").trim().to_string();
                if !name.is_empty() {
                    state.deploy_snapshots.insert(name, value);
                }
            }
            _ if k.starts_with("metric.") => {
                let label = k.trim_start_matches("metric.").trim().to_string();
                if label.is_empty() {
                    continue;
                }
                if let Some(spec) = CustomMetricSpec::parse(&value) {
                    state.custom_metrics.insert(label, spec);
                }
            }
            "hidden_cols" => {
                state.hidden_cols = value
                    .split(',')
                    .map(|s| s.trim().to_uppercase())
                    .filter(|s| !s.is_empty())
                    .collect();
            }
            _ => {}
        }
    }
    state
}

pub fn save(state: &PersistedState) {
    let path = state_path();
    // Parent-dir creation is handled by `write_atomic`. We just build
    // the body here and hand it off.
    let mut out = String::new();
    out.push_str("# ebman persisted state — managed by the app, edits will be overwritten\n");
    if let Some(p) = &state.profile {
        out.push_str(&format!("profile = \"{p}\"\n"));
    }
    if let Some(r) = &state.region {
        out.push_str(&format!("region = \"{r}\"\n"));
    }
    if let Some(f) = &state.filter {
        if !f.is_empty() {
            out.push_str(&format!("filter = \"{f}\"\n"));
        }
    }
    if let Some(s) = &state.sort {
        out.push_str(&format!("sort = \"{s}\"\n"));
    }
    if let Some(g) = state.grouped {
        out.push_str(&format!("grouped = {g}\n"));
    }
    if let Some(r) = state.redact {
        out.push_str(&format!("redact = {r}\n"));
    }
    if let Some(e) = state.events_visible {
        out.push_str(&format!("events_visible = {e}\n"));
    }
    if let Some(f) = state.event_time_format {
        out.push_str(&format!("event_time_format = \"{}\"\n", f.label()));
    }
    if let Some(s) = &state.selected_env {
        out.push_str(&format!("selected_env = \"{s}\"\n"));
    }
    if !state.pinned.is_empty() {
        let joined: Vec<&str> = state.pinned.iter().map(String::as_str).collect();
        out.push_str(&format!("pinned = \"{}\"\n", joined.join(",")));
    }
    if !state.pinned_apps.is_empty() {
        let joined: Vec<&str> = state.pinned_apps.iter().map(String::as_str).collect();
        out.push_str(&format!("pinned_apps = \"{}\"\n", joined.join(",")));
    }
    if let Some(b) = state.cost_enabled {
        out.push_str(&format!("cost_enabled = {b}\n"));
    }
    for (name, value) in &state.aliases {
        out.push_str(&format!("alias.{name} = \"{value}\"\n"));
    }
    for (name, value) in &state.saved_views {
        out.push_str(&format!("view.{name} = \"{value}\"\n"));
    }
    for (env, snap) in &state.deploy_snapshots {
        out.push_str(&format!("deploy_snapshot.{env} = \"{snap}\"\n"));
    }
    for (label, spec) in &state.custom_metrics {
        out.push_str(&format!("metric.{label} = \"{}\"\n", spec.serialize()));
    }
    if !state.hidden_cols.is_empty() {
        let joined: Vec<&str> = state.hidden_cols.iter().map(String::as_str).collect();
        out.push_str(&format!("hidden_cols = \"{}\"\n", joined.join(",")));
    }
    if let Err(e) = write_atomic(&path, &out) {
        tracing::warn!(error = %e, path = %path.display(), "failed to write state");
    }
}

fn state_path() -> PathBuf {
    config_file("state.toml")
}

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn parse_basic_fields() {
        let text = r#"
# comment
profile = "prod"
region = us-east-1
filter = "foo"
sort = "app:desc"
grouped = true
redact = off
events_visible = 1
selected_env = "my-env"
"#;
        let s = parse(text);
        assert_eq!(s.profile, Some("prod".into()));
        assert_eq!(s.region, Some("us-east-1".into()));
        assert_eq!(s.filter, Some("foo".into()));
        assert_eq!(s.sort, Some("app:desc".into()));
        assert_eq!(s.grouped, Some(true));
        assert_eq!(s.redact, Some(false));
        assert_eq!(s.events_visible, Some(true));
        assert_eq!(s.selected_env, Some("my-env".into()));
    }

    #[test]
    fn event_time_format_parses_each_value() {
        use crate::app::EventTimeFormat;
        assert_eq!(
            parse("event_time_format = \"utc\"\n").event_time_format,
            Some(EventTimeFormat::Utc)
        );
        assert_eq!(
            parse("event_time_format = \"local\"\n").event_time_format,
            Some(EventTimeFormat::Local)
        );
        assert_eq!(
            parse("event_time_format = \"age\"\n").event_time_format,
            Some(EventTimeFormat::Age)
        );
        // Absent key → None (app falls back to the EventTimeFormat default).
        assert_eq!(parse("region = \"x\"\n").event_time_format, None);
        // Garbage value → None, not a panic.
        assert_eq!(
            parse("event_time_format = \"bogus\"\n").event_time_format,
            None
        );
    }

    #[test]
    fn parse_legacy_filter_lines_promote_into_saved_views() {
        // Backward-compat: ebman ≤ 0.11 wrote `filter.NAME = "..."`
        // for saved filters; 0.12+ stores them as `view.NAME =
        // "filter=..."`. The parser promotes the legacy form into
        // `saved_views` using the filter-only encoding so existing
        // state.toml files keep working.
        let text = r#"
filter.dev = "production"
filter.prod = "live"
"#;
        let s = parse(text);
        assert_eq!(
            s.saved_views.get("dev").map(String::as_str),
            Some("filter=production")
        );
        assert_eq!(
            s.saved_views.get("prod").map(String::as_str),
            Some("filter=live")
        );
    }

    #[test]
    fn parse_explicit_view_wins_over_legacy_filter_for_same_name() {
        // If both `view.NAME = "..."` (new) and `filter.NAME = "..."`
        // (legacy) exist for the same NAME, the explicit `view.*`
        // form wins regardless of line order. This guards against
        // a mid-migration state.toml where the operator's full view
        // got overwritten by a legacy filter line.
        let text = r#"
filter.prod = "legacy-string"
view.prod = "filter=new-string;sort=app:asc"
"#;
        let s = parse(text);
        assert_eq!(
            s.saved_views.get("prod").map(String::as_str),
            Some("filter=new-string;sort=app:asc")
        );
        // And the reverse order — view.* first, filter.* second.
        let text = r#"
view.prod = "filter=new-string;sort=app:asc"
filter.prod = "legacy-string"
"#;
        let s = parse(text);
        assert_eq!(
            s.saved_views.get("prod").map(String::as_str),
            Some("filter=new-string;sort=app:asc")
        );
    }

    #[test]
    fn parse_collections() {
        let text = r#"
pinned = "prod-api,prod-worker"
pinned_apps = "billing,checkout"
alias.awseb-e-abc = "production"
alias.awseb-e-xyz = "staging"
view.dev = "filter=dev;sort=app:asc;grouped=false;scope=envs"
hidden_cols = "TREND,PLATFORM"
"#;
        let s = parse(text);
        assert!(s.pinned.contains("prod-api"));
        assert!(s.pinned.contains("prod-worker"));
        assert!(s.pinned_apps.contains("billing"));
        assert!(s.pinned_apps.contains("checkout"));
        assert_eq!(
            s.aliases.get("awseb-e-abc").map(String::as_str),
            Some("production")
        );
        assert!(s.saved_views.contains_key("dev"));
        assert!(s.hidden_cols.contains("TREND"));
        assert!(s.hidden_cols.contains("PLATFORM"));
    }

    #[test]
    fn parse_deploy_snapshots() {
        // `:deploy` captures these as a pre-rollback safety net;
        // persistence lets cross-session `:rollback` find them.
        let text = r#"
deploy_snapshot.prod-api = "build-823|2026-05-25T14:30:00+00:00"
deploy_snapshot.staging-api = "build-825|2026-05-25T15:00:00+00:00"
"#;
        let s = parse(text);
        assert_eq!(
            s.deploy_snapshots.get("prod-api").map(String::as_str),
            Some("build-823|2026-05-25T14:30:00+00:00")
        );
        assert_eq!(
            s.deploy_snapshots.get("staging-api").map(String::as_str),
            Some("build-825|2026-05-25T15:00:00+00:00")
        );
    }

    #[test]
    fn serialize_deploy_snapshots_round_trips() {
        // save() should emit deploy_snapshot.ENV lines that parse()
        // recognises. The intermediate file content isn't asserted
        // directly (avoids brittle string matching); instead we
        // round-trip via parse-after-save semantics.
        let mut state = PersistedState::default();
        state.deploy_snapshots.insert(
            "prod-api".into(),
            "build-823|2026-05-25T14:30:00+00:00".into(),
        );
        // Hand-construct the line save() would write so we can verify
        // it parses back without needing filesystem access.
        let line = format!(
            "deploy_snapshot.prod-api = \"{}\"\n",
            state.deploy_snapshots["prod-api"]
        );
        let reparsed = parse(&line);
        assert_eq!(
            reparsed.deploy_snapshots.get("prod-api"),
            state.deploy_snapshots.get("prod-api")
        );
    }

    #[test]
    fn parse_custom_metrics() {
        let text = r#"
metric.cpu = "AWS/EC2|CPUUtilization|Average"
metric.disk = "AWS/EC2|DiskReadOps|Sum"
"#;
        let s = parse(text);
        let cpu = s.custom_metrics.get("cpu").expect("cpu metric");
        assert_eq!(cpu.namespace, "AWS/EC2");
        assert_eq!(cpu.name, "CPUUtilization");
        assert_eq!(cpu.stat, "Average");
        assert!(s.custom_metrics.contains_key("disk"));
    }

    #[test]
    fn parse_custom_metric_drops_malformed_value() {
        // Wrong field count: silently dropped, no panic.
        let text = "metric.bad = \"only|two\"\n";
        let s = parse(text);
        assert!(s.custom_metrics.is_empty());
        // Empty field: also dropped.
        let text = "metric.bad = \"AWS/EC2||Average\"\n";
        let s = parse(text);
        assert!(s.custom_metrics.is_empty());
    }

    #[test]
    fn custom_metric_spec_round_trips() {
        let spec = CustomMetricSpec {
            namespace: "AWS/ApplicationELB".into(),
            name: "RequestCount".into(),
            stat: "Sum".into(),
            dimensions: Vec::new(),
        };
        assert_eq!(
            CustomMetricSpec::parse(&spec.serialize()).as_ref(),
            Some(&spec)
        );
    }

    #[test]
    fn custom_metric_spec_round_trips_with_dimensions() {
        let spec = CustomMetricSpec {
            namespace: "AWS/EC2".into(),
            name: "CPUUtilization".into(),
            stat: "Average".into(),
            dimensions: vec![("InstanceId".into(), "i-abc".into())],
        };
        let s = spec.serialize();
        assert!(s.contains("|InstanceId=i-abc"));
        assert_eq!(CustomMetricSpec::parse(&s).as_ref(), Some(&spec));
    }

    #[test]
    fn custom_metric_spec_parse_drops_malformed_dimension_pairs() {
        // The 'badkv' fragment is missing '='; the parser drops it but
        // keeps the well-formed pair.
        let s = "AWS/EC2|CPUUtilization|Average|InstanceId=i-abc;badkv";
        let spec = CustomMetricSpec::parse(s).expect("parse");
        assert_eq!(spec.dimensions, vec![("InstanceId".into(), "i-abc".into())]);
    }

    #[test]
    fn parse_skips_empty_and_unknown_keys() {
        let s = parse("");
        assert!(s.profile.is_none());
        let s = parse("# only comment\n  \nnonsense\n");
        assert!(s.profile.is_none());
        let s = parse("unknown = value\n");
        assert!(s.profile.is_none());
    }
}