ockam_command 0.104.0

use clap::Args;

use colorful::Colorful;
use miette::IntoDiagnostic;
use ockam::Context;
use ockam_api::InfluxDbTokenLease;
use time::format_description::well_known::Iso8601;
use time::PrimitiveDateTime;
use tokio::sync::Mutex;
use tokio::try_join;

use crate::identity::initialize_identity_if_default;
use crate::lease::authenticate;
use crate::terminal::OckamColor;
use crate::util::api::{CloudOpts, TrustContextOpts};
use crate::util::node_rpc;
use crate::{docs, CommandGlobalOpts};
use crate::{fmt_log, fmt_ok};

const HELP_DETAIL: &str = "";

/// Create a token within the lease token manager
#[derive(Clone, Debug, Args)]
#[command(help_template = docs::after_help(HELP_DETAIL))]
pub struct CreateCommand {}

impl CreateCommand {
    pub fn run(self, opts: CommandGlobalOpts, cloud_opts: CloudOpts, trust_opts: TrustContextOpts) {
        initialize_identity_if_default(&opts, &cloud_opts.identity);
        node_rpc(run_impl, (opts, cloud_opts, trust_opts));
    }
}

async fn run_impl(
    ctx: Context,
    (opts, cloud_opts, trust_opts): (CommandGlobalOpts, CloudOpts, TrustContextOpts),
) -> miette::Result<()> {
    opts.terminal
        .write_line(&fmt_log!("Creating influxdb token...\n"))?;

    let project_node = authenticate(&ctx, &opts, &cloud_opts, &trust_opts).await?;
    let is_finished: Mutex<bool> = Mutex::new(false);

    let send_req = async {
        let token = project_node.create_token(&ctx).await?;
        *is_finished.lock().await = true;
        Ok(token)
    };

    let output_messages = vec!["Creating influxdb token...".to_string()];

    let progress_output = opts
        .terminal
        .progress_output(&output_messages, &is_finished);

    let (resp_token, _) = try_join!(send_req, progress_output)?;

    opts.terminal
        .stdout()
        .machine(resp_token.token.to_string())
        .json(serde_json::to_string_pretty(&resp_token).into_diagnostic()?)
        .plain(
            fmt_ok!("Created influxdb token\n")
                + &fmt_log!(
                    "{}\n",
                    &resp_token
                        .token
                        .to_string()
                        .color(OckamColor::PrimaryResource.color())
                )
                + &fmt_log!(
                    "Id {}\n",
                    &resp_token
                        .id
                        .to_string()
                        .color(OckamColor::PrimaryResource.color())
                )
                + &fmt_log!(
                    "Expires at {}\n",
                    PrimitiveDateTime::parse(&resp_token.expires, &Iso8601::DEFAULT)
                        .into_diagnostic()?
                        .to_string()
                        .color(OckamColor::PrimaryResource.color())
                ),
        )
        .write_line()?;

    Ok(())
}