v0.1.3-rc.3

src/chat/route/usage.rs

@@ -0,0 +1,36 @@
+use crate::{
+    app::model::AppState,
+    common::{models::usage::GetUserInfo, utils::get_user_usage},
+};
+use axum::{
+    extract::{Query, State},
+    Json,
+};
+use serde::Deserialize;
+use std::sync::Arc;
+use tokio::sync::Mutex;
+
+#[derive(Deserialize)]
+pub struct GetUserInfoQuery {
+    alias: String,
+}
+
+pub async fn get_user_info(
+    State(state): State<Arc<Mutex<AppState>>>,
+    Query(query): Query<GetUserInfoQuery>,
+) -> Json<GetUserInfo> {
+    let token_infos = &state.lock().await.token_infos;
+    let token_info = token_infos
+        .iter()
+        .find(|token_info| token_info.alias == Some(query.alias.clone()));
+
+    let (auth_token, checksum) = match token_info {
+        Some(token_info) => (token_info.token.clone(), token_info.checksum.clone()),
+        None => return Json(GetUserInfo::Error("No data".to_string())),
+    };
+
+    match get_user_usage(&auth_token, &checksum).await {
+        Some(usage) => Json(GetUserInfo::Usage(usage)),
+        None => Json(GetUserInfo::Error("No data".to_string())),
+    }
+}