ic-query 0.2.17

Internet Computer query CLI for NNS, SNS, and related public network metadata
Documentation
//! Module: sns::commands::run::common
//!
//! Responsibility: provide shared runtime helpers for SNS command execution.
//! Does not own: command specs, report building, or cache policy.
//! Boundary: adapts CLI-level helpers into SNS command errors and request parts.

use crate::{
    cli::{
        clap::{
            OptionalSubcommand, parse_optional_subcommand_or_usage,
            parse_required_subcommand_or_usage,
        },
        common::{OutputFormat, current_unix_secs},
        help::collect_args_or_print_help_or_version,
    },
    project::icp_root,
    sns::commands::{SnsCommandError, options::SnsLookupOptions},
    version_text,
};
use clap::Command as ClapCommand;
use std::ffi::OsString;
use std::path::PathBuf;

pub(super) struct SnsLookupCommandParts {
    pub(super) format: OutputFormat,
    pub(super) network: String,
    pub(super) source_endpoint: String,
    pub(super) now_unix_secs: u64,
    pub(super) input: String,
}

pub(in crate::sns::commands::run) struct SnsCachedLookupCommandParts {
    pub(in crate::sns::commands::run) format: OutputFormat,
    pub(in crate::sns::commands::run) network: String,
    pub(in crate::sns::commands::run) source_endpoint: String,
    pub(in crate::sns::commands::run) now_unix_secs: u64,
    pub(in crate::sns::commands::run) input: String,
    pub(in crate::sns::commands::run) icp_root: PathBuf,
}

pub(in crate::sns::commands::run) struct SnsCacheCommandParts {
    pub(in crate::sns::commands::run) format: OutputFormat,
    pub(in crate::sns::commands::run) network: String,
    pub(in crate::sns::commands::run) icp_root: PathBuf,
}

pub(super) fn command_unix_secs() -> Result<u64, SnsCommandError> {
    current_unix_secs().map_err(SnsCommandError::Clock)
}

pub(super) fn command_args<I>(args: I, usage: impl FnOnce() -> String) -> Option<Vec<OsString>>
where
    I: IntoIterator<Item = OsString>,
{
    collect_args_or_print_help_or_version(args, usage, version_text())
}

pub(super) fn command_icp_root() -> Result<PathBuf, SnsCommandError> {
    icp_root().map_err(|err| SnsCommandError::Usage(err.to_string()))
}

pub(super) fn parse_required_command<I>(
    command: ClapCommand,
    args: I,
    usage: impl FnOnce() -> String,
) -> Result<(String, Vec<OsString>), SnsCommandError>
where
    I: IntoIterator<Item = OsString>,
{
    parse_required_subcommand_or_usage(command, args, usage).map_err(SnsCommandError::Usage)
}

pub(super) fn parse_optional_command<I>(
    command: ClapCommand,
    args: I,
    usage: impl FnOnce() -> String,
) -> Result<OptionalSubcommand, SnsCommandError>
where
    I: IntoIterator<Item = OsString>,
{
    parse_optional_subcommand_or_usage(command, args, usage).map_err(SnsCommandError::Usage)
}

pub(super) fn lookup_command_parts(
    options: SnsLookupOptions,
) -> Result<SnsLookupCommandParts, SnsCommandError> {
    Ok(SnsLookupCommandParts {
        format: options.format,
        network: options.network,
        source_endpoint: options.source_endpoint,
        now_unix_secs: command_unix_secs()?,
        input: options.input,
    })
}

pub(in crate::sns::commands::run) fn cached_lookup_command_parts(
    options: SnsLookupOptions,
) -> Result<SnsCachedLookupCommandParts, SnsCommandError> {
    let parts = lookup_command_parts(options)?;
    Ok(SnsCachedLookupCommandParts {
        format: parts.format,
        network: parts.network,
        source_endpoint: parts.source_endpoint,
        now_unix_secs: parts.now_unix_secs,
        input: parts.input,
        icp_root: command_icp_root()?,
    })
}

pub(in crate::sns::commands::run) fn cache_command_parts(
    format: OutputFormat,
    network: String,
) -> Result<SnsCacheCommandParts, SnsCommandError> {
    Ok(SnsCacheCommandParts {
        format,
        network,
        icp_root: command_icp_root()?,
    })
}