koi-net 0.4.1

Local network toolkit: service discovery, DNS, health monitoring, TLS proxy, and certificate mesh
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
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97

//! Admin CLI command handlers.
//!
//! These use `KoiClient` to manage a running daemon's registrations.

use crate::client::KoiClient;
use crate::format;

// ── Status ──────────────────────────────────────────────────────────

pub fn status(client: &KoiClient, json: bool) -> anyhow::Result<()> {
    let status = client.admin_status()?;
    if json {
        println!("{}", serde_json::to_string_pretty(&status)?);
    } else {
        println!("Koi daemon v{}", status.version);
        println!("  Platform: {}", status.platform);
        println!("  Uptime:   {}s", status.uptime_secs);
        let r = &status.registrations;
        println!(
            "  Services: {} total ({} alive, {} draining, {} permanent)",
            r.total, r.alive, r.draining, r.permanent,
        );
    }
    Ok(())
}

// ── List ─────────────────────────────────────────────────────────────

pub fn list(client: &KoiClient, json: bool) -> anyhow::Result<()> {
    let registrations = client.admin_registrations()?;
    if json {
        println!("{}", serde_json::to_string_pretty(&registrations)?);
    } else if registrations.is_empty() {
        println!("No registrations.");
    } else {
        println!(
            "{:<10} {:<20} {:<16} {:>5}  {:<10} {:<10}",
            "ID", "NAME", "TYPE", "PORT", "STATE", "MODE"
        );
        for reg in &registrations {
            print!("{}", format::registration_row(reg));
        }
    }
    Ok(())
}

// ── Inspect ──────────────────────────────────────────────────────────

pub fn inspect(client: &KoiClient, id: &str, json: bool) -> anyhow::Result<()> {
    let reg = client.admin_inspect(id)?;
    if json {
        let body = serde_json::to_string_pretty(&reg)?;
        println!("{body}");
    } else {
        print!("{}", format::registration_detail(&reg));
    }
    Ok(())
}

// ── Force unregister ─────────────────────────────────────────────────

pub fn unregister(client: &KoiClient, id: &str, json: bool) -> anyhow::Result<()> {
    client.admin_force_unregister(id)?;
    if json {
        println!(
            "{}",
            serde_json::json!({"unregistered": id, "source": "admin"})
        );
    } else {
        println!("Force-unregistered {id}");
    }
    Ok(())
}

// ── Drain ────────────────────────────────────────────────────────────

pub fn drain(client: &KoiClient, id: &str, json: bool) -> anyhow::Result<()> {
    client.admin_drain(id)?;
    if json {
        println!("{}", serde_json::json!({"drained": id}));
    } else {
        println!("Draining {id}");
    }
    Ok(())
}

// ── Revive ───────────────────────────────────────────────────────────

pub fn revive(client: &KoiClient, id: &str, json: bool) -> anyhow::Result<()> {
    client.admin_revive(id)?;
    if json {
        println!("{}", serde_json::json!({"revived": id}));
    } else {
        println!("Revived {id}");
    }
    Ok(())
}