wx-cli/src/cli/new_messages.rs

use super::output::{emit_warnings, print_response, OutputOpts};
use super::transport;
use crate::ipc::Request;
use anyhow::Result;
use std::collections::HashMap;

fn state_file() -> std::path::PathBuf {
    dirs::home_dir()
        .unwrap_or_else(|| std::path::PathBuf::from("."))
        .join(".wx-cli")
        .join("last_check.json")
}

/// 加载上次的 per-session 时间戳快照
/// 格式：{ "sessions": { "username": timestamp, ... } }
/// 旧格式（只有 timestamp 字段）直接丢弃，重新全量获取
fn load_state() -> Option<HashMap<String, i64>> {
    let data = std::fs::read_to_string(state_file()).ok()?;
    let v: serde_json::Value = serde_json::from_str(&data).ok()?;
    // 旧格式（只有 timestamp 字段）没有 sessions key → 返回 None 触发首次运行逻辑
    let map: HashMap<String, i64> = v
        .get("sessions")?
        .as_object()?
        .iter()
        .filter_map(|(k, v)| v.as_i64().map(|ts| (k.clone(), ts)))
        .collect();
    // 空 map 也是合法状态（账号无任何会话），返回 Some(empty) 而非 None
    // 这样不会误触发全量历史拉取
    Some(map)
}

fn save_state(new_state: &HashMap<String, i64>) -> Result<()> {
    let path = state_file();
    if let Some(parent) = path.parent() {
        std::fs::create_dir_all(parent)?;
    }
    std::fs::write(
        &path,
        serde_json::to_string(&serde_json::json!({ "sessions": new_state }))?,
    )?;
    Ok(())
}

pub fn cmd_new_messages(limit: usize, opts: OutputOpts) -> Result<()> {
    let state = load_state();
    let (with_meta, debug_source) = opts.request_flags();
    let resp = transport::send(Request::NewMessages {
        state,
        limit,
        with_meta,
        debug_source,
    })?;

    // 保存 daemon 返回的 new_state
    if let Some(obj) = resp.data.get("new_state").and_then(|v| v.as_object()) {
        let map: HashMap<String, i64> = obj
            .iter()
            .filter_map(|(k, v)| v.as_i64().map(|ts| (k.clone(), ts)))
            .collect();
        if !map.is_empty() {
            let _ = save_state(&map);
        }
    }

    emit_warnings(&resp.data);
    print_response(&resp.data, &opts)
}