gor-cli 0.1.0

A Rust CLI for GitHub — a 'gh' clone
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
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
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132

//! Implementation of the `gor copilot` subcommand.
//!
//! Provides Copilot status and usage information.

#![allow(clippy::print_stdout)]

use crate::cli::CopilotCommand;
use crate::client::Client;
use crate::output::print_json;
use anyhow::Context;

/// Run the `gor copilot` subcommand.
///
/// # Errors
///
/// Returns an error if the command execution fails.
pub fn run(cmd: CopilotCommand) -> anyhow::Result<()> {
    match cmd {
        CopilotCommand::Status { json, hostname } => status(json, hostname.as_deref()),
        CopilotCommand::Usage {
            org,
            json,
            hostname,
        } => usage(org.as_deref(), json, hostname.as_deref()),
    }
}

/// Show Copilot subscription status for the authenticated user.
///
/// # Errors
///
/// Returns an error if the API request fails.
fn status(json: Option<Vec<String>>, hostname: Option<&str>) -> anyhow::Result<()> {
    let host = hostname.unwrap_or("github.com");
    let client = Client::new(host).context("failed to create HTTP client")?;

    let response = client
        .get("/user/copilot/subscription")
        .context("failed to fetch Copilot subscription")?;

    let status_code = response.status();
    if status_code == reqwest::StatusCode::NOT_FOUND {
        println!("Copilot is not enabled for your account.");
        return Ok(());
    }
    if !status_code.is_success() {
        anyhow::bail!("failed to fetch Copilot status: HTTP {status_code}");
    }

    let sub: serde_json::Value = response.json().context("failed to parse response")?;

    if let Some(fields) = json {
        let fields_ref: Option<&[String]> = if fields.is_empty() {
            None
        } else {
            Some(&fields)
        };
        print_json(&sub, fields_ref);
        return Ok(());
    }

    let plan = sub["plan"].as_str().unwrap_or("—");
    let seat = sub["seat_status"].as_str().unwrap_or("—");
    let renewal = sub["renewal_date"].as_str().unwrap_or("—");

    println!("Copilot Status");
    println!("  Plan: {plan}");
    println!("  Seat: {seat}");
    println!("  Renewal: {renewal}");

    Ok(())
}

/// Show Copilot usage statistics for an organization.
///
/// # Errors
///
/// Returns an error if the API request fails.
fn usage(
    org: Option<&str>,
    json: Option<Vec<String>>,
    hostname: Option<&str>,
) -> anyhow::Result<()> {
    let host = hostname.unwrap_or("github.com");
    let client = Client::new(host).context("failed to create HTTP client")?;

    let org_name = org.ok_or_else(|| anyhow::anyhow!("--org is required for usage statistics"))?;

    let path = format!("/orgs/{org_name}/copilot/usage");
    let response = client.get(&path).context("failed to fetch Copilot usage")?;

    let status_code = response.status();
    if status_code == reqwest::StatusCode::NOT_FOUND {
        println!("Copilot is not enabled for organization '{org_name}'.");
        return Ok(());
    }
    if !status_code.is_success() {
        anyhow::bail!("failed to fetch Copilot usage: HTTP {status_code}");
    }

    let usage_data: serde_json::Value = response.json().context("failed to parse response")?;

    if let Some(fields) = json {
        let fields_ref: Option<&[String]> = if fields.is_empty() {
            None
        } else {
            Some(&fields)
        };
        print_json(&usage_data, fields_ref);
        return Ok(());
    }

    // Display usage summary
    let total = usage_data["total_active_users"].as_u64().unwrap_or(0);
    let breakdown = usage_data["breakdown"]
        .as_array()
        .map_or_else(Vec::new, Clone::clone);

    println!("Copilot Usage for {org_name}");
    println!("  Total active users: {total}");

    if !breakdown.is_empty() {
        println!("  Breakdown:");
        for entry in &breakdown {
            let lang = entry["language"].as_str().unwrap_or("—");
            let editors = entry["editor_count"].as_u64().unwrap_or(0);
            println!("    {lang}: {editors} editors");
        }
    }

    Ok(())
}