use crate::cli::QueryAction;
use crate::config::Config;
use crate::error::{BzrError, Result};
use crate::output;
use crate::types::{OutputFormat, QueryKind, SavedQuery};
pub async fn execute(
action: &QueryAction,
server: Option<&str>,
format: OutputFormat,
api: Option<crate::types::ApiMode>,
) -> Result<()> {
match action {
QueryAction::Save { .. } => handle_save(action, format),
QueryAction::List => handle_list(format),
QueryAction::Show { .. } => handle_show(action, format),
QueryAction::Delete { .. } => handle_delete(action, format),
QueryAction::Run { .. } => handle_run(action, server, format, api).await,
}
}
fn handle_save(action: &QueryAction, format: OutputFormat) -> Result<()> {
let QueryAction::Save {
name,
from_url,
search,
product,
component,
status,
assignee,
creator,
priority,
severity,
limit,
fields,
exclude_fields,
} = action
else {
unreachable!()
};
let (query, preloaded_config) = if let Some(url_str) = from_url {
let config = Config::load()?;
let parsed = crate::url_parser::parse_bugzilla_url(url_str, &config)?;
let mut query = parsed.query;
if let Some(limit) = limit {
query.limit = Some(*limit);
}
if let Some(f) = fields {
query.fields = Some(f.clone());
}
if let Some(ef) = exclude_fields {
query.exclude_fields = Some(ef.clone());
}
(query, Some(config))
} else {
let kind = if search.is_some() {
QueryKind::Search
} else {
QueryKind::List
};
let query = SavedQuery {
kind,
product: product.clone(),
component: component.clone(),
status: status.clone(),
assignee: assignee.clone(),
creator: creator.clone(),
priority: priority.clone(),
severity: severity.clone(),
quicksearch: search.clone(),
limit: *limit,
fields: fields.clone(),
exclude_fields: exclude_fields.clone(),
..SavedQuery::default()
};
(query, None)
};
if !query.has_filters() {
return Err(BzrError::InputValidation(
"query must have at least one filter set".into(),
));
}
let mut config = match preloaded_config {
Some(c) => c,
None => Config::load()?,
};
let is_update = config.queries.contains_key(name.as_str());
config.queries.insert(name.clone(), query);
config.save()?;
let verb = if is_update { "Updated" } else { "Saved" };
output::print_query_saved(name, verb, format);
Ok(())
}
fn handle_list(format: OutputFormat) -> Result<()> {
let config = Config::load()?;
output::print_query_list(&config.queries, format);
Ok(())
}
fn handle_show(action: &QueryAction, format: OutputFormat) -> Result<()> {
let QueryAction::Show { name } = action else {
unreachable!()
};
let config = Config::load()?;
let query = config
.queries
.get(name.as_str())
.ok_or_else(|| BzrError::config(format!("query '{name}' not found")))?;
output::print_query_detail(name, query, format);
Ok(())
}
fn handle_delete(action: &QueryAction, format: OutputFormat) -> Result<()> {
let QueryAction::Delete { name } = action else {
unreachable!()
};
let mut config = Config::load()?;
if config.queries.remove(name.as_str()).is_none() {
return Err(BzrError::config(format!("query '{name}' not found")));
}
config.save()?;
output::print_query_saved(name, "Deleted", format);
Ok(())
}
async fn handle_run(
action: &QueryAction,
server: Option<&str>,
format: OutputFormat,
api: Option<crate::types::ApiMode>,
) -> Result<()> {
let QueryAction::Run {
name,
limit,
fields,
exclude_fields,
server: server_override,
} = action
else {
unreachable!()
};
let config = Config::load()?;
let saved = config
.queries
.get(name.as_str())
.ok_or_else(|| BzrError::config(format!("query '{name}' not found")))?;
let effective_server = server
.or(server_override.as_deref())
.or(saved.server.as_deref());
let mut params = saved.to_search_params();
params.apply_overrides(*limit, fields.as_deref(), exclude_fields.as_deref());
let client = super::shared::connect_and_configure(effective_server, api).await?;
let bugs = client.search_bugs(¶ms).await?;
output::print_bugs(&bugs, format);
Ok(())
}
#[cfg(test)]
#[path = "query_tests.rs"]
mod tests;