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

use anyhow::{Context, Result};
use clap::Subcommand;
use serde::Serialize;

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

use super::CommandContext;

#[derive(Subcommand, Debug)]
pub enum AccountCommand {
    Me,
    Usage,
}

#[derive(Serialize)]
struct UsageSummary {
    jobs_visible: usize,
    assets_visible: usize,
}

pub async fn run(command: AccountCommand, ctx: &CommandContext) -> Result<()> {
    match command {
        AccountCommand::Me => me(ctx).await,
        AccountCommand::Usage => usage(ctx).await,
    }
}

async fn me(ctx: &CommandContext) -> Result<()> {
    let user = ctx
        .client()
        .get_current_user()
        .send()
        .await
        .context("fetching current user")?
        .into_inner();
    match ctx.format() {
        OutputFormat::Json => print_json(&user),
        OutputFormat::Text => {
            println!("{} {}", user.id, user.email);
            Ok(())
        }
    }
}

async fn usage(ctx: &CommandContext) -> Result<()> {
    let jobs = ctx
        .client()
        .list_jobs()
        .send()
        .await
        .context("listing jobs for usage")?
        .into_inner();
    let assets = ctx
        .client()
        .list_assets()
        .send()
        .await
        .context("listing assets for usage")?
        .into_inner();
    let summary = UsageSummary {
        jobs_visible: jobs.items.len(),
        assets_visible: assets.items.len(),
    };
    match ctx.format() {
        OutputFormat::Json => print_json(&summary),
        OutputFormat::Text => {
            println!(
                "jobs: {} assets: {}",
                summary.jobs_visible, summary.assets_visible
            );
            Ok(())
        }
    }
}