htb-cli 0.1.0

Hack The Box CLI
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

use clap::Subcommand;

use crate::api::HtbClient;
use crate::output::{self, OutputFormat};

#[derive(Subcommand)]
pub enum VpnCommand {
    /// Show current VPN connection status
    Status,
    /// List available VPN servers
    List,
    /// Switch VPN server
    Switch {
        /// Server ID
        server_id: u32,
    },
    /// Download .ovpn file
    Download {
        /// Server ID (uses default if omitted)
        server_id: Option<u32>,
    },
}

pub async fn handle(
    client: &HtbClient,
    cmd: VpnCommand,
    format: OutputFormat,
) -> anyhow::Result<()> {
    match cmd {
        VpnCommand::Status => {
            let status = client.vpn().status().await?;
            if status.is_empty() {
                output::print_message("Not connected to VPN.");
            } else {
                output::json::print_json(&status);
            }
        }

        VpnCommand::List => {
            let connections = client.vpn().connections().await?;
            output::print_list(&connections, format);
        }

        VpnCommand::Switch { server_id } => {
            let resp = client.vpn().switch(server_id).await?;
            if let Some(msg) = &resp.message {
                output::print_message(msg);
            }
            if let Some(server) = &resp.data {
                output::print_message(&format!(
                    "Switched to {} ({} clients)",
                    server.friendly_name, server.current_clients
                ));
            }
        }

        VpnCommand::Download { server_id } => {
            let sid = server_id.unwrap_or(1);
            let bytes = client.vpn().download_ovpn(sid).await?;
            let filename = format!("lab-vpn-{sid}.ovpn");
            std::fs::write(&filename, &bytes)?;
            output::print_message(&format!("Downloaded {} ({} bytes)", filename, bytes.len()));
        }
    }
    Ok(())
}