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

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

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

use super::CommandContext;

#[derive(Subcommand, Debug)]
pub enum BillingCommand {
    Subscription,
    Credits,
    Portal(PortalArgs),
}

#[derive(Args, Debug)]
pub struct PortalArgs {
    #[arg(long)]
    pub return_url: Option<String>,
}

pub async fn run(command: BillingCommand, ctx: &CommandContext) -> Result<()> {
    match command {
        BillingCommand::Subscription => subscription(ctx).await,
        BillingCommand::Credits => credits(ctx).await,
        BillingCommand::Portal(args) => portal(args, ctx).await,
    }
}

async fn credits(ctx: &CommandContext) -> Result<()> {
    let balance = ctx
        .client()
        .get_credits()
        .send()
        .await
        .context("fetching credit balance")?
        .into_inner();
    match ctx.format() {
        OutputFormat::Json => print_json(&balance),
        OutputFormat::Text => {
            println!(
                "subscription: {} (resets with plan)  api top-ups: {}",
                balance.app_subscription, balance.shared_topup
            );
            println!("total: {}", balance.total);
            Ok(())
        }
    }
}

async fn subscription(ctx: &CommandContext) -> Result<()> {
    let subscription = ctx
        .client()
        .get_subscription()
        .send()
        .await
        .context("fetching subscription")?
        .into_inner();
    match ctx.format() {
        OutputFormat::Json => print_json(&subscription),
        OutputFormat::Text => {
            println!("{} {}", subscription.tier, subscription.status);
            Ok(())
        }
    }
}

async fn portal(args: PortalArgs, ctx: &CommandContext) -> Result<()> {
    let body: nolgia_client::types::PortalLinkRequest =
        nolgia_client::types::PortalLinkRequest::builder()
            .return_url(args.return_url)
            .try_into()
            .context("building portal request")?;
    let portal = ctx
        .client()
        .create_portal_link()
        .body(body)
        .send()
        .await
        .context("creating portal link")?
        .into_inner();
    match ctx.format() {
        OutputFormat::Json => print_json(&portal),
        OutputFormat::Text => {
            println!("{}", portal.url);
            Ok(())
        }
    }
}