mimir-mem 0.7.0

//! `mimir sync` / `mimir serve` — the OPTIONAL centralized-sync layer.
//!
//! Two transports share one replication core (mimir_core::replicate):
//! `file` (a folder replicated by Syncthing/Dropbox/…) and `server` (a
//! `mimir serve` hub). Everything here is gated on `[sync] mode != "off"`.
//!
//! Transport functions return summaries instead of printing, so both the CLI
//! (which prints) and the MCP background loop (which logs via tracing, since
//! stdout is the MCP protocol channel) can reuse them.

mod file;
mod http;
mod server;

use anyhow::{bail, Context, Result};
use mimir_core::replicate::{self, ApplyStats};
use mimir_core::Mimir;

/// Watermarks are persisted minus this grace window so a record written during
/// a sync round (clock skew / same-second) re-delivers next time; idempotent
/// last-write-wins apply makes the re-delivery free.
const GRACE_SECS: i64 = 60;

fn require_enabled(m: &Mimir) -> Result<()> {
    if !m.config.sync.enabled() {
        bail!(
            "sync is off. Enable it in config.toml under [sync] (mode = \"file\" or \
             \"server\") — see docs/sync.md."
        );
    }
    Ok(())
}

/// Mode-appropriate full sync, returning a one-line summary (no printing).
/// Used by the CLI `sync` command and the MCP background loop.
pub(crate) fn perform(m: &mut Mimir) -> Result<String> {
    match m.config.sync.mode.as_str() {
        "file" => {
            let dir = file_dir(m)?;
            let s = file::sync(m, &dir)?;
            Ok(format!(
                "file: wrote {} memories, merged {} peer file(s) → {} applied, {} deletes, {} edges",
                s.wrote, s.peers_merged, s.applied.nodes_upserted, s.applied.tombstones, s.applied.edges_upserted
            ))
        }
        "server" => {
            let (sent, pushed) = srv_push(m)?;
            let pulled = srv_pull(m)?;
            Ok(format!(
                "server: pushed {sent} (hub applied {}), pulled {} applied, {} deletes, {} edges",
                pushed.applied.nodes_upserted,
                pulled.nodes_upserted,
                pulled.tombstones,
                pulled.edges_upserted
            ))
        }
        "off" => bail!("sync is off"),
        other => bail!("unknown sync mode '{other}' (use \"file\" or \"server\")"),
    }
}

/// Full sync (the default `mimir sync`).
pub fn sync() -> Result<()> {
    let mut m = Mimir::open()?;
    require_enabled(&m)?;
    println!("{}", perform(&mut m)?);
    Ok(())
}

pub fn push() -> Result<()> {
    let mut m = Mimir::open()?;
    require_enabled(&m)?;
    match m.config.sync.mode.as_str() {
        "file" => {
            let dir = file_dir(&m)?;
            let n = file::write_snapshot(&m, &dir)?;
            println!("wrote snapshot of {n} memories → {}", dir.display());
        }
        "server" => {
            let (sent, res) = srv_push(&mut m)?;
            println!(
                "pushed {sent} change(s) → hub applied {}",
                res.applied.nodes_upserted
            );
        }
        other => bail!("unknown sync mode '{other}'"),
    }
    Ok(())
}

pub fn pull() -> Result<()> {
    let mut m = Mimir::open()?;
    require_enabled(&m)?;
    match m.config.sync.mode.as_str() {
        "file" => {
            let dir = file_dir(&m)?;
            let (peers, s) = file::merge_peers(&mut m, &dir)?;
            let _ = m.embed_pending();
            println!(
                "merged {peers} peer file(s) → {} applied, {} deletes, {} edges",
                s.nodes_upserted, s.tombstones, s.edges_upserted
            );
        }
        "server" => {
            let s = srv_pull(&mut m)?;
            println!(
                "pulled → {} applied, {} deletes, {} edges",
                s.nodes_upserted, s.tombstones, s.edges_upserted
            );
        }
        other => bail!("unknown sync mode '{other}'"),
    }
    Ok(())
}

pub fn status() -> Result<()> {
    let m = Mimir::open()?;
    let sc = &m.config.sync;
    println!("mode      {}", sc.mode);
    if !sc.enabled() {
        println!("(sync is off — see docs/sync.md to enable)");
        return Ok(());
    }
    match sc.mode.as_str() {
        "file" => println!("dir       {}", sc.dir),
        "server" => {
            println!("endpoint  {}", sc.endpoint);
            println!(
                "token     {}",
                if std::env::var("MIMIR_SYNC_TOKEN")
                    .map(|t| !t.is_empty())
                    .unwrap_or(false)
                {
                    "set (MIMIR_SYNC_TOKEN)"
                } else {
                    "MISSING — set MIMIR_SYNC_TOKEN"
                }
            );
        }
        _ => {}
    }
    let pending =
        replicate::changes_since(&m.conn, replicate::get_watermark(&m.conn, "last_push")?)?;
    println!("pending   {} memory change(s) to push", pending.nodes.len());
    println!("auto      {} (every {} min)", sc.auto, sc.interval_mins);
    Ok(())
}

/// Run the `mimir serve` hub.
pub fn serve(bind: String) -> Result<()> {
    server::serve(&bind)
}

// ---- helpers ----

fn file_dir(m: &Mimir) -> Result<std::path::PathBuf> {
    if m.config.sync.dir.is_empty() {
        bail!("file sync needs `[sync] dir = \"...\"` in config.toml (a folder your file-sync tool replicates)");
    }
    Ok(std::path::PathBuf::from(expand_tilde(&m.config.sync.dir)))
}

fn expand_tilde(p: &str) -> String {
    if let Some(rest) = p.strip_prefix("~/") {
        if let Some(base) = directories::BaseDirs::new() {
            return base.home_dir().join(rest).to_string_lossy().into_owned();
        }
    }
    p.to_string()
}

fn server_token() -> Result<String> {
    std::env::var("MIMIR_SYNC_TOKEN")
        .ok()
        .filter(|t| !t.is_empty())
        .context("server-mode sync needs MIMIR_SYNC_TOKEN in the environment")
}

fn server_endpoint(m: &Mimir) -> Result<String> {
    if m.config.sync.endpoint.is_empty() {
        bail!("server sync needs `[sync] endpoint = \"http://...\"` in config.toml");
    }
    Ok(m.config.sync.endpoint.clone())
}

/// Push local changes; returns (count sent, hub result). Advances last_push.
fn srv_push(m: &mut Mimir) -> Result<(usize, http::PushResult)> {
    let (endpoint, token) = (server_endpoint(m)?, server_token()?);
    let since = replicate::get_watermark(&m.conn, "last_push")?;
    let batch = replicate::changes_since(&m.conn, since)?;
    let sent = batch.nodes.len();
    let res = http::push(&endpoint, &token, &batch)?;
    replicate::set_watermark(&m.conn, "last_push", (res.watermark - GRACE_SECS).max(0))?;
    Ok((sent, res))
}

/// Pull + apply remote changes; returns apply stats. Advances last_pull and
/// embeds new arrivals locally.
fn srv_pull(m: &mut Mimir) -> Result<ApplyStats> {
    let (endpoint, token) = (server_endpoint(m)?, server_token()?);
    let since = replicate::get_watermark(&m.conn, "last_pull")?;
    let batch = http::pull(&endpoint, &token, since)?;
    let applied = replicate::apply_changes(&m.conn, &batch)?;
    replicate::set_watermark(&m.conn, "last_pull", (batch.watermark - GRACE_SECS).max(0))?;
    let _ = m.embed_pending();
    Ok(applied)
}