feat: q builder id pro subscription flow

- Add better message to indicate when the user has reached their monthly query limit.
- Add new command `/subscribe` to initiate a subscription flow to paid usage for builder ID users
- IDC users cannot use this flow

Author: hayemaxi

crates/chat-cli/src/api_client/clients/client.rs

@@ -2,14 +2,12 @@ use amzn_codewhisperer_client::Client as CodewhispererClient;
 use amzn_codewhisperer_client::operation::create_subscription_token::CreateSubscriptionTokenOutput;
 use amzn_codewhisperer_client::types::{
     OptOutPreference,
-    SubscriptionStatus,
     TelemetryEvent,
     UserContext,
 };
 use tracing::error;
 
 use super::shared::bearer_sdk_config;
-use crate::api_client::consts::SUBSCRIPTION_STATUS_ACCOUNT_ID;
 use crate::api_client::interceptor::opt_out::OptOutInterceptor;
 use crate::api_client::{
     ApiClientError,
@@ -125,15 +123,12 @@ impl Client {
         }
     }
 
-    #[allow(dead_code)] // TODO: Remove
     pub async fn create_subscription_token(
         &self,
-        account_id: &str,
     ) -> Result<CreateSubscriptionTokenOutput, ApiClientError> {
         match &self.inner {
             inner::Inner::Codewhisperer(client) => client
                 .create_subscription_token()
-                .account_id(account_id)
                 .send()
                 .await
                 .map_err(ApiClientError::CreateSubscriptionToken),
@@ -142,18 +137,6 @@ impl Client {
                 .build()?),
         }
     }
-
-    #[allow(dead_code)] // TODO: Remove
-    pub async fn get_subscription_status(&self) -> Result<SubscriptionStatus, ApiClientError> {
-        match &self.inner {
-            inner::Inner::Codewhisperer(_) => Ok(self
-                .create_subscription_token(SUBSCRIPTION_STATUS_ACCOUNT_ID)
-                .await?
-                .status()
-                .clone()),
-            inner::Inner::Mock => Ok(SubscriptionStatus::Active),
-        }
-    }
 }
 
 #[cfg(test)]

crates/chat-cli/src/api_client/consts.rs

@@ -13,8 +13,3 @@ pub const PROD_CODEWHISPERER_FRA_ENDPOINT_REGION: Region = Region::from_static("
 
 // Opt out constants
 pub const X_AMZN_CODEWHISPERER_OPT_OUT_HEADER: &str = "x-amzn-codewhisperer-optout";
-
-// Dummy value that simply provides subscription status (for the holder of a given bearer token)
-// This is a design decision on the service-side to avoid creating a dedicated status API.
-#[allow(dead_code)] // TODO: Remove
-pub const SUBSCRIPTION_STATUS_ACCOUNT_ID: &str = "111111111111";

crates/chat-cli/src/auth/builder_id.rs

@@ -48,6 +48,10 @@ use tracing::{
     trace,
     warn,
 };
+use eyre::{
+    Result,
+    eyre,
+};
 
 use crate::api_client::clients::shared::stalled_stream_protection_config;
 use crate::auth::AuthError;
@@ -568,6 +572,14 @@ impl ResolveIdentity for BearerResolver {
     }
 }
 
+pub async fn is_idc_user(database: &Database) -> Result<bool> {
+    if let Ok(Some(token)) = BuilderIdToken::load(database).await {
+        return Ok(token.token_type() == TokenType::IamIdentityCenter);
+    } else {
+        return Err(eyre!("No auth token found - is the user signed in?"));
+    }
+}
+
 #[cfg(test)]
 mod tests {
     use super::*;

crates/chat-cli/src/cli/chat/command.rs

@@ -59,6 +59,7 @@ pub enum Command {
         force: bool,
     },
     Mcp,
+    Subscribe,
 }
 
 #[derive(Debug, Clone, PartialEq, Eq)]
@@ -839,6 +840,7 @@ impl Command {
                     Self::Save { path, force }
                 },
                 "mcp" => Self::Mcp,
+                "subscribe" => Self::Subscribe,
                 unknown_command => {
                     let looks_like_path = {
                         let after_slash_command_str = parts[1..].join(" ");