feat(auth): add email and billing management to CLI

cli/src/command/auth/billing.rs

@@ -0,0 +1,29 @@
+use anyhow::Result;
+use clap::Args;
+use slot::graphql::auth::{update_me::*, UpdateMe};
+use slot::graphql::GraphQLQuery;
+use slot::{api::Client, credential::Credentials};
+
+#[derive(Debug, Args)]
+#[command(next_help_heading = "Set billing options")]
+pub struct BillingArgs {
+    #[arg(help = "Enable slot billing for the authenticated user.")]
+    #[clap(long, short, action)]
+    pub enabled: bool,
+}
+
+impl BillingArgs {
+    pub async fn run(&self) -> Result<()> {
+        let credentials = Credentials::load()?;
+        let client = Client::new_with_token(credentials.access_token);
+
+        let request_body = UpdateMe::build_query(Variables {
+            email: None,
+            slot_billing: Some(self.enabled),
+        });
+        let res: ResponseData = client.query(&request_body).await?;
+        print!("{:?}", res);
+
+        Ok(())
+    }
+}

cli/src/command/auth/email.rs

@@ -0,0 +1,28 @@
+use anyhow::Result;
+use clap::Args;
+use slot::graphql::auth::{update_me::*, UpdateMe};
+use slot::graphql::GraphQLQuery;
+use slot::{api::Client, credential::Credentials};
+
+#[derive(Debug, Args)]
+#[command(next_help_heading = "Set email options")]
+pub struct EmailArgs {
+    #[arg(help = "The email address of the user.")]
+    pub email: String,
+}
+
+impl EmailArgs {
+    pub async fn run(&self) -> Result<()> {
+        let credentials = Credentials::load()?;
+        let client = Client::new_with_token(credentials.access_token);
+
+        let request_body = UpdateMe::build_query(Variables {
+            email: Some(self.email.clone()),
+            slot_billing: None,
+        });
+        let res: ResponseData = client.query(&request_body).await?;
+        print!("{:?}", res);
+
+        Ok(())
+    }
+}

cli/src/command/auth/mod.rs

@@ -1,8 +1,10 @@
+use self::{email::EmailArgs, info::InfoArgs, login::LoginArgs};
+use crate::command::auth::billing::BillingArgs;
 use anyhow::Result;
 use clap::Subcommand;
 
-use self::{info::InfoArgs, login::LoginArgs};
-
+mod billing;
+mod email;
 mod info;
 mod login;
 mod session;
@@ -11,8 +13,16 @@ mod session;
 pub enum Auth {
     #[command(about = "Login to your Cartridge account.")]
     Login(LoginArgs),
+
     #[command(about = "Display info about the authenticated user.")]
     Info(InfoArgs),
+
+    #[command(about = "Set the email address for the authenticated user.")]
+    SetEmail(EmailArgs),
+
+    #[command(about = "Manage slot billing for the authenticated user.")]
+    EnableSlotBilling(BillingArgs),
+
     // Mostly for testing purposes, will eventually turn it into a library call from `sozo`.
     #[command(hide = true)]
     CreateSession(session::CreateSession),
@@ -24,6 +34,8 @@ impl Auth {
             Auth::Login(args) => args.run().await,
             Auth::Info(args) => args.run().await,
             Auth::CreateSession(args) => args.run().await,
+            Auth::SetEmail(args) => args.run().await,
+            Auth::EnableSlotBilling(args) => args.run().await,
         }
     }
 }