manta-cli 2.0.0-beta.63

//! CLI context struct threaded through `manta-cli`'s call stack.
//!
//! Every dispatch handler in `crate::dispatch::*` receives an
//! `&AppContext` as its first parameter. It carries: the resolved
//! site name, the manta-server URL the CLI is bound to, optional
//! poll-tuning knobs from `cli.toml`, and the raw `config::Config`
//! handle for fields the typed struct doesn't surface.
//!
//! Authentication is *not* on the context — call
//! [`crate::common::authentication::get_api_token`] when a handler
//! needs a token, then thread the token into [`crate::http_client::MantaClient::from_app_ctx`].
//!
//! The server's analogous `InfraContext` (with backend dispatcher and
//! per-site URLs) lives in `manta_server::server::common::app_context`
//! — it depends on `StaticBackendDispatcher`, which the CLI never
//! touches.

use config::Config;

use crate::common::session::SessionContext;

/// Top-level CLI context, passed as `&AppContext` through CLI
/// handlers and commands.
///
/// # Example
///
/// Typical handler shape: resolve a token, build a client, call an
/// OpenAPI method.
///
/// ```ignore
/// use crate::common::app_context::AppContext;
/// use crate::common::authentication::get_api_token;
/// use crate::http_client::{MantaClient, OpenApiResultExt};
///
/// async fn list_groups(ctx: &AppContext<'_>) -> anyhow::Result<()> {
///   let token = get_api_token(ctx).await?;
///   let client = MantaClient::from_app_ctx(ctx, Some(&token))?;
///   let groups = client
///     .openapi
///     .get_groups(None, client.site_name())
///     .await
///     .into_anyhow()?;
///   println!("{groups:#?}");
///   Ok(())
/// }
/// ```
#[derive(Debug)]
pub struct AppContext<'a> {
  /// Resolved site (`--site` override, else `cli.toml`'s `site`), or
  /// `None` when neither was supplied. It's just the `X-Manta-Site`
  /// header on outbound `MantaClient` requests; commands that reach
  /// the server obtain it via [`AppContext::require_site`]. The
  /// purely-local `config set`/`config unset` commands never call
  /// that, so they work with no site configured.
  pub site_name: Option<&'a str>,
  /// URL of the manta HTTP server this CLI talks to. Required.
  pub manta_server_url: &'a str,
  /// Optional default group name from `cli.toml`'s
  /// `parent_group`; threaded into the typed `*Params`'
  /// `settings_group_name` field by every command that builds one.
  pub settings_group_name_opt: Option<&'a str>,
  /// Optional per-request HTTP timeout (seconds) for outbound
  /// `MantaClient` calls — read from `cli.toml`'s
  /// `request_timeout_secs`. Threaded into every
  /// [`crate::http_client::MantaClient::from_app_ctx`] call. `None`
  /// uses the per-client defaults set in
  /// [`crate::http_client::MantaClient::new_with_timeout`].
  pub request_timeout_secs: Option<u64>,
  /// Override (seconds) for the `manta power` poll interval. `None`
  /// keeps the dispatcher's compiled default.
  pub power_poll_interval_secs: Option<u64>,
  /// Override for the `manta power` max poll attempts. `None` keeps
  /// the dispatcher's compiled default.
  pub power_max_poll_attempts: Option<u32>,
  /// Override (seconds) for `manta apply sat-file`'s session poll
  /// interval. `None` keeps the dispatcher's compiled default.
  pub sat_file_poll_interval_secs: Option<u64>,
  /// Override (seconds) for the SAT-file monitor loop's hard cap.
  /// `None` keeps the dispatcher's compiled default.
  pub sat_file_poll_budget_secs: Option<u64>,
  /// Override (seconds) for the SAT-file "session not yet visible"
  /// cap. `None` keeps the dispatcher's compiled default.
  pub sat_file_not_visible_budget_secs: Option<u64>,
  /// CLI-local read-only flag from `cli.toml`'s `read_only`. Fed into
  /// [`crate::common::read_only::read_only_gate`] at the top of
  /// [`crate::dispatch::process::process_cli`] to refuse
  /// backend-mutating verbs before any HTTP request leaves the
  /// process. `false` by default.
  pub read_only: bool,
  /// Raw loaded `cli.toml` settings; held alongside the parsed
  /// `CliConfiguration` so handlers can read fields (e.g. `log`)
  /// that don't live on the typed struct.
  pub settings: &'a Config,
  /// Bearer token resolved at the top of
  /// [`crate::dispatch::process::process_cli`] for every command that
  /// authenticates. Handlers that call
  /// [`crate::common::authentication::get_api_token`] read this first
  /// via the short-circuit added in step 4 — that's how we avoid the
  /// second auth cascade per command. `None` for purely-local verbs.
  pub token: Option<String>,
  /// JWT-derived per-invocation facts, populated at the top of
  /// [`crate::dispatch::process::process_cli`] for every command that
  /// authenticates AND passes the read-only gate. `None` for
  /// purely-local verbs (`gen-autocomplete`, `gen-man`, `upgrade`,
  /// `config set site/log`, `config unset hsm/auth`) and for any
  /// command that was refused by the gate before the session's HTTP
  /// fetch ran.
  pub session: Option<SessionContext>,
}

impl<'a> AppContext<'a> {
  /// The site this invocation targets, or a user-facing error when
  /// neither `--site` nor `cli.toml`'s `site` was supplied. Call this
  /// at every point that needs to issue a request to `manta-server`;
  /// commands that touch only the local config file must not.
  ///
  /// # Errors
  ///
  /// Returns `Err` when `site_name` is `None` — the message names
  /// both ways to set it (`--site <name>` or the `site` key in
  /// `cli.toml`).
  pub fn require_site(&self) -> anyhow::Result<&'a str> {
    self.site_name.ok_or_else(|| {
      anyhow::anyhow!(
        "No site selected. Pass --site <name> or set `site` in cli.toml"
      )
    })
  }
}

#[cfg(test)]
mod tests {
  use super::*;

  fn ctx_with_site(site: Option<&'static str>) -> AppContext<'static> {
    // `settings` is borrowed, so leak a default Config for the test's
    // lifetime — cheap and avoids threading a real cli.toml through.
    let settings: &'static Config = Box::leak(Box::new(Config::default()));
    AppContext {
      site_name: site,
      manta_server_url: "https://example:8443",
      settings_group_name_opt: None,
      request_timeout_secs: None,
      power_poll_interval_secs: None,
      power_max_poll_attempts: None,
      sat_file_poll_interval_secs: None,
      sat_file_poll_budget_secs: None,
      sat_file_not_visible_budget_secs: None,
      read_only: false,
      settings,
      token: None,
      session: None,
    }
  }

  #[test]
  fn require_site_returns_the_name_when_set() {
    let ctx = ctx_with_site(Some("alps"));
    assert_eq!(ctx.require_site().unwrap(), "alps");
  }

  #[test]
  fn require_site_errors_when_unset() {
    let ctx = ctx_with_site(None);
    let err = ctx.require_site().unwrap_err().to_string();
    assert!(err.contains("No site selected"), "got: {err}");
  }
}