这是可回退普通版的提交

src/common/utils.rs

@@ -1,11 +1,12 @@
 mod checksum;
 pub use checksum::*;
-pub mod tokens;
+mod tokens;
+pub use tokens::*;
 use prost::Message as _;
 
 use crate::{app::constant::CURSOR_API2_GET_USER_INFO, chat::aiserver::v1::GetUserInfoResponse};
 
-use super::models::usage::UserUsageInfo;
+use super::models::usage::{StripeProfile, UserUsageInfo};
 
 pub fn parse_bool_from_env(key: &str, default: bool) -> bool {
     std::env::var(key)
@@ -43,8 +44,49 @@ pub async fn get_user_usage(auth_token: &str, checksum: &str) -> Option<UserUsag
         .ok()?;
     let user_info = GetUserInfoResponse::decode(response.as_ref()).ok()?;
 
+    let (mtype, trial_days) = get_stripe_profile(auth_token).await?;
+
     user_info.usage.map(|user_usage| UserUsageInfo {
         fast_requests: i32_to_u32(user_usage.gpt4_requests),
         max_fast_requests: i32_to_u32(user_usage.gpt4_max_requests),
+        mtype,
+        trial_days,
     })
 }
+
+pub async fn get_stripe_profile(auth_token: &str) -> Option<(String, u32)> {
+    let client = super::client::build_profile_client(auth_token);
+    let response = client.send().await.ok()?.json::<StripeProfile>().await.ok()?;
+    Some((response.membership_type, i32_to_u32(response.days_remaining_on_trial)))
+}
+
+pub fn validate_token_and_checksum(auth_token: &str) -> Option<(String, String, Option<String>)> {
+    // 提取 token、checksum 和可能的 alias
+    let (token, checksum, alias) = {
+        // 先尝试提取 alias
+        let (token_part, alias) = if let Some(pos) = auth_token.find("::") {
+            let (alias, rest) = auth_token.split_at(pos);
+            (&rest[2..], Some(alias))
+        } else if let Some(pos) = auth_token.find("%3A%3A") {
+            let (alias, rest) = auth_token.split_at(pos);
+            (&rest[6..], Some(alias))
+        } else {
+            (auth_token, None)
+        };
+
+        // 提取 token 和 checksum
+        if let Some(comma_pos) = token_part.find(',') {
+            let (token, checksum) = token_part.split_at(comma_pos);
+            (token, &checksum[1..], alias)
+        } else {
+            return None; // 缺少必要的 checksum
+        }
+    };
+
+    // 验证 token 和 checksum 有效性
+    if validate_token(token) && validate_checksum(checksum) {
+        Some((token.to_string(), checksum.to_string(), alias.map(String::from)))
+    } else {
+        None
+    }
+}