wm_tools/expansion/

association.rs

//! Association tools — associate_mine.

#![forbid(unsafe_code)]

use async_trait::async_trait;

use serde_json::{Value, json};
use std::sync::Arc;
use wm_core::{Context, EffectRow, Gana, Resource, Tool, ToolStats};
use wm_memory::{Association, AssociationStore, LinkType, MemoryStore};

use super::common::parse_galaxy;

pub struct MemoryAssociateMineTool {
    store: Arc<MemoryStore>,
    stats: ToolStats,
    effects: EffectRow,
}

impl MemoryAssociateMineTool {
    pub fn new(store: Arc<MemoryStore>) -> Self {
        Self {
            store,
            stats: ToolStats::default(),
            effects: EffectRow {
                writes: vec![Resource::Galaxy("associations".into())],
                ..Default::default()
            },
        }
    }
}

#[async_trait]
impl Tool for MemoryAssociateMineTool {
    fn input_schema(&self) -> Value {
        super::common::schema(
            &json!({
                "galaxy": super::common::str_prop("Galaxy to mine (optional; default codex)"),
                "limit": super::common::int_prop("Maximum memories to mine (default 50)"),
            }),
            &[],
        )
    }
    fn name(&self) -> &str {
        "memory.associate_mine"
    }
    fn gana(&self) -> Gana {
        Gana::Net
    }
    fn effects(&self) -> &EffectRow {
        &self.effects
    }
    fn description(&self) -> &str {
        "Mine associations across galaxies using keyword overlap"
    }
    async fn call(&self, _ctx: &mut Context, args: Value) -> wm_core::Result<Value> {
        let galaxy_name_str = args
            .get("galaxy")
            .and_then(|v| v.as_str())
            .unwrap_or("codex");
        let galaxy = parse_galaxy(galaxy_name_str)?;
        let limit = args
            .get("limit")
            .and_then(serde_json::Value::as_u64)
            .unwrap_or(50) as usize;
        let memories = self.store.scan(galaxy, limit)?;
        let env = self.store.env();
        let assoc_store = AssociationStore::open(env)?;
        let mut proposed = 0u32;
        for i in 0..memories.len() {
            for j in (i + 1)..memories.len() {
                let a = &memories[i];
                let b = &memories[j];
                let a_words: std::collections::HashSet<&str> =
                    a.content.split_whitespace().collect();
                let b_words: std::collections::HashSet<&str> =
                    b.content.split_whitespace().collect();
                let intersection = a_words.intersection(&b_words).count();
                let union = a_words.union(&b_words).count();
                if union > 0 && intersection > 2 {
                    let strength = intersection as f32 / union as f32;
                    if strength > 0.3 {
                        let assoc = Association::new(
                            a.metadata.id,
                            b.metadata.id,
                            LinkType::Related,
                            strength,
                        );
                        let _ = assoc_store.put(env, &assoc);
                        proposed += 1;
                    }
                }
            }
        }
        Ok(json!({
            "status": "success",
            "galaxy": galaxy_name_str,
            "scanned": memories.len(),
            "proposed_associations": proposed,
        }))
    }
    fn stats(&self) -> &ToolStats {
        &self.stats
    }
}

/// `memory.corroborate` — record a session's corroboration of a memory
/// (bridging consensus counter, 2nd PR after Slice B; maintainer ruling 2026-09-04).
///
/// A corroboration says: this session's agent independently stands behind
/// the memory. Sessions nest agents and machines — sessions are the working
/// unit for now; the record keeps only the session id (agent/machine can be
/// joined from session records later without a schema change).
///
/// Semantics:
/// - corroborator = explicit `session` UUID arg, else the dispatch
///   `Context.session_id`, else `InvalidArgs` (anonymous corroboration is
///   meaningless — independence requires an identity, and it is
///   client-asserted like `user_id`: attribution, never authentication).
/// - idempotent: a repeat corroboration by the same session reports
///   `already_recorded` and changes nothing.
/// - corroborating a non-current (validity-stamped) memory is allowed —
///   history accumulates; the recall surface still filters by validity
///   while enforced.
/// - ranking effect: none unless `WM_CORROBORATION_WEIGHT > 0` (the
///   saturating `corroboration_boost`, disclosed per-result).
pub struct MemoryCorroborateTool {
    store: Arc<MemoryStore>,
    stats: ToolStats,
    effects: EffectRow,
}

impl MemoryCorroborateTool {
    pub fn new(store: Arc<MemoryStore>) -> Self {
        Self {
            store,
            stats: ToolStats::default(),
            effects: EffectRow {
                writes: super::common::memory_galaxy_writes(),
                reads: super::common::memory_galaxy_reads(),
                ..Default::default()
            },
        }
    }
}

#[async_trait]
impl Tool for MemoryCorroborateTool {
    fn name(&self) -> &str {
        "memory.corroborate"
    }
    fn gana(&self) -> Gana {
        Gana::Net
    }
    fn effects(&self) -> &EffectRow {
        &self.effects
    }
    fn description(&self) -> &str {
        "Record this session's corroboration of a memory (bridging consensus counter input; ranking effect only with WM_CORROBORATION_WEIGHT > 0)"
    }
    fn input_schema(&self) -> Value {
        super::common::schema(
            &json!({
                "id": super::common::str_prop("Memory UUID to corroborate"),
                "session": super::common::str_prop("Corroborating session UUID (defaults to the dispatch session; required when the dispatch carries none)"),
            }),
            &["id"],
        )
    }
    async fn call(&self, ctx: &mut Context, args: Value) -> wm_core::Result<Value> {
        let id_str = args.get("id").and_then(Value::as_str).ok_or_else(|| {
            wm_core::CoreError::InvalidArgs("id (memory UUID string) required".into())
        })?;
        let id = uuid::Uuid::parse_str(id_str)
            .map_err(|e| wm_core::CoreError::InvalidArgs(format!("invalid memory UUID: {e}")))?;
        let session = args
            .get("session")
            .and_then(Value::as_str)
            .map(uuid::Uuid::parse_str)
            .transpose()
            .map_err(|e| wm_core::CoreError::InvalidArgs(format!("invalid session UUID: {e}")))?
            .or(ctx.session_id)
            .ok_or_else(|| {
                wm_core::CoreError::InvalidArgs(
                    "session (UUID string) required: the dispatch carries no session_id and none was passed"
                        .into(),
                )
            })?;
        // Galaxy-blind resolution (the graph.walk pattern): corroborate by
        // id wherever the memory lives.
        let mut found = None;
        for galaxy in wm_core::Galaxy::memory_galaxies() {
            if let Some(mem) = self.store.get(galaxy, id)? {
                found = Some((galaxy, mem));
                break;
            }
        }
        let (galaxy, mut mem) = found.ok_or_else(|| {
            wm_core::CoreError::NotFound(format!("memory {id} not found in any memory galaxy"))
        })?;
        if mem.metadata.corroborated_by.contains(&session) {
            return Ok(json!({
                "status": "already_recorded",
                "id": id,
                "galaxy": super::common::galaxy_name(galaxy),
                "corroboration_count": mem.metadata.corroborated_by.len(),
            }));
        }
        mem.metadata.corroborated_by.push(session);
        // No content change: the Tantivy index needs no update.
        self.store.put(galaxy, &mem)?;
        Ok(json!({
            "status": "recorded",
            "id": id,
            "galaxy": super::common::galaxy_name(galaxy),
            "session": session,
            "corroboration_count": mem.metadata.corroborated_by.len(),
        }))
    }
    fn stats(&self) -> &ToolStats {
        &self.stats
    }
}

/// `memory.relate` — typed-edge tool over the existing [`AssociationStore`]
/// (V8 S6: thin tool, not a new store).
///
/// Two actions:
/// - **relate** (default): create or re-activate a typed edge between two
///   memories. Existing edges are Hebbian-activated (`co_activation_count`
///   bumps, weight strengthens) instead of duplicated. The `follows`
///   link type maps onto `LinkType::Temporal` with the `follows`
///   association-type marker — session --follows--> session semantics.
/// - **derive_follows**: scan the Sessions galaxy (optionally filtered by
///   a workspace tag), sort session starts by creation time, and link
///   each consecutive pair with a `follows` edge — weight reflects the
///   shared tags (provenance signal), capped at 0.9.
///
/// Edges decay/prune via the store's existing Hebbian dynamics; the read
/// surface is `graph.walk` (and, with `WM_RECALL_GRAPH_WEIGHT > 0`, the
/// recall fusion itself).
pub struct MemoryRelateTool {
    store: Arc<MemoryStore>,
    stats: ToolStats,
    effects: EffectRow,
}

impl MemoryRelateTool {
    pub fn new(store: Arc<MemoryStore>) -> Self {
        Self {
            store,
            stats: ToolStats::default(),
            effects: EffectRow {
                writes: vec![Resource::Galaxy("associations".into())],
                ..Default::default()
            },
        }
    }

    /// Resolve a memory id across the memory galaxies.
    fn memory_exists(&self, id: uuid::Uuid) -> bool {
        wm_core::Galaxy::memory_galaxies()
            .iter()
            .any(|g| self.store.get(*g, id).ok().flatten().is_some())
    }

    fn relate_pair(
        &self,
        source: uuid::Uuid,
        target: uuid::Uuid,
        link_type: LinkType,
        marker: &str,
        weight: f32,
    ) -> wm_core::Result<Value> {
        if source == target {
            return Err(wm_core::CoreError::InvalidArgs(
                "source and target must differ".into(),
            ));
        }
        if !self.memory_exists(source) {
            return Err(wm_core::CoreError::NotFound(format!(
                "source memory {source} not found in any memory galaxy"
            )));
        }
        if !self.memory_exists(target) {
            return Err(wm_core::CoreError::NotFound(format!(
                "target memory {target} not found in any memory galaxy"
            )));
        }
        let env = self.store.env();
        let assoc_store = AssociationStore::open(env)?;
        if let Some(mut existing) = assoc_store.get(env, source, target)? {
            // Hebbian re-activation: the edge strengthens with co-use.
            existing.activate();
            if existing.link_type != link_type {
                existing.link_type = link_type;
                existing.association_type = marker.to_string();
            }
            assoc_store.put(env, &existing)?;
            return Ok(json!({
                "status": "activated",
                "source": source.to_string(),
                "target": target.to_string(),
                "link_type": existing.link_type.as_str(),
                "weight": existing.weight,
                "co_activation_count": existing.co_activation_count,
            }));
        }
        let mut assoc = Association::new(source, target, link_type, weight.clamp(0.0, 1.0));
        // `follows` rides the temporal type with its own marker (the
        // constructor derives the string from the type — override it).
        if marker != assoc.association_type {
            assoc.association_type = marker.to_string();
        }
        assoc_store.put(env, &assoc)?;
        Ok(json!({
            "status": "created",
            "source": source.to_string(),
            "target": target.to_string(),
            "link_type": assoc.link_type.as_str(),
            "association_type": assoc.association_type,
            "weight": assoc.weight,
        }))
    }

    fn derive_follows(&self, workspace_tag: Option<&str>) -> wm_core::Result<Value> {
        // Session start markers carry the `start` tag; sort by creation.
        let mut sessions: Vec<wm_memory::Memory> = self
            .store
            .scan_all(wm_core::Galaxy::Sessions)?
            .into_iter()
            .filter(|m| m.metadata.tags.contains(&"start".to_string()))
            .filter(|m| workspace_tag.is_none_or(|w| m.metadata.tags.contains(&w.to_string())))
            .collect();
        sessions.sort_by_key(|m| m.metadata.created_at);

        let env = self.store.env();
        let assoc_store = AssociationStore::open(env)?;
        let mut linked = 0usize;
        let mut activated = 0usize;
        for pair in sessions.windows(2) {
            let (earlier, later) = (&pair[0], &pair[1]);
            // Shared tags = the provenance signal for the follow.
            let shared = earlier
                .metadata
                .tags
                .iter()
                .filter(|t| later.metadata.tags.contains(t))
                .count();
            // Documented allow: mul_add changes float rounding; the weight is a
            // deterministic provenance signal, not a hot-path score.
            #[allow(clippy::suboptimal_flops)]
            let weight = (0.5 + 0.05 * shared as f32).min(0.9);
            let source = earlier.metadata.id;
            let target = later.metadata.id;
            if let Some(mut existing) = assoc_store.get(env, source, target)? {
                existing.activate();
                assoc_store.put(env, &existing)?;
                activated += 1;
            } else {
                let mut assoc = Association::new(source, target, LinkType::Temporal, weight);
                assoc.association_type = "follows".to_string();
                assoc_store.put(env, &assoc)?;
                linked += 1;
            }
        }
        Ok(json!({
            "status": "success",
            "sessions": sessions.len(),
            "workspace_tag": workspace_tag,
            "follows_created": linked,
            "follows_activated": activated,
        }))
    }

    /// Extract `@token` mentions from text (`@` + 2+ word chars/hyphens,
    /// lowercased, deduped, in first-seen order). No regex dependency —
    /// a manual scan keeps this crate's dependency surface unchanged.
    fn extract_mentions(text: &str) -> Vec<String> {
        let mut out = Vec::new();
        let bytes = text.as_bytes();
        let mut i = 0;
        while i < bytes.len() {
            if bytes[i] == b'@' {
                let mut j = i + 1;
                while j < bytes.len()
                    && (bytes[j].is_ascii_alphanumeric() || bytes[j] == b'_' || bytes[j] == b'-')
                {
                    j += 1;
                }
                if j - (i + 1) >= 2 {
                    let token = text[i + 1..j].to_lowercase();
                    if !out.contains(&token) {
                        out.push(token);
                    }
                }
                i = j;
            } else {
                i += 1;
            }
        }
        out
    }

    /// Extract ISO calendar dates (`YYYY-MM-DD`) validated by chrono —
    /// bare digit runs that are not real dates (month 13, day 40) are
    /// refused rather than linked.
    fn extract_iso_dates(text: &str) -> Vec<String> {
        let mut out = Vec::new();
        let bytes = text.as_bytes();
        let mut i = 0;
        while i + 10 <= bytes.len() {
            let s = &text[i..i + 10];
            let is_shape = s.as_bytes()[4] == b'-'
                && s.as_bytes()[7] == b'-'
                && s[..4].bytes().all(|b| b.is_ascii_digit())
                && s[5..7].bytes().all(|b| b.is_ascii_digit())
                && s[8..10].bytes().all(|b| b.is_ascii_digit());
            if is_shape && chrono::NaiveDate::parse_from_str(s, "%Y-%m-%d").is_ok() {
                let date = s.to_string();
                if !out.contains(&date) {
                    out.push(date);
                }
                i += 10;
            } else {
                i += 1;
            }
        }
        out
    }

    /// `derive_mentions` — close the S6 `mentions` gap beside
    /// `derive_follows`. For each `@token` in a memory's content/title,
    /// link to memories whose title contains the token or whose tags
    /// equal it (case-insensitive) with a `(Related, "mentions")` edge —
    /// the same marker-reuse trick as `follows` (no DB migration; the
    /// recall graph phase already walks any edge type).
    ///
    /// Conservative by construction: tokens resolve only onto memories in
    /// the same scanned galaxy, at most 5 targets per token (oldest
    /// first), and re-runs Hebbian-activate instead of duplicating.
    fn derive_mentions(&self, galaxy_arg: Option<&str>) -> wm_core::Result<Value> {
        let galaxy = super::common::parse_galaxy_or(galaxy_arg, wm_core::Galaxy::Codex)?;
        let mut mems = self.store.scan(galaxy, 10_000)?;
        mems.sort_by_key(|m| m.metadata.created_at);
        let mut created = 0usize;
        let mut activated = 0usize;
        let mut scanned = 0usize;
        for source in &mems {
            scanned += 1;
            let haystack = match &source.metadata.title {
                Some(title) => format!("{}\n{title}", source.content),
                None => source.content.clone(),
            };
            for token in Self::extract_mentions(&haystack).iter().take(20) {
                let mut targets = 0;
                for target in &mems {
                    if target.metadata.id == source.metadata.id || targets >= 5 {
                        continue;
                    }
                    let title_hit = target
                        .metadata
                        .title
                        .as_deref()
                        .unwrap_or_default()
                        .to_lowercase()
                        .contains(token);
                    let tag_hit = target
                        .metadata
                        .tags
                        .iter()
                        .any(|t| t.to_lowercase() == *token);
                    if !(title_hit || tag_hit) {
                        continue;
                    }
                    targets += 1;
                    let out = self.relate_pair(
                        source.metadata.id,
                        target.metadata.id,
                        LinkType::Related,
                        "mentions",
                        0.6,
                    )?;
                    if out.get("status").and_then(Value::as_str) == Some("activated") {
                        activated += 1;
                    } else {
                        created += 1;
                    }
                }
            }
        }
        Ok(json!({
            "status": "success",
            "galaxy": super::common::galaxy_name(galaxy),
            "scanned": scanned,
            "mentions_created": created,
            "mentions_activated": activated,
        }))
    }

    /// `derive_at` — close the S6 `at` gap beside `derive_follows`.
    /// Memories are not time nodes, so there is nothing to point at:
    /// instead, memories sharing an exact ISO date string are linked
    /// consecutive-by-time with a `(Temporal, "at")` edge (same
    /// marker-reuse trick). Exact-date co-occurrence only — no fuzzy
    /// temporal inference, no speculative links.
    fn derive_at(&self, galaxy_arg: Option<&str>) -> wm_core::Result<Value> {
        let galaxy = super::common::parse_galaxy_or(galaxy_arg, wm_core::Galaxy::Codex)?;
        let mut mems = self.store.scan(galaxy, 10_000)?;
        mems.sort_by_key(|m| m.metadata.created_at);
        let mut by_date: std::collections::BTreeMap<String, Vec<uuid::Uuid>> =
            std::collections::BTreeMap::new();
        for mem in &mems {
            for date in Self::extract_iso_dates(&mem.content) {
                by_date.entry(date).or_default().push(mem.metadata.id);
            }
        }
        let mut created = 0usize;
        let mut activated = 0usize;
        let mut dates_linked = 0usize;
        for ids in by_date.values().filter(|ids| ids.len() >= 2) {
            dates_linked += 1;
            for pair in ids.windows(2) {
                let out = self.relate_pair(pair[0], pair[1], LinkType::Temporal, "at", 0.5)?;
                if out.get("status").and_then(Value::as_str) == Some("activated") {
                    activated += 1;
                } else {
                    created += 1;
                }
            }
        }
        Ok(json!({
            "status": "success",
            "galaxy": super::common::galaxy_name(galaxy),
            "scanned": mems.len(),
            "dates_linked": dates_linked,
            "at_created": created,
            "at_activated": activated,
        }))
    }
}

#[async_trait]
impl Tool for MemoryRelateTool {
    fn name(&self) -> &str {
        "memory.relate"
    }
    fn gana(&self) -> Gana {
        Gana::Net
    }
    fn effects(&self) -> &EffectRow {
        &self.effects
    }
    fn input_schema(&self) -> Value {
        super::common::schema(
            &json!({
                "action": super::common::str_prop("relate (default) | derive_follows | derive_mentions | derive_at"),
                "source": super::common::str_prop("Source memory UUID (relate action)"),
                "target": super::common::str_prop("Target memory UUID (relate action)"),
                "link_type": super::common::str_prop("related | extends | contradicts | supersedes | temporal | causal | cascade | follows (default related; follows = temporal + follows marker)"),
                "weight": super::common::str_prop("Edge weight 0.0-1.0 (default 0.6; follows derivation derives it from shared tags)"),
                "workspace_tag": super::common::str_prop("derive_follows: only link sessions carrying this tag"),
                "galaxy": super::common::str_prop("derive_mentions / derive_at: galaxy to scan (default codex)"),
            }),
            &[],
        )
    }
    fn description(&self) -> &str {
        "Relate two memories with a typed, Hebbian edge (related/extends/contradicts/supersedes/temporal/causal/cascade/follows), or derive session --follows--> session threads (action: derive_follows), @mention links (action: derive_mentions), or shared-date links (action: derive_at). Edges feed graph.walk and the recall fusion graph phase."
    }
    async fn call(&self, _ctx: &mut Context, args: Value) -> wm_core::Result<Value> {
        match args
            .get("action")
            .and_then(Value::as_str)
            .unwrap_or("relate")
        {
            "derive_follows" => {
                let workspace_tag = args
                    .get("workspace_tag")
                    .and_then(Value::as_str)
                    .filter(|s| !s.is_empty());
                self.derive_follows(workspace_tag)
            }
            "derive_mentions" => self.derive_mentions(args.get("galaxy").and_then(Value::as_str)),
            "derive_at" => self.derive_at(args.get("galaxy").and_then(Value::as_str)),
            "relate" => {
                let source_str = args.get("source").and_then(Value::as_str).ok_or_else(|| {
                    wm_core::CoreError::InvalidArgs("source (UUID string) required".into())
                })?;
                let target_str = args.get("target").and_then(Value::as_str).ok_or_else(|| {
                    wm_core::CoreError::InvalidArgs("target (UUID string) required".into())
                })?;
                let source = uuid::Uuid::parse_str(source_str).map_err(|e| {
                    wm_core::CoreError::InvalidArgs(format!("invalid source UUID: {e}"))
                })?;
                let target = uuid::Uuid::parse_str(target_str).map_err(|e| {
                    wm_core::CoreError::InvalidArgs(format!("invalid target UUID: {e}"))
                })?;
                let weight = args
                    .get("weight")
                    .and_then(Value::as_f64)
                    .map_or(0.6, |w| w as f32);
                let link_str = args
                    .get("link_type")
                    .and_then(Value::as_str)
                    .unwrap_or("related");
                // `follows` is a semantic marker on the temporal type.
                let (link_type, marker) = if link_str == "follows" {
                    (LinkType::Temporal, "follows")
                } else {
                    (LinkType::from_str_lossy(link_str), link_str)
                };
                self.relate_pair(source, target, link_type, marker, weight)
            }
            other => Err(wm_core::CoreError::InvalidArgs(format!(
                "unknown action '{other}' (relate | derive_follows | derive_mentions | derive_at)"
            ))),
        }
    }
    fn stats(&self) -> &ToolStats {
        &self.stats
    }
}

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

    fn test_store() -> Arc<MemoryStore> {
        let dir = tempfile::tempdir().unwrap();
        let path = dir.path().join("lmdb");
        std::fs::create_dir_all(&path).unwrap();
        // Leak the tempdir for the test's lifetime (store outlives it).
        std::mem::forget(dir);
        Arc::new(MemoryStore::open_default(path).unwrap())
    }

    #[tokio::test]
    async fn relate_creates_then_activates_the_same_edge() {
        let store = test_store();
        let a = Memory::new(wm_core::Galaxy::Codex, "decision alpha".into());
        let b = Memory::new(wm_core::Galaxy::Codex, "decision beta extends alpha".into());
        store.put(wm_core::Galaxy::Codex, &a).unwrap();
        store.put(wm_core::Galaxy::Codex, &b).unwrap();

        let tool = MemoryRelateTool::new(store.clone());
        let mut ctx = Context::default();
        let args = json!({
            "source": a.metadata.id.to_string(),
            "target": b.metadata.id.to_string(),
            "link_type": "extends",
            "weight": 0.7,
        });
        let first = tool.call(&mut ctx, args.clone()).await.unwrap();
        assert_eq!(first["status"], "created");
        assert_eq!(first["link_type"], "extends");
        assert!((first["weight"].as_f64().unwrap() - 0.7).abs() < 1e-6);

        let second = tool.call(&mut ctx, args).await.unwrap();
        assert_eq!(second["status"], "activated");
        assert_eq!(second["co_activation_count"], 1);
        // Hebbian re-activation strengthened the edge.
        assert!(second["weight"].as_f64().unwrap() > 0.7);

        // Exactly one edge exists.
        let env = store.env();
        let assocs = AssociationStore::open(env).unwrap();
        assert_eq!(assocs.count(env).unwrap(), 1);
    }

    #[tokio::test]
    async fn relate_follows_maps_to_temporal_with_marker() {
        let store = test_store();
        let s1 = Memory::new(wm_core::Galaxy::Sessions, "session one start".into());
        let s2 = Memory::new(wm_core::Galaxy::Sessions, "session two start".into());
        store.put(wm_core::Galaxy::Sessions, &s1).unwrap();
        store.put(wm_core::Galaxy::Sessions, &s2).unwrap();

        let tool = MemoryRelateTool::new(store);
        let mut ctx = Context::default();
        let out = tool
            .call(
                &mut ctx,
                json!({
                    "source": s1.metadata.id.to_string(),
                    "target": s2.metadata.id.to_string(),
                    "link_type": "follows",
                }),
            )
            .await
            .unwrap();
        assert_eq!(out["link_type"], "temporal");
        assert_eq!(out["association_type"], "follows");
    }

    #[tokio::test]
    async fn relate_validates_endpoints() {
        let store = test_store();
        let a = Memory::new(wm_core::Galaxy::Codex, "existing memory".into());
        store.put(wm_core::Galaxy::Codex, &a).unwrap();
        let tool = MemoryRelateTool::new(store);
        let mut ctx = Context::default();
        let err = tool
            .call(
                &mut ctx,
                json!({
                    "source": a.metadata.id.to_string(),
                    "target": uuid::Uuid::new_v4().to_string(),
                }),
            )
            .await
            .unwrap_err();
        assert!(err.to_string().contains("target memory"), "{err}");

        // Self-edges are refused.
        let err2 = tool
            .call(
                &mut ctx,
                json!({
                    "source": a.metadata.id.to_string(),
                    "target": a.metadata.id.to_string(),
                }),
            )
            .await
            .unwrap_err();
        assert!(err2.to_string().contains("must differ"), "{err2}");
    }

    #[tokio::test]
    async fn derive_follows_links_consecutive_sessions_in_order() {
        let store = test_store();
        // Three sessions, deliberately created out of chronological order.
        let mk = |content: &str, offset_days: i64, workspace: Option<&str>| {
            let mut m = Memory::new(wm_core::Galaxy::Sessions, content.to_string());
            m.metadata.tags = vec!["start".to_string(), "session".to_string()];
            if let Some(w) = workspace {
                m.metadata.tags.push(w.to_string());
            }
            m.metadata.created_at = chrono::Utc::now() - chrono::Duration::days(offset_days);
            m.metadata.accessed_at = m.metadata.created_at;
            m
        };
        let s1 = mk("first session", 3, Some("wmv5"));
        let s2 = mk("second session", 2, Some("wmv5"));
        let s3 = mk("third session", 1, None);
        // Insert out of order; the derivation sorts by created_at.
        for m in [&s3, &s1, &s2] {
            store.put(wm_core::Galaxy::Sessions, m).unwrap();
        }

        let tool = MemoryRelateTool::new(store.clone());
        let mut ctx = Context::default();

        // Workspace-filtered: only s1 --follows--> s2 (both carry "wmv5").
        let out = tool
            .call(
                &mut ctx,
                json!({"action": "derive_follows", "workspace_tag": "wmv5"}),
            )
            .await
            .unwrap();
        assert_eq!(out["sessions"], 2);
        assert_eq!(out["follows_created"], 1);

        let env = store.env();
        let assocs = AssociationStore::open(env).unwrap();
        let from_s1 = assocs.find_from(env, s1.metadata.id).unwrap();
        assert_eq!(from_s1.len(), 1);
        assert_eq!(from_s1[0].target, s2.metadata.id);
        assert_eq!(from_s1[0].link_type, LinkType::Temporal);
        assert_eq!(from_s1[0].association_type, "follows");

        // Unfiltered re-run: s2 --follows--> s3 is new; s1→s2 re-activates.
        let out2 = tool
            .call(&mut ctx, json!({"action": "derive_follows"}))
            .await
            .unwrap();
        assert_eq!(out2["sessions"], 3);
        assert_eq!(out2["follows_created"], 1);
        assert_eq!(out2["follows_activated"], 1);
        assert_eq!(assocs.count(env).unwrap(), 2);

        // The acceptance query: "what led to s3" walks back through the
        // follows chain to s1.
        let chain = assocs.find_to(env, s3.metadata.id).unwrap();
        assert_eq!(chain[0].source, s2.metadata.id);
        let earlier = assocs.find_to(env, chain[0].source).unwrap();
        assert_eq!(earlier[0].source, s1.metadata.id);
    }

    /// S6: `@token` mentions resolve onto titled/tagged memories with a
    /// `(Related, "mentions")` edge; unresolvable tokens link nothing.
    #[tokio::test]
    async fn derive_mentions_links_at_tokens_to_titled_memories() {
        let store = test_store();
        let mut aria = Memory::new(wm_core::Galaxy::Codex, "persona notes".into());
        aria.metadata.title = Some("Aria Essays".into());
        store.put(wm_core::Galaxy::Codex, &aria).unwrap();
        let mentioner = Memory::new(
            wm_core::Galaxy::Codex,
            "as @aria wrote, memory persists".into(),
        );
        store.put(wm_core::Galaxy::Codex, &mentioner).unwrap();
        let lonely = Memory::new(
            wm_core::Galaxy::Codex,
            "shouting into the void @nobodyhere".into(),
        );
        store.put(wm_core::Galaxy::Codex, &lonely).unwrap();

        let tool = MemoryRelateTool::new(store.clone());
        let mut ctx = Context::default();
        let out = tool
            .call(&mut ctx, json!({"action": "derive_mentions"}))
            .await
            .unwrap();
        assert_eq!(out["status"], "success");
        assert_eq!(out["scanned"], 3);
        assert_eq!(out["mentions_created"], 1);

        let env = store.env();
        let assocs = AssociationStore::open(env).unwrap();
        let edges = assocs.find_from(env, mentioner.metadata.id).unwrap();
        assert_eq!(edges.len(), 1);
        assert_eq!(edges[0].target, aria.metadata.id);
        assert_eq!(edges[0].link_type, LinkType::Related);
        assert_eq!(edges[0].association_type, "mentions");
        // Unresolvable token: no edge.
        assert!(
            assocs
                .find_from(env, lonely.metadata.id)
                .unwrap()
                .is_empty()
        );

        // Re-run Hebbian-activates instead of duplicating.
        let out2 = tool
            .call(&mut ctx, json!({"action": "derive_mentions"}))
            .await
            .unwrap();
        assert_eq!(out2["mentions_created"], 0);
        assert_eq!(out2["mentions_activated"], 1);
        assert_eq!(assocs.count(env).unwrap(), 1);
    }

    /// S6: shared exact ISO dates link consecutive-by-time with a
    /// `(Temporal, "at")` edge; non-dates (`2026-13-40`) never link.
    #[tokio::test]
    async fn derive_at_links_shared_dates_only() {
        let store = test_store();
        let a = Memory::new(
            wm_core::Galaxy::Codex,
            "shipped the slice on 2026-09-04".into(),
        );
        store.put(wm_core::Galaxy::Codex, &a).unwrap();
        let b = Memory::new(
            wm_core::Galaxy::Codex,
            "retro on 2026-09-04 went well".into(),
        );
        store.put(wm_core::Galaxy::Codex, &b).unwrap();
        let c = Memory::new(
            wm_core::Galaxy::Codex,
            "impossible date 2026-13-40 here".into(),
        );
        store.put(wm_core::Galaxy::Codex, &c).unwrap();

        let tool = MemoryRelateTool::new(store.clone());
        let mut ctx = Context::default();
        let out = tool
            .call(&mut ctx, json!({"action": "derive_at"}))
            .await
            .unwrap();
        assert_eq!(out["status"], "success");
        assert_eq!(out["dates_linked"], 1);
        assert_eq!(out["at_created"], 1);

        let env = store.env();
        let assocs = AssociationStore::open(env).unwrap();
        let edges = assocs.find_from(env, a.metadata.id).unwrap();
        assert_eq!(edges.len(), 1);
        assert_eq!(edges[0].target, b.metadata.id);
        assert_eq!(edges[0].link_type, LinkType::Temporal);
        assert_eq!(edges[0].association_type, "at");
        assert!(assocs.find_from(env, c.metadata.id).unwrap().is_empty());
    }

    /// Bridging counter: recording is idempotent per session, distinct
    /// sessions accumulate, and anonymous corroboration is refused.
    #[tokio::test]
    async fn corroborate_records_distinct_sessions_idempotently() {
        let store = test_store();
        let mem = Memory::new(wm_core::Galaxy::Codex, "a claim".into());
        let id = mem.metadata.id;
        store.put(wm_core::Galaxy::Codex, &mem).unwrap();

        let tool = MemoryCorroborateTool::new(store.clone());
        let mut ctx = Context::default();
        let s1 = uuid::Uuid::new_v4().to_string();
        let s2 = uuid::Uuid::new_v4().to_string();

        let first = tool
            .call(&mut ctx, json!({"id": id.to_string(), "session": s1}))
            .await
            .unwrap();
        assert_eq!(first["status"], "recorded");
        assert_eq!(first["corroboration_count"], 1);

        // Same session again: no-op, count unchanged.
        let repeat = tool
            .call(&mut ctx, json!({"id": id.to_string(), "session": s1}))
            .await
            .unwrap();
        assert_eq!(repeat["status"], "already_recorded");
        assert_eq!(repeat["corroboration_count"], 1);

        // Distinct session accumulates.
        let second = tool
            .call(&mut ctx, json!({"id": id.to_string(), "session": s2}))
            .await
            .unwrap();
        assert_eq!(second["status"], "recorded");
        assert_eq!(second["corroboration_count"], 2);

        // Persisted on the record.
        let stored = store.get(wm_core::Galaxy::Codex, id).unwrap().unwrap();
        assert_eq!(stored.metadata.corroborated_by.len(), 2);
    }

    #[tokio::test]
    async fn corroborate_uses_dispatch_session_then_refuses_anonymous() {
        let store = test_store();
        let mem = Memory::new(wm_core::Galaxy::Codex, "a claim".into());
        let id = mem.metadata.id;
        store.put(wm_core::Galaxy::Codex, &mem).unwrap();

        let tool = MemoryCorroborateTool::new(store.clone());
        // Dispatch session fills the corroborator.
        let session = uuid::Uuid::new_v4();
        let mut ctx = Context {
            session_id: Some(session),
            ..Default::default()
        };
        let out = tool
            .call(&mut ctx, json!({"id": id.to_string()}))
            .await
            .unwrap();
        assert_eq!(out["status"], "recorded");
        assert_eq!(out["session"], session.to_string());

        // No session anywhere: refused, not minted anonymously.
        let mut bare = Context::default();
        let err = tool
            .call(&mut bare, json!({"id": id.to_string()}))
            .await
            .unwrap_err();
        assert!(matches!(err, wm_core::CoreError::InvalidArgs(_)));

        // Unknown memory: not found.
        let err = tool
            .call(
                &mut bare,
                json!({"id": uuid::Uuid::new_v4().to_string(), "session": session.to_string()}),
            )
            .await
            .unwrap_err();
        assert!(matches!(err, wm_core::CoreError::NotFound(_)));
    }
}