use clap::Args;
use colorful::Colorful;
use console::Term;
use miette::IntoDiagnostic;
use ockam::Context;
use ockam_api::nodes::models::portal::InletList;
use ockam_api::nodes::service::portals::Inlets;
use ockam_api::nodes::BackgroundNodeClient;
use ockam_core::api::Request;
use ockam_core::AsyncTryClone;
use crate::node::NodeOpts;
use crate::tcp::util::alias_parser;
use crate::terminal::tui::DeleteCommandTui;
use crate::terminal::PluralTerm;
use crate::util::async_cmd;
use crate::{color, docs, fmt_ok, CommandGlobalOpts, OckamColor, Terminal, TerminalStream};
const AFTER_LONG_HELP: &str = include_str!("./static/delete/after_long_help.txt");
#[derive(Clone, Debug, Args)]
#[command(after_long_help = docs::after_help(AFTER_LONG_HELP))]
pub struct DeleteCommand {
#[arg(display_order = 900, id = "ALIAS", value_parser = alias_parser)]
alias: Option<String>,
#[command(flatten)]
node_opts: NodeOpts,
#[arg(display_order = 901, long, short)]
yes: bool,
#[arg(long, short, group = "tcp-inlets")]
all: bool,
}
impl DeleteCommand {
pub fn run(self, opts: CommandGlobalOpts) -> miette::Result<()> {
async_cmd(&self.name(), opts.clone(), |ctx| async move {
self.async_run(&ctx, opts).await
})
}
pub fn name(&self) -> String {
"delete tcp inlet".into()
}
pub async fn async_run(&self, ctx: &Context, opts: CommandGlobalOpts) -> miette::Result<()> {
DeleteTui::run(
ctx.async_try_clone().await.into_diagnostic()?,
opts,
self.clone(),
)
.await
}
}
struct DeleteTui {
ctx: Context,
opts: CommandGlobalOpts,
node: BackgroundNodeClient,
cmd: DeleteCommand,
}
impl DeleteTui {
pub async fn run(
ctx: Context,
opts: CommandGlobalOpts,
cmd: DeleteCommand,
) -> miette::Result<()> {
let node = BackgroundNodeClient::create(&ctx, &opts.state, &cmd.node_opts.at_node).await?;
let tui = Self {
ctx,
opts,
node,
cmd,
};
tui.delete().await
}
}
#[ockam_core::async_trait]
impl DeleteCommandTui for DeleteTui {
const ITEM_NAME: PluralTerm = PluralTerm::Inlet;
fn cmd_arg_item_name(&self) -> Option<String> {
self.cmd.alias.clone()
}
fn cmd_arg_delete_all(&self) -> bool {
self.cmd.all
}
fn cmd_arg_confirm_deletion(&self) -> bool {
self.cmd.yes
}
fn terminal(&self) -> Terminal<TerminalStream<Term>> {
self.opts.terminal.clone()
}
async fn list_items_names(&self) -> miette::Result<Vec<String>> {
let inlets: InletList = self
.node
.ask(&self.ctx, Request::get("/node/inlet"))
.await?;
let names = inlets.list.into_iter().map(|i| i.alias).collect();
Ok(names)
}
async fn delete_single(&self, item_name: &str) -> miette::Result<()> {
let node_name = self.node.node_name();
self.node.delete_inlet(&self.ctx, item_name).await?;
self.terminal()
.stdout()
.plain(fmt_ok!(
"TCP inlet with alias {} on Node {} has been deleted",
color!(item_name, OckamColor::PrimaryResource),
color!(node_name, OckamColor::PrimaryResource)
))
.write_line()?;
Ok(())
}
}