recall_echo/

graph_cli.rs

//! Graph memory CLI subcommands (behind `graph` feature flag).

use std::path::{Path, PathBuf};

use crate::graph::traverse::format_traversal;
use crate::graph::types::*;
use crate::graph::GraphMemory;

const GREEN: &str = "\x1b[32m";
const CYAN: &str = "\x1b[36m";
const YELLOW: &str = "\x1b[33m";
const BOLD: &str = "\x1b[1m";
const DIM: &str = "\x1b[2m";
const RESET: &str = "\x1b[0m";

/// Initialize the graph store at {memory_dir}/graph/.
pub fn init(memory_dir: &Path) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;
        println!(
            "{GREEN}✓{RESET} Graph store initialized at {}",
            graph_dir.display()
        );
        Ok(())
    })
}

/// Show graph stats.
pub fn graph_status(memory_dir: &Path) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }
    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;
        let stats = gm.stats().await.map_err(|e| e.to_string())?;

        println!("{BOLD}Graph Memory Status{RESET}");
        println!("  Entities:      {}", stats.entity_count);
        println!("  Relationships: {}", stats.relationship_count);
        println!("  Episodes:      {}", stats.episode_count);

        if !stats.entity_type_counts.is_empty() {
            println!("\n  {DIM}By type:{RESET}");
            let mut types: Vec<_> = stats.entity_type_counts.iter().collect();
            types.sort_by(|a, b| b.1.cmp(a.1));
            for (t, count) in types {
                println!("    {t}: {count}");
            }
        }
        Ok(())
    })
}

/// Add an entity to the graph.
pub fn add_entity(
    memory_dir: &Path,
    name: &str,
    entity_type: &str,
    abstract_text: &str,
    overview: Option<&str>,
    source: Option<&str>,
) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    let et: EntityType = entity_type.parse().map_err(|e: String| e)?;

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let entity = gm
            .add_entity(NewEntity {
                name: name.to_string(),
                entity_type: et,
                abstract_text: abstract_text.to_string(),
                overview: overview.map(String::from),
                content: None,
                attributes: None,
                source: source.map(String::from),
            })
            .await
            .map_err(|e| e.to_string())?;

        println!(
            "{GREEN}✓{RESET} Created entity: {BOLD}{}{RESET} ({}) [{}]",
            entity.name,
            entity.entity_type,
            entity.id_string()
        );
        Ok(())
    })
}

/// Create a relationship between two entities.
pub fn relate(
    memory_dir: &Path,
    from: &str,
    rel_type: &str,
    to: &str,
    description: Option<&str>,
    source: Option<&str>,
) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let rel = gm
            .add_relationship(NewRelationship {
                from_entity: from.to_string(),
                to_entity: to.to_string(),
                rel_type: rel_type.to_string(),
                description: description.map(String::from),
                confidence: None,
                source: source.map(String::from),
            })
            .await
            .map_err(|e| e.to_string())?;

        println!(
            "{GREEN}✓{RESET} {from} {CYAN}—[{rel_type}]→{RESET} {to} [{}]",
            rel.id_string()
        );
        Ok(())
    })
}

/// Semantic search across entities.
pub fn search(
    memory_dir: &Path,
    query: &str,
    limit: usize,
    entity_type: Option<&str>,
    keyword: Option<&str>,
) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let options = SearchOptions {
            limit,
            entity_type: entity_type.map(String::from),
            keyword: keyword.map(String::from),
        };

        let results = gm
            .search_with_options(query, &options)
            .await
            .map_err(|e| e.to_string())?;

        if results.is_empty() {
            println!("{YELLOW}No results.{RESET}");
            return Ok(());
        }

        for (i, r) in results.iter().enumerate() {
            println!(
                "{BOLD}{}. {}{RESET} ({}) — score: {:.3}",
                i + 1,
                r.entity.name,
                r.entity.entity_type,
                r.score
            );
            println!("   {DIM}{}{RESET}", r.entity.abstract_text);
        }
        Ok(())
    })
}

/// Ingest a single archive file into the graph (episodes only, no LLM extraction).
pub fn ingest(memory_dir: &Path, archive_path: &Path) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }

    let content = std::fs::read_to_string(archive_path)
        .map_err(|e| format!("Failed to read {}: {e}", archive_path.display()))?;

    // Extract session_id and log_number from frontmatter if available
    let (session_id, log_number) = extract_archive_metadata(&content, archive_path);

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let report = gm
            .ingest_archive(&content, &session_id, log_number, None)
            .await
            .map_err(|e| e.to_string())?;

        println!(
            "{GREEN}✓{RESET} Ingested {}: {} episodes created",
            archive_path.display(),
            report.episodes_created
        );
        if !report.errors.is_empty() {
            for err in &report.errors {
                println!("  {YELLOW}warning:{RESET} {err}");
            }
        }
        Ok(())
    })
}

/// Ingest all un-ingested archives in conversations/.
pub fn ingest_all(memory_dir: &Path) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }

    let conversations_dir = find_conversations_dir(memory_dir)?;

    // Collect all conversation files, sorted
    let mut files: Vec<_> = std::fs::read_dir(&conversations_dir)
        .map_err(|e| e.to_string())?
        .filter_map(|e| e.ok())
        .filter(|e| {
            let name = e.file_name().to_string_lossy().to_string();
            name.starts_with("conversation-") || name.starts_with("archive-log-")
        })
        .collect();
    files.sort_by_key(|e| e.file_name());

    if files.is_empty() {
        println!("{YELLOW}No conversation archives found.{RESET}");
        return Ok(());
    }

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let mut total_episodes = 0u32;
        let mut ingested = 0u32;
        let mut skipped = 0u32;

        for entry in &files {
            let path = entry.path();
            let content = std::fs::read_to_string(&path).map_err(|e| e.to_string())?;

            let (session_id, log_number) = extract_archive_metadata(&content, &path);

            // Check if already ingested (has episodes for this log_number)
            if let Some(ln) = log_number {
                if let Ok(Some(_)) = gm.get_episode_by_log_number(ln).await {
                    skipped += 1;
                    continue;
                }
            }

            let report = gm
                .ingest_archive(&content, &session_id, log_number, None)
                .await
                .map_err(|e| e.to_string())?;

            total_episodes += report.episodes_created;
            ingested += 1;

            println!(
                "  {GREEN}✓{RESET} {} — {} episodes",
                path.file_name().unwrap_or_default().to_string_lossy(),
                report.episodes_created
            );
        }

        println!(
            "\n{GREEN}✓{RESET} Ingested {ingested} archives ({total_episodes} episodes), skipped {skipped} already ingested"
        );
        Ok(())
    })
}

/// Extract session_id and log_number from a conversation archive's frontmatter.
fn extract_archive_metadata(content: &str, path: &Path) -> (String, Option<u32>) {
    let mut session_id = "unknown".to_string();
    let mut log_number: Option<u32> = None;

    // Try to extract log number from filename
    if let Some(name) = path.file_stem().and_then(|s| s.to_str()) {
        let num_str = name
            .strip_prefix("conversation-")
            .or_else(|| name.strip_prefix("archive-log-"));
        if let Some(num_str) = num_str {
            if let Ok(n) = num_str.parse::<u32>() {
                log_number = Some(n);
            }
        }
    }

    // Try to extract session_id from frontmatter
    if let Some(stripped) = content.strip_prefix("---") {
        if let Some(end) = stripped.find("---") {
            let frontmatter = &stripped[..end];
            for line in frontmatter.lines() {
                let line = line.trim();
                if let Some(val) = line.strip_prefix("session_id:") {
                    session_id = val.trim().trim_matches('"').to_string();
                }
            }
        }
    }

    (session_id, log_number)
}

/// Traverse the graph from an entity.
pub fn traverse(
    memory_dir: &Path,
    entity_name: &str,
    depth: u32,
    type_filter: Option<&str>,
) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let tree = gm
            .traverse_filtered(entity_name, depth, type_filter)
            .await
            .map_err(|e| e.to_string())?;

        let output = format_traversal(&tree, 0);
        print!("{output}");
        Ok(())
    })
}

/// Hybrid query: semantic + graph expansion + optional episodes.
pub fn hybrid_query(
    memory_dir: &Path,
    query: &str,
    limit: usize,
    entity_type: Option<&str>,
    keyword: Option<&str>,
    depth: u32,
    episodes: bool,
) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let options = QueryOptions {
            limit,
            entity_type: entity_type.map(String::from),
            keyword: keyword.map(String::from),
            graph_depth: depth,
            include_episodes: episodes,
        };

        let result = gm.query(query, &options).await.map_err(|e| e.to_string())?;

        if result.entities.is_empty() && result.episodes.is_empty() {
            println!("{YELLOW}No results.{RESET}");
            return Ok(());
        }

        if !result.entities.is_empty() {
            println!("{BOLD}Entities:{RESET}");
            for (i, r) in result.entities.iter().enumerate() {
                let source_tag = match &r.source {
                    MatchSource::Semantic => "semantic".to_string(),
                    MatchSource::Graph { parent, rel_type } => {
                        format!("graph: {parent} —[{rel_type}]")
                    }
                    MatchSource::Keyword => "keyword".to_string(),
                };
                println!(
                    "  {BOLD}{}. {}{RESET} ({}) — {:.3} [{DIM}{source_tag}{RESET}]",
                    i + 1,
                    r.entity.name,
                    r.entity.entity_type,
                    r.score
                );
                println!("     {DIM}{}{RESET}", r.entity.abstract_text);
            }
        }

        if !result.episodes.is_empty() {
            println!("\n{BOLD}Episodes:{RESET}");
            for (i, ep) in result.episodes.iter().enumerate() {
                let log = ep
                    .episode
                    .log_number
                    .map(|n| format!("#{n}"))
                    .unwrap_or_default();
                println!(
                    "  {BOLD}{}. {}{RESET} ({}) — {:.3}",
                    i + 1,
                    ep.episode.session_id,
                    log,
                    ep.score
                );
                println!("     {DIM}{}{RESET}", ep.episode.abstract_text);
            }
        }

        Ok(())
    })
}

/// Extract entities from already-ingested archives using an LLM.
#[cfg(feature = "llm")]
pub fn extract(
    memory_dir: &Path,
    log: Option<u32>,
    all: bool,
    dry_run: bool,
    model_override: Option<String>,
    provider_override: Option<String>,
    delay_ms: u64,
) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        // Determine which log numbers to process
        let log_numbers: Vec<u32> = if let Some(ln) = log {
            vec![ln]
        } else if all {
            gm.unextracted_log_numbers()
                .await
                .map_err(|e| e.to_string())?
                .into_iter()
                .map(|n| n as u32)
                .collect()
        } else {
            return Err("Specify --log <N> or --all".into());
        };

        if log_numbers.is_empty() {
            println!("{YELLOW}No unextracted archives found.{RESET}");
            return Ok(());
        }

        // Find conversations directory
        let conversations_dir = find_conversations_dir(memory_dir)?;

        if dry_run {
            println!(
                "{BOLD}Dry run — {}{RESET} archives to extract",
                log_numbers.len()
            );
            for ln in &log_numbers {
                let path = find_archive_file(&conversations_dir, *ln);
                let label = match &path {
                    Ok(p) => {
                        p.file_name()
                            .unwrap_or_default()
                            .to_string_lossy()
                            .to_string()
                    }
                    Err(_) => format!("log {ln:03} (file not found)"),
                };
                println!("  {label}");
            }
            return Ok(());
        }

        // Build LLM provider from .recall-echo.toml (CLI flags override)
        let (llm, model_name) = crate::llm_provider::create_provider(
            memory_dir,
            provider_override.as_deref(),
            model_override.as_deref(),
        )?;

        println!(
            "{BOLD}Extracting entities from {} archives using {model_name}{RESET}",
            log_numbers.len(),
        );

        let mut total_entities_created = 0u32;
        let mut total_entities_merged = 0u32;
        let mut total_entities_skipped = 0u32;
        let mut total_relationships = 0u32;
        let mut total_errors = Vec::new();
        let mut processed = 0u32;

        for ln in &log_numbers {
            let archive_path = match find_archive_file(&conversations_dir, *ln) {
                Ok(p) => p,
                Err(e) => {
                    println!("  {YELLOW}⚠{RESET} log {ln:03}: {e}");
                    total_errors.push(format!("log {ln:03}: {e}"));
                    continue;
                }
            };

            let content = std::fs::read_to_string(&archive_path)
                .map_err(|e| format!("read {}: {e}", archive_path.display()))?;

            let (session_id, _) = extract_archive_metadata(&content, &archive_path);

            let report = gm
                .extract_from_archive(&content, &session_id, Some(*ln), &*llm)
                .await
                .map_err(|e| format!("extraction log {ln:03}: {e}"))?;

            println!(
                "  {GREEN}✓{RESET} log {ln:03}: +{} entities, ~{} merged, -{} skipped, {} rels",
                report.entities_created,
                report.entities_merged,
                report.entities_skipped,
                report.relationships_created,
            );

            gm.mark_extracted(*ln).await.map_err(|e| e.to_string())?;

            total_entities_created += report.entities_created;
            total_entities_merged += report.entities_merged;
            total_entities_skipped += report.entities_skipped;
            total_relationships += report.relationships_created;
            total_errors.extend(report.errors);
            processed += 1;

            // Rate limiting between archives
            if delay_ms > 0 && *ln != *log_numbers.last().unwrap() {
                tokio::time::sleep(std::time::Duration::from_millis(delay_ms)).await;
            }
        }

        println!(
            "\n{GREEN}✓{RESET} Done: {processed} archives — +{total_entities_created} created, ~{total_entities_merged} merged, -{total_entities_skipped} skipped, {total_relationships} relationships"
        );

        if !total_errors.is_empty() {
            println!("\n{YELLOW}Warnings ({}):{RESET}", total_errors.len());
            for err in total_errors.iter().take(10) {
                println!("  {DIM}{err}{RESET}");
            }
            if total_errors.len() > 10 {
                println!("  {DIM}... and {} more{RESET}", total_errors.len() - 10);
            }
        }

        Ok(())
    })
}

// ── Vigil sync commands ──────────────────────────────────────────────

/// Sync vigil-pulse signals and outcomes into the graph.
pub fn vigil_sync(
    memory_dir: &Path,
    signals_path: Option<&Path>,
    outcomes_path: Option<&Path>,
) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }

    // Default paths: look for vigil/ and caliber/ relative to memory_dir's parent (entity root)
    let entity_root = memory_dir.parent().unwrap_or(memory_dir);

    let default_signals = entity_root.join("vigil").join("signals.json");
    let default_outcomes = entity_root.join("caliber").join("outcomes.json");

    let sig_path = signals_path.unwrap_or(&default_signals);
    let out_path = outcomes_path.unwrap_or(&default_outcomes);

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let report = gm
            .sync_vigil(sig_path, out_path)
            .await
            .map_err(|e| e.to_string())?;

        println!("{BOLD}Vigil Sync{RESET}");
        println!("  Measurements: +{}", report.measurements_created);
        println!("  Outcomes:     +{}", report.outcomes_created);
        println!("  Relationships: +{}", report.relationships_created);
        println!("  Skipped:       {}", report.skipped);

        if !report.errors.is_empty() {
            println!("\n  {YELLOW}Warnings:{RESET}");
            for err in &report.errors {
                println!("    {DIM}{err}{RESET}");
            }
        }

        if report.measurements_created == 0 && report.outcomes_created == 0 {
            println!("\n  {DIM}No new data — graph is in sync.{RESET}");
        }

        Ok(())
    })
}

// ── Pipeline commands ──────────────────────────────────────────────────

/// Sync pipeline documents into the graph.
pub fn pipeline_sync(memory_dir: &Path, docs_dir_override: Option<&Path>) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }

    // Resolve docs directory: CLI flag > config > error
    let docs_dir = if let Some(d) = docs_dir_override {
        d.to_path_buf()
    } else {
        let cfg = crate::config::load_from_dir(memory_dir);
        match cfg.pipeline.and_then(|p| p.docs_dir) {
            Some(d) => {
                let path = PathBuf::from(shellexpand(&d));
                if !path.exists() {
                    return Err(format!(
                        "Configured docs_dir does not exist: {}",
                        path.display()
                    ));
                }
                path
            }
            None => {
                return Err(
                    "No docs directory specified. Use --docs-dir or set [pipeline] docs_dir in config.".into(),
                );
            }
        }
    };

    // Read pipeline documents
    let docs = read_pipeline_docs(&docs_dir)?;

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let report = gm.sync_pipeline(&docs).await.map_err(|e| e.to_string())?;

        println!("{BOLD}Pipeline Sync{RESET}");
        println!("  Created:      {}", report.entities_created);
        println!("  Updated:      {}", report.entities_updated);
        println!("  Archived:     {}", report.entities_archived);
        println!(
            "  Relationships: +{} / ~{} skipped",
            report.relationships_created, report.relationships_skipped
        );

        if !report.errors.is_empty() {
            println!("\n  {YELLOW}Warnings:{RESET}");
            for err in &report.errors {
                println!("    {DIM}{err}{RESET}");
            }
        }

        if report.entities_created == 0
            && report.entities_updated == 0
            && report.entities_archived == 0
        {
            println!("\n  {DIM}No changes — graph is in sync.{RESET}");
        }

        Ok(())
    })
}

/// Show pipeline health stats.
pub fn pipeline_status(memory_dir: &Path, staleness_days: u32) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let stats = gm
            .pipeline_stats(staleness_days)
            .await
            .map_err(|e| e.to_string())?;

        println!(
            "{BOLD}Pipeline Status{RESET} ({} entities)",
            stats.total_entities
        );

        if stats.by_stage.is_empty() {
            println!(
                "  {DIM}No pipeline entities in graph. Run `graph pipeline sync` first.{RESET}"
            );
            return Ok(());
        }

        // Display stages in pipeline order
        let stage_order = ["learning", "thoughts", "curiosity", "reflections", "praxis"];
        for stage in &stage_order {
            if let Some(statuses) = stats.by_stage.get(*stage) {
                println!("\n  {CYAN}{}{RESET}", stage.to_uppercase());
                let mut items: Vec<_> = statuses.iter().collect();
                items.sort_by_key(|(s, _)| (*s).clone());
                for (status, count) in items {
                    println!("    {status}: {count}");
                }
            }
        }

        if !stats.stale_thoughts.is_empty() {
            println!("\n  {YELLOW}Stale thoughts (>{staleness_days}d):{RESET}");
            for entity in &stats.stale_thoughts {
                println!("    {DIM}•{RESET} {}", entity.name);
            }
        }

        if !stats.stale_questions.is_empty() {
            println!(
                "\n  {YELLOW}Stale questions (>{}d):{RESET}",
                staleness_days * 2
            );
            for entity in &stats.stale_questions {
                println!("    {DIM}•{RESET} {}", entity.name);
            }
        }

        if let Some(ref last) = stats.last_movement {
            println!("\n  {DIM}Last movement: {last}{RESET}");
        }

        Ok(())
    })
}

/// Trace pipeline flow for an entity.
pub fn pipeline_flow(memory_dir: &Path, entity_name: &str) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let chain = gm
            .pipeline_flow(entity_name)
            .await
            .map_err(|e| e.to_string())?;

        if chain.is_empty() {
            println!("{YELLOW}No pipeline relationships found for \"{entity_name}\".{RESET}");
            return Ok(());
        }

        println!("{BOLD}Pipeline Flow: {entity_name}{RESET}\n");
        for (source, rel_type, target) in &chain {
            println!(
                "  {} ({}) {CYAN}—[{rel_type}]→{RESET} {} ({})",
                source.name, source.entity_type, target.name, target.entity_type
            );
        }

        Ok(())
    })
}

/// List stale pipeline entities.
pub fn pipeline_stale(memory_dir: &Path, staleness_days: u32) -> Result<(), String> {
    let graph_dir = memory_dir.join("graph");
    if !graph_dir.exists() {
        return Err("Graph store not initialized. Run `recall-echo graph init` first.".into());
    }

    let rt = tokio::runtime::Runtime::new().map_err(|e| e.to_string())?;
    rt.block_on(async {
        let gm = GraphMemory::open(&graph_dir)
            .await
            .map_err(|e| e.to_string())?;

        let stats = gm
            .pipeline_stats(staleness_days)
            .await
            .map_err(|e| e.to_string())?;

        let total_stale = stats.stale_thoughts.len() + stats.stale_questions.len();
        if total_stale == 0 {
            println!("{GREEN}✓{RESET} No stale pipeline entities.");
            return Ok(());
        }

        println!("{BOLD}Stale Pipeline Entities{RESET}\n");

        if !stats.stale_thoughts.is_empty() {
            println!("  {YELLOW}Thoughts (>{staleness_days} days):{RESET}");
            for entity in &stats.stale_thoughts {
                println!("    • {} {DIM}({}){RESET}", entity.name, entity.entity_type);
            }
        }

        if !stats.stale_questions.is_empty() {
            println!("  {YELLOW}Questions (>{} days):{RESET}", staleness_days * 2);
            for entity in &stats.stale_questions {
                println!("    • {} {DIM}({}){RESET}", entity.name, entity.entity_type);
            }
        }

        Ok(())
    })
}

/// Read pipeline documents from a directory.
fn read_pipeline_docs(dir: &Path) -> Result<PipelineDocuments, String> {
    let read_or_empty = |name: &str| -> String {
        let path = dir.join(name);
        std::fs::read_to_string(&path).unwrap_or_default()
    };

    Ok(PipelineDocuments {
        learning: read_or_empty("LEARNING.md"),
        thoughts: read_or_empty("THOUGHTS.md"),
        curiosity: read_or_empty("CURIOSITY.md"),
        reflections: read_or_empty("REFLECTIONS.md"),
        praxis: read_or_empty("PRAXIS.md"),
    })
}

/// Expand ~ to home directory in paths.
fn shellexpand(path: &str) -> String {
    if let Some(rest) = path.strip_prefix("~/") {
        if let Some(home) = dirs::home_dir() {
            return home.join(rest).to_string_lossy().to_string();
        }
    }
    path.to_string()
}

/// Find the conversations directory — checks memory_dir/conversations/ then parent/conversations/.
fn find_conversations_dir(memory_dir: &Path) -> Result<PathBuf, String> {
    let conv = memory_dir.join("conversations");
    if conv.exists() {
        return Ok(conv);
    }
    if let Some(parent) = memory_dir.parent() {
        let parent_conv = parent.join("conversations");
        if parent_conv.exists() {
            return Ok(parent_conv);
        }
    }
    Err("conversations/ directory not found".into())
}

/// Find the archive file for a given log number.
#[cfg(feature = "llm")]
fn find_archive_file(conversations_dir: &Path, log_number: u32) -> Result<PathBuf, String> {
    // Try both naming conventions
    let patterns = [
        format!("conversation-{log_number:03}.md"),
        format!("conversation-{log_number}.md"),
        format!("archive-log-{log_number:03}.md"),
        format!("archive-log-{log_number}.md"),
    ];

    for name in &patterns {
        let path = conversations_dir.join(name);
        if path.exists() {
            return Ok(path);
        }
    }

    Err(format!("no archive file for log {log_number:03}"))
}