use crate::Client;
use reqwest::Method;
use serde_json::json;
use shared_entity::{
  dto::billing_dto::{
    RecurringInterval, SubscriptionPlan, WorkspaceSubscriptionStatus, WorkspaceUsage,
    WorkspaceUsageLimit,
  },
  response::{AppResponse, AppResponseError},
};

impl Client {
  pub async fn customer_id(&self) -> Result<String, AppResponseError> {
    let url = format!("{}/billing/api/v1/customer-id", &self.base_url);
    let resp = self
      .http_client_with_auth(Method::GET, &url)
      .await?
      .send()
      .await?;

    AppResponse::<String>::from_response(resp)
      .await?
      .into_data()
  }

  pub async fn create_subscription(
    &self,
    workspace_id: &str,
    recurring_interval: RecurringInterval,
    workspace_subscription_plan: SubscriptionPlan,
    success_url: &str,
  ) -> Result<String, AppResponseError> {
    let url = format!("{}/billing/api/v1/subscription-link", &self.base_url,);
    let resp = self
      .http_client_with_auth(Method::GET, &url)
      .await?
      .query(&[
        ("workspace_id", workspace_id),
        ("recurring_interval", recurring_interval.as_str()),
        (
          "workspace_subscription_plan",
          &workspace_subscription_plan.as_str(),
        ),
        ("success_url", success_url),
      ])
      .send()
      .await?;

    AppResponse::<String>::from_response(resp)
      .await?
      .into_data()
  }

  pub async fn cancel_subscription(&self, workspace_id: &str) -> Result<(), AppResponseError> {
    let url = format!("{}/billing/api/v1/cancel-subscription", &self.base_url);
    let resp = self
      .http_client_with_auth(Method::POST, &url)
      .await?
      .json(&json!({ "workspace_id": workspace_id }))
      .send()
      .await?;
    AppResponse::<()>::from_response(resp).await?.into_error()
  }

  pub async fn list_subscription(
    &self,
  ) -> Result<Vec<WorkspaceSubscriptionStatus>, AppResponseError> {
    let url = format!("{}/billing/api/v1/subscription-status", &self.base_url,);
    let resp = self
      .http_client_with_auth(Method::GET, &url)
      .await?
      .send()
      .await?;

    AppResponse::<Vec<WorkspaceSubscriptionStatus>>::from_response(resp)
      .await?
      .into_data()
  }

  pub async fn get_billing_workspace_usage(
    &self,
    workspace_id: &str,
  ) -> Result<WorkspaceUsage, AppResponseError> {
    let num_members = self.get_workspace_members(workspace_id).await?.len();
    let limits = self.get_workspace_limits(workspace_id).await?;
    let doc_usage = self.get_workspace_usage(workspace_id).await?;

    let workspace_usage = WorkspaceUsage {
      member_count: num_members,
      member_count_limit: limits.member_count,
      total_blob_bytes: doc_usage.consumed_capacity as _,
      total_blob_bytes_limit: limits.total_blob_size,
    };
    Ok(workspace_usage)
  }

  pub async fn get_portal_session_link(&self) -> Result<String, AppResponseError> {
    let url = format!("{}/billing/api/v1/portal-session-link", &self.base_url,);
    let portal_url = self
      .http_client_with_auth(Method::GET, &url)
      .await?
      .send()
      .await?
      .error_for_status()?
      .json::<AppResponse<String>>()
      .await?
      .into_data()?;
    Ok(portal_url)
  }

  async fn get_workspace_limits(
    &self,
    workspace_id: &str,
  ) -> Result<WorkspaceUsageLimit, AppResponseError> {
    let url = format!("{}/api/workspace/{}/limit", &self.base_url, workspace_id);
    self
      .http_client_with_auth(Method::GET, &url)
      .await?
      .send()
      .await?
      .error_for_status()?
      .json::<AppResponse<WorkspaceUsageLimit>>()
      .await?
      .into_data()
  }
}