nolgia-cli 0.2.3

CLI for the Nolgia generative media platform (image, audio, video)
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

use anyhow::{Context, Result};
use clap::{Args, Subcommand};
use uuid::Uuid;

use crate::output::{OutputFormat, print_json};

use super::CommandContext;

#[derive(Subcommand, Debug)]
pub enum PatCommand {
    Create(CreatePatArgs),
    List,
    Revoke(RevokePatArgs),
}

#[derive(Args, Debug)]
pub struct CreatePatArgs {
    #[arg(long)]
    pub name: String,
}

#[derive(Args, Debug)]
pub struct RevokePatArgs {
    pub pat_id: Uuid,
}

pub async fn run(command: PatCommand, ctx: &CommandContext) -> Result<()> {
    match command {
        PatCommand::Create(args) => create(args, ctx).await,
        PatCommand::List => list(ctx).await,
        PatCommand::Revoke(args) => revoke(args, ctx).await,
    }
}

async fn create(args: CreatePatArgs, ctx: &CommandContext) -> Result<()> {
    let body: nolgia_client::types::CreatePatRequest =
        nolgia_client::types::CreatePatRequest::builder()
            .name(args.name)
            .try_into()
            .context("building create-pat request")?;
    let created = ctx
        .client()
        .create_pat()
        .body(body)
        .send()
        .await
        .context("creating personal access token")?
        .into_inner();
    match ctx.format() {
        OutputFormat::Json => print_json(&created),
        OutputFormat::Text => {
            println!("created {} ({})", created.pat.id, created.pat.name.as_str());
            println!("token: {}", created.token);
            println!("warning: this token will not be shown again; store it securely");
            Ok(())
        }
    }
}

async fn list(ctx: &CommandContext) -> Result<()> {
    let page = ctx
        .client()
        .list_pats()
        .send()
        .await
        .context("listing personal access tokens")?
        .into_inner();
    match ctx.format() {
        OutputFormat::Json => print_json(&page),
        OutputFormat::Text => {
            for pat in page.items {
                let last_used = pat
                    .last_used_at
                    .map(|at| at.to_rfc3339())
                    .unwrap_or_else(|| "never".to_string());
                println!(
                    "{} {} {} created {} last used {}",
                    pat.id,
                    pat.name.as_str(),
                    pat.prefix.as_str(),
                    pat.created_at.to_rfc3339(),
                    last_used
                );
            }
            Ok(())
        }
    }
}

async fn revoke(args: RevokePatArgs, ctx: &CommandContext) -> Result<()> {
    ctx.client()
        .revoke_pat()
        .id(args.pat_id)
        .send()
        .await
        .context("revoking personal access token")?;
    match ctx.format() {
        OutputFormat::Json => print_json(&serde_json::json!({ "revoked": args.pat_id })),
        OutputFormat::Text => {
            println!("revoked {}", args.pat_id);
            Ok(())
        }
    }
}