ockam_command 0.150.0

End-to-end encryption and mutual authentication for distributed applications.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116

use clap::{Args, Subcommand};

use ockam_api::nodes::InMemoryNode;
use ockam_api::orchestrator::project::ProjectsOrchestratorApi;
use ockam_node::Context;

use crate::operation::util::check_for_operation_completion;
use crate::project::addon::configure_influxdb::AddonConfigureInfluxdbSubcommand;
use crate::project::addon::configure_kafka::{
    AddonConfigureAivenSubcommand, AddonConfigureConfluentSubcommand,
    AddonConfigureInstaclustrSubcommand, AddonConfigureKafkaSubcommand,
    AddonConfigureRedpandaSubcommand, AddonConfigureWarpstreamSubcommand,
};
use crate::project::addon::configure_okta::AddonConfigureOktaSubcommand;
use crate::project::addon::disable::AddonDisableSubcommand;
use crate::project::addon::list::AddonListSubcommand;
use crate::project::util::check_project_readiness;
use crate::shared_args::IdentityOpts;
use crate::{CommandGlobalOpts, Result};

mod configure_influxdb;
mod configure_kafka;
mod configure_okta;
mod disable;
mod list;

/// Manage addons for a Project
#[derive(Clone, Debug, Args)]
#[command(arg_required_else_help = true, subcommand_required = true)]
pub struct AddonCommand {
    #[command(subcommand)]
    subcommand: AddonSubcommand,
    #[command(flatten)]
    identity_opts: IdentityOpts,
}

#[derive(Clone, Debug, Subcommand)]
pub enum AddonSubcommand {
    List(AddonListSubcommand),
    Disable(AddonDisableSubcommand),
    #[command(subcommand)]
    Configure(ConfigureAddonCommand),
}

impl AddonCommand {
    pub async fn run(self, ctx: &Context, opts: CommandGlobalOpts) -> miette::Result<()> {
        match self.subcommand {
            AddonSubcommand::List(cmd) => cmd.run(ctx, opts).await,
            AddonSubcommand::Disable(cmd) => cmd.run(ctx, opts).await,
            AddonSubcommand::Configure(cmd) => cmd.run(ctx, opts).await,
        }
    }

    pub fn name(&self) -> String {
        match &self.subcommand {
            AddonSubcommand::List(c) => c.name(),
            AddonSubcommand::Disable(c) => c.name(),
            AddonSubcommand::Configure(c) => c.name(),
        }
    }
}

/// Configure an addon for a project
#[derive(Clone, Debug, Subcommand)]
pub enum ConfigureAddonCommand {
    Okta(AddonConfigureOktaSubcommand),
    Influxdb(AddonConfigureInfluxdbSubcommand),
    Confluent(AddonConfigureConfluentSubcommand),
    InstaclustrKafka(AddonConfigureInstaclustrSubcommand),
    AivenKafka(AddonConfigureAivenSubcommand),
    Redpanda(AddonConfigureRedpandaSubcommand),
    Warpstream(AddonConfigureWarpstreamSubcommand),
    Kafka(AddonConfigureKafkaSubcommand),
}

impl ConfigureAddonCommand {
    pub async fn run(self, ctx: &Context, opts: CommandGlobalOpts) -> miette::Result<()> {
        match self {
            ConfigureAddonCommand::Okta(cmd) => cmd.run(ctx, opts).await,
            ConfigureAddonCommand::Influxdb(cmd) => cmd.run(ctx, opts).await,
            ConfigureAddonCommand::Confluent(cmd) => cmd.run(ctx, opts).await,
            ConfigureAddonCommand::InstaclustrKafka(cmd) => cmd.run(ctx, opts).await,
            ConfigureAddonCommand::AivenKafka(cmd) => cmd.run(ctx, opts).await,
            ConfigureAddonCommand::Redpanda(cmd) => cmd.run(ctx, opts).await,
            ConfigureAddonCommand::Warpstream(cmd) => cmd.run(ctx, opts).await,
            ConfigureAddonCommand::Kafka(cmd) => cmd.run(ctx, opts, "Apache Kafka").await,
        }
    }

    pub fn name(&self) -> String {
        match &self {
            ConfigureAddonCommand::Okta(c) => c.name(),
            ConfigureAddonCommand::Influxdb(c) => c.name(),
            ConfigureAddonCommand::Confluent(c) => c.name(),
            ConfigureAddonCommand::InstaclustrKafka(c) => c.name(),
            ConfigureAddonCommand::AivenKafka(c) => c.name(),
            ConfigureAddonCommand::Redpanda(c) => c.name(),
            ConfigureAddonCommand::Warpstream(c) => c.name(),
            ConfigureAddonCommand::Kafka(c) => c.name(),
        }
    }
}

async fn check_configuration_completion(
    opts: &CommandGlobalOpts,
    ctx: &Context,
    node: &InMemoryNode,
    project_id: &str,
    operation_id: &str,
) -> Result<()> {
    check_for_operation_completion(opts, ctx, node, operation_id, "the addon configuration")
        .await?;
    let project = node.get_project(ctx, project_id).await?;
    let _ = check_project_readiness(opts, ctx, node, project).await?;
    Ok(())
}