use super::{output_mode, should_spin};
use crate::cli::{Cli, MetaArgs, parse_where_clause};
use crate::config::{self, RefreshLockPolicy, 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::{
MetaInput, PositiveCount, ResponseMeta, ScopeFilter, TalonEnvelope, TalonResponseData,
open_database, open_database_read_only, query_meta,
};
pub(super) async fn emit(args: &MetaArgs, cli: &Cli) -> Result<()> {
let where_clauses: Vec<talon_core::WhereClause> = args
.where_
.iter()
.map(|s| parse_where_clause(s).map_err(|e| eyre::eyre!("invalid --where: {s}: {e}")))
.collect::<Result<Vec<_>>>()?;
let limit_default = if args.tag_counts {
u16::MAX
} else {
talon_core::constants::DEFAULT_LIMIT
};
let input = MetaInput {
where_: where_clauses,
since: args.since.clone(),
scope: args.scope.scope.clone(),
scope_only: args.scope.scope_only.clone(),
scope_all: args.scope.scope_all,
select: args.select.clone(),
tag_counts: args.tag_counts,
sources: args.sources.clone(),
limit: PositiveCount::new(args.limit.unwrap_or(limit_default), "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 since_str = input.since.clone();
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 fast = cli.fast;
let started = Instant::now();
let work = async move {
let mut conn = if fast {
open_database_read_only(&db_path)
} else {
open_database(&db_path)
}
.wrap_err_with(|| format!("opening index at {}", db_path.display()))?;
crate::config::refresh_index_if_needed(
&config,
&mut conn,
fast,
RefreshLockPolicy::ErrorIfBusy,
)?;
let mut response = query_meta(&conn, &input, Some(&config));
response.vault = vault;
let result_count = count_u32(response.entries.len());
let meta = ResponseMeta {
duration_ms: elapsed_ms(started),
result_count: Some(result_count),
warnings: Vec::new(),
scope_set,
since: since_str,
};
let data = TalonResponseData::Meta(response);
Ok::<TalonEnvelope, eyre::Report>(TalonEnvelope::ok("meta", data, meta))
};
let response = if should_spin(cli) {
spinner::with_spinner("Querying frontmatter...".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))
}