ASCII Art, delete an api key, new key, get usage, account balance

Author: crypdoughdoteth

src/commands/account_info.rs

@@ -0,0 +1,37 @@
+use reqwest::Client;
+use serde::Deserialize;
+
+#[derive(Debug, Deserialize)]
+pub struct UserBalances {
+    pub calls: i64,
+    pub balance: i64,
+}
+
+pub async fn get_rpc_calls(client: &Client) -> Result<(), Box<dyn std::error::Error>> {
+    let res = client
+        .get("https://api.cloud.developerdao.com/api/balances")
+        .send()
+        .await?
+        .error_for_status()?
+        .json::<UserBalances>()
+        .await?;
+
+    println!("You've made {} RPC calls this month\n", res.calls);
+    Ok(())
+}
+
+pub async fn get_balance(client: &Client) -> Result<(), Box<dyn std::error::Error>> {
+    let res = client
+        .get("https://api.cloud.developerdao.com/api/balances")
+        .send()
+        .await?
+        .error_for_status()?
+        .json::<UserBalances>()
+        .await?;
+
+    println!(
+        "Your account has a balance of ${}\n",
+        res.balance as f64 / 100.0
+    );
+    Ok(())
+}

src/commands/keys.rs

@@ -1,6 +1,6 @@
 use arboard::Clipboard;
 use console::Term;
-use dialoguer::{FuzzySelect, theme::ColorfulTheme};
+use dialoguer::{Confirm, FuzzySelect, theme::ColorfulTheme};
 use reqwest::Client;
 
 use crate::types::{CHAINS, Key, RedactedKey};
@@ -106,3 +106,74 @@ pub async fn get_keys_interactive(
 
     Ok(())
 }
+
+pub async fn new_api_key(client: &Client) -> Result<(), Box<dyn std::error::Error>> {
+    let mut clipboard = Clipboard::new()?;
+    let new_key = Key {
+        apikey: client
+            .post("https://api.cloud.developerdao.com/api/keys")
+            .send()
+            .await?
+            .error_for_status()?
+            .text()
+            .await?,
+    };
+
+    let redacted_key = new_key.as_redacted();
+    clipboard.set_text(new_key.apikey)?;
+
+    println!("Your new API key:\n{redacted_key}\nSuccessfully created and copied to clipboard!");
+    Ok(())
+}
+
+pub async fn delete_api_key(client: &Client) -> Result<(), Box<dyn std::error::Error>> {
+    let api_keys: Vec<Key> = client
+        .get("https://api.cloud.developerdao.com/api/keys")
+        .send()
+        .await?
+        .error_for_status()?
+        .json::<Vec<Key>>()
+        .await?;
+
+    if api_keys.is_empty() {
+        Err("Err: deletion failed. No API keys found.")?
+    }
+
+    let redacted_keys = api_keys
+        .iter()
+        .map(|e| e.as_redacted())
+        .collect::<Vec<RedactedKey>>();
+
+    let index = dialoguer::Select::with_theme(&ColorfulTheme::default())
+        .items(&redacted_keys)
+        .with_prompt("Select an API key to delete")
+        .interact()?;
+
+    let confirmation = Confirm::new()
+        .with_prompt(format!(
+            "Are you want you want to delete {}?",
+            redacted_keys[index].redacted
+        ))
+        .interact()
+        .unwrap();
+
+    if !confirmation {
+        Err("API key deletion aborted")?
+    }
+
+    client
+        .delete(format!(
+            "https://api.cloud.developerdao.com/api/keys/{}",
+            api_keys[index].apikey
+        ))
+        .send()
+        .await?
+        .error_for_status()?;
+
+    println!(
+        "Successfully deleted api key: {}",
+        redacted_keys[index].redacted
+    );
+
+    Ok(())
+}

src/commands/mod.rs

@@ -1 +1,2 @@
 pub mod keys;
+pub mod account_info;

src/main.rs

@@ -1,5 +1,11 @@
-use crate::types::LoginRequest;
-use clap::{Parser, Subcommand, ValueEnum};
+use crate::{
+    commands::{
+        account_info::{get_balance, get_rpc_calls},
+        keys::{delete_api_key, new_api_key},
+    },
+    types::{ASCII_ART, LoginRequest},
+};
+use clap::{Parser, Subcommand};
 use commands::keys::get_keys_interactive;
 use console::Term;
 use dialoguer::{Input, Password, theme::ColorfulTheme};
@@ -18,29 +24,13 @@ pub enum Command {
     GetApiKey {
         #[arg(short, long)]
         unsafe_print: bool,
-    },
-    DeleteApiKey {
         #[arg(short, long)]
-        key: String,
+        no_url: bool,
     },
+    DeleteApiKey,
     NewApiKey,
-    TrackUsage {
-        #[arg(short)]
-        service: Service,
-    },
+    TrackUsage,
     Balance,
-    GetPaymentInfo,
-    SubmitPayment {
-        #[arg(short, long)]
-        hash: String,
-        #[arg(short, long)]
-        plan: Option<String>,
-    },
-}
-
-#[derive(Debug, Copy, Clone, ValueEnum)]
-pub enum Service {
-    Rpc,
 }
 
 #[tokio::main]
@@ -49,7 +39,8 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let mut term = Term::stdout();
     term.clear_screen()?;
     term.set_title("Developer DAO Cloud");
-    term.write_line("Login to Developer DAO Cloud")?;
+    term.write_line(ASCII_ART)?;
+    term.write_line("Login")?;
     term.write_line("")?;
 
     let email: String = Input::with_theme(&ColorfulTheme::default())
@@ -71,18 +62,17 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         .json(&login)
         .send()
         .await?
-        .error_for_status()?;
+        .error_for_status()
+        .map_err(|_| "Invalid username or password")?;
 
     match cli_input.cmd {
-        Command::GetApiKey { unsafe_print } => {
+        Command::GetApiKey { unsafe_print, .. } => {
             get_keys_interactive(&client, &mut term, unsafe_print).await?
         }
-        Command::DeleteApiKey { key: _ } => todo!(),
-        Command::NewApiKey => todo!(),
-        Command::TrackUsage { service: _ } => todo!(),
-        Command::Balance => todo!(),
-        Command::GetPaymentInfo => todo!(),
-        Command::SubmitPayment { hash: _, plan: _ } => todo!(),
+        Command::DeleteApiKey => delete_api_key(&client).await?,
+        Command::NewApiKey => new_api_key(&client).await?,
+        Command::TrackUsage => get_rpc_calls(&client).await?,
+        Command::Balance => get_balance(&client).await?,
     };
 
     Ok(())

src/types.rs

@@ -1,6 +1,18 @@
 use std::fmt::Display;
 
 use serde::{Deserialize, Serialize};
+
+pub const ASCII_ART: &'static str = {
+    r#"
+██████╗         ██████╗      ██████╗██╗      ██████╗ ██╗   ██╗██████╗ 
+██╔══██╗        ██╔══██╗    ██╔════╝██║     ██╔═══██╗██║   ██║██╔══██╗
+██║  ██║        ██║  ██║    ██║     ██║     ██║   ██║██║   ██║██║  ██║
+██║  ██║        ██║  ██║    ██║     ██║     ██║   ██║██║   ██║██║  ██║
+██████╔╝███████╗██████╔╝    ╚██████╗███████╗╚██████╔╝╚██████╔╝██████╔╝
+╚═════╝ ╚══════╝╚═════╝      ╚═════╝╚══════╝ ╚═════╝  ╚═════╝ ╚═════╝ 
+    "#
+};
+
 pub static CHAINS: &[Chains] = &[
     Chains::Ethereum,
     Chains::Base,