lab-ops 0.1.23

Personal utility tools for my homelab
Documentation
use std::path::PathBuf;

use clap::Parser;
use clap::Subcommand;
use clap::ValueEnum;

use crate::consts::CMD_AUTO_DISCOVER;
use crate::consts::CMD_CF2ANSIBLE;
use crate::consts::CMD_CF2TERRA;
use crate::consts::CMD_COMPLETIONS;
use crate::consts::CMD_DOCKERNET;
use crate::consts::CMD_NATMAP;

/// Top-level CLI argument parser for `lab-ops`.
#[derive(Parser)]
#[command(version = env!("CARGO_PKG_VERSION"))]
#[command(name = crate::consts::CMD, about = "Lab operations toolkit")]
pub struct Cli {
    #[command(subcommand)]
    pub command: Command,

    /// Verbosity level (-v, -vv, -vvv, -vvvv)
    #[arg(short, long, global = true, action = clap::ArgAction::Count)]
    pub verbose: u8,

    /// Color output mode [auto|always|never]
    #[arg(
        long,
        global = true,
        default_value = "auto",
        value_enum,
        hide_possible_values = true
    )]
    pub color: ColorMode,
}

/// Controls ANSI color output in logs and error formatting.
#[derive(Clone, Debug, Default, ValueEnum)]
pub enum ColorMode {
    #[default]
    Auto,
    Always,
    Never,
}

/// All supported subcommands of lab-ops.
#[derive(Subcommand)]
pub enum Command {
    /// Converts BIND DNS zone files to Ansible YAML tasks.
    #[command(name = CMD_CF2ANSIBLE)]
    Cf2Ansible {
        /// Path to the DNS zone file to parse.
        zone_file: PathBuf,
        /// Zone name override; defaults to the name extracted from the SOA record.
        zone_name: Option<String>,
    },
    /// Converts BIND DNS zone files to Terraform Cloudflare DNS resources.
    #[command(name = CMD_CF2TERRA)]
    Cf2Terra {
        /// Path to the DNS zone file to parse.
        zone_file: PathBuf,
        /// Zone name override; defaults to the name extracted from the SOA record.
        zone_name: Option<String>,
        /// Terraform variable for the Cloudflare Zone ID.
        #[arg(long, default_value = "var.cloudflare_zone_id")]
        zone_id_var: String,
    },
    /// Displays IP addresses and port bindings of Docker containers.
    #[command(name = CMD_DOCKERNET)]
    DockerNet,
    /// Manages iptables NAT rules for static VMs and dynamic Docker containers.
    #[command(name = CMD_NATMAP)]
    NatMap {
        #[command(flatten)]
        args: lab_ops_natmap::cli::Cli,
    },
    /// Service discovery daemon: watches Docker events, manages port forwarding,
    /// registers services with Consul, and generates nginx configs.
    #[command(name = CMD_AUTO_DISCOVER)]
    AutoDiscover {
        #[command(flatten)]
        args: lab_ops_auto_discover::cli::Cli,
    },
    /// Generate shell completion scripts.
    #[command(name = CMD_COMPLETIONS)]
    Completions {
        /// Target shell [bash, elvish, fish, powershell, zsh]
        #[arg(value_enum, hide_possible_values = true)]
        shell: clap_complete::Shell,
        /// Output directory (e.g., ~/.zfunc)
        #[arg(long)]
        dir: Option<PathBuf>,
    },
}