talon-cli 0.4.2

use super::{output_mode, should_spin};
use crate::cli::{Cli, InspectArgs, InspectCheck};
use crate::config::{self, vault_container_path};
use crate::output::emit_response;
use crate::spinner;
use crate::telemetry::{count_u32, elapsed_ms};
use eyre::{Result, WrapErr as _};
use std::path::PathBuf;
use std::time::Instant;
use talon_core::{
    InspectInput, ResponseMeta, ScopeFilter, SyncLockError, TalonEnvelope, TalonResponseData,
    acquire_sync_lock, open_database, open_database_read_only, query_inspect,
    vec_ext::register_sqlite_vec,
};

pub(super) async fn emit(args: &InspectArgs, cli: &Cli) -> Result<()> {
    let check = args.check.unwrap_or(InspectCheck::All).into();

    let input = InspectInput {
        check,
        scope: args.scope.scope.clone(),
        scope_only: args.scope.scope_only.clone(),
        scope_all: args.scope.scope_all,
        limit: args.limit,
    };

    let config = config::load_config(cli.config_file.as_deref())?;
    let db_path: PathBuf = config.db_path.clone();
    let vault = vault_container_path(Some(&config));
    let scope_set = Some(
        ScopeFilter::from_args(&config, &input.scope, &input.scope_only, input.scope_all)
            .map_err(|e| eyre::eyre!("{e}"))?
            .resolved_set(),
    );

    let started = Instant::now();
    let work = async move {
        tokio::task::spawn_blocking(move || {
            register_sqlite_vec().wrap_err("registering sqlite-vec extension")?;

            // Try to acquire the sync lock for a fresh index. If busy (another
            // sync is running), fall back to read-only + fast mode — same as
            // search does.
            let lock_path = crate::config::sync_lock_path(&config);
            let conn = match acquire_sync_lock(&lock_path) {
                Ok(lock) => {
                    let mut conn = open_database(&db_path)
                        .wrap_err_with(|| format!("opening index at {}", db_path.display()))?;
                    crate::config::refresh_index_with_lock(&config, &mut conn, lock)?;
                    conn
                }
                Err(SyncLockError::Busy) => open_database_read_only(&db_path)
                    .wrap_err_with(|| format!("opening index at {}", db_path.display()))?,
                Err(SyncLockError::Io(err)) => {
                    return Err(err).wrap_err("acquiring sync lock for inspect");
                }
                Err(err) => return Err(eyre::eyre!("acquiring sync lock for inspect: {err}")),
            };

            let mut response = query_inspect(&conn, &input, Some(&config));
            response.vault = vault;
            let result_count = count_u32(response.findings.len());
            let meta = ResponseMeta {
                duration_ms: elapsed_ms(started),
                result_count: Some(result_count),
                warnings: Vec::new(),
                scope_set,
                since: None,
            };
            let data = TalonResponseData::Inspect(response);
            Ok::<TalonEnvelope, eyre::Report>(TalonEnvelope::ok("inspect", data, meta))
        })
        .await
        .wrap_err("inspect task join failed")?
        .wrap_err("inspect failed")
    };
    let response = if should_spin(cli) {
        spinner::with_spinner("Inspecting vault...".to_string(), work).await?
    } else {
        work.await?
    };
    if crate::banner::should_clear_fancy_prelude(cli) {
        crate::banner::clear_fancy_prelude();
    }
    emit_response(&response, output_mode(cli))
}