v0.1.3-rc.2

.gitignore

@@ -16,6 +16,4 @@ node_modules
 /cursor-api.exe
 /release
 
-/static/readme.html
 /*.py
-/src/decoder.rs

Cargo.lock

@@ -292,7 +292,7 @@ dependencies = [
 
 [[package]]
 name = "cursor-api"
-version = "0.1.3-rc.1"
+version = "0.1.3-rc.2"
 dependencies = [
  "axum",
  "base64",
@@ -305,6 +305,7 @@ dependencies = [
  "hex",
  "image",
  "lazy_static",
+ "paste",
  "prost",
  "prost-build",
  "rand",
@@ -1108,6 +1109,12 @@ dependencies = [
  "vcpkg",
 ]
 
+[[package]]
+name = "paste"
+version = "1.0.15"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "57c0d7b74b563b49d38dae00a0c37d4d6de9b432382b2892f0574ddcae73fd0a"
+
 [[package]]
 name = "percent-encoding"
 version = "2.3.1"

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "cursor-api"
-version = "0.1.3-rc.1"
+version = "0.1.3-rc.2"
 edition = "2021"
 authors = ["wisdgod <nav@wisdgod.com>"]
 # license = "MIT"
@@ -26,6 +26,7 @@ gif = { version = "0.13.1", default-features = false, features = ["std"] }
 hex = { version = "0.4.3", default-features = false, features = ["std"] }
 image = { version = "0.25.5", default-features = false, features = ["jpeg", "png", "gif", "webp"] }
 lazy_static = "1.5.0"
+paste = "1.0.15"
 prost = "0.13.4"
 rand = { version = "0.8.5", default-features = false, features = ["std", "std_rng"] }
 regex = { version = "1.11.1", default-features = false, features = ["std", "perf"] }

README.md

@@ -2,9 +2,9 @@
 
 ## 获取key
 
-1. 访问 [www.cursor.com](https://www.cursor.com) 并完成注册登录（赠送 250 次快速响应，可通过删除账号再注册重置）
+1. 访问 [www.cursor.com](https://www.cursor.com) 并完成注册登录
 2. 在浏览器中打开开发者工具（F12）
-3. 找到 Application-Cookies 中名为 `WorkosCursorSessionToken` 的值并复制其第3个字段，%3A%3A是::的编码，cookie用:分隔值
+3. 在 Application-Cookies 中查找名为 `WorkosCursorSessionToken` 的条目，并复制其第三个字段。请注意，%3A%3A 是 :: 的 URL 编码形式，cookie 的值使用冒号 (:) 进行分隔。
 
 ## 接口说明
 
@@ -107,7 +107,6 @@ gpt-4o
 claude-3-opus
 cursor-fast
 cursor-small
-gpt-3.5
 gpt-3.5-turbo
 gpt-4-turbo-2024-04-09
 gpt-4o-128k

build.rs

@@ -139,7 +139,7 @@ fn minify_assets() -> Result<()> {
 
 fn main() -> Result<()> {
     // Proto 文件处理
-    println!("cargo:rerun-if-changed=src/aiserver/v1/aiserver.proto");
+    println!("cargo:rerun-if-changed=src/chat/aiserver/v1/aiserver.proto");
     let mut config = prost_build::Config::new();
     // config.type_attribute(".", "#[derive(serde::Serialize, serde::Deserialize)]");
     // config.type_attribute(
@@ -147,7 +147,7 @@ fn main() -> Result<()> {
     //     "#[derive(serde::Serialize, serde::Deserialize)]"
     // );
     config
-        .compile_protos(&["src/aiserver/v1/aiserver.proto"], &["src/aiserver/v1/"])
+        .compile_protos(&["src/chat/aiserver/v1/aiserver.proto"], &["src/chat/aiserver/v1/"])
         .unwrap();
 
     // 静态资源文件处理

src/app.rs

@@ -1,5 +1,7 @@
-pub mod models;
+pub mod client;
+pub mod config;
 pub mod constant;
+pub mod models;
+pub mod statics;
 pub mod token;
 pub mod utils;
-pub mod client;

src/app/config.rs

@@ -0,0 +1,301 @@
+use super::{
+    constant::*,
+    models::{AppConfig, AppState},
+    statics::*,
+};
+use crate::common::models::{
+    config::{ConfigData, ConfigUpdateRequest},
+    ApiStatus, ErrorResponse, NormalResponse,
+};
+use axum::{
+    extract::State,
+    http::{HeaderMap, StatusCode},
+    Json,
+};
+use std::sync::Arc;
+use tokio::sync::Mutex;
+
+pub async fn handle_config_update(
+    State(_state): State<Arc<Mutex<AppState>>>,
+    headers: HeaderMap,
+    Json(request): Json<ConfigUpdateRequest>,
+) -> Result<Json<NormalResponse<ConfigData>>, (StatusCode, Json<ErrorResponse>)> {
+    let auth_header = headers
+        .get(HEADER_NAME_AUTHORIZATION)
+        .and_then(|h| h.to_str().ok())
+        .and_then(|h| h.strip_prefix(AUTHORIZATION_BEARER_PREFIX))
+        .ok_or((
+            StatusCode::UNAUTHORIZED,
+            Json(ErrorResponse {
+                status: ApiStatus::Failed,
+                code: Some(401),
+                error: Some("未提供认证令牌".to_string()),
+                message: None,
+            }),
+        ))?;
+
+    if auth_header != get_auth_token() {
+        return Err((
+            StatusCode::UNAUTHORIZED,
+            Json(ErrorResponse {
+                status: ApiStatus::Failed,
+                code: Some(401),
+                error: Some("无效的认证令牌".to_string()),
+                message: None,
+            }),
+        ));
+    }
+
+    match request.action.as_str() {
+        "get" => Ok(Json(NormalResponse {
+            status: ApiStatus::Success,
+            data: Some(ConfigData {
+                page_content: AppConfig::get_page_content(&request.path),
+                enable_stream_check: AppConfig::get_stream_check(),
+                include_stop_stream: AppConfig::get_stop_stream(),
+                vision_ability: AppConfig::get_vision_ability(),
+                enable_slow_pool: AppConfig::get_slow_pool(),
+                enable_all_claude: AppConfig::get_allow_claude(),
+                check_usage_models: AppConfig::get_usage_check(),
+            }),
+            message: None,
+        })),
+
+        "update" => {
+            // 处理页面内容更新
+            if !request.path.is_empty() && request.content.is_some() {
+                let content = request.content.unwrap();
+
+                if let Err(e) = AppConfig::update_page_content(&request.path, content) {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("更新页面内容失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 处理 enable_stream_check 更新
+            if let Some(enable_stream_check) = request.enable_stream_check {
+                if let Err(e) = AppConfig::update_stream_check(enable_stream_check) {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("更新 enable_stream_check 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 处理 include_stop_stream 更新
+            if let Some(include_stop_stream) = request.include_stop_stream {
+                if let Err(e) = AppConfig::update_stop_stream(include_stop_stream) {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("更新 include_stop_stream 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 处理 vision_ability 更新
+            if let Some(vision_ability) = request.vision_ability {
+                if let Err(e) = AppConfig::update_vision_ability(vision_ability) {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("更新 vision_ability 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 处理 enable_slow_pool 更新
+            if let Some(enable_slow_pool) = request.enable_slow_pool {
+                if let Err(e) = AppConfig::update_slow_pool(enable_slow_pool) {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("更新 enable_slow_pool 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 处理 enable_all_claude 更新
+            if let Some(enable_all_claude) = request.enable_all_claude {
+                if let Err(e) = AppConfig::update_allow_claude(enable_all_claude) {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("更新 enable_all_claude 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 处理 check_usage_models 更新
+            if let Some(check_usage_models) = request.check_usage_models {
+                if let Err(e) = AppConfig::update_usage_check(check_usage_models) {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("更新 check_usage_models 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            Ok(Json(NormalResponse {
+                status: ApiStatus::Success,
+                data: None,
+                message: Some("配置已更新".to_string()),
+            }))
+        }
+
+        "reset" => {
+            // 重置页面内容
+            if !request.path.is_empty() {
+                if let Err(e) = AppConfig::reset_page_content(&request.path) {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("重置页面内容失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 重置 enable_stream_check
+            if request.enable_stream_check.is_some() {
+                if let Err(e) = AppConfig::reset_stream_check() {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("重置 enable_stream_check 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 重置 include_stop_stream
+            if request.include_stop_stream.is_some() {
+                if let Err(e) = AppConfig::reset_stop_stream() {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("重置 include_stop_stream 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 重置 vision_ability
+            if request.vision_ability.is_some() {
+                if let Err(e) = AppConfig::reset_vision_ability() {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("重置 vision_ability 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 重置 enable_slow_pool
+            if request.enable_slow_pool.is_some() {
+                if let Err(e) = AppConfig::reset_slow_pool() {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("重置 enable_slow_pool 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 重置 enable_all_claude
+            if request.enable_all_claude.is_some() {
+                if let Err(e) = AppConfig::reset_allow_claude() {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("重置 enable_slow_pool 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+
+            // 重置 check_usage_models
+            if request.check_usage_models.is_some() {
+                if let Err(e) = AppConfig::reset_usage_check() {
+                    return Err((
+                        StatusCode::INTERNAL_SERVER_ERROR,
+                        Json(ErrorResponse {
+                            status: ApiStatus::Failed,
+                            code: Some(500),
+                            error: Some(format!("重置 check_usage_models 失败: {}", e)),
+                            message: None,
+                        }),
+                    ));
+                }
+            }
+            Ok(Json(NormalResponse {
+                status: ApiStatus::Success,
+                data: None,
+                message: Some("配置已重置".to_string()),
+            }))
+        }
+
+        _ => Err((
+            StatusCode::BAD_REQUEST,
+            Json(ErrorResponse {
+                status: ApiStatus::Failed,
+                code: Some(400),
+                error: Some("无效的操作类型".to_string()),
+                message: None,
+            }),
+        )),
+    }
+}

src/app/constant.rs

@@ -1,64 +1,71 @@
-pub const PKG_VERSION: &str = env!("CARGO_PKG_VERSION");
-pub const PKG_NAME: &str = env!("CARGO_PKG_NAME");
-pub const PKG_DESCRIPTION: &str = env!("CARGO_PKG_DESCRIPTION");
-pub const PKG_AUTHORS: &str = env!("CARGO_PKG_AUTHORS");
-pub const PKG_REPOSITORY: &str = env!("CARGO_PKG_REPOSITORY");
+macro_rules! def_pub_const {
+    ($name:ident, $value:expr) => {
+        pub const $name: &'static str = $value;
+    };
+}
 
-pub const ROUTER_ROOT_PATH: &str = "/";
-pub const ROUTER_HEALTH_PATH: &str = "/health";
-pub const ROUTER_GET_CHECKSUM: &str = "/get-checksum";
-pub const ROUTER_GET_USER_INFO_PATH: &str = "/get-user-info";
-pub const ROUTER_LOGS_PATH: &str = "/logs";
-pub const ROUTER_CONFIG_PATH: &str = "/config";
-pub const ROUTER_TOKENINFO_PATH: &str = "/tokeninfo";
-pub const ROUTER_GET_TOKENINFO_PATH: &str = "/get-tokeninfo";
-pub const ROUTER_UPDATE_TOKENINFO_PATH: &str = "/update-tokeninfo";
-pub const ROUTER_ENV_EXAMPLE_PATH: &str = "/env-example";
-pub const ROUTER_SHARED_STYLES_PATH: &str = "/static/shared-styles.css";
-pub const ROUTER_SHARED_JS_PATH: &str = "/static/shared.js";
+def_pub_const!(PKG_VERSION, env!("CARGO_PKG_VERSION"));
+def_pub_const!(PKG_NAME, env!("CARGO_PKG_NAME"));
+def_pub_const!(PKG_DESCRIPTION, env!("CARGO_PKG_DESCRIPTION"));
+def_pub_const!(PKG_AUTHORS, env!("CARGO_PKG_AUTHORS"));
+def_pub_const!(PKG_REPOSITORY, env!("CARGO_PKG_REPOSITORY"));
 
-pub const STATUS: &str = "status";
-pub const MESSAGE: &str = "message";
-pub const ERROR: &str = "error";
+def_pub_const!(EMPTY_STRING, "");
 
-pub const TOKEN_FILE: &str = "token_file";
-pub const TOKEN_LIST_FILE: &str = "token_list_file";
-pub const TOKENS: &str = "tokens";
-pub const TOKEN_LIST: &str = "token_list";
+def_pub_const!(ROUTE_ROOT_PATH, "/");
+def_pub_const!(ROUTE_HEALTH_PATH, "/health");
+def_pub_const!(ROUTE_GET_CHECKSUM, "/get-checksum");
+def_pub_const!(ROUTE_GET_USER_INFO_PATH, "/get-user-info");
+def_pub_const!(ROUTE_LOGS_PATH, "/logs");
+def_pub_const!(ROUTE_CONFIG_PATH, "/config");
+def_pub_const!(ROUTE_TOKENINFO_PATH, "/tokeninfo");
+def_pub_const!(ROUTE_GET_TOKENINFO_PATH, "/get-tokeninfo");
+def_pub_const!(ROUTE_UPDATE_TOKENINFO_PATH, "/update-tokeninfo");
+def_pub_const!(ROUTE_ENV_EXAMPLE_PATH, "/env-example");
+def_pub_const!(ROUTE_STATIC_PATH, "/static/:path");
+def_pub_const!(ROUTE_SHARED_STYLES_PATH, "/static/shared-styles.css");
+def_pub_const!(ROUTE_SHARED_JS_PATH, "/static/shared.js");
+def_pub_const!(ROUTE_ABOUT_PATH, "/about");
+def_pub_const!(ROUTE_README_PATH, "/readme");
 
-pub const STATUS_SUCCESS: &str = "success";
-pub const STATUS_FAILED: &str = "failed";
+def_pub_const!(STATUS, "status");
+def_pub_const!(MESSAGE, "message");
+def_pub_const!(ERROR, "error");
 
-pub const HEADER_NAME_CONTENT_TYPE: &str = "content-type";
-pub const HEADER_NAME_AUTHORIZATION: &str = "Authorization";
+def_pub_const!(TOKEN_FILE, "token_file");
+def_pub_const!(DEFAULT_TOKEN_FILE_NAME, ".token");
+def_pub_const!(TOKEN_LIST_FILE, "token_list_file");
+def_pub_const!(DEFAULT_TOKEN_LIST_FILE_NAME, ".token-list");
+def_pub_const!(TOKENS, "tokens");
+def_pub_const!(TOKEN_LIST, "token_list");
 
-pub const CONTENT_TYPE_PROTO: &str = "application/proto";
-pub const CONTENT_TYPE_CONNECT_PROTO: &str = "application/connect+proto";
-pub const CONTENT_TYPE_TEXT_HTML_WITH_UTF8: &str = "text/html;charset=utf-8";
-pub const CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8: &str = "text/plain;charset=utf-8";
+def_pub_const!(STATUS_SUCCESS, "success");
+def_pub_const!(STATUS_FAILED, "failed");
 
-pub const AUTHORIZATION_BEARER_PREFIX: &str = "Bearer ";
+def_pub_const!(HEADER_NAME_CONTENT_TYPE, "content-type");
+def_pub_const!(HEADER_NAME_AUTHORIZATION, "authorization");
+def_pub_const!(HEADER_NAME_LOCATION, "Location");
 
-pub const OBJECT_CHAT_COMPLETION: &str = "chat.completion";
-pub const OBJECT_CHAT_COMPLETION_CHUNK: &str = "chat.completion.chunk";
+def_pub_const!(CONTENT_TYPE_PROTO, "application/proto");
+def_pub_const!(CONTENT_TYPE_CONNECT_PROTO, "application/connect+proto");
+def_pub_const!(CONTENT_TYPE_TEXT_HTML_WITH_UTF8, "text/html;charset=utf-8");
+def_pub_const!(CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8, "text/plain;charset=utf-8");
+def_pub_const!(CONTENT_TYPE_TEXT_CSS_WITH_UTF8, "text/css;charset=utf-8");
+def_pub_const!(CONTENT_TYPE_TEXT_JS_WITH_UTF8, "text/javascript;charset=utf-8");
 
-pub const CURSOR_API2_HOST: &str = "api2.cursor.sh";
-pub const CURSOR_API2_BASE_URL: &str = "https://api2.cursor.sh/aiserver.v1.AiService/";
+def_pub_const!(AUTHORIZATION_BEARER_PREFIX, "Bearer ");
 
-pub const CURSOR_API2_STREAM_CHAT: &str = "StreamChat";
-pub const CURSOR_API2_GET_USER_INFO: &str = "GetUserInfo";
+def_pub_const!(OBJECT_CHAT_COMPLETION, "chat.completion");
+def_pub_const!(OBJECT_CHAT_COMPLETION_CHUNK, "chat.completion.chunk");
 
-pub const FINISH_REASON_STOP: &str = "stop";
+def_pub_const!(CURSOR_API2_HOST, "api2.cursor.sh");
+def_pub_const!(CURSOR_API2_BASE_URL, "https://api2.cursor.sh/aiserver.v1.AiService/");
 
-pub const LONG_CONTEXT_MODELS: [&str; 4] = [
-    "gpt-4o-128k",
-    "gemini-1.5-flash-500k",
-    "claude-3-haiku-200k",
-    "claude-3-5-sonnet-200k",
-];
+def_pub_const!(CURSOR_API2_STREAM_CHAT, "StreamChat");
+def_pub_const!(CURSOR_API2_GET_USER_INFO, "GetUserInfo");
 
-pub const MODEL_OBJECT: &str = "model";
-pub const ANTHROPIC: &str = "anthropic";
-pub const CURSOR: &str = "cursor";
-pub const GOOGLE: &str = "google";
-pub const OPENAI: &str = "openai";
+def_pub_const!(FINISH_REASON_STOP, "stop");
+
+def_pub_const!(ERR_UPDATE_CONFIG, "无法更新配置");
+def_pub_const!(ERR_RESET_CONFIG, "无法重置配置");
+def_pub_const!(ERR_INVALID_PATH, "无效的路径");

src/app/models.rs

@@ -1,6 +1,5 @@
 use super::{constant::*, token::UserUsageInfo};
-use crate::message::Message;
-use chrono::{DateTime, Local};
+use crate::chat::models::Message;
 use lazy_static::lazy_static;
 use serde::{Deserialize, Serialize};
 use std::sync::RwLock;
@@ -23,20 +22,19 @@ impl Default for PageContent {
     }
 }
 
+mod usage_check;
+pub use usage_check::UsageCheck;
+
 // 静态配置
 #[derive(Clone)]
 pub struct AppConfig {
-    enable_stream_check: bool,
-    include_stop_stream: bool,
+    stream_check: bool,
+    stop_stream: bool,
     vision_ability: VisionAbility,
-    enable_slow_pool: bool,
+    slow_pool: bool,
     allow_claude: bool,
-    auth_token: String,
-    token_file: String,
-    token_list_file: String,
-    route_prefix: String,
-    pub start_time: chrono::DateTime<chrono::Local>,
     pages: Pages,
+    usage_check: UsageCheck,
 }
 
 #[derive(Serialize, Deserialize, Clone)]
@@ -50,12 +48,12 @@ pub enum VisionAbility {
 }
 
 impl VisionAbility {
-    pub fn from_str(s: &str) -> Result<Self, &'static str> {
+    pub fn from_str(s: &str) -> Self {
         match s.to_lowercase().as_str() {
-            "none" | "disabled" => Ok(Self::None),
-            "base64" | "base64-only" => Ok(Self::Base64),
-            "all" | "base64-http" => Ok(Self::All),
-            _ => Err("Invalid VisionAbility value"),
+            "none" | "disabled" => Self::None,
+            "base64" | "base64-only" => Self::Base64,
+            "all" | "base64-http" => Self::All,
+            _ => Self::default(),
         }
     }
 }
@@ -66,7 +64,7 @@ impl Default for VisionAbility {
     }
 }
 
-#[derive(Clone)]
+#[derive(Clone, Default)]
 pub struct Pages {
     pub root_content: PageContent,
     pub logs_content: PageContent,
@@ -74,19 +72,8 @@ pub struct Pages {
     pub tokeninfo_content: PageContent,
     pub shared_styles_content: PageContent,
     pub shared_js_content: PageContent,
-}
-
-impl Default for Pages {
-    fn default() -> Self {
-        Self {
-            root_content: PageContent::Default,
-            logs_content: PageContent::Default,
-            config_content: PageContent::Default,
-            tokeninfo_content: PageContent::Default,
-            shared_styles_content: PageContent::Default,
-            shared_js_content: PageContent::Default,
-        }
-    }
+    pub about_content: PageContent,
+    pub readme_content: PageContent,
 }
 
 // 运行时状态
@@ -105,58 +92,72 @@ lazy_static! {
 impl Default for AppConfig {
     fn default() -> Self {
         Self {
-            enable_stream_check: true,
-            include_stop_stream: true,
+            stream_check: true,
+            stop_stream: true,
             vision_ability: VisionAbility::Base64,
-            enable_slow_pool: false,
+            slow_pool: false,
             allow_claude: false,
-            auth_token: String::new(),
-            token_file: ".token".to_string(),
-            token_list_file: ".token-list".to_string(),
-            route_prefix: String::new(),
-            start_time: chrono::Local::now(),
             pages: Pages::default(),
+            usage_check: UsageCheck::default(),
         }
     }
 }
 
+macro_rules! config_methods {
+    ($($field:ident: $type:ty, $default:expr;)*) => {
+        $(
+            paste::paste! {
+                pub fn [<get_ $field>]() -> $type {
+                    APP_CONFIG
+                        .read()
+                        .map(|config| config.$field.clone())
+                        .unwrap_or($default)
+                }
+                
+                pub fn [<update_ $field>](value: $type) -> Result<(), &'static str> {
+                    if let Ok(mut config) = APP_CONFIG.write() {
+                        config.$field = value;
+                        Ok(())
+                    } else {
+                        Err(ERR_UPDATE_CONFIG)
+                    }
+                }
+                
+                pub fn [<reset_ $field>]() -> Result<(), &'static str> {
+                    if let Ok(mut config) = APP_CONFIG.write() {
+                        config.$field = $default;
+                        Ok(())
+                    } else {
+                        Err(ERR_RESET_CONFIG)
+                    }
+                }
+            }
+        )*
+    };
+}
+
 impl AppConfig {
     pub fn init(
-        enable_stream_check: bool,
-        include_stop_stream: bool,
+        stream_check: bool,
+        stop_stream: bool,
         vision_ability: VisionAbility,
-        enable_slow_pool: bool,
+        slow_pool: bool,
         allow_claude: bool,
-        auth_token: String,
-        token_file: String,
-        token_list_file: String,
-        route_prefix: String,
     ) {
         if let Ok(mut config) = APP_CONFIG.write() {
-            config.enable_stream_check = enable_stream_check;
-            config.include_stop_stream = include_stop_stream;
+            config.stream_check = stream_check;
+            config.stop_stream = stop_stream;
             config.vision_ability = vision_ability;
-            config.enable_slow_pool = enable_slow_pool;
+            config.slow_pool = slow_pool;
             config.allow_claude = allow_claude;
-            config.auth_token = auth_token;
-            config.token_file = token_file;
-            config.token_list_file = token_list_file;
-            config.route_prefix = route_prefix;
         }
     }
 
-    pub fn get_stream_check() -> bool {
-        APP_CONFIG
-            .read()
-            .map(|config| config.enable_stream_check)
-            .unwrap_or(true)
-    }
-
-    pub fn get_stop_stream() -> bool {
-        APP_CONFIG
-            .read()
-            .map(|config| config.include_stop_stream)
-            .unwrap_or(true)
+    config_methods! {
+        stream_check: bool, true;
+        stop_stream: bool, true;
+        slow_pool: bool, false;
+        allow_claude: bool, false;
     }
 
     pub fn get_vision_ability() -> VisionAbility {
@@ -166,137 +167,62 @@ impl AppConfig {
             .unwrap_or_default()
     }
 
-    pub fn get_slow_pool() -> bool {
-        APP_CONFIG
-            .read()
-            .map(|config| config.enable_slow_pool)
-            .unwrap_or(false)
-    }
-
-    pub fn get_allow_claude() -> bool {
-        APP_CONFIG
-            .read()
-            .map(|config| config.allow_claude)
-            .unwrap_or(false)
-    }
-
-    pub fn get_auth_token() -> String {
-        APP_CONFIG
-            .read()
-            .map(|config| config.auth_token.clone())
-            .unwrap_or_default()
-    }
-
-    pub fn get_token_file() -> String {
-        APP_CONFIG
-            .read()
-            .map(|config| config.token_file.clone())
-            .unwrap_or_default()
-    }
-
-    pub fn get_token_list_file() -> String {
-        APP_CONFIG
-            .read()
-            .map(|config| config.token_list_file.clone())
-            .unwrap_or_default()
-    }
-
-    pub fn get_route_prefix() -> String {
-        APP_CONFIG
-            .read()
-            .map(|config| config.route_prefix.clone())
-            .unwrap_or_default()
-    }
-
     pub fn get_page_content(path: &str) -> Option<PageContent> {
         APP_CONFIG.read().ok().map(|config| match path {
-            ROUTER_ROOT_PATH => config.pages.root_content.clone(),
-            ROUTER_LOGS_PATH => config.pages.logs_content.clone(),
-            ROUTER_CONFIG_PATH => config.pages.config_content.clone(),
-            ROUTER_TOKENINFO_PATH => config.pages.tokeninfo_content.clone(),
-            ROUTER_SHARED_STYLES_PATH => config.pages.shared_styles_content.clone(),
-            ROUTER_SHARED_JS_PATH => config.pages.shared_js_content.clone(),
-            _ => PageContent::Default,
+            ROUTE_ROOT_PATH => config.pages.root_content.clone(),
+            ROUTE_LOGS_PATH => config.pages.logs_content.clone(),
+            ROUTE_CONFIG_PATH => config.pages.config_content.clone(),
+            ROUTE_TOKENINFO_PATH => config.pages.tokeninfo_content.clone(),
+            ROUTE_SHARED_STYLES_PATH => config.pages.shared_styles_content.clone(),
+            ROUTE_SHARED_JS_PATH => config.pages.shared_js_content.clone(),
+            ROUTE_ABOUT_PATH => config.pages.about_content.clone(),
+            ROUTE_README_PATH => config.pages.readme_content.clone(),
+            _ => PageContent::default(),
         })
     }
 
-    pub fn update_stream_check(enable: bool) -> Result<(), &'static str> {
-        if let Ok(mut config) = APP_CONFIG.write() {
-            config.enable_stream_check = enable;
-            Ok(())
-        } else {
-            Err("无法更新配置")
-        }
+    pub fn get_usage_check() -> UsageCheck {
+        APP_CONFIG
+            .read()
+            .map(|config| config.usage_check.clone())
+            .unwrap_or_default()
     }
 
-    pub fn update_stop_stream(enable: bool) -> Result<(), &'static str> {
-        if let Ok(mut config) = APP_CONFIG.write() {
-            config.include_stop_stream = enable;
-            Ok(())
-        } else {
-            Err("无法更新配置")
-        }
-    }
 
     pub fn update_vision_ability(new_ability: VisionAbility) -> Result<(), &'static str> {
         if let Ok(mut config) = APP_CONFIG.write() {
             config.vision_ability = new_ability;
             Ok(())
         } else {
-            Err("无法更新配置")
-        }
-    }
-
-    pub fn update_slow_pool(enable: bool) -> Result<(), &'static str> {
-        if let Ok(mut config) = APP_CONFIG.write() {
-            config.enable_slow_pool = enable;
-            Ok(())
-        } else {
-            Err("无法更新配置")
-        }
-    }
-
-    pub fn update_allow_claude(enable: bool) -> Result<(), &'static str> {
-        if let Ok(mut config) = APP_CONFIG.write() {
-            config.allow_claude = enable;
-            Ok(())
-        } else {
-            Err("无法更新配置")
+            Err(ERR_UPDATE_CONFIG)
         }
     }
 
     pub fn update_page_content(path: &str, content: PageContent) -> Result<(), &'static str> {
         if let Ok(mut config) = APP_CONFIG.write() {
             match path {
-                ROUTER_ROOT_PATH => config.pages.root_content = content,
-                ROUTER_LOGS_PATH => config.pages.logs_content = content,
-                ROUTER_CONFIG_PATH => config.pages.config_content = content,
-                ROUTER_TOKENINFO_PATH => config.pages.tokeninfo_content = content,
-                ROUTER_SHARED_STYLES_PATH => config.pages.shared_styles_content = content,
-                ROUTER_SHARED_JS_PATH => config.pages.shared_js_content = content,
-                _ => return Err("无效的路径"),
+                ROUTE_ROOT_PATH => config.pages.root_content = content,
+                ROUTE_LOGS_PATH => config.pages.logs_content = content,
+                ROUTE_CONFIG_PATH => config.pages.config_content = content,
+                ROUTE_TOKENINFO_PATH => config.pages.tokeninfo_content = content,
+                ROUTE_SHARED_STYLES_PATH => config.pages.shared_styles_content = content,
+                ROUTE_SHARED_JS_PATH => config.pages.shared_js_content = content,
+                ROUTE_ABOUT_PATH => config.pages.about_content = content,
+                ROUTE_README_PATH => config.pages.readme_content = content,
+                _ => return Err(ERR_INVALID_PATH),
             }
             Ok(())
         } else {
-            Err("无法更新配置")
+            Err(ERR_UPDATE_CONFIG)
         }
     }
 
-    pub fn reset_stream_check() -> Result<(), &'static str> {
+    pub fn update_usage_check(rule: UsageCheck) -> Result<(), &'static str> {
         if let Ok(mut config) = APP_CONFIG.write() {
-            config.enable_stream_check = true;
+            config.usage_check = rule;
             Ok(())
         } else {
-            Err("无法重置配置")
-        }
-    }
-
-    pub fn reset_stop_stream() -> Result<(), &'static str> {
-        if let Ok(mut config) = APP_CONFIG.write() {
-            config.include_stop_stream = true;
-            Ok(())
-        } else {
-            Err("无法重置配置")
+            Err(ERR_UPDATE_CONFIG)
         }
     }
 
@@ -305,44 +231,37 @@ impl AppConfig {
             config.vision_ability = VisionAbility::Base64;
             Ok(())
         } else {
-            Err("无法重置配置")
-        }
-    }
-
-    pub fn reset_slow_pool() -> Result<(), &'static str> {
-        if let Ok(mut config) = APP_CONFIG.write() {
-            config.enable_slow_pool = false;
-            Ok(())
-        } else {
-            Err("无法重置配置")
-        }
-    }
-
-    pub fn reset_allow_claude() -> Result<(), &'static str> {
-        if let Ok(mut config) = APP_CONFIG.write() {
-            config.allow_claude = false;
-            Ok(())
-        } else {
-            Err("无法重置配置")
+            Err(ERR_RESET_CONFIG)
         }
     }
 
     pub fn reset_page_content(path: &str) -> Result<(), &'static str> {
         if let Ok(mut config) = APP_CONFIG.write() {
             match path {
-                ROUTER_ROOT_PATH => config.pages.root_content = PageContent::Default,
-                ROUTER_LOGS_PATH => config.pages.logs_content = PageContent::Default,
-                ROUTER_CONFIG_PATH => config.pages.config_content = PageContent::Default,
-                ROUTER_TOKENINFO_PATH => config.pages.tokeninfo_content = PageContent::Default,
-                ROUTER_SHARED_STYLES_PATH => {
-                    config.pages.shared_styles_content = PageContent::Default
+                ROUTE_ROOT_PATH => config.pages.root_content = PageContent::default(),
+                ROUTE_LOGS_PATH => config.pages.logs_content = PageContent::default(),
+                ROUTE_CONFIG_PATH => config.pages.config_content = PageContent::default(),
+                ROUTE_TOKENINFO_PATH => config.pages.tokeninfo_content = PageContent::default(),
+                ROUTE_SHARED_STYLES_PATH => {
+                    config.pages.shared_styles_content = PageContent::default()
                 }
-                ROUTER_SHARED_JS_PATH => config.pages.shared_js_content = PageContent::Default,
-                _ => return Err("无效的路径"),
+                ROUTE_SHARED_JS_PATH => config.pages.shared_js_content = PageContent::default(),
+                ROUTE_ABOUT_PATH => config.pages.about_content = PageContent::default(),
+                ROUTE_README_PATH => config.pages.readme_content = PageContent::default(),
+                _ => return Err(ERR_INVALID_PATH),
             }
             Ok(())
         } else {
-            Err("无法重置配置")
+            Err(ERR_RESET_CONFIG)
+        }
+    }
+
+    pub fn reset_usage_check() -> Result<(), &'static str> {
+        if let Ok(mut config) = APP_CONFIG.write() {
+            config.usage_check = UsageCheck::default();
+            Ok(())
+        } else {
+            Err(ERR_RESET_CONFIG)
         }
     }
 }
@@ -362,31 +281,16 @@ impl AppState {
     }
 }
 
-// 模型定义
-#[derive(Serialize, Clone)]
-pub struct Model {
-    pub id: String,
-    pub created: i64,
-    pub object: String,
-    pub owned_by: String,
-}
-
-// impl Model {
-//     pub fn is_pesticide(&self) -> bool {
-//         !(self.owned_by.as_str() == CURSOR || self.id.as_str() == "gpt-4o-mini") 
-//     }
-// }
-
 // 请求日志
 #[derive(Serialize, Clone)]
 pub struct RequestLog {
-    pub timestamp: DateTime<Local>,
+    pub timestamp: chrono::DateTime<chrono::Local>,
     pub model: String,
     pub token_info: TokenInfo,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub prompt: Option<String>,
     pub stream: bool,
-    pub status: String,
+    pub status: &'static str,
     #[serde(skip_serializing_if = "Option::is_none")]
     pub error: Option<String>,
 }
@@ -424,24 +328,3 @@ pub struct TokenUpdateRequest {
     #[serde(default)]
     pub token_list: Option<String>,
 }
-
-// 添加用于接收更新请求的结构体
-#[derive(Deserialize)]
-pub struct ConfigUpdateRequest {
-    #[serde(default)]
-    pub action: String, // "get", "update", "reset"
-    #[serde(default)]
-    pub path: String,
-    #[serde(default)]
-    pub content: Option<PageContent>, // "default", "text", "html"
-    #[serde(default)]
-    pub enable_stream_check: Option<bool>,
-    #[serde(default)]
-    pub include_stop_stream: Option<bool>,
-    #[serde(default)]
-    pub vision_ability: Option<VisionAbility>,
-    #[serde(default)]
-    pub enable_slow_pool: Option<bool>,
-    #[serde(default)]
-    pub enable_all_claude: Option<bool>,
-}

src/app/models/usage_check.rs

@@ -0,0 +1,91 @@
+use crate::chat::constant::AVAILABLE_MODELS;
+use serde::{Deserialize, Serialize};
+
+#[derive(Clone)]
+pub enum UsageCheck {
+    None,
+    Default,
+    All,
+    Custom(Vec<&'static str>),
+}
+
+impl Default for UsageCheck {
+    fn default() -> Self {
+        Self::Default
+    }
+}
+
+impl Serialize for UsageCheck {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: serde::Serializer,
+    {
+        use serde::ser::SerializeStruct;
+        let mut state = serializer.serialize_struct("UsageCheck", 1)?;
+        match self {
+            UsageCheck::None => {
+                state.serialize_field("type", "none")?;
+            }
+            UsageCheck::Default => {
+                state.serialize_field("type", "default")?;
+            }
+            UsageCheck::All => {
+                state.serialize_field("type", "all")?;
+            }
+            UsageCheck::Custom(models) => {
+                state.serialize_field("type", "list")?;
+                state.serialize_field("content", &models.join(","))?;
+            }
+        }
+        state.end()
+    }
+}
+
+impl<'de> Deserialize<'de> for UsageCheck {
+    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
+    where
+        D: serde::Deserializer<'de>,
+    {
+        #[derive(Deserialize)]
+        #[serde(tag = "type", content = "content")]
+        enum UsageCheckHelper {
+            #[serde(rename = "none")]
+            None,
+            #[serde(rename = "default")]
+            Default,
+            #[serde(rename = "all")]
+            All,
+            #[serde(rename = "list")]
+            Custom(String),
+        }
+
+        let helper = UsageCheckHelper::deserialize(deserializer)?;
+        Ok(match helper {
+            UsageCheckHelper::None => UsageCheck::None,
+            UsageCheckHelper::Default => UsageCheck::Default,
+            UsageCheckHelper::All => UsageCheck::All,
+            UsageCheckHelper::Custom(list) => {
+                if list.is_empty() {
+                    return Ok(UsageCheck::None);
+                }
+
+                let models: Vec<&'static str> = list
+                    .split(',')
+                    .filter_map(|model| {
+                        let model = model.trim();
+                        AVAILABLE_MODELS
+                            .iter()
+                            .find(|m| m.id == model)
+                            .map(|m| m.id)
+                    })
+                    .collect();
+
+                if models.is_empty() {
+                    UsageCheck::None
+                } else {
+                    UsageCheck::Custom(models)
+                }
+            }
+        })
+    }
+}

src/app/statics.rs

@@ -0,0 +1,52 @@
+use super::{
+    constant::{DEFAULT_TOKEN_FILE_NAME, DEFAULT_TOKEN_LIST_FILE_NAME, EMPTY_STRING},
+    utils::parse_string_from_env,
+};
+use std::sync::LazyLock;
+
+macro_rules! def_pub_static {
+    // 基础版本：直接存储 String
+    ($name:ident, $value:expr) => {
+        pub static $name: LazyLock<String> = LazyLock::new(|| $value);
+
+        def_pub_static_getter!($name);
+    };
+
+    // 环境变量版本
+    ($name:ident, env: $env_key:expr, default: $default:expr) => {
+        pub static $name: LazyLock<String> =
+            LazyLock::new(|| parse_string_from_env($env_key, $default).trim().to_string());
+
+        def_pub_static_getter!($name);
+    };
+}
+
+macro_rules! def_pub_static_getter {
+    ($name:ident) => {
+        paste::paste! {
+            pub fn [<get_ $name:lower>]() -> String {
+                (*$name).clone()
+            }
+        }
+    };
+}
+
+def_pub_static!(ROUTE_PREFIX, env: "ROUTE_PREFIX", default: EMPTY_STRING);
+def_pub_static!(AUTH_TOKEN, env: "AUTH_TOKEN", default: EMPTY_STRING);
+def_pub_static!(TOKEN_FILE, env: "TOKEN_FILE", default: DEFAULT_TOKEN_FILE_NAME);
+def_pub_static!(TOKEN_LIST_FILE, env: "TOKEN_LIST_FILE", default: DEFAULT_TOKEN_LIST_FILE_NAME);
+def_pub_static!(
+    ROUTE_MODELS_PATH,
+    format!("{}/v1/models", ROUTE_PREFIX.as_str())
+);
+def_pub_static!(
+    ROUTE_CHAT_PATH,
+    format!("{}/v1/chat/completions", ROUTE_PREFIX.as_str())
+);
+
+pub static START_TIME: LazyLock<chrono::DateTime<chrono::Local>> =
+    LazyLock::new(chrono::Local::now);
+
+pub fn get_start_time() -> chrono::DateTime<chrono::Local> {
+    *START_TIME
+}

src/app/token.rs

@@ -1,9 +1,10 @@
 use super::{
     constant::*,
-    models::{AppConfig, AppState, TokenInfo, TokenUpdateRequest},
-    utils::i32_to_u32,
+    models::{AppState, TokenInfo, TokenUpdateRequest},
+    statics::*,
+    utils::{generate_checksum, generate_hash, i32_to_u32},
 };
-use crate::aiserver::v1::GetUserInfoResponse;
+use crate::{chat::aiserver::v1::GetUserInfoResponse, common::models::{ApiStatus, NormalResponseNoData}};
 use axum::http::HeaderMap;
 use axum::{
     extract::{Query, State},
@@ -41,13 +42,13 @@ fn parse_token_alias(token_part: &str, line: &str) -> Option<(String, Option<Str
 
 // Token 加载函数
 pub fn load_tokens() -> Vec<TokenInfo> {
-    let token_file = AppConfig::get_token_file();
-    let token_list_file = AppConfig::get_token_list_file();
+    let token_file = get_token_file();
+    let token_list_file = get_token_list_file();
 
     // 确保文件存在
     for file in [&token_file, &token_list_file] {
         if !std::path::Path::new(file).exists() {
-            if let Err(e) = std::fs::write(file, "") {
+            if let Err(e) = std::fs::write(file, EMPTY_STRING) {
                 eprintln!("警告: 无法创建文件 '{}': {}", file, e);
             }
         }
@@ -118,8 +119,7 @@ pub fn load_tokens() -> Vec<TokenInfo> {
             }
         } else {
             // 为新token生成checksum
-            let checksum =
-                crate::generate_checksum(&crate::generate_hash(), Some(&crate::generate_hash()));
+            let checksum = generate_checksum(&generate_hash(), Some(&generate_hash()));
             token_map.insert(token, (checksum, alias));
         }
     }
@@ -153,10 +153,20 @@ pub fn load_tokens() -> Vec<TokenInfo> {
         .collect()
 }
 
+#[derive(Serialize)]
+pub struct ChecksumResponse {
+    pub checksum: String,
+}
+
+pub async fn handle_get_checksum() -> Json<ChecksumResponse> {
+    let checksum = generate_checksum(&generate_hash(), Some(&generate_hash()));
+    Json(ChecksumResponse { checksum })
+}
+
 // 更新 TokenInfo 处理
 pub async fn handle_update_tokeninfo(
     State(state): State<Arc<Mutex<AppState>>>,
-) -> Json<serde_json::Value> {
+) -> Json<NormalResponseNoData> {
     // 重新加载 tokens
     let token_infos = load_tokens();
 
@@ -166,20 +176,20 @@ pub async fn handle_update_tokeninfo(
         state.token_infos = token_infos;
     }
 
-    Json(serde_json::json!({
-        STATUS: STATUS_SUCCESS,
-        MESSAGE: "Token list has been reloaded"
-    }))
+    Json(NormalResponseNoData {
+        status: ApiStatus::Success,
+        message: Some("Token list has been reloaded".to_string()),
+    })
 }
 
 // 获取 TokenInfo 处理
 pub async fn handle_get_tokeninfo(
     State(_state): State<Arc<Mutex<AppState>>>,
     headers: HeaderMap,
-) -> Result<Json<serde_json::Value>, StatusCode> {
-    let auth_token = AppConfig::get_auth_token();
-    let token_file = AppConfig::get_token_file();
-    let token_list_file = AppConfig::get_token_list_file();
+) -> Result<Json<TokenInfoResponse>, StatusCode> {
+    let auth_token = get_auth_token();
+    let token_file = get_token_file();
+    let token_list_file = get_token_list_file();
 
     // 验证 AUTH_TOKEN
     let auth_header = headers
@@ -196,23 +206,40 @@ pub async fn handle_get_tokeninfo(
     let tokens = std::fs::read_to_string(&token_file).unwrap_or_else(|_| String::new());
     let token_list = std::fs::read_to_string(&token_list_file).unwrap_or_else(|_| String::new());
 
-    Ok(Json(serde_json::json!({
-        STATUS: STATUS_SUCCESS,
-        "token_file": token_file,
-        "token_list_file": token_list_file,
-        "tokens": tokens,
-        "token_list": token_list
-    })))
+    Ok(Json(TokenInfoResponse {
+        status: ApiStatus::Success,
+        token_file: token_file.clone(),
+        token_list_file: token_list_file.clone(),
+        tokens: Some(tokens.clone()),
+        tokens_count: Some(tokens.len()),
+        token_list: Some(token_list),
+        message: None,
+    }))
+}
+
+#[derive(Serialize)]
+pub struct TokenInfoResponse {
+    pub status: ApiStatus,
+    pub token_file: String,
+    pub token_list_file: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub tokens: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub tokens_count: Option<usize>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub token_list: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub message: Option<String>,
 }
 
 pub async fn handle_update_tokeninfo_post(
     State(state): State<Arc<Mutex<AppState>>>,
     headers: HeaderMap,
     Json(request): Json<TokenUpdateRequest>,
-) -> Result<Json<serde_json::Value>, StatusCode> {
-    let auth_token = AppConfig::get_auth_token();
-    let token_file = AppConfig::get_token_file();
-    let token_list_file = AppConfig::get_token_list_file();
+) -> Result<Json<TokenInfoResponse>, StatusCode> {
+    let auth_token = get_auth_token();
+    let token_file = get_token_file();
+    let token_list_file = get_token_list_file();
 
     // 验证 AUTH_TOKEN
     let auth_header = headers
@@ -244,13 +271,15 @@ pub async fn handle_update_tokeninfo_post(
         state.token_infos = token_infos;
     }
 
-    Ok(Json(serde_json::json!({
-        STATUS: STATUS_SUCCESS,
-        MESSAGE: "Token files have been updated and reloaded",
-        "token_file": token_file,
-        "token_list_file": token_list_file,
-        "token_count": token_infos_len
-    })))
+    Ok(Json(TokenInfoResponse {
+        status: ApiStatus::Success,
+        token_file: token_file.clone(),
+        token_list_file: token_list_file.clone(),
+        tokens: None,
+        tokens_count: Some(token_infos_len),
+        token_list: None,
+        message: Some("Token files have been updated and reloaded".to_string()),
+    }))
 }
 
 #[derive(Deserialize)]
@@ -262,14 +291,13 @@ pub async fn get_user_info(
     State(state): State<Arc<Mutex<AppState>>>,
     Query(query): Query<GetUserInfoQuery>,
 ) -> Json<GetUserInfo> {
-    let (auth_token, checksum) = match {
-        let app_token_infos = &state.lock().await.token_infos;
-        app_token_infos
-            .iter()
-            .find(|token_info| token_info.alias == Some(query.alias.clone()))
-            .map(|token_info| (token_info.token.clone(), token_info.checksum.clone()))
-    } {
-        Some(token) => token,
+    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())),
     };
 
@@ -300,9 +328,9 @@ pub async fn get_user_usage(auth_token: &str, checksum: &str) -> Option<UserUsag
 
 #[derive(Serialize)]
 pub enum GetUserInfo {
-    #[serde[rename = "usage"]]
+    #[serde(rename = "usage")]
     Usage(UserUsageInfo),
-    #[serde[rename = "error"]]
+    #[serde(rename = "error")]
     Error(String),
 }
 

src/app/utils.rs

@@ -1,3 +1,6 @@
+mod checksum;
+pub use checksum::*;
+
 pub fn parse_bool_from_env(key: &str, default: bool) -> bool {
     std::env::var(key)
         .ok()

src/app/utils/checksum.rs

@@ -0,0 +1,44 @@
+use base64::{engine::general_purpose::STANDARD as BASE64, Engine as _};
+use rand::Rng;
+use sha2::{Digest, Sha256};
+
+pub fn generate_hash() -> String {
+    let random_bytes = rand::thread_rng().gen::<[u8; 32]>();
+    let mut hasher = Sha256::new();
+    hasher.update(random_bytes);
+    hex::encode(hasher.finalize())
+}
+
+fn obfuscate_bytes(bytes: &mut [u8]) {
+    let mut prev: u8 = 165;
+    for (idx, byte) in bytes.iter_mut().enumerate() {
+        let old_value = *byte;
+        *byte = (old_value ^ prev).wrapping_add((idx % 256) as u8);
+        prev = *byte;
+    }
+}
+
+pub fn generate_checksum(device_id: &str, mac_addr: Option<&str>) -> String {
+    let timestamp = std::time::SystemTime::now()
+        .duration_since(std::time::UNIX_EPOCH)
+        .unwrap()
+        .as_millis()
+        / 1_000_000;
+
+    let mut timestamp_bytes = vec![
+        ((timestamp >> 40) & 255) as u8,
+        ((timestamp >> 32) & 255) as u8,
+        ((timestamp >> 24) & 255) as u8,
+        ((timestamp >> 16) & 255) as u8,
+        ((timestamp >> 8) & 255) as u8,
+        (255 & timestamp) as u8,
+    ];
+
+    obfuscate_bytes(&mut timestamp_bytes);
+    let encoded = BASE64.encode(&timestamp_bytes);
+
+    match mac_addr {
+        Some(mac) => format!("{}{}/{}", encoded, device_id, mac),
+        None => format!("{}{}", encoded, device_id),
+    }
+}

src/chat.rs

@@ -1,2 +1,6 @@
-pub mod stream;
+pub mod aiserver;
+pub mod constant;
 pub mod error;
+pub mod models;
+pub mod service;
+pub mod stream;

src/chat/constant.rs

@@ -0,0 +1,192 @@
+use super::models::Model;
+
+macro_rules! def_pub_const {
+    ($name:ident, $value:expr) => {
+        pub const $name: &'static str = $value;
+    };
+}
+def_pub_const!(ERR_UNSUPPORTED_GIF, "不支持动态 GIF");
+def_pub_const!(ERR_UNSUPPORTED_IMAGE_FORMAT, "不支持的图片格式，仅支持 PNG、JPEG、WEBP 和非动态 GIF");
+
+const MODEL_OBJECT: &str = "model";
+const CREATED: i64 = 1706659200;
+
+def_pub_const!(ANTHROPIC, "anthropic");
+def_pub_const!(CURSOR, "cursor");
+def_pub_const!(GOOGLE, "google");
+def_pub_const!(OPENAI, "openai");
+
+def_pub_const!(CLAUDE_3_5_SONNET, "claude-3.5-sonnet");
+def_pub_const!(GPT_4, "gpt-4");
+def_pub_const!(GPT_4O, "gpt-4o");
+def_pub_const!(CLAUDE_3_OPUS, "claude-3-opus");
+def_pub_const!(CURSOR_FAST, "cursor-fast");
+def_pub_const!(CURSOR_SMALL, "cursor-small");
+def_pub_const!(GPT_3_5_TURBO, "gpt-3.5-turbo");
+def_pub_const!(GPT_4_TURBO_2024_04_09, "gpt-4-turbo-2024-04-09");
+def_pub_const!(GPT_4O_128K, "gpt-4o-128k");
+def_pub_const!(GEMINI_1_5_FLASH_500K, "gemini-1.5-flash-500k");
+def_pub_const!(CLAUDE_3_HAIKU_200K, "claude-3-haiku-200k");
+def_pub_const!(CLAUDE_3_5_SONNET_200K, "claude-3-5-sonnet-200k");
+def_pub_const!(CLAUDE_3_5_SONNET_20241022, "claude-3-5-sonnet-20241022");
+def_pub_const!(GPT_4O_MINI, "gpt-4o-mini");
+def_pub_const!(O1_MINI, "o1-mini");
+def_pub_const!(O1_PREVIEW, "o1-preview");
+def_pub_const!(O1, "o1");
+def_pub_const!(CLAUDE_3_5_HAIKU, "claude-3.5-haiku");
+def_pub_const!(GEMINI_EXP_1206, "gemini-exp-1206");
+def_pub_const!(
+    GEMINI_2_0_FLASH_THINKING_EXP,
+    "gemini-2.0-flash-thinking-exp"
+);
+def_pub_const!(GEMINI_2_0_FLASH_EXP, "gemini-2.0-flash-exp");
+
+pub const AVAILABLE_MODELS: &[Model] = &[
+    Model {
+        id: CLAUDE_3_5_SONNET,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: ANTHROPIC,
+    },
+    Model {
+        id: GPT_4,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: GPT_4O,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: CLAUDE_3_OPUS,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: ANTHROPIC,
+    },
+    Model {
+        id: CURSOR_FAST,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: CURSOR,
+    },
+    Model {
+        id: CURSOR_SMALL,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: CURSOR,
+    },
+    Model {
+        id: GPT_3_5_TURBO,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: GPT_4_TURBO_2024_04_09,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: GPT_4O_128K,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: GEMINI_1_5_FLASH_500K,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: GOOGLE,
+    },
+    Model {
+        id: CLAUDE_3_HAIKU_200K,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: ANTHROPIC,
+    },
+    Model {
+        id: CLAUDE_3_5_SONNET_200K,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: ANTHROPIC,
+    },
+    Model {
+        id: CLAUDE_3_5_SONNET_20241022,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: ANTHROPIC,
+    },
+    Model {
+        id: GPT_4O_MINI,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: O1_MINI,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: O1_PREVIEW,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: O1,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: OPENAI,
+    },
+    Model {
+        id: CLAUDE_3_5_HAIKU,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: ANTHROPIC,
+    },
+    Model {
+        id: GEMINI_EXP_1206,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: GOOGLE,
+    },
+    Model {
+        id: GEMINI_2_0_FLASH_THINKING_EXP,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: GOOGLE,
+    },
+    Model {
+        id: GEMINI_2_0_FLASH_EXP,
+        created: CREATED,
+        object: MODEL_OBJECT,
+        owned_by: GOOGLE,
+    },
+];
+
+pub const USAGE_CHECK_MODELS: [&str; 11] = [
+    CLAUDE_3_5_SONNET_20241022,
+    CLAUDE_3_5_SONNET,
+    GEMINI_EXP_1206,
+    GPT_4,
+    GPT_4_TURBO_2024_04_09,
+    GPT_4O,
+    CLAUDE_3_5_HAIKU,
+    GPT_4O_128K,
+    GEMINI_1_5_FLASH_500K,
+    CLAUDE_3_HAIKU_200K,
+    CLAUDE_3_5_SONNET_200K,
+];
+
+pub const LONG_CONTEXT_MODELS: [&str; 4] = [
+    GPT_4O_128K,
+    GEMINI_1_5_FLASH_500K,
+    CLAUDE_3_HAIKU_200K,
+    CLAUDE_3_5_SONNET_200K,
+];

src/chat/error.rs

@@ -1,4 +1,4 @@
-use crate::aiserver::v1::throw_error_check_request::Error as ErrorType;
+use super::aiserver::v1::throw_error_check_request::Error as ErrorType;
 use reqwest::StatusCode;
 use serde::{Deserialize, Serialize};
 
@@ -31,9 +31,9 @@ pub struct ErrorDebug {
 }
 
 impl ErrorDebug {
-    pub fn is_valid(&self) -> bool {
-        ErrorType::from_str_name(&self.error).is_some()
-    }
+    // pub fn is_valid(&self) -> bool {
+    //     ErrorType::from_str_name(&self.error).is_some()
+    // }
 
     pub fn status_code(&self) -> u16 {
         match ErrorType::from_str_name(&self.error) {
@@ -83,6 +83,8 @@ pub struct ErrorDetails {
     pub is_retryable: bool,
 }
 
+use crate::common::models::{ApiStatus, ErrorResponse as CommonErrorResponse};
+
 impl ChatError {
     pub fn to_json(&self) -> serde_json::Value {
         serde_json::to_value(self).unwrap()
@@ -135,6 +137,15 @@ impl ErrorResponse {
     pub fn native_code(&self) -> String {
         self.code.replace("_", " ").to_lowercase()
     }
+
+    pub fn to_common(self) -> CommonErrorResponse {
+        CommonErrorResponse {
+            status: ApiStatus::Error,
+            code: Some(self.status),
+            error: self.error.as_ref().map(|error| error.message.clone()).or(Some(self.code.clone())),
+            message: self.error.as_ref().map(|error| error.details.clone()),
+        }
+    }
 }
 
 pub enum StreamError {

src/chat/models.rs

@@ -76,3 +76,32 @@ pub struct Usage {
     pub completion_tokens: i32,
     pub total_tokens: i32,
 }
+
+// 模型定义
+#[derive(Serialize, Clone)]
+pub struct Model {
+    pub id: &'static str,
+    pub created: i64,
+    pub object: &'static str,
+    pub owned_by: &'static str,
+}
+
+use crate::{AppConfig, UsageCheck};
+use super::constant::USAGE_CHECK_MODELS;
+
+impl Model {
+    pub fn is_usage_check(&self) -> bool {
+        match AppConfig::get_usage_check() {
+            UsageCheck::None => false,
+            UsageCheck::Default => USAGE_CHECK_MODELS.contains(&self.id),
+            UsageCheck::All => true,
+            UsageCheck::Custom(models) => models.contains(&self.id),
+        }
+    }
+}
+
+#[derive(Serialize)]
+pub struct ModelsResponse {
+    pub object: &'static str,
+    pub data: &'static [Model],
+}

src/chat/service.rs

@@ -0,0 +1,506 @@
+use axum::{
+  body::Body,
+  extract::State,
+  http::{HeaderMap, StatusCode},
+  response::Response,
+  Json,
+};
+use bytes::Bytes;
+use crate::{
+  app::{
+      client::build_client,
+      constant::*,
+      models::*,
+      statics::*,
+      token::get_user_usage,
+  }, chat::{
+      error::StreamError,
+      models::*,
+      stream::{parse_stream_data, StreamMessage},
+  }, common::models::{error::ChatError, ErrorResponse}
+};
+use super::constant::AVAILABLE_MODELS;
+use futures::{Stream, StreamExt};
+use std::{
+    convert::Infallible, sync::{atomic::AtomicBool, Arc}
+};
+use std::{
+    pin::Pin,
+    sync::atomic::{AtomicUsize, Ordering},
+};
+use tokio::sync::Mutex;
+use uuid::Uuid;
+
+// 模型列表处理
+pub async fn handle_models() -> Json<ModelsResponse> {
+  Json(ModelsResponse {
+      object: "list",
+      data: AVAILABLE_MODELS,
+  })
+}
+
+// 聊天处理函数的签名
+pub async fn handle_chat(
+  State(state): State<Arc<Mutex<AppState>>>,
+  headers: HeaderMap,
+  Json(request): Json<ChatRequest>,
+) -> Result<Response<Body>, (StatusCode, Json<ErrorResponse>)> {
+  let allow_claude = AppConfig::get_allow_claude();
+  // 验证模型是否支持并获取模型信息
+  let model = AVAILABLE_MODELS.iter().find(|m| m.id == request.model);
+  let model_supported = model.is_some();
+
+  if !(model_supported || allow_claude && request.model.starts_with("claude")) {
+      return Err((
+          StatusCode::BAD_REQUEST,
+          Json(ChatError::ModelNotSupported(request.model).to_json()),
+      ));
+  }
+
+  let request_time = chrono::Local::now();
+
+  // 验证请求
+  if request.messages.is_empty() {
+      return Err((
+          StatusCode::BAD_REQUEST,
+          Json(ChatError::EmptyMessages.to_json()),
+      ));
+  }
+
+  // 获取并处理认证令牌
+  let auth_token = headers
+      .get(axum::http::header::AUTHORIZATION)
+      .and_then(|h| h.to_str().ok())
+      .and_then(|h| h.strip_prefix(AUTHORIZATION_BEARER_PREFIX))
+      .ok_or((
+          StatusCode::UNAUTHORIZED,
+          Json(ChatError::Unauthorized.to_json()),
+      ))?;
+
+  // 验证 AuthToken
+  if auth_token != get_auth_token() {
+      return Err((
+          StatusCode::UNAUTHORIZED,
+          Json(ChatError::Unauthorized.to_json()),
+      ));
+  }
+
+  // 完整的令牌处理逻辑和对应的 checksum
+  let (auth_token, checksum, alias) = {
+      static CURRENT_KEY_INDEX: AtomicUsize = AtomicUsize::new(0);
+      let state_guard = state.lock().await;
+      let token_infos = &state_guard.token_infos;
+
+      if token_infos.is_empty() {
+          return Err((
+              StatusCode::SERVICE_UNAVAILABLE,
+              Json(ChatError::NoTokens.to_json()),
+          ));
+      }
+
+      let index = CURRENT_KEY_INDEX.fetch_add(1, Ordering::SeqCst) % token_infos.len();
+      let token_info = &token_infos[index];
+      (
+          token_info.token.clone(),
+          token_info.checksum.clone(),
+          token_info.alias.clone(),
+      )
+  };
+
+  // 更新请求日志
+  {
+      let state_clone = state.clone();
+      let mut state = state.lock().await;
+      state.total_requests += 1;
+      state.active_requests += 1;
+
+      // 如果有model且需要获取使用情况,创建后台任务获取
+      if let Some(model) = model {
+          if model.is_usage_check() {
+              let auth_token_clone = auth_token.clone();
+              let checksum_clone = checksum.clone();
+              let state_clone = state_clone.clone();
+
+              tokio::spawn(async move {
+                  let usage = get_user_usage(&auth_token_clone, &checksum_clone).await;
+                  let mut state = state_clone.lock().await;
+                  // 根据时间戳找到对应的日志
+                  if let Some(log) = state
+                      .request_logs
+                      .iter_mut()
+                      .find(|log| log.timestamp == request_time)
+                  {
+                      log.token_info.usage = usage;
+                  }
+              });
+          }
+      }
+
+      state.request_logs.push(RequestLog {
+          timestamp: request_time,
+          model: request.model.clone(),
+          token_info: TokenInfo {
+              token: auth_token.clone(),
+              checksum: checksum.clone(),
+              alias: alias.clone(),
+              usage: None,
+          },
+          prompt: None,
+          stream: request.stream,
+          status: "pending",
+          error: None,
+      });
+
+      if state.request_logs.len() > 100 {
+          state.request_logs.remove(0);
+      }
+  }
+
+  // 将消息转换为hex格式
+  let hex_data = crate::encode_chat_message(request.messages, &request.model)
+      .await
+      .map_err(|_| {
+          (
+              StatusCode::INTERNAL_SERVER_ERROR,
+              Json(
+                  ChatError::RequestFailed("Failed to encode chat message".to_string()).to_json(),
+              ),
+          )
+      })?;
+
+  // 构建请求客户端
+  let client = build_client(&auth_token, &checksum, CURSOR_API2_STREAM_CHAT);
+  let response = client.body(hex_data).send().await;
+
+  // 处理请求结果
+  let response = match response {
+      Ok(resp) => {
+          // 更新请求日志为成功
+          {
+              let mut state = state.lock().await;
+              state.request_logs.last_mut().unwrap().status = STATUS_SUCCESS;
+          }
+          resp
+      }
+      Err(e) => {
+          // 更新请求日志为失败
+          {
+              let mut state = state.lock().await;
+              if let Some(last_log) = state.request_logs.last_mut() {
+                  last_log.status = STATUS_FAILED;
+                  last_log.error = Some(e.to_string());
+              }
+          }
+          return Err((
+              StatusCode::INTERNAL_SERVER_ERROR,
+              Json(ChatError::RequestFailed(e.to_string()).to_json()),
+          ));
+      }
+  };
+
+  // 释放活动请求计数
+  {
+      let mut state = state.lock().await;
+      state.active_requests -= 1;
+  }
+
+  if request.stream {
+      let response_id = format!("chatcmpl-{}", Uuid::new_v4().simple());
+      let full_text = Arc::new(Mutex::new(String::with_capacity(1024)));
+      let is_start = Arc::new(AtomicBool::new(true));
+
+      let stream = {
+          // 创建新的 stream
+          let mut stream = response.bytes_stream();
+
+          let enable_stream_check = AppConfig::get_stream_check();
+
+          if enable_stream_check {
+              // 检查第一个 chunk
+              match stream.next().await {
+                  Some(first_chunk) => {
+                      let chunk = first_chunk.map_err(|e| {
+                          let error_message = format!("Failed to read response chunk: {}", e);
+                          // 理论上，若程序正常，必定成功，因为前面判断过了
+                          (
+                              StatusCode::INTERNAL_SERVER_ERROR,
+                              Json(ChatError::RequestFailed(error_message).to_json()),
+                          )
+                      })?;
+
+                      match parse_stream_data(&chunk) {
+                          Err(StreamError::ChatError(error)) => {
+                              let error_respone = error.to_error_response();
+                              // 更新请求日志为失败
+                              {
+                                  let mut state = state.lock().await;
+                                  if let Some(last_log) = state.request_logs.last_mut() {
+                                      last_log.status = STATUS_FAILED;
+                                      last_log.error = Some(error_respone.native_code());
+                                  }
+                              }
+                              return Err((
+                                  error_respone.status_code(),
+                                  Json(error_respone.to_common()),
+                              ));
+                          }
+                          Ok(_) | Err(_) => {
+                              // 创建一个包含第一个 chunk 的 stream
+                              Box::pin(
+                                  futures::stream::once(async move { Ok(chunk) }).chain(stream),
+                              )
+                                  as Pin<
+                                      Box<
+                                          dyn Stream<Item = Result<Bytes, reqwest::Error>> + Send,
+                                      >,
+                                  >
+                          }
+                      }
+                  }
+                  None => {
+                      // Box::pin(stream)
+                      //     as Pin<Box<dyn Stream<Item = Result<Bytes, reqwest::Error>> + Send>>
+                      // 更新请求日志为失败
+                      {
+                          let mut state = state.lock().await;
+                          if let Some(last_log) = state.request_logs.last_mut() {
+                              last_log.status = STATUS_FAILED;
+                              last_log.error = Some("Empty stream response".to_string());
+                          }
+                      }
+                      return Err((
+                          StatusCode::INTERNAL_SERVER_ERROR,
+                          Json(
+                              ChatError::RequestFailed("Empty stream response".to_string())
+                                  .to_json(),
+                          ),
+                      ));
+                  }
+              }
+          } else {
+              Box::pin(stream)
+                  as Pin<Box<dyn Stream<Item = Result<Bytes, reqwest::Error>> + Send>>
+          }
+      }
+      .then(move |chunk| {
+          let response_id = response_id.clone();
+          let model = request.model.clone();
+          let is_start = is_start.clone();
+          let full_text = full_text.clone();
+          let state = state.clone();
+
+          async move {
+              let chunk = chunk.unwrap_or_default();
+              match parse_stream_data(&chunk) {
+                  Ok(StreamMessage::Content(texts)) => {
+                      let mut response_data = String::new();
+
+                      for text in texts {
+                          let mut text_guard = full_text.lock().await;
+                          text_guard.push_str(&text);
+                          let is_first = is_start.load(Ordering::SeqCst);
+
+                          let response = ChatResponse {
+                              id: response_id.clone(),
+                              object: OBJECT_CHAT_COMPLETION_CHUNK.to_string(),
+                              created: chrono::Utc::now().timestamp(),
+                              model: if is_first { Some(model.clone()) } else { None },
+                              choices: vec![Choice {
+                                  index: 0,
+                                  message: None,
+                                  delta: Some(Delta {
+                                      role: if is_first {
+                                          is_start.store(false, Ordering::SeqCst);
+                                          Some(Role::Assistant)
+                                      } else {
+                                          None
+                                      },
+                                      content: Some(text),
+                                  }),
+                                  finish_reason: None,
+                              }],
+                              usage: None,
+                          };
+
+                          response_data.push_str(&format!(
+                              "data: {}\n\n",
+                              serde_json::to_string(&response).unwrap()
+                          ));
+                      }
+
+                      Ok::<_, Infallible>(Bytes::from(response_data))
+                  }
+                  Ok(StreamMessage::StreamStart) => {
+                      // 发送初始响应，包含模型信息
+                      let response = ChatResponse {
+                          id: response_id.clone(),
+                          object: OBJECT_CHAT_COMPLETION_CHUNK.to_string(),
+                          created: chrono::Utc::now().timestamp(),
+                          model: {
+                              is_start.store(true, Ordering::SeqCst);
+                              Some(model.clone())
+                          },
+                          choices: vec![Choice {
+                              index: 0,
+                              message: None,
+                              delta: Some(Delta {
+                                  role: Some(Role::Assistant),
+                                  content: Some(String::new()),
+                              }),
+                              finish_reason: None,
+                          }],
+                          usage: None,
+                      };
+
+                      Ok(Bytes::from(format!(
+                          "data: {}\n\n",
+                          serde_json::to_string(&response).unwrap()
+                      )))
+                  }
+                  Ok(StreamMessage::StreamEnd) => {
+                      // 根据配置决定是否发送最后的 finish_reason
+                      let include_finish_reason = AppConfig::get_stop_stream();
+
+                      if include_finish_reason {
+                          let response = ChatResponse {
+                              id: response_id.clone(),
+                              object: OBJECT_CHAT_COMPLETION_CHUNK.to_string(),
+                              created: chrono::Utc::now().timestamp(),
+                              model: None,
+                              choices: vec![Choice {
+                                  index: 0,
+                                  message: None,
+                                  delta: Some(Delta {
+                                      role: None,
+                                      content: None,
+                                  }),
+                                  finish_reason: Some(FINISH_REASON_STOP.to_string()),
+                              }],
+                              usage: None,
+                          };
+                          Ok(Bytes::from(format!(
+                              "data: {}\n\ndata: [DONE]\n\n",
+                              serde_json::to_string(&response).unwrap()
+                          )))
+                      } else {
+                          Ok(Bytes::from("data: [DONE]\n\n"))
+                      }
+                  }
+                  Ok(StreamMessage::Debug(debug_prompt)) => {
+                      if let Ok(mut state) = state.try_lock() {
+                          if let Some(last_log) = state.request_logs.last_mut() {
+                              last_log.prompt = Some(debug_prompt.clone());
+                          }
+                      }
+                      Ok(Bytes::new())
+                  }
+                  Err(StreamError::ChatError(error)) => {
+                      eprintln!("Stream error occurred: {}", error.to_json());
+                      Ok(Bytes::new())
+                  }
+                  Err(e) => {
+                      eprintln!("[警告] Stream error: {}", e);
+                      Ok(Bytes::new())
+                  }
+              }
+          }
+      });
+
+      Ok(Response::builder()
+          .header("Cache-Control", "no-cache")
+          .header("Connection", "keep-alive")
+          .header(HEADER_NAME_CONTENT_TYPE, "text/event-stream")
+          .body(Body::from_stream(stream))
+          .unwrap())
+  } else {
+      // 非流式响应
+      let mut full_text = String::with_capacity(1024); // 预分配合适的容量
+      let mut stream = response.bytes_stream();
+      let mut prompt = None;
+
+      while let Some(chunk) = stream.next().await {
+          let chunk = chunk.map_err(|e| {
+              (
+                  StatusCode::INTERNAL_SERVER_ERROR,
+                  Json(
+                      ChatError::RequestFailed(format!("Failed to read response chunk: {}", e))
+                          .to_json(),
+                  ),
+              )
+          })?;
+
+          match parse_stream_data(&chunk) {
+              Ok(StreamMessage::Content(texts)) => {
+                  for text in texts {
+                      full_text.push_str(&text);
+                  }
+              }
+              Ok(StreamMessage::Debug(debug_prompt)) => {
+                  prompt = Some(debug_prompt);
+              }
+              Ok(StreamMessage::StreamStart) | Ok(StreamMessage::StreamEnd) => {}
+              Err(StreamError::ChatError(error)) => {
+                  return Err((
+                      StatusCode::from_u16(error.error.details[0].debug.status_code())
+                          .unwrap_or(StatusCode::INTERNAL_SERVER_ERROR),
+                      Json(error.to_error_response().to_common()),
+                  ));
+              }
+              Err(_) => continue,
+          }
+      }
+
+      // 检查响应是否为空
+      if full_text.is_empty() {
+          // 更新请求日志为失败
+          {
+              let mut state = state.lock().await;
+              if let Some(last_log) = state.request_logs.last_mut() {
+                  last_log.status = STATUS_FAILED;
+                  last_log.error = Some("Empty response received".to_string());
+                  if let Some(p) = prompt {
+                      last_log.prompt = Some(p);
+                  }
+              }
+          }
+          return Err((
+              StatusCode::INTERNAL_SERVER_ERROR,
+              Json(ChatError::RequestFailed("Empty response received".to_string()).to_json()),
+          ));
+      }
+
+      // 更新请求日志提示词
+      {
+          let mut state = state.lock().await;
+          if let Some(last_log) = state.request_logs.last_mut() {
+              last_log.prompt = prompt;
+          }
+      }
+
+      let response_data = ChatResponse {
+          id: format!("chatcmpl-{}", Uuid::new_v4().simple()),
+          object: OBJECT_CHAT_COMPLETION.to_string(),
+          created: chrono::Utc::now().timestamp(),
+          model: Some(request.model),
+          choices: vec![Choice {
+              index: 0,
+              message: Some(Message {
+                  role: Role::Assistant,
+                  content: MessageContent::Text(full_text),
+              }),
+              delta: None,
+              finish_reason: Some(FINISH_REASON_STOP.to_string()),
+          }],
+          usage: Some(Usage {
+              prompt_tokens: 0,
+              completion_tokens: 0,
+              total_tokens: 0,
+          }),
+      };
+
+      Ok(Response::builder()
+          .header(HEADER_NAME_CONTENT_TYPE, "application/json")
+          .body(Body::from(serde_json::to_string(&response_data).unwrap()))
+          .unwrap())
+  }
+}

src/chat/stream.rs

@@ -1,4 +1,4 @@
-use crate::aiserver::v1::StreamChatResponse;
+use super::aiserver::v1::StreamChatResponse;
 use flate2::read::GzDecoder;
 use prost::Message;
 use std::io::Read;

src/common.rs

@@ -0,0 +1 @@
+pub mod models;

src/common/models.rs

@@ -0,0 +1,70 @@
+pub mod error;
+pub mod health;
+pub mod config;
+
+use config::ConfigData;
+
+use serde::Serialize;
+
+#[derive(Serialize)]
+pub enum ApiStatus {
+    #[serde(rename = "healthy")]
+    Healthy,
+    #[serde(rename = "success")]
+    Success,
+    #[serde(rename = "error")]
+    Error,
+    #[serde(rename = "failed")]
+    Failed,
+}
+
+// #[derive(Serialize)]
+// #[serde(untagged)]
+// pub enum ApiResponse {
+//     HealthCheck(HealthCheckResponse),
+//     ConfigData(NormalResponse<ConfigData>),
+//     Error(ErrorResponse),
+// }
+
+// impl ApiResponse {
+//     pub fn to_string(&self) -> String {
+//         serde_json::to_string(self).unwrap()
+//     }
+// }
+
+#[derive(Serialize)]
+pub struct NormalResponse<T> {
+    pub status: ApiStatus,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub data: Option<T>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub message: Option<String>,
+}
+
+impl std::fmt::Display for NormalResponse<ConfigData> {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        write!(f, "{}", serde_json::to_string(self).unwrap())
+    }
+}
+
+#[derive(Serialize)]
+pub struct NormalResponseNoData {
+    pub status: ApiStatus,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub message: Option<String>,
+}
+
+#[derive(Serialize)]
+pub struct ErrorResponse {
+    // status -> 成功 / 失败
+    pub status: ApiStatus,
+    // HTTP 请求的状态码
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub code: Option<u16>,
+    // HTTP 请求的错误码
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub error: Option<String>,
+    // 错误详情
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub message: Option<String>,
+}

src/common/models/config.rs

@@ -0,0 +1,36 @@
+use serde::{Deserialize, Serialize};
+
+use crate::{PageContent, UsageCheck, VisionAbility};
+
+#[derive(Serialize)]
+pub struct ConfigData {
+    pub page_content: Option<PageContent>,
+    pub enable_stream_check: bool,
+    pub include_stop_stream: bool,
+    pub vision_ability: VisionAbility,
+    pub enable_slow_pool: bool,
+    pub enable_all_claude: bool,
+    pub check_usage_models: UsageCheck,
+}
+
+#[derive(Deserialize)]
+pub struct ConfigUpdateRequest {
+    #[serde(default)]
+    pub action: String, // "get", "update", "reset"
+    #[serde(default)]
+    pub path: String,
+    #[serde(default)]
+    pub content: Option<PageContent>, // "default", "text", "html"
+    #[serde(default)]
+    pub enable_stream_check: Option<bool>,
+    #[serde(default)]
+    pub include_stop_stream: Option<bool>,
+    #[serde(default)]
+    pub vision_ability: Option<VisionAbility>,
+    #[serde(default)]
+    pub enable_slow_pool: Option<bool>,
+    #[serde(default)]
+    pub enable_all_claude: Option<bool>,
+    #[serde(default)]
+    pub check_usage_models: Option<UsageCheck>,
+}

src/common/models/error.rs

@@ -0,0 +1,34 @@
+use super::ErrorResponse;
+
+pub enum ChatError {
+    ModelNotSupported(String),
+    EmptyMessages,
+    NoTokens,
+    RequestFailed(String),
+    Unauthorized,
+}
+
+impl ChatError {
+    pub fn to_json(&self) -> ErrorResponse {
+        let (error, message) = match self {
+            ChatError::ModelNotSupported(model) => (
+                "model_not_supported",
+                format!("Model '{}' is not supported", model),
+            ),
+            ChatError::EmptyMessages => (
+                "empty_messages",
+                "Message array cannot be empty".to_string(),
+            ),
+            ChatError::NoTokens => ("no_tokens", "No available tokens".to_string()),
+            ChatError::RequestFailed(err) => ("request_failed", format!("Request failed: {}", err)),
+            ChatError::Unauthorized => ("unauthorized", "Invalid authorization token".to_string()),
+        };
+
+        ErrorResponse {
+          status: super::ApiStatus::Error,
+          code: None,
+          error: Some(error.to_string()),
+          message: Some(message),
+        }
+    }
+}

src/common/models/health.rs

@@ -0,0 +1,37 @@
+use serde::Serialize;
+
+use super::ApiStatus;
+
+#[derive(Serialize)]
+pub struct HealthCheckResponse {
+    pub status: ApiStatus,
+    pub version: &'static str,
+    pub uptime: i64,
+    pub stats: SystemStats,
+    pub models: Vec<&'static str>,
+    pub endpoints: Vec<&'static str>,
+}
+
+#[derive(Serialize)]
+pub struct SystemStats {
+    pub started: String,
+    pub total_requests: u64,
+    pub active_requests: u64,
+    pub system: SystemInfo,
+}
+
+#[derive(Serialize)]
+pub struct SystemInfo {
+    pub memory: MemoryInfo,
+    pub cpu: CpuInfo,
+}
+
+#[derive(Serialize)]
+pub struct MemoryInfo {
+    pub rss: u64, // 物理内存使用量(字节)
+}
+
+#[derive(Serialize)]
+pub struct CpuInfo {
+    pub usage: f32, // CPU 使用率(百分比)
+}

src/lib.rs

@@ -1,23 +1,25 @@
 use base64::{engine::general_purpose::STANDARD as BASE64, Engine as _};
 use image::guess_format;
 use prost::Message as _;
-use rand::Rng;
-use sha2::{Digest, Sha256};
 use uuid::Uuid;
 
-mod aiserver;
-use aiserver::v1::*;
-
-pub mod message;
-use message::*;
+pub mod common;
 
 pub mod app;
-use app::{models::*,constant::*};
+use app::{constant::EMPTY_STRING, models::*};
 
 pub mod chat;
+use chat::{
+    aiserver::v1::{
+        conversation_message, image_proto, ConversationMessage, ExplicitContext, GetChatRequest,
+        ImageProto, ModelDetails,
+    },
+    constant::{LONG_CONTEXT_MODELS, ERR_UNSUPPORTED_GIF, ERR_UNSUPPORTED_IMAGE_FORMAT},
+    models::{Message, MessageContent, Role},
+};
 
 async fn process_chat_inputs(inputs: Vec<Message>) -> (String, Vec<ConversationMessage>) {
-    // 收集 system 和 developer 指令
+    // 收集 system 指令
     let instructions = inputs
         .iter()
         .filter(|input| input.role == Role::System)
@@ -56,7 +58,7 @@ async fn process_chat_inputs(inputs: Vec<Message>) -> (String, Vec<ConversationM
         return (
             instructions,
             vec![ConversationMessage {
-                text: " ".to_string(),
+                text: EMPTY_STRING.into(),
                 r#type: conversation_message::MessageType::Human as i32,
                 attached_code_chunks: vec![],
                 codebase_context_chunks: vec![],
@@ -101,7 +103,7 @@ async fn process_chat_inputs(inputs: Vec<Message>) -> (String, Vec<ConversationM
             0,
             Message {
                 role: Role::User,
-                content: MessageContent::Text(" ".to_string()),
+                content: MessageContent::Text(EMPTY_STRING.into()),
             },
         );
     }
@@ -119,7 +121,7 @@ async fn process_chat_inputs(inputs: Vec<Message>) -> (String, Vec<ConversationM
                 i,
                 Message {
                     role: insert_role,
-                    content: MessageContent::Text(" ".to_string()),
+                    content: MessageContent::Text(EMPTY_STRING.into()),
                 },
             );
         }
@@ -133,7 +135,7 @@ async fn process_chat_inputs(inputs: Vec<Message>) -> (String, Vec<ConversationM
     {
         chat_inputs.push(Message {
             role: Role::User,
-            content: MessageContent::Text(" ".to_string()),
+            content: MessageContent::Text(EMPTY_STRING.into()),
         });
     }
 
@@ -259,7 +261,7 @@ fn process_base64_image(
         && !format.contains("webp")
         && !format.contains("gif")
     {
-        return Err("不支持的图片格式，仅支持 PNG、JPEG、WEBP 和非动态 GIF".into());
+        return Err(ERR_UNSUPPORTED_IMAGE_FORMAT.into());
     }
 
     let image_data = BASE64.decode(parts[1])?;
@@ -268,7 +270,7 @@ fn process_base64_image(
     if format.contains("gif") {
         if let Ok(frames) = gif::DecodeOptions::new().read_info(std::io::Cursor::new(&image_data)) {
             if frames.into_iter().count() > 1 {
-                return Err("不支持动态 GIF".into());
+                return Err(ERR_UNSUPPORTED_GIF.into());
             }
         }
     }
@@ -304,11 +306,11 @@ async fn process_http_image(
                 gif::DecodeOptions::new().read_info(std::io::Cursor::new(&image_data))
             {
                 if frames.into_iter().count() > 1 {
-                    return Err("不支持动态 GIF".into());
+                    return Err(ERR_UNSUPPORTED_GIF.into());
                 }
             }
         }
-        _ => return Err("不支持的图片格式，仅支持 PNG、JPEG、WEBP 和非动态 GIF".into()),
+        _ => return Err(ERR_UNSUPPORTED_IMAGE_FORMAT.into()),
     }
 
     // 获取图片尺寸
@@ -399,44 +401,3 @@ pub async fn encode_chat_message(
 
     Ok(hex::decode(len_prefix + &content)?)
 }
-
-pub fn generate_hash() -> String {
-    let random_bytes = rand::thread_rng().gen::<[u8; 32]>();
-    let mut hasher = Sha256::new();
-    hasher.update(random_bytes);
-    hex::encode(hasher.finalize())
-}
-
-fn obfuscate_bytes(bytes: &mut [u8]) {
-    let mut prev: u8 = 165;
-    for (idx, byte) in bytes.iter_mut().enumerate() {
-        let old_value = *byte;
-        *byte = (old_value ^ prev).wrapping_add((idx % 256) as u8);
-        prev = *byte;
-    }
-}
-
-pub fn generate_checksum(device_id: &str, mac_addr: Option<&str>) -> String {
-    let timestamp = std::time::SystemTime::now()
-        .duration_since(std::time::UNIX_EPOCH)
-        .unwrap()
-        .as_millis()
-        / 1_000_000;
-
-    let mut timestamp_bytes = vec![
-        ((timestamp >> 40) & 255) as u8,
-        ((timestamp >> 32) & 255) as u8,
-        ((timestamp >> 24) & 255) as u8,
-        ((timestamp >> 16) & 255) as u8,
-        ((timestamp >> 8) & 255) as u8,
-        (255 & timestamp) as u8,
-    ];
-
-    obfuscate_bytes(&mut timestamp_bytes);
-    let encoded = BASE64.encode(&timestamp_bytes);
-
-    match mac_addr {
-        Some(mac) => format!("{}{}/{}", encoded, device_id, mac),
-        None => format!("{}{}", encoded, device_id),
-    }
-}

src/message.proto

@@ -1,53 +0,0 @@
-syntax = "proto3";
-
-package cursor;
-
-message ChatMessage {
-  message FileContent {
-    message Position {
-      int32 line = 1;
-      int32 column = 2;
-    }
-    message Range {
-      Position start = 1;
-      Position end = 2;
-    }
-
-    string filename = 1;
-    string content = 2;
-    Position position = 3;
-    string language = 5;
-    Range range = 6;
-    int32 length = 8;
-    int32 type = 9;
-    int32 error_code = 11;
-  }
-
-  message Message {
-    string content = 1;
-    int32 role = 2;
-    string message_id = 13;
-  }
-
-  message Instructions {
-    string content = 1;
-  }
-
-  message Model {
-    string name = 1;
-    string empty = 4;
-  }
-
-  // repeated FileContent files = 1;
-  repeated Message messages = 2;
-  Instructions instructions = 4;
-  string projectPath = 5;
-  Model model = 7;
-  string requestId = 9;
-  string summary = 11; // 或许是空的，描述会话做了什么事情，但是不是标题 或许可以当作额外的设定来用
-  string conversationId = 15; // 又来一个uuid
-}
-
-message ResMessage {
-  string msg = 1;
-}

src/models.rs

@@ -1,141 +0,0 @@
-use crate::Model;
-use std::sync::LazyLock;
-
-use super::{ANTHROPIC, CURSOR, GOOGLE, MODEL_OBJECT, OPENAI};
-
-pub static AVAILABLE_MODELS: LazyLock<Vec<Model>> = LazyLock::new(|| {
-    vec![
-        Model {
-            id: "claude-3.5-sonnet".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: ANTHROPIC.into(),
-        },
-        Model {
-            id: "gpt-3.5".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "gpt-4".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "gpt-4o".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "claude-3-opus".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: ANTHROPIC.into(),
-        },
-        Model {
-            id: "cursor-fast".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: CURSOR.into(),
-        },
-        Model {
-            id: "cursor-small".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: CURSOR.into(),
-        },
-        Model {
-            id: "gpt-3.5-turbo".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "gpt-4-turbo-2024-04-09".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "gpt-4o-128k".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "gemini-1.5-flash-500k".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: GOOGLE.into(),
-        },
-        Model {
-            id: "claude-3-haiku-200k".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: ANTHROPIC.into(),
-        },
-        Model {
-            id: "claude-3-5-sonnet-200k".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: ANTHROPIC.into(),
-        },
-        Model {
-            id: "claude-3-5-sonnet-20241022".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: ANTHROPIC.into(),
-        },
-        Model {
-            id: "gpt-4o-mini".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "o1-mini".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "o1-preview".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "o1".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: OPENAI.into(),
-        },
-        Model {
-            id: "claude-3.5-haiku".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: ANTHROPIC.into(),
-        },
-        Model {
-            id: "gemini-exp-1206".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: GOOGLE.into(),
-        },
-        Model {
-            id: "gemini-2.0-flash-thinking-exp".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: GOOGLE.into(),
-        },
-        Model {
-            id: "gemini-2.0-flash-exp".into(),
-            created: 1706659200,
-            object: MODEL_OBJECT.into(),
-            owned_by: GOOGLE.into(),
-        },
-    ]
-});

start_instruction

@@ -1,6 +0,0 @@
-当前版本已稳定，若发现响应出现缺字漏字，与本程序无关。
-若发现首字慢，与本程序无关。
-若发现响应出现乱码，也与本程序无关。
-属于官方的问题，请不要像作者反馈。
-本程序拥有堪比客户端原本的速度，甚至可能更快。
-本程序的性能是非常厉害的。

static/config.html

@@ -23,13 +23,15 @@
         <option value="/tokeninfo">Token 信息页面 (/tokeninfo)</option>
         <option value="/static/shared-styles.css">共享样式 (/static/shared-styles.css)</option>
         <option value="/static/shared.js">共享脚本 (/static/shared.js)</option>
+        <option value="/about">关于页面 (/about)</option>
+        <option value="/readme">ReadMe文档 (/readme)</option>
       </select>
     </div>
 
     <div class="form-group">
       <label>内容类型:</label>
       <select id="content_type">
-        <option value="default">默认行为</option>
+        <option value="default">默认</option>
         <option value="text">纯文本</option>
         <option value="html">HTML</option>
       </select>
@@ -91,6 +93,18 @@
       <input type="password" id="authToken">
     </div>
 
+    <div class="form-group">
+      <label>使用量检查模型规则:</label>
+      <select id="check_usage_models_type">
+        <option value="">保持不变</option>
+        <option value="none">禁用</option>
+        <option value="default">默认</option>
+        <option value="all">所有</option>
+        <option value="list">自定义列表</option>
+      </select>
+      <input type="text" id="check_usage_models_list" placeholder="模型列表，以逗号分隔" style="display: none;">
+    </div>
+
     <div class="button-group">
       <button onclick="updateConfig('get')">获取配置</button>
       <button onclick="updateConfig('update')">更新配置</button>
@@ -147,6 +161,8 @@
           parseStringFromBoolean(data.data.enable_slow_pool, '');
         document.getElementById('enable_all_claude').value =
           parseStringFromBoolean(data.data.enable_all_claude, '');
+        document.getElementById('check_usage_models_type').value = data.data.check_usage_models?.type || '';
+        document.getElementById('check_usage_models_list').value = data.data.check_usage_models?.type === 'list' ? data.data.check_usage_models?.content || '' : document.getElementById('check_usage_models_list').value;
       }
     }
 
@@ -186,6 +202,14 @@
         }),
         ...(document.getElementById('enable_all_claude').value && {
           enable_all_claude: parseBooleanFromString(document.getElementById('enable_all_claude').value)
+        }),
+        ...(document.getElementById('check_usage_models_type').value && {
+          check_usage_models: {
+            type: document.getElementById('check_usage_models_type').value,
+            ...(document.getElementById('check_usage_models_type').value === 'list' && {
+              content: document.getElementById('check_usage_models_list').value
+            })
+          }
         })
       };
 
@@ -220,6 +244,12 @@
 
     // 初始化 token 处理
     initializeTokenHandling('authToken');
+
+    // 添加使用量检查模型类型变更处理
+    document.getElementById('check_usage_models_type').addEventListener('change', function() {
+      const input = document.getElementById('check_usage_models_list');
+      input.style.display = this.value === 'list' ? 'inline-block' : 'none';
+    });
   </script>
 </body>
 

static/readme.html

@@ -0,0 +1,399 @@
+<h1>cursor-api</h1>
+
+<h2>说明</h2>
+
+<ul>
+<li>当前版本已稳定，若发现响应出现缺字漏字，与本程序无关。</li>
+<li>若发现首字慢，与本程序无关。</li>
+<li>若发现响应出现乱码，也与本程序无关。</li>
+<li>属于官方的问题，请不要像作者反馈。</li>
+<li>本程序拥有堪比客户端原本的速度，甚至可能更快。</li>
+<li>本程序的性能是非常厉害的。</li>
+</ul>
+
+<h2>获取key</h2>
+
+<ol>
+<li>访问 <a href="https://www.cursor.com">www.cursor.com</a> 并完成注册登录</li>
+<li>在浏览器中打开开发者工具（F12）</li>
+<li>在 Application-Cookies 中查找名为 <code>WorkosCursorSessionToken</code> 的条目，并复制其第三个字段。请注意，%3A%3A 是 :: 的 URL 编码形式，cookie 的值使用冒号 (:) 进行分隔。</li>
+</ol>
+
+<h2>配置说明</h2>
+
+<h3>环境变量</h3>
+
+<ul>
+<li><code>PORT</code>: 服务器端口号（默认：3000）</li>
+<li><code>AUTH_TOKEN</code>: 认证令牌（必须，用于API认证）</li>
+<li><code>ROUTE_PREFIX</code>: 路由前缀（可选）</li>
+<li><code>TOKEN_FILE</code>: token文件路径（默认：.token）</li>
+<li><code>TOKEN_LIST_FILE</code>: token列表文件路径（默认：.token-list）</li>
+</ul>
+
+<p>更多请查看 <code>/env-example</code></p>
+
+<h3>Token文件格式</h3>
+
+<ol>
+<li>
+<p><code>.token</code> 文件：每行一个token，支持以下格式：</p>
+
+<pre><code># 这是注释
+token1
+# alias与标签的作用差不多
+alias::token2
+</code></pre>
+
+<p>alias 可以是任意值，用于区分不同的 token，更方便管理，WorkosCursorSessionToken 是相同格式<br>
+该文件将自动向.token-list文件中追加token，同时自动生成checksum</p>
+</li>
+
+<li>
+<p><code>.token-list</code> 文件：每行为token和checksum的对应关系：</p>
+
+<pre><code># 这里的#表示这行在下次读取要删除
+token1,checksum1
+# 支持像.token一样的alias，冲突时以.token为准
+alias::token2,checksum2
+</code></pre>
+
+<p>该文件可以被自动管理，但用户仅可在确认自己拥有修改能力时修改，一般仅有以下情况需要手动修改：</p>
+
+<ul>
+<li>需要删除某个 token</li>
+<li>需要使用已有 checksum 来对应某一个 token</li>
+</ul>
+</li>
+</ol>
+
+<h3>模型列表</h3>
+
+<p>写死了，后续也不会会支持自定义模型列表</p>
+
+<pre><code>claude-3.5-sonnet
+gpt-4
+gpt-4o
+claude-3-opus
+cursor-fast
+cursor-small
+gpt-3.5-turbo
+gpt-4-turbo-2024-04-09
+gpt-4o-128k
+gemini-1.5-flash-500k
+claude-3-haiku-200k
+claude-3-5-sonnet-200k
+claude-3-5-sonnet-20241022
+gpt-4o-mini
+o1-mini
+o1-preview
+o1
+claude-3.5-haiku
+gemini-exp-1206
+gemini-2.0-flash-thinking-exp
+gemini-2.0-flash-exp
+</code></pre>
+
+<h1>接口说明</h1>
+
+<h2>基础对话</h2>
+
+<ul>
+<li>接口地址: <code>/v1/chat/completions</code></li>
+<li>请求方法: POST</li>
+<li>认证方式: Bearer Token
+<ol>
+<li>使用环境变量 <code>AUTH_TOKEN</code> 进行认证</li>
+<li>使用 <code>.token</code> 文件中的令牌列表进行轮询认证</li>
+</ol></li>
+</ul>
+
+<h3>请求格式</h3>
+
+<pre><code class="language-json">{
+  "model": "string",
+  "messages": [
+    {
+      "role": "system" | "user" | "assistant", // 也可以是 "developer" | "human" | "ai"
+      "content": "string" | [
+        {
+          "type": "text" | "image_url",
+          "text": "string",
+          "image_url": {
+            "url": "string"
+          }
+        }
+      ]
+    }
+  ],
+  "stream": boolean
+}
+</code></pre>
+
+<h3>响应格式</h3>
+
+<p>如果 <code>stream</code> 为 <code>false</code>:</p>
+
+<pre><code class="language-json">{
+  "id": "string",
+  "object": "chat.completion",
+  "created": number,
+  "model": "string",
+  "choices": [
+    {
+      "index": number,
+      "message": {
+        "role": "assistant",
+        "content": "string"
+      },
+      "finish_reason": "stop" | "length"
+    }
+  ],
+  "usage": {
+    "prompt_tokens": number,
+    "completion_tokens": number,
+    "total_tokens": number
+  }
+}
+</code></pre>
+
+<p>如果 <code>stream</code> 为 <code>true</code>:</p>
+
+<pre><code>data: {"id":"string","object":"chat.completion.chunk","created":number,"model":"string","choices":[{"index":number,"delta":{"role":"assistant","content":"string"},"finish_reason":null}]}
+
+data: {"id":"string","object":"chat.completion.chunk","created":number,"model":"string","choices":[{"index":number,"delta":{"content":"string"},"finish_reason":null}]}
+
+data: {"id":"string","object":"chat.completion.chunk","created":number,"model":"string","choices":[{"index":number,"delta":{},"finish_reason":"stop"}]}
+
+data: [DONE]
+</code></pre>
+
+<h2>Token管理接口</h2>
+
+<h3>简易Token信息管理页面</h3>
+
+<ul>
+<li>接口地址: <code>/tokeninfo</code></li>
+<li>请求方法: GET</li>
+<li>响应格式: HTML页面</li>
+<li>功能: 获取 .token 和 .token-list 文件内容,并允许用户方便地使用 API 修改文件内容</li>
+</ul>
+
+<h3>更新Token信息 (GET)</h3>
+
+<ul>
+<li>接口地址: <code>/update-tokeninfo</code></li>
+<li>请求方法: GET</li>
+<li>认证方式: 不需要</li>
+<li>功能: 请求内容不包括文件内容,直接修改文件,调用重载函数</li>
+</ul>
+
+<h3>更新Token信息 (POST)</h3>
+
+<ul>
+<li>接口地址: <code>/update-tokeninfo</code></li>
+<li>请求方法: POST</li>
+<li>认证方式: Bearer Token</li>
+<li>请求格式:</li>
+</ul>
+
+<pre><code class="language-json">{
+  "tokens": "string",
+  "token_list": "string"
+}
+</code></pre>
+
+<ul>
+<li>响应格式:</li>
+</ul>
+
+<pre><code class="language-json">{
+  "status": "success",
+  "message": "Token files have been updated and reloaded",
+  "token_file": "string",
+  "token_list_file": "string",
+  "token_count": number
+}
+</code></pre>
+
+<h3>获取Token信息</h3>
+
+<ul>
+<li>接口地址: <code>/get-tokeninfo</code></li>
+<li>请求方法: POST</li>
+<li>认证方式: Bearer Token</li>
+<li>响应格式:</li>
+</ul>
+
+<pre><code class="language-json">{
+  "status": "success",
+  "token_file": "string",
+  "token_list_file": "string",
+  "tokens": "string",
+  "token_list": "string"
+}
+</code></pre>
+
+<h2>配置管理接口</h2>
+
+<h3>配置页面</h3>
+
+<ul>
+<li>接口地址: <code>/config</code></li>
+<li>请求方法: GET</li>
+<li>响应格式: HTML页面</li>
+<li>功能: 提供配置管理界面,可以修改页面内容和系统配置</li>
+</ul>
+
+<h3>更新配置</h3>
+
+<ul>
+<li>接口地址: <code>/config</code></li>
+<li>请求方法: POST</li>
+<li>认证方式: Bearer Token</li>
+<li>请求格式:</li>
+</ul>
+
+<pre><code class="language-json">{
+  "action": "get" | "update" | "reset",
+  "path": "string",
+  "content": "string",
+  "content_type": "default" | "text" | "html",
+  "enable_stream_check": boolean,
+  "enable_stream_check": boolean,
+  "vision_ability": "none" | "base64" | "all", // "disabled" | "base64-only" | "base64-http"
+  "enable_slow_pool": boolean,
+  "enable_slow_pool": boolean
+}
+</code></pre>
+
+<ul>
+<li>响应格式:</li>
+</ul>
+
+<pre><code class="language-json">{
+  "status": "success",
+  "message": "string",
+  "data": {
+    "page_content": {
+      "type": "default" | "text" | "html",
+      "content": "string"
+    },
+    "enable_stream_check": boolean,
+    "vision_ability": "base64" | "url" | "none", 
+    "enable_slow_pool": boolean
+  }
+}
+</code></pre>
+
+<h2>静态资源接口</h2>
+
+<h3>获取共享样式</h3>
+
+<ul>
+<li>接口地址: <code>/static/shared-styles.css</code></li>
+<li>请求方法: GET</li>
+<li>响应格式: CSS文件</li>
+<li>功能: 获取共享样式表</li>
+</ul>
+
+<h3>获取共享脚本</h3>
+
+<ul>
+<li>接口地址: <code>/static/shared.js</code></li>
+<li>请求方法: GET</li>
+<li>响应格式: JavaScript文件</li>
+<li>功能: 获取共享JavaScript代码</li>
+</ul>
+
+<h3>环境变量示例</h3>
+
+<ul>
+<li>接口地址: <code>/env-example</code></li>
+<li>请求方法: GET</li>
+<li>响应格式: 文本文件</li>
+<li>功能: 获取环境变量配置示例</li>
+</ul>
+
+<h2>其他接口</h2>
+
+<h3>获取模型列表</h3>
+
+<ul>
+<li>接口地址: <code>/v1/models</code></li>
+<li>请求方法: GET</li>
+<li>响应格式:</li>
+</ul>
+
+<pre><code class="language-json">{
+  "object": "list",
+  "data": [
+    {
+      "id": "string",
+      "object": "model",
+      "created": number,
+      "owned_by": "string"
+    }
+  ]
+}
+</code></pre>
+
+<h3>获取随机checksum</h3>
+
+<ul>
+<li>接口地址: <code>/checksum</code></li>
+<li>请求方法: GET</li>
+<li>响应格式:</li>
+</ul>
+
+<pre><code class="language-json">{
+  "checksum": "string"
+}
+</code></pre>
+
+<h3>健康检查接口</h3>
+
+<ul>
+<li>接口地址: <code>/health</code> 或 <code>/</code>(重定向)</li>
+<li>请求方法: GET</li>
+<li>响应格式: 根据配置返回不同的内容类型(默认、文本或HTML)</li>
+</ul>
+
+<h3>获取日志接口</h3>
+
+<ul>
+<li>接口地址: <code>/logs</code></li>
+<li>请求方法: GET</li>
+<li>响应格式: 根据配置返回不同的内容类型(默认、文本或HTML)</li>
+</ul>
+
+<h3>获取日志数据</h3>
+
+<ul>
+<li>接口地址: <code>/logs</code></li>
+<li>请求方法: POST</li>
+<li>认证方式: Bearer Token</li>
+<li>响应格式:</li>
+</ul>
+
+<pre><code class="language-json">{
+  "total": number,
+  "logs": [
+    {
+      "timestamp": "string",
+      "model": "string",
+      "token_info": {
+        "token": "string",
+        "checksum": "string",
+        "alias": "string"
+      },
+      "prompt": "string",
+      "stream": boolean,
+      "status": "string",
+      "error": "string"
+    }
+  ],
+  "timestamp": "string",
+  "status": "success"
+}
+</code></pre>

static/shared-styles.css

@@ -44,25 +44,38 @@ label {
   font-weight: 500;
 }
 
-input[type="text"],
+/* input[type="text"], 由于minify.js会删除input[type="text"]，所以改为input */
+input,
 input[type="password"],
 select,
 textarea {
   width: 100%;
-  padding: 8px 12px;
+  padding: 10px 12px;
   border: 1px solid #ddd;
   border-radius: 4px;
   font-size: 14px;
-  transition: border-color 0.2s;
+  transition: border-color 0.2s, box-shadow 0.2s;
+  background: white;
+  color: #333;
+  appearance: none;
 }
 
-input[type="text"]:focus,
+/* input[type="text"]:focus, 由于minify.js会删除input[type="text"]:focus，所以改为input:focus */
+input:focus,
 input[type="password"]:focus,
 select:focus,
 textarea:focus {
   border-color: var(--primary-color);
   outline: none;
-  box-shadow: 0 0 0 2px rgba(33, 150, 243, 0.1);
+  box-shadow: 0 0 0 2px rgba(33, 150, 243, 0.2);
+}
+
+select {
+  background-image: url("data:image/svg+xml,%3Csvg xmlns='http://www.w3.org/2000/svg' viewBox='0 0 24 24' fill='%23757575'%3E%3Cpath d='M7 10l5 5 5-5H7z'/%3E%3C/svg%3E");
+  background-repeat: no-repeat;
+  background-position: right 8px center;
+  background-size: 20px;
+  padding-right: 36px;
 }
 
 textarea {