127 lines
3.6 KiB
Rust
127 lines
3.6 KiB
Rust
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()
|
|
}
|
|
}
|