necronux_cli_app 0.3.0-rc.2

Provides the CLI implementation for Necronux.
Documentation 
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

// ==----------------------------------------------------------------== //
// SPDX-FileCopyrightText: © 2024-2025 Nayan Patil <nayantsg@proton.me>
//
// SPDX-License-Identifier: GPL-3.0-or-later
// ==----------------------------------------------------------------== //

use crate::{
    commands::lvl_1::infra::{InfraCommand, InfraSubCommands},
    flow_controller::Cli,
    handlers::infra::infra_run::handle_infra_run,
};
use clap::CommandFactory;
use color_eyre::eyre::{Ok, Result};
use log::{debug, info};

pub fn handle_infra(infra_command: &InfraCommand) -> Result<()> {
    debug!("Handling infra command subcommands");
    let command_handled = handle_infra_command(infra_command)?;
    if !command_handled {
        debug!("No infra subcommand or valid argument provided, displaying help message");
        let mut cli = Cli::command();
        if let Some(infra_command) = cli.find_subcommand_mut("infra") {
            infra_command.print_help()?;
        }
    }

    Ok(())
}

pub fn handle_infra_command(infra_command: &InfraCommand) -> Result<bool> {
    if let Some(subcommand) = &infra_command.infra_subcommand {
        match subcommand {
            InfraSubCommands::Run(args) => {
                info!("'infra run' command was provided");
                debug!("Handling infra run command");
                handle_infra_run(args)?;
                Ok(true)
            }
            InfraSubCommands::ListHosts => {
                info!("'list-hosts' flag was provided");
                debug!("Displaying list of hosts");
                println!("trinity-infinity,\ntrinity-heliolens");
                Ok(true)
            }
        }
    } else {
        Ok(false)
    }
}