ockam_command 0.117.0

use crate::util::async_cmd;
use crate::{color, docs, fmt_info, fmt_ok, fmt_warn, CommandGlobalOpts, OckamColor};

use clap::Args;
use colorful::Colorful;
use miette::miette;

const LONG_ABOUT: &str = include_str!("./static/stop/long_about.txt");
const PREVIEW_TAG: &str = include_str!("../static/preview_tag.txt");
const AFTER_LONG_HELP: &str = include_str!("./static/stop/after_long_help.txt");

/// Stop a running node
#[derive(Clone, Debug, Args)]
#[command(
    long_about = docs::about(LONG_ABOUT),
    before_help = docs::before_help(PREVIEW_TAG),
    after_long_help = docs::after_help(AFTER_LONG_HELP)
)]
pub struct StopCommand {
    /// Name of the node.
    node_name: Option<String>,

    /// Whether to use the SIGTERM or SIGKILL signal to stop the node
    #[arg(short, long)]
    force: bool,
}

impl StopCommand {
    pub fn run(self, opts: CommandGlobalOpts) -> miette::Result<()> {
        async_cmd(&self.name(), opts.clone(), |_ctx| async move {
            self.async_run(opts).await
        })
    }

    pub fn name(&self) -> String {
        "stop node".into()
    }

    async fn async_run(&self, opts: CommandGlobalOpts) -> miette::Result<()> {
        let running_nodes = opts
            .state
            .get_nodes()
            .await?
            .iter()
            .filter(|node| node.is_running())
            .map(|node| node.name())
            .collect::<Vec<String>>();
        if running_nodes.is_empty() {
            opts.terminal
                .stdout()
                .plain(fmt_info!("There are no nodes running"))
                .write_line()?;
            return Ok(());
        }

        if self.node_name.is_some() || !opts.terminal.can_ask_for_user_input() {
            let node_name = opts
                .state
                .get_node_or_default(&self.node_name)
                .await?
                .name();
            if !running_nodes.contains(&node_name) {
                return Err(miette!(
                    "The node {} was not found",
                    node_name.light_magenta()
                ));
            }
            stop_node(opts, &node_name, self.force).await?;
            return Ok(());
        }

        match running_nodes.len() {
            0 => {
                unreachable!("this case is already handled above");
            }
            1 => {
                let node_name = running_nodes[0].as_str();
                stop_node(opts, node_name, self.force).await?;
            }
            _ => {
                let selected_item_names = opts.terminal.select_multiple(
                    "Select one or more nodes that you want to stop".to_string(),
                    running_nodes,
                );
                match selected_item_names.len() {
                    0 => {
                        opts.terminal
                            .stdout()
                            .plain(fmt_info!("No nodes selected to stop"))
                            .write_line()?;
                    }
                    1 => {
                        let node_name = selected_item_names[0].as_str();
                        stop_node(opts, node_name, self.force).await?;
                    }
                    _ => {
                        for item_name in selected_item_names {
                            stop_node(opts.clone(), &item_name, self.force).await?;
                        }
                    }
                }
            }
        }
        Ok(())
    }
}

async fn stop_node(opts: CommandGlobalOpts, node_name: &str, force: bool) -> miette::Result<()> {
    let res = opts.state.stop_node(node_name, force).await;
    let output = if res.is_ok() {
        fmt_ok!(
            "Node with name {} was stopped",
            color!(node_name, OckamColor::PrimaryResource)
        )
    } else {
        fmt_warn!(
            "Failed to delete node with name {}",
            color!(node_name, OckamColor::PrimaryResource)
        )
    };
    opts.terminal.stdout().plain(output).write_line()?;
    Ok(())
}