use crate::{
cli::common::write_text_or_json,
sns::commands::{
SnsCommandError,
options::{SnsNeuronsCacheListOptions, SnsNeuronsCacheStatusOptions},
run::common::{cache_command_parts, command_args, parse_required_command},
spec::{
sns_neuron_cache_command, sns_neuron_cache_list_usage, sns_neuron_cache_status_usage,
sns_neuron_cache_usage,
},
},
};
use ic_query::sns::{
SnsCacheListRequest, SnsCacheStatusRequest, build_sns_neurons_cache_list_report,
build_sns_neurons_cache_status_report, sns_neurons_cache_list_report_text,
sns_neurons_cache_status_report_text,
};
use std::ffi::OsString;
pub(super) fn run_sns_neuron_cache<I>(args: I) -> Result<(), SnsCommandError>
where
I: IntoIterator<Item = OsString>,
{
let Some(args) = command_args(args, sns_neuron_cache_usage) else {
return Ok(());
};
let (command, args) =
parse_required_command(sns_neuron_cache_command(), args, sns_neuron_cache_usage)?;
match command.as_str() {
"list" => run_sns_neuron_cache_list(args),
"status" => run_sns_neuron_cache_status(args),
_ => unreachable!("sns neuron cache dispatch command only defines known commands"),
}
}
fn run_sns_neuron_cache_list<I>(args: I) -> Result<(), SnsCommandError>
where
I: IntoIterator<Item = OsString>,
{
let Some(args) = command_args(args, sns_neuron_cache_list_usage) else {
return Ok(());
};
let options = SnsNeuronsCacheListOptions::parse(args)?;
let parts = cache_command_parts(options.format, options.network)?;
let request = SnsCacheListRequest {
network: parts.network,
cache_root: parts.cache_root,
};
let report = build_sns_neurons_cache_list_report(&request)?;
write_text_or_json(parts.format, &report, sns_neurons_cache_list_report_text)
}
fn run_sns_neuron_cache_status<I>(args: I) -> Result<(), SnsCommandError>
where
I: IntoIterator<Item = OsString>,
{
let Some(args) = command_args(args, sns_neuron_cache_status_usage) else {
return Ok(());
};
let options = SnsNeuronsCacheStatusOptions::parse(args)?;
let parts = cache_command_parts(options.format, options.network)?;
let request = SnsCacheStatusRequest {
network: parts.network,
cache_root: parts.cache_root,
input: options.input,
};
let report = build_sns_neurons_cache_status_report(&request)?;
write_text_or_json(parts.format, &report, sns_neurons_cache_status_report_text)
}