up

.env.example

@@ -1,5 +1,100 @@
+# 当前配置为默认值，请根据需要修改
+
+# 服务器监听端口
 PORT=3000
+
+# 路由前缀，必须以 / 开头（如果不为空）
 ROUTE_PREFIX=
+
+# 最高权限的认证令牌，必填
 AUTH_TOKEN=
-TOKEN_FILE=.token
-TOKEN_LIST_FILE=.token-list
+
+# 共享的认证令牌，仅Chat端点权限(轮询与AUTH_TOKEN同步)，无其余权限
+SHARED_TOKEN=
+
+# 启用流式响应检查，关闭则无法响应错误，代价是会对第一个块解析2次(已弃用)
+# 新版本已经完成优化
+# ENABLE_STREAM_CHECK=true
+
+# 流式消息结束后发送包含"finish_reason"为"stop"的空消息块(已弃用)
+# INCLUDE_STOP_REASON_STREAM=true
+
+# 令牌文件路径(已弃用)
+# TOKEN_FILE=.token
+
+# 令牌列表文件路径
+TOKEN_LIST_FILE=.tokens
+
+# （实验性）是否启用慢速池（true/false）
+ENABLE_SLOW_POOL=false
+
+# 允许claude开头的模型请求绕过内置模型限制（true/false）
+PASS_ANY_CLAUDE=false
+
+# 图片处理能力配置
+# 可选值:
+# - none 或 disabled：禁用图片功能
+# - base64 或 base64-only：仅支持 base64 编码的图片
+# - all 或 base64-http：支持 base64 和 HTTP 图片
+#   注意：启用 HTTP 支持可能会暴露服务器 IP
+VISION_ABILITY=base64
+
+# 额度检查配置
+# 可选值:
+# - none 或 disabled：禁用额度检查
+# - default：详见 README
+# - all 或 everything：额度无条件检查
+# - 以,分隔的模型列表，为空时使用默认值
+USAGE_CHECK=default
+
+# 是否允许使用动态(自定义)配置的 API Key
+DYNAMIC_KEY=false
+
+# 动态 Key 的标识前缀
+KEY_PREFIX=sk-
+
+# 默认提示词
+DEFAULT_INSTRUCTIONS="Respond in Chinese by default"
+
+# 反向代理服务器主机名
+REVERSE_PROXY_HOST=
+
+# 代理地址配置说明
+# - 留空或 `no`: 不使用任何代理
+# - `system`: 使用系统代理（变量不存在时的默认值）
+# - 代理地址: 支持以下格式
+#   - 多个代理: `http://localhost:7890,https://username:password@localhost:1234`
+#     没有轮询，只是选择第一个格式正确的
+#   - 支持的协议: http, https, socks4, socks5, socks5h
+PROXIES=
+
+# 请求体大小限制（单位为MB）
+# 默认为2MB (2,097,152 字节)
+REQUEST_BODY_LIMIT_MB=2
+
+# OpenAI 请求时，token 和 checksum 的分隔符
+TOKEN_DELIMITER=,
+
+# 同时兼容默认的,作为分隔符
+USE_COMMA_DELIMITER=true
+
+# 调试
+DEBUG=false
+
+# 调试文件
+DEBUG_LOG_FILE=debug.log
+
+# 日志储存条数(最大值2000)
+REQUEST_LOGS_LIMIT=100
+
+# Cursor 服务超时(秒)(最大值600)
+SERVICE_TIMEOUT=30
+
+# 包含网络引用
+INCLUDE_WEB_REFERENCES=false
+
+# 持久化日志文件路径
+LOGS_FILE_PATH=logs.bin
+
+# 持久化页面配置文件路径
+PAGES_FILE_PATH=pages.bin

.gitattributes

@@ -1,35 +1,6 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz filter=lfs diff=lfs merge=lfs -text
-*.zip filter=lfs diff=lfs merge=lfs -text
-*.zst filter=lfs diff=lfs merge=lfs -text
-*tfevents* filter=lfs diff=lfs merge=lfs -text
+# 统一使用 LF
+* text=auto eol=lf
+
+# 对特定文件类型设置
+*.bat text eol=crlf
+*.ps1 text eol=crlf

.gitignore

@@ -1,14 +1,26 @@
 /target
-/get-token/target
+/tools/*/target
 /*.log
 /*.env
-/static/tokeninfo.min.html
+/static/*.min.html
+/static/*.min.css
+/static/*.min.js
+/scripts/.asset-hashes.json
 node_modules
 .DS_Store
 /.vscode
 /.cargo
 /.token
 /.token-list
+/.tokens
 /cursor-api
 /cursor-api.exe
 /release
+
+/*.py
+/logs
+/dev*
+/build*
+/*.bin
+/result.txt
+tools/tokenizer/

.license-compliance

@@ -0,0 +1,16 @@
+# 合规使用指南
+attribution_rules:
+  required_disclaimer: 
+    text: "基于第三方技术构建，与原始开发者无关联"
+    placement: 
+      - documentation
+      - marketing_materials
+      - about_sections
+  prohibited_actions:
+    - using_author_name_in_press_releases
+    - claiming_official_support
+    - using_project_logo_as_endorsement
+
+enforcement:
+  grace_period: 72h
+  compliance_check: https://api.wisdgod.com/license/validate

Cargo.lock

@@ -17,6 +17,17 @@ version = "2.0.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "512761e0bb2578dd7380c6baaa0f4ce03e84f95e960231d1dec8bf4d7d6e2627"
 
+[[package]]
+name = "ahash"
+version = "0.7.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "891477e0c6a8957309ee5c45a6368af3ae14bb510732d2684ffa19af310920f9"
+dependencies = [
+ "getrandom",
+ "once_cell",
+ "version_check",
+]
+
 [[package]]
 name = "aho-corasick"
 version = "1.1.3"
@@ -26,6 +37,21 @@ dependencies = [
  "memchr",
 ]
 
+[[package]]
+name = "alloc-no-stdlib"
+version = "2.0.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cc7bb162ec39d46ab1ca8c77bf72e890535becd1751bb45f64c597edb4c8c6b3"
+
+[[package]]
+name = "alloc-stdlib"
+version = "0.2.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94fb8275041c72129eb51b7d0322c29b8387a0386127718b096429201a5d6ece"
+dependencies = [
+ "alloc-no-stdlib",
+]
+
 [[package]]
 name = "android-tzdata"
 version = "0.1.1"
@@ -53,6 +79,7 @@ version = "0.4.18"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "df895a515f70646414f4b45c0b79082783b80552b373a68283012928df56f522"
 dependencies = [
+ "brotli",
  "flate2",
  "futures-core",
  "memchr",
@@ -60,17 +87,6 @@ dependencies = [
  "tokio",
 ]
 
-[[package]]
-name = "async-trait"
-version = "0.1.83"
-source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "721cae7de5c34fbb2acd27e21e6d2cf7b886dce0c27388d46c4e6c47ea4318dd"
-dependencies = [
- "proc-macro2",
- "quote",
- "syn",
-]
-
 [[package]]
 name = "atomic-waker"
 version = "1.1.2"
@@ -85,13 +101,13 @@ checksum = "ace50bade8e6234aa140d9a2f552bbee1db4d353f69b8217bc503490fc1a9f26"
 
 [[package]]
 name = "axum"
-version = "0.7.9"
+version = "0.8.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "edca88bc138befd0323b20752846e6587272d3b03b0343c8ea28a6f819e6e71f"
+checksum = "6d6fd624c75e18b3b4c6b9caf42b1afe24437daaee904069137d8bab077be8b8"
 dependencies = [
- "async-trait",
  "axum-core",
  "bytes",
+ "form_urlencoded",
  "futures-util",
  "http",
  "http-body",
@@ -119,11 +135,10 @@ dependencies = [
 
 [[package]]
 name = "axum-core"
-version = "0.4.5"
+version = "0.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "09f2bd6146b97ae3359fa0cc6d6b376d9539582c7b4220f041a33ec24c226199"
+checksum = "df1362f362fd16024ae199c1970ce98f9661bf5ef94b9808fee734bc3698b733"
 dependencies = [
- "async-trait",
  "bytes",
  "futures-util",
  "http",
@@ -161,9 +176,27 @@ checksum = "72b3254f16251a8381aa12e40e3c4d2f0199f8c6508fbecb9d91f575e0fbb8c6"
 
 [[package]]
 name = "bitflags"
-version = "2.6.0"
+version = "1.3.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bef38d45163c2f1dde094a7dfd33ccf595c92905c8f8f4fdc18d06fb1037718a"
+
+[[package]]
+name = "bitflags"
+version = "2.8.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "b048fb63fd8b5923fc5aa7b340d8e156aec7ec02f0c78fa8a6ddc2613f6f71de"
+checksum = "8f68f53c83ab957f72c32642f3868eec03eb974d1fb82e453128456482613d36"
+
+[[package]]
+name = "bitvec"
+version = "1.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1bc2832c24239b0141d5674bb9174f9d68a8b5b3f2753311927c172ca46f7e9c"
+dependencies = [
+ "funty",
+ "radium",
+ "tap",
+ "wyz",
+]
 
 [[package]]
 name = "block-buffer"
@@ -174,18 +207,73 @@ dependencies = [
  "generic-array",
 ]
 
+[[package]]
+name = "brotli"
+version = "7.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cc97b8f16f944bba54f0433f07e30be199b6dc2bd25937444bbad560bcea29bd"
+dependencies = [
+ "alloc-no-stdlib",
+ "alloc-stdlib",
+ "brotli-decompressor",
+]
+
+[[package]]
+name = "brotli-decompressor"
+version = "4.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9a45bd2e4095a8b518033b128020dd4a55aab1c0a381ba4404a472630f4bc362"
+dependencies = [
+ "alloc-no-stdlib",
+ "alloc-stdlib",
+]
+
 [[package]]
 name = "bumpalo"
 version = "3.16.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "79296716171880943b8470b5f8d03aa55eb2e645a4874bdbb28adb49162e012c"
 
+[[package]]
+name = "bytecheck"
+version = "0.6.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "23cdc57ce23ac53c931e88a43d06d070a6fd142f2617be5855eb75efc9beb1c2"
+dependencies = [
+ "bytecheck_derive",
+ "ptr_meta 0.1.4",
+ "simdutf8",
+]
+
+[[package]]
+name = "bytecheck_derive"
+version = "0.6.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3db406d29fbcd95542e92559bed4d8ad92636d1ca8b3b72ede10b4bcc010e659"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "bytemuck"
+version = "1.21.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ef657dfab802224e671f5818e9a4935f9b1957ed18e58292690cc39e7a4092a3"
+
 [[package]]
 name = "byteorder"
 version = "1.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "1fd0f2584146f6f2ef48085050886acf353beff7305ebd1ae69500e27c67f64b"
 
+[[package]]
+name = "byteorder-lite"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8f1fe948ff07f4bd06c30984e69f5b4899c516a3ef74f34df92a2df2ab535495"
+
 [[package]]
 name = "bytes"
 version = "1.9.0"
@@ -194,9 +282,9 @@ checksum = "325918d6fe32f23b19878fe4b34794ae41fc19ddbe53b10571a4874d44ffd39b"
 
 [[package]]
 name = "cc"
-version = "1.2.5"
+version = "1.2.10"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "c31a0499c1dc64f458ad13872de75c0eb7e3fdb0e67964610c914b034fc5956e"
+checksum = "13208fcbb66eaeffe09b99fffbe1af420f00a7b35aa99ad683dfc1aa76145229"
 dependencies = [
  "shlex",
 ]
@@ -215,13 +303,18 @@ checksum = "7e36cc9d416881d2e24f9a963be5fb1cd90966419ac844274161d10488b3e825"
 dependencies = [
  "android-tzdata",
  "iana-time-zone",
- "js-sys",
  "num-traits",
+ "rkyv 0.7.45",
  "serde",
- "wasm-bindgen",
  "windows-targets",
 ]
 
+[[package]]
+name = "color_quant"
+version = "1.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3d7b894f5411737b7867f4827955924d7c254fc9f4d91a6aad6b097804b1018b"
+
 [[package]]
 name = "core-foundation"
 version = "0.9.4"
@@ -268,7 +361,7 @@ dependencies = [
 
 [[package]]
 name = "cursor-api"
-version = "0.1.0"
+version = "0.1.3-rc.4.3"
 dependencies = [
  "axum",
  "base64",
@@ -277,18 +370,27 @@ dependencies = [
  "dotenvy",
  "flate2",
  "futures",
+ "gif",
  "hex",
+ "image",
+ "memmap2",
+ "parking_lot",
+ "paste",
  "prost",
  "prost-build",
  "rand",
  "regex",
  "reqwest",
+ "rkyv 0.7.45",
  "serde",
  "serde_json",
  "sha2",
+ "sonic-rs",
+ "sysinfo",
  "tokio",
  "tokio-stream",
  "tower-http",
+ "url",
  "uuid",
 ]
 
@@ -310,7 +412,7 @@ checksum = "97369cbbc041bc366949bc74d34658d6cda5621039731c6310521892a3a20ae0"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
@@ -356,6 +458,27 @@ version = "2.3.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "37909eebbb50d72f9059c3b6d82c0463f2ff062c9e95845c43a6c9c0355411be"
 
+[[package]]
+name = "faststr"
+version = "0.2.27"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9154486833a83cb5d99de8c4d831314b8ae810dd4ef18d89ceb7a9c7c728dd74"
+dependencies = [
+ "bytes",
+ "rkyv 0.8.10",
+ "serde",
+ "simdutf8",
+]
+
+[[package]]
+name = "fdeflate"
+version = "0.3.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1e6853b52649d4ac5c0bd02320cddc5ba956bdb407c4b75a2c6b75bf51500f8c"
+dependencies = [
+ "simd-adler32",
+]
+
 [[package]]
 name = "fixedbitset"
 version = "0.4.2"
@@ -402,6 +525,12 @@ dependencies = [
  "percent-encoding",
 ]
 
+[[package]]
+name = "funty"
+version = "2.0.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e6d5a32815ae3f33302d95fdcb2ce17862f8c65363dcfd29360480ba1001fc9c"
+
 [[package]]
 name = "futures"
 version = "0.3.31"
@@ -446,7 +575,7 @@ checksum = "162ee34ebcb7c64a8abebc059ce0fee27c2262618d7b60ed8faf72fef13c3650"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
@@ -500,6 +629,16 @@ dependencies = [
  "wasi",
 ]
 
+[[package]]
+name = "gif"
+version = "0.13.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3fb2d69b19215e18bb912fa30f7ce15846e301408695e44e0ef719f1da9e19f2"
+dependencies = [
+ "color_quant",
+ "weezl",
+]
+
 [[package]]
 name = "gimli"
 version = "0.31.1"
@@ -525,6 +664,15 @@ dependencies = [
  "tracing",
 ]
 
+[[package]]
+name = "hashbrown"
+version = "0.12.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "8a9ee70c43aaf417c914396645a0fa852624801b24ebb7ae78fe8272889ac888"
+dependencies = [
+ "ahash",
+]
+
 [[package]]
 name = "hashbrown"
 version = "0.15.2"
@@ -673,7 +821,7 @@ dependencies = [
  "iana-time-zone-haiku",
  "js-sys",
  "wasm-bindgen",
- "windows-core",
+ "windows-core 0.52.0",
 ]
 
 [[package]]
@@ -800,7 +948,7 @@ checksum = "1ec89e9337638ecdc08744df490b221a7399bf8d164eb52a665454e60e075ad6"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
@@ -824,21 +972,48 @@ dependencies = [
  "icu_properties",
 ]
 
+[[package]]
+name = "image"
+version = "0.25.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "cd6f44aed642f18953a158afeb30206f4d50da59fbc66ecb53c66488de73563b"
+dependencies = [
+ "bytemuck",
+ "byteorder-lite",
+ "color_quant",
+ "gif",
+ "image-webp",
+ "num-traits",
+ "png",
+ "zune-core",
+ "zune-jpeg",
+]
+
+[[package]]
+name = "image-webp"
+version = "0.2.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b77d01e822461baa8409e156015a1d91735549f0f2c17691bd2d996bef238f7f"
+dependencies = [
+ "byteorder-lite",
+ "quick-error",
+]
+
 [[package]]
 name = "indexmap"
-version = "2.7.0"
+version = "2.7.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "62f822373a4fe84d4bb149bf54e584a7f4abec90e072ed49cda0edea5b95471f"
+checksum = "8c9c992b02b5b4c94ea26e32fe5bccb7aa7d9f390ab5c1221ff895bc7ea8b652"
 dependencies = [
  "equivalent",
- "hashbrown",
+ "hashbrown 0.15.2",
 ]
 
 [[package]]
 name = "ipnet"
-version = "2.10.1"
+version = "2.11.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "ddc24109865250148c2e0f3d25d4f0f479571723792d3802153c60922a4fb708"
+checksum = "469fb0b9cefa57e3ef31275ee7cacb78f2fdca44e4765491884a2b119d4eb130"
 
 [[package]]
 name = "itertools"
@@ -857,9 +1032,9 @@ checksum = "d75a2a4b1b190afb6f5425f10f6a8f959d2ea0b9c2b1d79553551850539e4674"
 
 [[package]]
 name = "js-sys"
-version = "0.3.76"
+version = "0.3.77"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "6717b6b5b077764fb5966237269cb3c64edddde4b14ce42647430a78ced9e7b7"
+checksum = "1cfaf33c695fc6e08064efbc1f72ec937429614f25eef83af942d0e227c3a28f"
 dependencies = [
  "once_cell",
  "wasm-bindgen",
@@ -873,9 +1048,9 @@ checksum = "b5aba8db14291edd000dfcc4d620c7ebfb122c613afb886ca8803fa4e128a20a"
 
 [[package]]
 name = "linux-raw-sys"
-version = "0.4.14"
+version = "0.4.15"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "78b3ae25bc7c8c38cec158d1f2757ee79e9b3740fbc7ccf0e59e4b08d793fa89"
+checksum = "d26c52dbd32dccf2d10cac7725f8eae5296885fb5703b261f7d0a0739ec807ab"
 
 [[package]]
 name = "litemap"
@@ -883,17 +1058,27 @@ version = "0.7.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "4ee93343901ab17bd981295f2cf0026d4ad018c7c31ba84549a4ddbb47a45104"
 
+[[package]]
+name = "lock_api"
+version = "0.4.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "07af8b9cdd281b7915f413fa73f29ebd5d55d0d3f0155584dade1ff18cea1b17"
+dependencies = [
+ "autocfg",
+ "scopeguard",
+]
+
 [[package]]
 name = "log"
-version = "0.4.22"
+version = "0.4.25"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "a7a70ba024b9dc04c27ea2f0c0548feb474ec5c54bba33a7f72f873a39d07b24"
+checksum = "04cbf5b083de1c7e0222a7a51dbfdba1cbe1c6ab0b15e29fff3f6c077fd9cd9f"
 
 [[package]]
 name = "matchit"
-version = "0.7.3"
+version = "0.8.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0e7465ac9959cc2b1404e8e2367b43684a6d13790fe23056cc8c6c5a6b7bcb94"
+checksum = "47e1ffaa40ddd1f3ed91f717a33c8c0ee23fff369e3aa8772b9605cc1d22f4c3"
 
 [[package]]
 name = "memchr"
@@ -901,6 +1086,15 @@ version = "2.7.4"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "78ca9ab1a0babb1e7d5695e3530886289c18cf2f87ec19a575a0abdce112e3a3"
 
+[[package]]
+name = "memmap2"
+version = "0.9.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fd3f7eed9d3848f8b98834af67102b720745c4ec028fcd0aa0239277e7de374f"
+dependencies = [
+ "libc",
+]
+
 [[package]]
 name = "mime"
 version = "0.3.17"
@@ -909,11 +1103,12 @@ checksum = "6877bb514081ee2a7ff5ef9de3281f14a4dd4bceac4c09388074a6b5df8a139a"
 
 [[package]]
 name = "miniz_oxide"
-version = "0.8.2"
+version = "0.8.3"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "4ffbe83022cedc1d264172192511ae958937694cd57ce297164951b8b3568394"
+checksum = "b8402cab7aefae129c6977bb0ff1b8fd9a04eb5b51efc50a70bea51cda0c7924"
 dependencies = [
  "adler2",
+ "simd-adler32",
 ]
 
 [[package]]
@@ -933,6 +1128,26 @@ version = "0.10.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "defc4c55412d89136f966bbb339008b474350e5e6e78d2714439c386b3137a03"
 
+[[package]]
+name = "munge"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "64142d38c84badf60abf06ff9bd80ad2174306a5b11bd4706535090a30a419df"
+dependencies = [
+ "munge_macro",
+]
+
+[[package]]
+name = "munge_macro"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1bb5c1d8184f13f7d0ccbeeca0def2f9a181bce2624302793005f5ca8aa62e5e"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.96",
+]
+
 [[package]]
 name = "native-tls"
 version = "0.2.12"
@@ -950,6 +1165,15 @@ dependencies = [
  "tempfile",
 ]
 
+[[package]]
+name = "ntapi"
+version = "0.4.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e8a3895c6391c39d7fe7ebc444a87eb2991b2a0bc718fdabd071eec617fc68e4"
+dependencies = [
+ "winapi",
+]
+
 [[package]]
 name = "num-traits"
 version = "0.2.19"
@@ -980,7 +1204,7 @@ version = "0.10.68"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "6174bc48f102d208783c2c84bf931bb75927a617866870de8a4ea85597f871f5"
 dependencies = [
- "bitflags",
+ "bitflags 2.8.0",
  "cfg-if",
  "foreign-types",
  "libc",
@@ -997,7 +1221,7 @@ checksum = "a948666b637a0f465e8564c73e89d4dde00d72d4d473cc972f390fc3dcee7d9c"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
@@ -1018,6 +1242,35 @@ dependencies = [
  "vcpkg",
 ]
 
+[[package]]
+name = "parking_lot"
+version = "0.12.3"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "f1bf18183cf54e8d6059647fc3063646a1801cf30896933ec2311622cc4b9a27"
+dependencies = [
+ "lock_api",
+ "parking_lot_core",
+]
+
+[[package]]
+name = "parking_lot_core"
+version = "0.9.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1e401f977ab385c9e4e3ab30627d6f26d00e2c73eef317493c4ec6d468726cf8"
+dependencies = [
+ "cfg-if",
+ "libc",
+ "redox_syscall",
+ "smallvec",
+ "windows-targets",
+]
+
+[[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"
@@ -1036,9 +1289,9 @@ dependencies = [
 
 [[package]]
 name = "pin-project-lite"
-version = "0.2.15"
+version = "0.2.16"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "915a1e146535de9163f3987b8944ed8cf49a18bb0056bcebcdcece385cece4ff"
+checksum = "3b3cff922bd51709b605d9ead9aa71031d81447142d828eb4a6eba76fe619f9b"
 
 [[package]]
 name = "pin-utils"
@@ -1052,6 +1305,19 @@ version = "0.3.31"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "953ec861398dccce10c670dfeaf3ec4911ca479e9c02154b3a215178c5f566f2"
 
+[[package]]
+name = "png"
+version = "0.17.16"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "82151a2fc869e011c153adc57cf2789ccb8d9906ce52c0b39a6b5697749d7526"
+dependencies = [
+ "bitflags 1.3.2",
+ "crc32fast",
+ "fdeflate",
+ "flate2",
+ "miniz_oxide",
+]
+
 [[package]]
 name = "ppv-lite86"
 version = "0.2.20"
@@ -1063,19 +1329,19 @@ dependencies = [
 
 [[package]]
 name = "prettyplease"
-version = "0.2.25"
+version = "0.2.29"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "64d1ec885c64d0457d564db4ec299b2dae3f9c02808b8ad9c3a089c591b18033"
+checksum = "6924ced06e1f7dfe3fa48d57b9f74f55d8915f5036121bef647ef4b204895fac"
 dependencies = [
  "proc-macro2",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
 name = "proc-macro2"
-version = "1.0.92"
+version = "1.0.93"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "37d3544b3f2748c54e147655edb5025752e2303145b5aefb3c3ea2c78b973bb0"
+checksum = "60946a68e5f9d28b0dc1c21bb8a97ee7d018a8b322fa57838ba31cc878e22d99"
 dependencies = [
  "unicode-ident",
 ]
@@ -1106,7 +1372,7 @@ dependencies = [
  "prost",
  "prost-types",
  "regex",
- "syn",
+ "syn 2.0.96",
  "tempfile",
 ]
 
@@ -1120,7 +1386,7 @@ dependencies = [
  "itertools",
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
@@ -1132,15 +1398,76 @@ dependencies = [
  "prost",
 ]
 
+[[package]]
+name = "ptr_meta"
+version = "0.1.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0738ccf7ea06b608c10564b31debd4f5bc5e197fc8bfe088f68ae5ce81e7a4f1"
+dependencies = [
+ "ptr_meta_derive 0.1.4",
+]
+
+[[package]]
+name = "ptr_meta"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "fe9e76f66d3f9606f44e45598d155cb13ecf09f4a28199e48daf8c8fc937ea90"
+dependencies = [
+ "ptr_meta_derive 0.3.0",
+]
+
+[[package]]
+name = "ptr_meta_derive"
+version = "0.1.4"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "16b845dbfca988fa33db069c0e230574d15a3088f147a87b64c7589eb662c9ac"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "ptr_meta_derive"
+version = "0.3.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ca414edb151b4c8d125c12566ab0d74dc9cdba36fb80eb7b848c15f495fd32d1"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.96",
+]
+
+[[package]]
+name = "quick-error"
+version = "2.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a993555f31e5a609f617c12db6250dedcac1b0a85076912c436e6fc9b2c8e6a3"
+
 [[package]]
 name = "quote"
-version = "1.0.37"
+version = "1.0.38"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "b5b9d34b8991d19d98081b46eacdd8eb58c6f2b201139f7c5f643cc155a633af"
+checksum = "0e4dccaaaf89514f546c693ddc140f729f958c247918a13380cccc6078391acc"
 dependencies = [
  "proc-macro2",
 ]
 
+[[package]]
+name = "radium"
+version = "0.7.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "dc33ff2d4973d518d823d61aa239014831e521c75da58e3df4840d3f47749d09"
+
+[[package]]
+name = "rancor"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "caf5f7161924b9d1cea0e4cabc97c372cea92b5f927fc13c6bca67157a0ad947"
+dependencies = [
+ "ptr_meta 0.3.0",
+]
+
 [[package]]
 name = "rand"
 version = "0.8.5"
@@ -1171,6 +1498,35 @@ dependencies = [
  "getrandom",
 ]
 
+[[package]]
+name = "redox_syscall"
+version = "0.5.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "03a862b389f93e68874fbf580b9de08dd02facb9a788ebadaf4a3fd33cf58834"
+dependencies = [
+ "bitflags 2.8.0",
+]
+
+[[package]]
+name = "ref-cast"
+version = "1.0.23"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ccf0a6f84d5f1d581da8b41b47ec8600871962f2a528115b542b362d4b744931"
+dependencies = [
+ "ref-cast-impl",
+]
+
+[[package]]
+name = "ref-cast-impl"
+version = "1.0.23"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "bcc303e793d3734489387d205e9b186fac9c6cfacedd98cbb2e8a5943595f3e6"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.96",
+]
+
 [[package]]
 name = "regex"
 version = "1.11.1"
@@ -1200,11 +1556,26 @@ version = "0.8.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "2b15c43186be67a4fd63bee50d0303afffcef381492ebe2c5d87f324e1b8815c"
 
+[[package]]
+name = "rend"
+version = "0.4.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "71fe3824f5629716b1589be05dacd749f6aa084c87e00e016714a8cdfccc997c"
+dependencies = [
+ "bytecheck",
+]
+
+[[package]]
+name = "rend"
+version = "0.5.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a35e8a6bf28cd121053a66aa2e6a2e3eaffad4a60012179f0e864aa5ffeff215"
+
 [[package]]
 name = "reqwest"
-version = "0.12.9"
+version = "0.12.12"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "a77c62af46e79de0a562e1a9849205ffcb7fc1238876e9bd743357570e04046f"
+checksum = "43e734407157c3c2034e0258f5e4473ddb361b1e85f95a66690d67264d7cd1da"
 dependencies = [
  "async-compression",
  "base64",
@@ -1236,7 +1607,9 @@ dependencies = [
  "system-configuration",
  "tokio",
  "tokio-native-tls",
+ "tokio-socks",
  "tokio-util",
+ "tower",
  "tower-service",
  "url",
  "wasm-bindgen",
@@ -1261,6 +1634,64 @@ dependencies = [
  "windows-sys 0.52.0",
 ]
 
+[[package]]
+name = "rkyv"
+version = "0.7.45"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9008cd6385b9e161d8229e1f6549dd23c3d022f132a2ea37ac3a10ac4935779b"
+dependencies = [
+ "bitvec",
+ "bytecheck",
+ "bytes",
+ "hashbrown 0.12.3",
+ "ptr_meta 0.1.4",
+ "rend 0.4.2",
+ "rkyv_derive 0.7.45",
+ "seahash",
+ "tinyvec",
+ "uuid",
+]
+
+[[package]]
+name = "rkyv"
+version = "0.8.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1e147371c75553e1e2fcdb483944a8540b8438c31426279553b9a8182a9b7b65"
+dependencies = [
+ "bytes",
+ "hashbrown 0.15.2",
+ "indexmap",
+ "munge",
+ "ptr_meta 0.3.0",
+ "rancor",
+ "rend 0.5.2",
+ "rkyv_derive 0.8.10",
+ "tinyvec",
+ "uuid",
+]
+
+[[package]]
+name = "rkyv_derive"
+version = "0.7.45"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "503d1d27590a2b0a3a4ca4c94755aa2875657196ecbf401a42eff41d7de532c0"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 1.0.109",
+]
+
+[[package]]
+name = "rkyv_derive"
+version = "0.8.10"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "246b40ac189af6c675d124b802e8ef6d5246c53e17367ce9501f8f66a81abb7a"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.96",
+]
+
 [[package]]
 name = "rustc-demangle"
 version = "0.1.24"
@@ -1269,11 +1700,11 @@ checksum = "719b953e2095829ee67db738b3bfa9fa368c94900df327b3f07fe6e794d2fe1f"
 
 [[package]]
 name = "rustix"
-version = "0.38.42"
+version = "0.38.43"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f93dc38ecbab2eb790ff964bb77fa94faf256fd3e73285fd7ba0903b76bedb85"
+checksum = "a78891ee6bf2340288408954ac787aa063d8e8817e9f53abb37c695c6d834ef6"
 dependencies = [
- "bitflags",
+ "bitflags 2.8.0",
  "errno",
  "libc",
  "linux-raw-sys",
@@ -1282,9 +1713,9 @@ dependencies = [
 
 [[package]]
 name = "rustls"
-version = "0.23.20"
+version = "0.23.21"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "5065c3f250cbd332cd894be57c40fa52387247659b14a2d6041d121547903b1b"
+checksum = "8f287924602bf649d949c63dc8ac8b235fa5387d394020705b80c4eb597ce5b8"
 dependencies = [
  "once_cell",
  "rustls-pki-types",
@@ -1321,9 +1752,9 @@ dependencies = [
 
 [[package]]
 name = "rustversion"
-version = "1.0.18"
+version = "1.0.19"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0e819f2bc632f285be6d7cd36e25940d45b2391dd6d9b939e79de557f7014248"
+checksum = "f7c45b9784283f1b2e7fb61b42047c2fd678ef0960d4f6f1eba131594cc369d4"
 
 [[package]]
 name = "ryu"
@@ -1340,13 +1771,25 @@ dependencies = [
  "windows-sys 0.59.0",
 ]
 
+[[package]]
+name = "scopeguard"
+version = "1.2.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "94143f37725109f92c262ed2cf5e59bce7498c01bcc1502d7b9afe439a4e9f49"
+
+[[package]]
+name = "seahash"
+version = "4.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1c107b6f4780854c8b126e228ea8869f4d7b71260f962fefb57b996b8959ba6b"
+
 [[package]]
 name = "security-framework"
 version = "2.11.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "897b2245f0b511c87893af39b033e5ca9cce68824c4d7e7630b5a1d339658d02"
 dependencies = [
- "bitflags",
+ "bitflags 2.8.0",
  "core-foundation",
  "core-foundation-sys",
  "libc",
@@ -1355,9 +1798,9 @@ dependencies = [
 
 [[package]]
 name = "security-framework-sys"
-version = "2.13.0"
+version = "2.14.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "1863fd3768cd83c56a7f60faa4dc0d403f1b6df0a38c3c25f44b7894e45370d5"
+checksum = "49db231d56a190491cb4aeda9527f1ad45345af50b0851622a7adb8c03b01c32"
 dependencies = [
  "core-foundation-sys",
  "libc",
@@ -1365,29 +1808,29 @@ dependencies = [
 
 [[package]]
 name = "serde"
-version = "1.0.216"
+version = "1.0.217"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "0b9781016e935a97e8beecf0c933758c97a5520d32930e460142b4cd80c6338e"
+checksum = "02fc4265df13d6fa1d00ecff087228cc0a2b5f3c0e87e258d8b94a156e984c70"
 dependencies = [
  "serde_derive",
 ]
 
 [[package]]
 name = "serde_derive"
-version = "1.0.216"
+version = "1.0.217"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "46f859dbbf73865c6627ed570e78961cd3ac92407a2d117204c49232485da55e"
+checksum = "5a9bf7cf98d04a2b28aead066b7496853d4779c9cc183c440dbac457641e19a0"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
 name = "serde_json"
-version = "1.0.134"
+version = "1.0.137"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d00f4175c42ee48b15416f6193a959ba3a0d67fc699a0db9ad12df9f83991c7d"
+checksum = "930cfb6e6abf99298aaad7d29abbef7a9999a9a8806a40088f55f0dcec03146b"
 dependencies = [
  "itoa",
  "memchr",
@@ -1434,6 +1877,27 @@ version = "1.3.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "0fda2ff0d084019ba4d7c6f371c95d8fd75ce3524c3cb8fb653a3023f6323e64"
 
+[[package]]
+name = "signal-hook-registry"
+version = "1.4.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a9e9e0b4211b72e7b8b6e85c807d36c212bdb33ea8587f7569562a84df5465b1"
+dependencies = [
+ "libc",
+]
+
+[[package]]
+name = "simd-adler32"
+version = "0.3.7"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d66dc143e6b11c1eddc06d5c423cfc97062865baf299914ab64caa38182078fe"
+
+[[package]]
+name = "simdutf8"
+version = "0.1.5"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "e3a9fe34e3e7a50316060351f37187a3f546bce95496156754b601a5fa71b76e"
+
 [[package]]
 name = "slab"
 version = "0.4.9"
@@ -1459,6 +1923,44 @@ dependencies = [
  "windows-sys 0.52.0",
 ]
 
+[[package]]
+name = "sonic-number"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "a8a74044c092f4f43ca7a6cfd62854cf9fb5ac8502b131347c990bf22bef1dfe"
+dependencies = [
+ "cfg-if",
+]
+
+[[package]]
+name = "sonic-rs"
+version = "0.3.17"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0275f9f2f07d47556fe60c2759da8bc4be6083b047b491b2d476aa0bfa558eb1"
+dependencies = [
+ "bumpalo",
+ "bytes",
+ "cfg-if",
+ "faststr",
+ "itoa",
+ "ref-cast",
+ "ryu",
+ "serde",
+ "simdutf8",
+ "sonic-number",
+ "sonic-simd",
+ "thiserror 2.0.11",
+]
+
+[[package]]
+name = "sonic-simd"
+version = "0.1.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "940a24e82c9a97483ef66cef06b92160a8fa5cd74042c57c10b24d99d169d2fc"
+dependencies = [
+ "cfg-if",
+]
+
 [[package]]
 name = "spin"
 version = "0.9.8"
@@ -1479,9 +1981,20 @@ checksum = "13c2bddecc57b384dee18652358fb23172facb8a2c51ccc10d74c157bdea3292"
 
 [[package]]
 name = "syn"
-version = "2.0.91"
+version = "1.0.109"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "72b64191b275b66ffe2469e8af2c1cfe3bafa67b529ead792a6d0160888b4237"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "unicode-ident",
+]
+
+[[package]]
+name = "syn"
+version = "2.0.96"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "d53cbcb5a243bd33b7858b1d7f4aca2153490815872d86d955d6ea29f743c035"
+checksum = "d5d0adab1ae378d7f53bdebc67a39f1f151407ef230f0ce2883572f5d8985c80"
 dependencies = [
  "proc-macro2",
  "quote",
@@ -1505,7 +2018,20 @@ checksum = "c8af7666ab7b6390ab78131fb5b0fce11d6b7a6951602017c35fa82800708971"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
+]
+
+[[package]]
+name = "sysinfo"
+version = "0.33.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4fc858248ea01b66f19d8e8a6d55f41deaf91e9d495246fd01368d99935c6c01"
+dependencies = [
+ "core-foundation-sys",
+ "libc",
+ "memchr",
+ "ntapi",
+ "windows",
 ]
 
 [[package]]
@@ -1514,7 +2040,7 @@ version = "0.6.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "3c879d448e9d986b661742763247d3693ed13609438cf3d006f51f5368a5ba6b"
 dependencies = [
- "bitflags",
+ "bitflags 2.8.0",
  "core-foundation",
  "system-configuration-sys",
 ]
@@ -1529,19 +2055,66 @@ dependencies = [
  "libc",
 ]
 
+[[package]]
+name = "tap"
+version = "1.0.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "55937e1799185b12863d447f42597ed69d9928686b8d88a1df17376a097d8369"
+
 [[package]]
 name = "tempfile"
-version = "3.14.0"
+version = "3.15.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "28cce251fcbc87fac86a866eeb0d6c2d536fc16d06f184bb61aeae11aa4cee0c"
+checksum = "9a8a559c81686f576e8cd0290cd2a24a2a9ad80c98b3478856500fcbd7acd704"
 dependencies = [
  "cfg-if",
  "fastrand",
+ "getrandom",
  "once_cell",
  "rustix",
  "windows-sys 0.59.0",
 ]
 
+[[package]]
+name = "thiserror"
+version = "1.0.69"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "b6aaf5339b578ea85b50e080feb250a3e8ae8cfcdff9a461c9ec2904bc923f52"
+dependencies = [
+ "thiserror-impl 1.0.69",
+]
+
+[[package]]
+name = "thiserror"
+version = "2.0.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d452f284b73e6d76dd36758a0c8684b1d5be31f92b89d07fd5822175732206fc"
+dependencies = [
+ "thiserror-impl 2.0.11",
+]
+
+[[package]]
+name = "thiserror-impl"
+version = "1.0.69"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "4fee6c4efc90059e10f81e6d42c60a18f76588c3d74cb83a0b242a2b6c7504c1"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.96",
+]
+
+[[package]]
+name = "thiserror-impl"
+version = "2.0.11"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "26afc1baea8a989337eeb52b6e72a039780ce45c3edfcc9c5b9d112feeb173c2"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.96",
+]
+
 [[package]]
 name = "tinystr"
 version = "0.7.6"
@@ -1552,17 +2125,33 @@ dependencies = [
  "zerovec",
 ]
 
+[[package]]
+name = "tinyvec"
+version = "1.8.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "022db8904dfa342efe721985167e9fcd16c29b226db4397ed752a761cfce81e8"
+dependencies = [
+ "tinyvec_macros",
+]
+
+[[package]]
+name = "tinyvec_macros"
+version = "0.1.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "1f3ccbac311fea05f86f61904b462b55fb3df8837a366dfc601a0161d0532f20"
+
 [[package]]
 name = "tokio"
-version = "1.42.0"
+version = "1.43.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "5cec9b21b0450273377fc97bd4c33a8acffc8c996c987a7c5b319a0083707551"
+checksum = "3d61fa4ffa3de412bfea335c6ecff681de2b609ba3c77ef3e00e521813a9ed9e"
 dependencies = [
  "backtrace",
  "bytes",
  "libc",
  "mio",
  "pin-project-lite",
+ "signal-hook-registry",
  "socket2",
  "tokio-macros",
  "windows-sys 0.52.0",
@@ -1570,13 +2159,13 @@ dependencies = [
 
 [[package]]
 name = "tokio-macros"
-version = "2.4.0"
+version = "2.5.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "693d596312e88961bc67d7f1f97af8a70227d9f90c31bba5806eec004978d752"
+checksum = "6e06d43f1345a3bcd39f6a56dbb7dcab2ba47e68e8ac134855e7e2bdbaf8cab8"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
@@ -1599,6 +2188,18 @@ dependencies = [
  "tokio",
 ]
 
+[[package]]
+name = "tokio-socks"
+version = "0.5.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "0d4770b8024672c1101b3f6733eab95b18007dbe0847a8afe341fcf79e06043f"
+dependencies = [
+ "either",
+ "futures-util",
+ "thiserror 1.0.69",
+ "tokio",
+]
+
 [[package]]
 name = "tokio-stream"
 version = "0.1.17"
@@ -1645,9 +2246,11 @@ version = "0.6.2"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "403fa3b783d4b626a8ad51d766ab03cb6d2dbfc46b1c5d4448395e6628dc9697"
 dependencies = [
- "bitflags",
+ "bitflags 2.8.0",
  "bytes",
  "http",
+ "http-body",
+ "http-body-util",
  "pin-project-lite",
  "tower-layer",
  "tower-service",
@@ -1734,9 +2337,9 @@ checksum = "b6c140620e7ffbb22c2dee59cafe6084a59b5ffc27a8859a5f0d494b5d52b6be"
 
 [[package]]
 name = "uuid"
-version = "1.11.0"
+version = "1.12.1"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "f8c5f0a0af699448548ad1a2fbf920fb4bee257eae39953ba95cb84891a0446a"
+checksum = "b3758f5e68192bb96cc8f9b7e2c2cfdabb435499a28499a42f8f984092adad4b"
 dependencies = [
  "getrandom",
 ]
@@ -1770,34 +2373,35 @@ checksum = "9c8d87e72b64a3b4db28d11ce29237c246188f4f51057d65a7eab63b7987e423"
 
 [[package]]
 name = "wasm-bindgen"
-version = "0.2.99"
+version = "0.2.100"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "a474f6281d1d70c17ae7aa6a613c87fce69a127e2624002df63dcb39d6cf6396"
+checksum = "1edc8929d7499fc4e8f0be2262a241556cfc54a0bea223790e71446f2aab1ef5"
 dependencies = [
  "cfg-if",
  "once_cell",
+ "rustversion",
  "wasm-bindgen-macro",
 ]
 
 [[package]]
 name = "wasm-bindgen-backend"
-version = "0.2.99"
+version = "0.2.100"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "5f89bb38646b4f81674e8f5c3fb81b562be1fd936d84320f3264486418519c79"
+checksum = "2f0a0651a5c2bc21487bde11ee802ccaf4c51935d0d3d42a6101f98161700bc6"
 dependencies = [
  "bumpalo",
  "log",
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
  "wasm-bindgen-shared",
 ]
 
 [[package]]
 name = "wasm-bindgen-futures"
-version = "0.4.49"
+version = "0.4.50"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "38176d9b44ea84e9184eff0bc34cc167ed044f816accfe5922e54d84cf48eca2"
+checksum = "555d470ec0bc3bb57890405e5d4322cc9ea83cebb085523ced7be4144dac1e61"
 dependencies = [
  "cfg-if",
  "js-sys",
@@ -1808,9 +2412,9 @@ dependencies = [
 
 [[package]]
 name = "wasm-bindgen-macro"
-version = "0.2.99"
+version = "0.2.100"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "2cc6181fd9a7492eef6fef1f33961e3695e4579b9872a6f7c83aee556666d4fe"
+checksum = "7fe63fc6d09ed3792bd0897b314f53de8e16568c2b3f7982f468c0bf9bd0b407"
 dependencies = [
  "quote",
  "wasm-bindgen-macro-support",
@@ -1818,22 +2422,25 @@ dependencies = [
 
 [[package]]
 name = "wasm-bindgen-macro-support"
-version = "0.2.99"
+version = "0.2.100"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "30d7a95b763d3c45903ed6c81f156801839e5ee968bb07e534c44df0fcd330c2"
+checksum = "8ae87ea40c9f689fc23f209965b6fb8a99ad69aeeb0231408be24920604395de"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
  "wasm-bindgen-backend",
  "wasm-bindgen-shared",
 ]
 
 [[package]]
 name = "wasm-bindgen-shared"
-version = "0.2.99"
+version = "0.2.100"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "943aab3fdaaa029a6e0271b35ea10b72b943135afe9bffca82384098ad0e06a6"
+checksum = "1a05d73b933a847d6cccdda8f838a22ff101ad9bf93e33684f39c1f5f0eece3d"
+dependencies = [
+ "unicode-ident",
+]
 
 [[package]]
 name = "wasm-streams"
@@ -1850,14 +2457,52 @@ dependencies = [
 
 [[package]]
 name = "web-sys"
-version = "0.3.76"
+version = "0.3.77"
 source = "registry+https://github.com/rust-lang/crates.io-index"
-checksum = "04dd7223427d52553d3702c004d3b2fe07c148165faa56313cb00211e31c12bc"
+checksum = "33b6dd2ef9186f1f2072e409e99cd22a975331a6b3591b12c764e0e55c60d5d2"
 dependencies = [
  "js-sys",
  "wasm-bindgen",
 ]
 
+[[package]]
+name = "weezl"
+version = "0.1.8"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "53a85b86a771b1c87058196170769dd264f66c0782acf1ae6cc51bfd64b39082"
+
+[[package]]
+name = "winapi"
+version = "0.3.9"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5c839a674fcd7a98952e593242ea400abe93992746761e38641405d28b00f419"
+dependencies = [
+ "winapi-i686-pc-windows-gnu",
+ "winapi-x86_64-pc-windows-gnu",
+]
+
+[[package]]
+name = "winapi-i686-pc-windows-gnu"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "ac3b87c63620426dd9b991e5ce0329eff545bccbbb34f3be09ff6fb6ab51b7b6"
+
+[[package]]
+name = "winapi-x86_64-pc-windows-gnu"
+version = "0.4.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "712e227841d057c1ee1cd2fb22fa7e5a5461ae8e48fa2ca79ec42cfc1931183f"
+
+[[package]]
+name = "windows"
+version = "0.57.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "12342cb4d8e3b046f3d80effd474a7a02447231330ef77d71daa6fbc40681143"
+dependencies = [
+ "windows-core 0.57.0",
+ "windows-targets",
+]
+
 [[package]]
 name = "windows-core"
 version = "0.52.0"
@@ -1867,17 +2512,60 @@ dependencies = [
  "windows-targets",
 ]
 
+[[package]]
+name = "windows-core"
+version = "0.57.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "d2ed2439a290666cd67ecce2b0ffaad89c2a56b976b736e6ece670297897832d"
+dependencies = [
+ "windows-implement",
+ "windows-interface",
+ "windows-result 0.1.2",
+ "windows-targets",
+]
+
+[[package]]
+name = "windows-implement"
+version = "0.57.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "9107ddc059d5b6fbfbffdfa7a7fe3e22a226def0b2608f72e9d552763d3e1ad7"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.96",
+]
+
+[[package]]
+name = "windows-interface"
+version = "0.57.0"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "29bee4b38ea3cde66011baa44dba677c432a78593e202392d1e9070cf2a7fca7"
+dependencies = [
+ "proc-macro2",
+ "quote",
+ "syn 2.0.96",
+]
+
 [[package]]
 name = "windows-registry"
 version = "0.2.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "e400001bb720a623c1c69032f8e3e4cf09984deec740f007dd2b03ec864804b0"
 dependencies = [
- "windows-result",
+ "windows-result 0.2.0",
  "windows-strings",
  "windows-targets",
 ]
 
+[[package]]
+name = "windows-result"
+version = "0.1.2"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "5e383302e8ec8515204254685643de10811af0ed97ea37210dc26fb0032647f8"
+dependencies = [
+ "windows-targets",
+]
+
 [[package]]
 name = "windows-result"
 version = "0.2.0"
@@ -1893,7 +2581,7 @@ version = "0.1.0"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "4cd9b125c486025df0eabcb585e62173c6c9eddcec5d117d3b6e8c30e2ee4d10"
 dependencies = [
- "windows-result",
+ "windows-result 0.2.0",
  "windows-targets",
 ]
 
@@ -1991,6 +2679,15 @@ version = "0.5.5"
 source = "registry+https://github.com/rust-lang/crates.io-index"
 checksum = "1e9df38ee2d2c3c5948ea468a8406ff0db0b29ae1ffde1bcf20ef305bcc95c51"
 
+[[package]]
+name = "wyz"
+version = "0.5.1"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "05f360fc0b24296329c78fda852a1e9ae82de9cf7b27dae4b7f62f118f77b9ed"
+dependencies = [
+ "tap",
+]
+
 [[package]]
 name = "yoke"
 version = "0.7.5"
@@ -2011,7 +2708,7 @@ checksum = "2380878cad4ac9aac1e2435f3eb4020e8374b5f13c296cb75b4620ff8e229154"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
  "synstructure",
 ]
 
@@ -2033,7 +2730,7 @@ checksum = "fa4f8080344d4671fb4e831a13ad1e68092748387dfc4f55e356242fae12ce3e"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
 ]
 
 [[package]]
@@ -2053,7 +2750,7 @@ checksum = "595eed982f7d355beb85837f651fa22e90b3c044842dc7f2c2842c086f295808"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
  "synstructure",
 ]
 
@@ -2082,5 +2779,20 @@ checksum = "6eafa6dfb17584ea3e2bd6e76e0cc15ad7af12b09abdd1ca55961bed9b1063c6"
 dependencies = [
  "proc-macro2",
  "quote",
- "syn",
+ "syn 2.0.96",
+]
+
+[[package]]
+name = "zune-core"
+version = "0.4.12"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "3f423a2c17029964870cfaabb1f13dfab7d092a62a29a89264f4d36990ca414a"
+
+[[package]]
+name = "zune-jpeg"
+version = "0.4.14"
+source = "registry+https://github.com/rust-lang/crates.io-index"
+checksum = "99a5bab8d7dedf81405c4bb1f2b83ea057643d9cb28778cea9eecddeedd2e028"
+dependencies = [
+ "zune-core",
 ]

Cargo.toml

@@ -1,50 +1,55 @@
 [package]
 name = "cursor-api"
-version = "0.1.0"
+version = "0.1.3-rc.4.3"
 edition = "2021"
 authors = ["wisdgod <nav@wisdgod.com>"]
+description = "OpenAI format compatibility layer for the Cursor API"
+repository = "https://github.com/wisdgod/cursor-api"
 
 [build-dependencies]
 prost-build = "0.13.4"
+sha2 = { version = "0.10.8", default-features = false }
+serde_json = "1.0.134"
 
 [dependencies]
-axum = { version = "0.7.9", features = ["json"] }
+axum = { version = "0.8.1", features = ["json"] }
 base64 = { version = "0.22.1", default-features = false, features = ["std"] }
+# brotli = { version = "7.0.0", default-features = false, features = ["std"] }
 bytes = "1.9.0"
-chrono = { version = "0.4.39", features = ["serde"] }
+chrono = { version = "0.4.39", default-features = false, features = ["std", "clock", "now", "serde", "rkyv-64"] }
 dotenvy = "0.15.7"
 flate2 = { version = "1.0.35", default-features = false, features = ["rust_backend"] }
 futures = { version = "0.3.31", default-features = false, features = ["std"] }
+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"] }
+memmap2 = "0.9.5"
+# openssl = { version = "0.10.68", features = ["vendored"] }
+parking_lot = "0.12.3"
+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"] }
-reqwest = { version = "0.12.9", features = ["json", "gzip", "stream"] }
-serde = { version = "1.0.216", features = ["derive"] }
-serde_json = { version = "1.0.134", features = ["std"] }
+reqwest = { version = "0.12.12", default-features = false, features = ["gzip", "brotli", "json", "stream", "socks", "__tls", "charset", "default-tls", "h2", "http2", "macos-system-configuration"] }
+rkyv = { version = "0.7.45", default-features = false, features = ["alloc", "std", "bytecheck", "size_64", "validation", "std"] }
+serde = { version = "1.0.217", default-features = false, features = ["std", "derive"] }
+serde_json = { package = "sonic-rs", version = "0.3.17" }
+# serde_json = "1.0.137"
 sha2 = { version = "0.10.8", default-features = false }
-tokio = { version = "1.42.0", features = ["rt-multi-thread", "macros", "net", "sync", "time"] }
+sysinfo = { version = "0.33.1", default-features = false, features = ["system"] }
+tokio = { version = "1.43.0", features = ["rt-multi-thread", "macros", "net", "sync", "time", "fs", "signal"] }
 tokio-stream = { version = "0.1.17", features = ["time"] }
-tower-http = { version = "0.6.2", features = ["cors"] }
-uuid = { version = "1.11.0", features = ["v4"] }
+tower-http = { version = "0.6.2", features = ["cors", "limit"] }
+url = { version = "2.5.4", default-features = false }
+uuid = { version = "1.12.1", features = ["v4"] }
 
-# 优化设置
 [profile.release]
-lto = true           # 启用链接时优化
-codegen-units = 1    # 减少并行编译单元以提高优化
-panic = 'abort'      # 在 panic 时直接终止，减小二进制大小
-strip = true         # 移除调试符号
-opt-level = 3        # 最高优化级别
-
-# 构建脚本设置
-[package.metadata.cross.target.x86_64-unknown-linux-gnu]
-image = "ghcr.io/cross-rs/x86_64-unknown-linux-gnu:main"
-
-[package.metadata.cross.target.aarch64-unknown-linux-gnu]
-image = "ghcr.io/cross-rs/aarch64-unknown-linux-gnu:main"
-
-[package.metadata.cross.target.x86_64-apple-darwin]
-image = "ghcr.io/cross-rs/x86_64-apple-darwin:main"
+lto = true
+codegen-units = 1
+panic = 'abort'
+strip = true
+opt-level = 3
 
-[package.metadata.cross.target.aarch64-apple-darwin]
-image = "ghcr.io/cross-rs/aarch64-apple-darwin:main"
+[features]
+default = []
+use-minified = []

Cross.toml

@@ -0,0 +1,5 @@
+[target.x86_64-unknown-linux-gnu]
+dockerfile = "Dockerfile.cross"
+
+[target.aarch64-unknown-linux-gnu]
+dockerfile = "Dockerfile.cross.arm64"

Cursor API.md

@@ -0,0 +1,179 @@
+# Cursor API
+
+## 项目说明
+
+### 版本声明
+- 当前版本已进入稳定阶段
+- 以下问题与程序无关，请勿反馈：
+  - 响应缺字漏字
+  - 首字延迟现象
+  - 响应出现乱码
+- 性能优势：
+  - 达到原生客户端响应速度
+  - 部分场景下表现更优
+- 开源协议要求：
+  - Fork 项目禁止以原作者名义进行宣传推广
+  - 禁止发布任何形式的官方声明
+
+![Cursor API 架构示意图](https://via.placeholder.com/800x400.png?text=Cursor+API+Architecture)
+
+## 快速入门
+
+### 密钥获取
+1. 访问 [Cursor 官网](https://www.cursor.com) 完成注册登录
+2. 开启浏览器开发者工具 (F12)
+3. 在 Application → Cookies 中定位 `WorkosCursorSessionToken`
+4. 复制第三个字段值（注意：`%3A%3A` 为 `::` 的 URL 编码形式）
+
+## 配置指南
+
+### 环境变量
+| 变量名 | 类型 | 默认值 | 说明 |
+|--------|------|--------|-----|
+| PORT   | int  | 3000   | 服务端口号 |
+| AUTH_TOKEN | string | 无 | 认证令牌（必需） |
+| ROUTE_PREFIX | string | 无 | 路由前缀 |
+| TOKEN_LIST_FILE | string | .tokens | Token 存储文件 |
+
+完整配置参见 [env-example](/env-example)
+
+### Token 文件规范
+`.tokens` 文件格式：
+```plaintext
+# 注释行将在下次读取时自动删除
+token1,checksum1
+token2,checksum2
+```
+
+文件管理原则：
+- 系统自动维护文件内容
+- 仅以下情况需要手动编辑：
+  - 删除特定 token
+  - 绑定已有 checksum 到指定 token
+
+## 模型支持列表
+```json
+[
+  "claude-3.5-sonnet",
+  "gpt-4",
+  "gpt-4o",
+  "cursor-fast",
+  "gpt-4o-mini",
+  "deepseek-v3"
+]
+```
+*注：模型列表为固定配置，暂不支持自定义扩展*
+
+## API 文档
+
+### 基础对话接口
+**Endpoint**
+`POST /v1/chat/completions`
+
+**认证方式**
+`Bearer Token` 三级认证机制：
+1. 环境变量 `AUTH_TOKEN`
+2. `.token` 文件轮询
+3. 直接 token,checksum 认证（v0.1.3-rc.3+）
+
+**请求示例**
+```json
+{
+  "model": "gpt-4",
+  "messages": [
+    {
+      "role": "user",
+      "content": "解释量子计算的基本原理"
+    }
+  ],
+  "stream": false
+}
+```
+
+**响应示例（非流式）**
+```json
+{
+  "id": "chatcmpl-9Xy...",
+  "object": "chat.completion",
+  "created": 1628063500,
+  "model": "gpt-4",
+  "choices": [{
+    "index": 0,
+    "message": {
+      "role": "assistant",
+      "content": "量子计算基于量子比特..."
+    },
+    "finish_reason": "stop"
+  }]
+}
+```
+
+### Token 管理接口
+| 端点 | 方法 | 功能 |
+|------|------|-----|
+| `/tokens` | GET | Token 信息管理界面 |
+| `/tokens/update` | POST | 批量更新 Token 列表 |
+| `/tokens/add` | POST | 增量添加 Token |
+| `/tokens/delete` | POST | 删除指定 Token |
+
+```mermaid
+sequenceDiagram
+    participant Client
+    participant API
+    Client->>API: POST /tokens/add
+    API->>API: 验证Token有效性
+    API->>File: 写入.tokens
+    API-->>Client: 返回更新结果
+```
+
+## 高级功能
+
+### 动态密钥生成
+**Endpoint**
+`POST /build-key`
+
+**优势对比**
+| 特性 | 传统模式 | 动态密钥 |
+|------|---------|---------|
+| 密钥长度 | 较长 | 优化缩短 |
+| 配置扩展 | 无 | 支持自定义 |
+| 安全等级 | 基础 | 增强编码 |
+| 验证效率 | 预校验耗时 | 即时验证 |
+
+## 系统监控
+
+### 健康检查
+**Endpoint**
+`GET /health`
+
+**响应示例**
+```json
+{
+  "status": "success",
+  "version": "1.2.0",
+  "uptime": 86400,
+  "models": ["gpt-4", "claude-3.5"],
+  "endpoints": ["/v1/chat", "/tokens"]
+}
+```
+
+## 生态工具
+
+### 开发辅助工具
+- [Token 获取工具](https://github.com/wisdgod/cursor-api/tree/main/tools/get-token)
+  支持 Windows/Linux/macOS 系统
+- [遥测数据重置工具](https://github.com/wisdgod/cursor-api/tree/main/tools/reset-telemetry)
+  清除用户使用数据记录
+
+## 致谢声明
+本项目的发展离不开以下开源项目的启发：
+- [zhx47/cursor-api](https://github.com/zhx47/cursor-api) - 基础架构参考
+- [cursorToApi](https://github.com/luolazyandlazy/cursorToApi) - 认证机制优化方案
+
+--- 
+
+> **项目维护说明**
+> 我们欢迎社区贡献，但请注意：
+> 1. 功能请求需附带使用场景说明
+> 2. Bug 报告请提供复现步骤和环境信息
+> 3. 重要变更需通过 CI/CD 测试流程

Dockerfile

@@ -1,48 +1,36 @@
-# 构建阶段
-FROM rust:1.83.0-slim-bookworm as builder
+ARG TARGETARCH
+FROM --platform=linux/${TARGETARCH} rust:1.84.0-slim-bookworm as builder
 
-WORKDIR /app
+ARG TARGETARCH
 
-# 安装构建依赖
+WORKDIR /app
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
-    build-essential \
-    protobuf-compiler \
-    pkg-config \
-    libssl-dev \
-    nodejs \
-    npm \
+    build-essential protobuf-compiler pkg-config libssl-dev nodejs npm openssl \
     && rm -rf /var/lib/apt/lists/*
 
-# 复制项目文件
 COPY . .
-
-# 构建
-RUN rustup target add x86_64-unknown-linux-gnu && \
-    cargo build --target x86_64-unknown-linux-gnu --release && \
-    cp target/x86_64-unknown-linux-gnu/release/cursor-api /app/cursor-api
+RUN case "$TARGETARCH" in \
+      amd64) TARGET_CPU="x86-64-v3" ;; \
+      arm64) TARGET_CPU="neoverse-n1" ;; \
+      *) echo "Unsupported architecture: $TARGETARCH" && exit 1 ;; \
+    esac && \
+    RUSTFLAGS="-C link-arg=-s -C target-cpu=$TARGET_CPU" cargo build --release && \
+    cp target/release/cursor-api /app/cursor-api
 
 # 运行阶段
-FROM debian:bookworm-slim
+FROM --platform=linux/${TARGETARCH} debian:bookworm-slim
 
 WORKDIR /app
-
 ENV TZ=Asia/Shanghai
 
-# 安装运行时依赖
 RUN apt-get update && \
     apt-get install -y --no-install-recommends \
-    ca-certificates \
-    tzdata \
+    ca-certificates tzdata openssl \
     && rm -rf /var/lib/apt/lists/*
 
-# 复制构建产物
 COPY --from=builder /app/cursor-api .
 
-# 设置默认端口
 ENV PORT=3000
-
-# 动态暴露端口
 EXPOSE ${PORT}
-
 CMD ["./cursor-api"]

Dockerfile.cross

@@ -0,0 +1,31 @@
+# Dockerfile.cross
+
+FROM --platform=linux/amd64 rust:1.84.0-slim-bookworm
+
+WORKDIR /app
+
+# 安装必要的软件包
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    build-essential \
+    pkg-config \
+    libssl-dev \
+    protobuf-compiler \
+    openssl \
+    && rm -rf /var/lib/apt/lists/*
+
+# 设置环境变量 (如果需要)
+# ENV RUSTFLAGS="-C link-arg=-s"
+
+# 设置 PROTOC 环境变量 (因为你的 build.rs 需要)
+ENV PROTOC=/usr/bin/protoc
+
+# 安装特定版本的 protoc (如果你需要特定版本，例如 29.3；否则可以删除这部分)
+# ENV PROTOC_VERSION=29.3
+# ENV PROTOC_ZIP=protoc-${PROTOC_VERSION}-linux-x86_64.zip
+# RUN wget https://github.com/protocolbuffers/protobuf/releases/download/v${PROTOC_VERSION}/${PROTOC_ZIP} -O /tmp/${PROTOC_ZIP} && \
+#     unzip /tmp/${PROTOC_ZIP} -d /usr && \
+#     rm /tmp/${PROTOC_ZIP}
+
+# 验证安装
+RUN protoc --version

Dockerfile.cross.arm64

@@ -0,0 +1,31 @@
+# Dockerfile.cross
+
+FROM --platform=linux/arm64 rust:1.84.0-slim-bookworm
+
+WORKDIR /app
+
+# 安装必要的软件包
+RUN apt-get update && \
+    apt-get install -y --no-install-recommends \
+    build-essential \
+    pkg-config \
+    libssl-dev \
+    protobuf-compiler \
+    openssl \
+    && rm -rf /var/lib/apt/lists/*
+
+# 设置环境变量 (如果需要)
+# ENV RUSTFLAGS="-C link-arg=-s"
+
+# 设置 PROTOC 环境变量 (因为你的 build.rs 需要)
+ENV PROTOC=/usr/bin/protoc
+
+# 安装特定版本的 protoc (如果你需要特定版本，例如 29.3；否则可以删除这部分)
+# ENV PROTOC_VERSION=29.3
+# ENV PROTOC_ZIP=protoc-${PROTOC_VERSION}-linux-x86_64.zip
+# RUN wget https://github.com/protocolbuffers/protobuf/releases/download/v${PROTOC_VERSION}/${PROTOC_ZIP} -O /tmp/${PROTOC_ZIP} && \
+#     unzip /tmp/${PROTOC_ZIP} -d /usr && \
+#     rm /tmp/${PROTOC_ZIP}
+
+# 验证安装
+RUN protoc --version

LICENSE

@@ -0,0 +1,38 @@
+MIT License with Attribution Restrictions (MIT-AR)
+Copyright (c) 2025 wisdgod <nav@wisdgod.com>
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+1. The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+2. Any public reference to this software in promotional materials must include
+the following disclaimer in a prominent position:
+   "This product utilizes components developed by third-party contributors. 
+   There is no affiliation, endorsement, or sponsorship by the original author."
+
+3. Explicit prohibition against:
+   a) Using the author's name/alias in marketing collateral
+   b) Suggesting official certification or partnership
+   c) Using project name as technical endorsement
+
+4. Violation of these terms automatically terminates granted rights and
+   requires immediate cessation of software use within 72 hours.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.
+
+--- Special Provisions ---
+* This is a modified MIT license approved by SPDX as "MIT-AR" (Attribution-Restricted)
+* Commercial users may request certification waiver via nav@wisdgod.com
+* Community projects may display "Powered by" logo pack available at /branding

README.md

@@ -1,10 +0,0 @@
----
-title: Cursor
-emoji: 👁
-colorFrom: green
-colorTo: red
-sdk: docker
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

build.rs

@@ -1,15 +1,27 @@
+#[cfg(not(any(feature = "use-minified")))]
+use sha2::{Digest, Sha256};
+#[cfg(not(any(feature = "use-minified")))]
+use std::collections::HashMap;
+#[cfg(not(any(feature = "use-minified")))]
+use std::fs;
 use std::io::Result;
+#[cfg(not(any(feature = "use-minified")))]
 use std::path::Path;
+use std::path::PathBuf;
+#[cfg(not(any(feature = "use-minified")))]
 use std::process::Command;
 
+// 支持的文件类型
+#[cfg(not(any(feature = "use-minified")))]
+const SUPPORTED_EXTENSIONS: [&str; 4] = ["html", "js", "css", "md"];
+
+#[cfg(not(any(feature = "use-minified")))]
 fn check_and_install_deps() -> Result<()> {
     let scripts_dir = Path::new("scripts");
     let node_modules = scripts_dir.join("node_modules");
 
-    // 如果 node_modules 不存在，运行 npm install
     if !node_modules.exists() {
-        println!("cargo:warning=Installing HTML minifier dependencies...");
-
+        println!("cargo:warning=Installing minifier dependencies...");
         let status = Command::new("npm")
             .current_dir(scripts_dir)
             .arg("install")
@@ -23,38 +35,186 @@ fn check_and_install_deps() -> Result<()> {
     Ok(())
 }
 
-fn minify_html() -> Result<()> {
-    println!("cargo:warning=Minifying HTML files...");
+#[cfg(not(any(feature = "use-minified")))]
+fn get_files_hash() -> Result<HashMap<PathBuf, String>> {
+    let mut file_hashes = HashMap::new();
+    let static_dir = Path::new("static");
+
+    // 首先处理 README.md
+    let readme_path = Path::new("README.md");
+    if readme_path.exists() {
+        let content = fs::read(readme_path)?;
+        let mut hasher = Sha256::new();
+        hasher.update(&content);
+        let hash = format!("{:x}", hasher.finalize());
+        file_hashes.insert(readme_path.to_path_buf(), hash);
+    }
+
+    if static_dir.exists() {
+        for entry in fs::read_dir(static_dir)? {
+            let entry = entry?;
+            let path = entry.path();
+
+            // 检查是否是支持的文件类型，且不是已经压缩的文件
+            if let Some(ext) = path.extension().and_then(|e| e.to_str()) {
+                if SUPPORTED_EXTENSIONS.contains(&ext) && !path.to_string_lossy().contains(".min.")
+                {
+                    let content = fs::read(&path)?;
+                    let mut hasher = Sha256::new();
+                    hasher.update(&content);
+                    let hash = format!("{:x}", hasher.finalize());
+                    file_hashes.insert(path, hash);
+                }
+            }
+        }
+    }
+
+    Ok(file_hashes)
+}
+
+#[cfg(not(any(feature = "use-minified")))]
+fn load_saved_hashes() -> Result<HashMap<PathBuf, String>> {
+    let hash_file = Path::new("scripts/.asset-hashes.json");
+    if hash_file.exists() {
+        let content = fs::read_to_string(hash_file)?;
+        let hash_map: HashMap<String, String> = serde_json::from_str(&content)?;
+        Ok(hash_map
+            .into_iter()
+            .map(|(k, v)| (PathBuf::from(k), v))
+            .collect())
+    } else {
+        Ok(HashMap::new())
+    }
+}
+
+#[cfg(not(any(feature = "use-minified")))]
+fn save_hashes(hashes: &HashMap<PathBuf, String>) -> Result<()> {
+    let hash_file = Path::new("scripts/.asset-hashes.json");
+    let string_map: HashMap<String, String> = hashes
+        .iter()
+        .map(|(k, v)| (k.to_string_lossy().into_owned(), v.clone()))
+        .collect();
+    let content = serde_json::to_string_pretty(&string_map)?;
+    fs::write(hash_file, content)?;
+    Ok(())
+}
+
+#[cfg(not(any(feature = "use-minified")))]
+fn minify_assets() -> Result<()> {
+    // 获取现有文件的哈希
+    let current_hashes = get_files_hash()?;
 
+    if current_hashes.is_empty() {
+        println!("cargo:warning=No files to minify");
+        return Ok(());
+    }
+
+    // 加载保存的哈希值
+    let saved_hashes = load_saved_hashes()?;
+
+    // 找出需要更新的文件
+    let files_to_update: Vec<_> = current_hashes
+        .iter()
+        .filter(|(path, current_hash)| {
+            let is_readme = path.file_name().map_or(false, |f| f == "README.md");
+            let ext = path.extension().and_then(|e| e.to_str()).unwrap_or("");
+
+            // 为 README.md 和其他文件使用不同的输出路径检查
+            let min_path = if is_readme {
+                PathBuf::from("static/readme.min.html")
+            } else {
+                path.with_file_name(format!(
+                    "{}.min.{}",
+                    path.file_stem().unwrap().to_string_lossy(),
+                    ext
+                ))
+            };
+
+            // 检查压缩/转换后的文件是否存在
+            if !min_path.exists() {
+                return true;
+            }
+
+            // 检查原始文件是否发生变化
+            saved_hashes
+                .get(*path)
+                .map_or(true, |saved_hash| saved_hash != *current_hash)
+        })
+        .map(|(path, _)| path.file_name().unwrap().to_string_lossy().into_owned())
+        .collect();
+
+    if files_to_update.is_empty() {
+        println!("cargo:warning=No files need to be updated");
+        return Ok(());
+    }
+
+    println!("cargo:warning=Minifying {} files...", files_to_update.len());
+
+    // 运行压缩脚本
     let status = Command::new("node")
-        .args(&["scripts/minify-html.js"])
+        .arg("scripts/minify.js")
+        .args(&files_to_update)
         .status()?;
 
     if !status.success() {
-        panic!("HTML minification failed");
+        panic!("Asset minification failed");
     }
+
+    // 保存新的哈希值
+    save_hashes(&current_hashes)?;
+
     Ok(())
 }
 
 fn main() -> Result<()> {
     // Proto 文件处理
-    println!("cargo:rerun-if-changed=src/message.proto");
+    println!("cargo:rerun-if-changed=src/chat/aiserver/v1/lite.proto");
+    println!("cargo:rerun-if-changed=src/chat/config/key.proto");
+    // 获取环境变量 PROTOC
+    let protoc_path = match std::env::var_os("PROTOC") {
+        Some(path) => PathBuf::from(path),
+        None => {
+            println!("cargo:warning=PROTOC environment variable not set, using default protoc.");
+            // 如果 PROTOC 未设置，则返回一个空的 PathBuf，prost-build 会尝试使用默认的 protoc
+            PathBuf::new()
+        }
+    };
     let mut config = prost_build::Config::new();
-    config.type_attribute(".", "#[derive(serde::Serialize, serde::Deserialize)]");
+    // 如果 protoc_path 不为空，则配置使用指定的 protoc
+    if !protoc_path.as_os_str().is_empty() {
+        config.protoc_executable(protoc_path);
+    }
+    // config.type_attribute(".", "#[derive(serde::Serialize, serde::Deserialize)]");
+    config
+        .compile_protos(
+            &["src/chat/aiserver/v1/lite.proto"],
+            &["src/chat/aiserver/v1/"],
+        )
+        .unwrap();
     config
-        .compile_protos(&["src/message.proto"], &["src/"])
+        .compile_protos(&["src/chat/config/key.proto"], &["src/chat/config/"])
         .unwrap();
 
-    // HTML 文件处理
-    println!("cargo:rerun-if-changed=static/tokeninfo.html");
-    println!("cargo:rerun-if-changed=scripts/minify-html.js");
+    // 静态资源文件处理
+    println!("cargo:rerun-if-changed=scripts/minify.js");
     println!("cargo:rerun-if-changed=scripts/package.json");
+    println!("cargo:rerun-if-changed=static/api.html");
+    println!("cargo:rerun-if-changed=static/build_key.html");
+    println!("cargo:rerun-if-changed=static/config.html");
+    println!("cargo:rerun-if-changed=static/logs.html");
+    println!("cargo:rerun-if-changed=static/shared-styles.css");
+    println!("cargo:rerun-if-changed=static/shared.js");
+    println!("cargo:rerun-if-changed=static/tokens.html");
+    println!("cargo:rerun-if-changed=README.md");
 
-    // 检查并安装依赖
-    check_and_install_deps()?;
+    #[cfg(not(any(feature = "use-minified")))]
+    {
+        // 检查并安装依赖
+        check_and_install_deps()?;
 
-    // 运行 HTML 压缩
-    minify_html()?;
+        // 运行资源压缩
+        minify_assets()?;
+    }
 
     Ok(())
 }

scripts/build.ps1

@@ -1,158 +1,126 @@
-# ��ɫ�������
-function Write-Info { Write-Host "[INFO] $args" -ForegroundColor Blue }
-function Write-Warn { Write-Host "[WARN] $args" -ForegroundColor Yellow }
-function Write-Error { Write-Host "[ERROR] $args" -ForegroundColor Red; exit 1 }
-
-# ����Ҫ�Ĺ���
-function Test-Requirements {
-  $tools = @("cargo", "protoc", "npm", "node")
-  $missing = @()
-
-  foreach ($tool in $tools) {
-    if (!(Get-Command $tool -ErrorAction SilentlyContinue)) {
-      $missing += $tool
+# 参数处理
+param(
+    [switch]$Static,
+    [switch]$Help,
+    [ValidateSet("x86_64", "aarch64", "i686")]
+    [string]$Architecture
+)
+
+# 设置错误时停止执行
+$ErrorActionPreference = "Stop"
+
+# 颜色输出函数
+function Write-Info  { param($Message) Write-Host "[INFO] $Message" -ForegroundColor Blue }
+function Write-Warn  { param($Message) Write-Host "[WARN] $Message" -ForegroundColor Yellow }
+function Write-Error { param($Message) Write-Host "[ERROR] $Message" -ForegroundColor Red; exit 1 }
+
+# 检查必要的工具
+function Check-Requirements {
+    $tools = @("cargo", "protoc", "npm", "node")
+    $missing = @()
+
+    foreach ($tool in $tools) {
+        if (-not (Get-Command $tool -ErrorAction SilentlyContinue)) {
+            $missing += $tool
+        }
     }
-  }
 
-  if ($missing.Count -gt 0) {
-    Write-Error "ȱ�ٱ�Ҫ����: $($missing -join ', ')"
-  }
+    if ($missing.Count -gt 0) {
+        Write-Error "缺少必要工具: $($missing -join ', ')"
+    }
 }
 
-# �� Test-Requirements �����������º���
-function Initialize-VSEnvironment {
-    Write-Info "���ڳ�ʼ�� Visual Studio ����..."
-
-    # ֱ��ʹ����֪�� vcvarsall.bat ·��
-    $vcvarsallPath = "E:\Program Files (x86)\Microsoft Visual Studio\2022\BuildTools\VC\Auxiliary\Build\vcvarsall.bat"
-
-    if (-not (Test-Path $vcvarsallPath)) {
-        Write-Error "δ�ҵ� vcvarsall.bat: $vcvarsallPath"
-        return
-    }
+# 帮助信息
+function Show-Help {
+    Write-Host @"
+用法: $(Split-Path $MyInvocation.ScriptName -Leaf) [选项]
 
-    Write-Info "ʹ�� vcvarsall.bat ·��: $vcvarsallPath"
+选项:
+  -Static        使用静态链接（默认动态链接）
+  -Help          显示此帮助信息
 
-    # ��ȡ��������
-    $archArg = "x64"
-    $command = "`"$vcvarsallPath`" $archArg && set"
+不带参数时使用默认配置构建
+"@
+}
 
-    try {
-        $output = cmd /c "$command" 2>&1
+# 构建函数
+function Build-Target {
+    param (
+        [string]$Target,
+        [string]$RustFlags
+    )
 
-        # ��������Ƿ�ɹ�ִ��
-        if ($LASTEXITCODE -ne 0) {
-            Write-Error "vcvarsall.bat ִ��ʧ�ܣ��˳���: $LASTEXITCODE"
-            return
-        }
+    Write-Info "正在构建 $Target..."
 
-        # ���µ�ǰ PowerShell �Ự�Ļ�������
-        foreach ($line in $output) {
-            if ($line -match "^([^=]+)=(.*)$") {
-                $name = $matches[1]
-                $value = $matches[2]
-                if (![string]::IsNullOrEmpty($name)) {
-                    Set-Item -Path "env:$name" -Value $value -ErrorAction SilentlyContinue
-                }
-            }
-        }
+    # 设置环境变量
+    $env:RUSTFLAGS = $RustFlags
 
-        Write-Info "Visual Studio ������ʼ�����"
+    # 构建
+    if ($Target -ne (rustc -Vv | Select-String "host: (.*)" | ForEach-Object { $_.Matches.Groups[1].Value })) {
+        cargo build --target $Target --release
+    } else {
+        cargo build --release
     }
-    catch {
-        Write-Error "��ʼ�� Visual Studio ����ʱ��������: $_"
-    }
-}
 
-# ������Ϣ
-function Show-Help {
-  Write-Host @"
-�÷�: $(Split-Path $MyInvocation.MyCommand.Path -Leaf) [ѡ��]
+    # 移动编译产物到 release 目录
+    $binaryName = "cursor-api"
+    if ($Static) {
+        $binaryName += "-static"
+    }
 
-ѡ��:
-  --static        ʹ�þ�̬���ӣ�Ĭ�϶�̬���ӣ�
-  --help          ��ʾ�˰�����Ϣ
+    $binaryPath = if ($Target -eq (rustc -Vv | Select-String "host: (.*)" | ForEach-Object { $_.Matches.Groups[1].Value })) {
+        "target/release/cursor-api.exe"
+    } else {
+        "target/$Target/release/cursor-api.exe"
+    }
 
-Ĭ�ϱ������� Windows ֧�ֵļܹ� (x64 �� arm64)
-"@
-}
+    if (Test-Path $binaryPath) {
+        Copy-Item $binaryPath "release/$binaryName-$Target.exe"
+        Write-Info "完成构建 $Target"
+    } else {
+        Write-Warn "构建产物未找到: $Target"
+        Write-Warn "查找路径: $binaryPath"
+        Write-Warn "当前目录内容:"
+        Get-ChildItem -Recurse target/
+        return $false
+    }
 
-# ��������
-function New-Target {
-  param (
-    [string]$target,
-    [string]$rustflags
-  )
-
-  Write-Info "���ڹ��� $target..."
-
-  # ���û���������ִ�й���
-  $env:RUSTFLAGS = $rustflags
-  cargo build --target $target --release
-
-  # �ƶ���������
-  $binaryName = "cursor-api"
-  if ($UseStatic) {
-    $binaryName += "-static"
-  }
-
-  $sourcePath = "target/$target/release/cursor-api.exe"
-  $targetPath = "release/${binaryName}-${target}.exe"
-
-  if (Test-Path $sourcePath) {
-    Copy-Item $sourcePath $targetPath -Force
-    Write-Info "��ɹ��� $target"
-  }
-  else {
-    Write-Warn "��������δ�ҵ�: $target"
-    return $false
-  }
-  return $true
+    return $true
 }
 
-# ��������
-$UseStatic = $false
-
-foreach ($arg in $args) {
-  switch ($arg) {
-    "--static" { $UseStatic = $true }
-    "--help" { Show-Help; exit 0 }
-    default { Write-Error "δ֪����: $arg" }
-  }
+if ($Help) {
+    Show-Help
+    exit 0
 }
 
-# ������
-try {
-  # �������
-  Test-Requirements
-
-  # ��ʼ�� Visual Studio ����
-  Initialize-VSEnvironment
+# 检查依赖
+Check-Requirements
 
-  # ���� release Ŀ¼
-  New-Item -ItemType Directory -Force -Path "release" | Out-Null
+# 创建 release 目录
+New-Item -ItemType Directory -Force -Path release | Out-Null
 
-  # ����Ŀ��ƽ̨
-  $targets = @(
-    "x86_64-pc-windows-msvc",
-    "aarch64-pc-windows-msvc"
-  )
-
-  # ���þ�̬���ӱ�־
-  $rustflags = ""
-  if ($UseStatic) {
-    $rustflags = "-C target-feature=+crt-static"
-  }
-
-  Write-Info "��ʼ����..."
+# 设置静态链接标志
+$rustFlags = ""
+if ($Static) {
+    $rustFlags = "-C target-feature=+crt-static"
+}
 
-  # ��������Ŀ��
-  foreach ($target in $targets) {
-    New-Target -target $target -rustflags $rustflags
-  }
+# 获取目标架构
+$arch = if ($Architecture) {
+    $Architecture
+} else {
+    switch ($env:PROCESSOR_ARCHITECTURE) {
+        "AMD64" { "x86_64" }
+        "ARM64" { "aarch64" }
+        "X86" { "i686" }
+        default { Write-Error "不支持的架构: $env:PROCESSOR_ARCHITECTURE" }
+    }
+}
+$target = "$arch-pc-windows-msvc"
 
-  Write-Info "������ɣ�"
+Write-Info "开始构建..."
+if (-not (Build-Target -Target $target -RustFlags $rustFlags)) {
+    Write-Error "构建失败"
 }
-catch {
-  Write-Error "���������������: $_"
-}
+
+Write-Info "构建完成！"

scripts/build.sh

@@ -6,11 +6,6 @@ info() { echo -e "\033[1;34m[INFO]\033[0m $*"; }
 warn() { echo -e "\033[1;33m[WARN]\033[0m $*"; }
 error() { echo -e "\033[1;31m[ERROR]\033[0m $*" >&2; exit 1; }
 
-# 检查是否在 Linux 环境
-is_linux() {
-    [ "$(uname -s)" = "Linux" ]
-}
-
 # 检查必要的工具
 check_requirements() {
     local missing_tools=()
@@ -22,23 +17,29 @@ check_requirements() {
         fi
     done
 
-    # Linux 特定检查
-    if [[ $USE_CROSS == true ]] && ! command -v cross &>/dev/null; then
-        missing_tools+=("cross")
-    fi
-
     if [[ ${#missing_tools[@]} -gt 0 ]]; then
         error "缺少必要工具: ${missing_tools[*]}"
     fi
 }
 
+# 解析参数
+USE_STATIC=false
+
+while [[ $# -gt 0 ]]; do
+    case $1 in
+        --static) USE_STATIC=true ;;
+        --help) show_help; exit 0 ;;
+        *) error "未知参数: $1" ;;
+    esac
+    shift
+done
+
 # 帮助信息
 show_help() {
     cat << EOF
 用法: $(basename "$0") [选项]
 
 选项:
-  --cross         使用 cross 进行交叉编译（仅在 Linux 上有效）
   --static        使用静态链接（默认动态链接）
   --help          显示此帮助信息
 
@@ -57,42 +58,37 @@ build_target() {
     # 确定文件后缀
     [[ $target == *"windows"* ]] && extension=".exe"
 
-    # 设置目标特定的环境变量
-    local build_env=()
-    if [[ $target == "aarch64-unknown-linux-gnu" ]]; then
-        build_env+=(
-            "CC_aarch64_unknown_linux_gnu=aarch64-linux-gnu-gcc"
-            "CXX_aarch64_unknown_linux_gnu=aarch64-linux-gnu-g++"
-            "CARGO_TARGET_AARCH64_UNKNOWN_LINUX_GNU_LINKER=aarch64-linux-gnu-gcc"
-            "PKG_CONFIG_PATH=/usr/lib/aarch64-linux-gnu/pkgconfig"
-            "PKG_CONFIG_ALLOW_CROSS=1"
-            "OPENSSL_DIR=/usr"
-            "OPENSSL_INCLUDE_DIR=/usr/include"
-            "OPENSSL_LIB_DIR=/usr/lib/aarch64-linux-gnu"
-        )
-    fi
-
-    # 判断是否使用 cross（仅在 Linux 上）
+    # 构建
     if [[ $target != "$CURRENT_TARGET" ]]; then
-        env ${build_env[@]+"${build_env[@]}"} RUSTFLAGS="$rustflags" cargo build --target "$target" --release
+        env RUSTFLAGS="$rustflags" cargo build --target "$target" --release
     else
-        env ${build_env[@]+"${build_env[@]}"} RUSTFLAGS="$rustflags" cargo build --release
+        env RUSTFLAGS="$rustflags" cargo build --release
     fi
 
     # 移动编译产物到 release 目录
     local binary_name="cursor-api"
     [[ $USE_STATIC == true ]] && binary_name+="-static"
 
-    if [[ -f "target/$target/release/cursor-api$extension" ]]; then
-        cp "target/$target/release/cursor-api$extension" "release/${binary_name}-$target$extension"
+    local binary_path
+    if [[ $target == "$CURRENT_TARGET" ]]; then
+        binary_path="target/release/cursor-api$extension"
+    else
+        binary_path="target/$target/release/cursor-api$extension"
+    fi
+
+    if [[ -f "$binary_path" ]]; then
+        cp "$binary_path" "release/${binary_name}-$target$extension"
         info "完成构建 $target"
     else
         warn "构建产物未找到: $target"
+        warn "查找路径: $binary_path"
+        warn "当前目录内容:"
+        ls -R target/
         return 1
     fi
 }
 
-# 获取 CPU 架构
+# 获取 CPU 架构和操作系统
 ARCH=$(uname -m | sed 's/^aarch64\|arm64$/aarch64/;s/^x86_64\|x86-64\|x64\|amd64$/x86_64/')
 OS=$(uname -s)
 
@@ -102,9 +98,15 @@ get_target() {
     local os=$2
     case "$os" in
         "Darwin") echo "${arch}-apple-darwin" ;;
-        "Linux") echo "${arch}-unknown-linux-gnu" ;;
+        "Linux") 
+            if [[ $USE_STATIC == true ]]; then
+                echo "${arch}-unknown-linux-musl"
+            else
+                echo "${arch}-unknown-linux-gnu"
+            fi
+            ;;
         "MINGW"*|"MSYS"*|"CYGWIN"*|"Windows_NT") echo "${arch}-pc-windows-msvc" ;;
-        "FreeBSD") echo "x86_64-unknown-freebsd" ;;
+        "FreeBSD") echo "${arch}-unknown-freebsd" ;;
         *) error "不支持的系统: $os" ;;
     esac
 }
@@ -118,52 +120,52 @@ CURRENT_TARGET=$(get_target "$ARCH" "$OS")
 # 获取系统对应的所有目标
 get_targets() {
     case "$1" in
-        "linux") echo "x86_64-unknown-linux-gnu aarch64-unknown-linux-gnu" ;;
-        "windows") echo "x86_64-pc-windows-msvc aarch64-pc-windows-msvc" ;;
-        "macos") echo "x86_64-apple-darwin aarch64-apple-darwin" ;;
-        "freebsd") echo "x86_64-unknown-freebsd" ;;
+        "linux")
+            # Linux 只构建当前架构
+            echo "$CURRENT_TARGET"
+            ;;
+        "freebsd")
+            # FreeBSD 只构建当前架构
+            echo "$CURRENT_TARGET"
+            ;;
+        "windows")
+            # Windows 只构建当前架构
+            echo "$CURRENT_TARGET"
+            ;;
+        "macos")
+            # macOS 构建所有 macOS 目标
+            echo "x86_64-apple-darwin aarch64-apple-darwin"
+            ;;
         *) error "不支持的系统组: $1" ;;
     esac
 }
 
-# 解析参数
-USE_CROSS=false
-USE_STATIC=false
-
-while [[ $# -gt 0 ]]; do
-    case $1 in
-        --cross) USE_CROSS=true ;;
-        --static) USE_STATIC=true ;;
-        --help) show_help; exit 0 ;;
-        *) error "未知参数: $1" ;;
-    esac
-    shift
-done
-
 # 检查依赖
 check_requirements
 
 # 确定要构建的目标
-if [[ $USE_CROSS == true ]] && is_linux; then
-    # 只在 Linux 上使用 cross 进行多架构构建
-    TARGETS=($(get_targets "linux"))
-else
-    # 其他系统或不使用 cross 时只构建当前系统的所有架构
-    case "$OS" in
-        "Darwin") TARGETS=($(get_targets "macos")) ;;
-        "Linux") TARGETS=("$CURRENT_TARGET") ;;
-        "MINGW"*|"MSYS"*|"CYGWIN"*|"Windows_NT") TARGETS=($(get_targets "windows")) ;;
-        "FreeBSD") TARGETS=("$CURRENT_TARGET") ;;
-        *) error "不支持的系统: $OS" ;;
-    esac
-fi
+case "$OS" in
+    Darwin) 
+        TARGETS=($(get_targets "macos"))
+        ;;
+    Linux)
+        TARGETS=($(get_targets "linux"))
+        ;;
+    FreeBSD)
+        TARGETS=($(get_targets "freebsd"))
+        ;;
+    MINGW*|MSYS*|CYGWIN*|Windows_NT)
+        TARGETS=($(get_targets "windows"))
+        ;;
+    *) error "不支持的系统: $OS" ;;
+esac
 
 # 创建 release 目录
 mkdir -p release
 
 # 设置静态链接标志
-RUSTFLAGS=""
-[[ $USE_STATIC == true ]] && RUSTFLAGS="-C target-feature=+crt-static"
+RUSTFLAGS="-C link-arg=-s"
+[[ $USE_STATIC == true ]] && RUSTFLAGS="-C target-feature=+crt-static -C link-arg=-s"
 
 # 并行构建所有目标
 info "开始构建..."

scripts/minify-html.js

@@ -1,49 +0,0 @@
-#!/usr/bin/env node
-
-const { minify } = require('html-minifier-terser');
-const fs = require('fs');
-const path = require('path');
-
-// 配置选项
-const options = {
-  collapseWhitespace: true,
-  removeComments: true,
-  removeEmptyAttributes: true,
-  removeOptionalTags: true,
-  removeRedundantAttributes: true,
-  removeScriptTypeAttributes: true,
-  removeStyleLinkTypeAttributes: true,
-  minifyCSS: true,
-  minifyJS: true,
-  processScripts: ['application/json'],
-};
-
-// 处理文件
-async function minifyFile(inputPath, outputPath) {
-  try {
-    const html = fs.readFileSync(inputPath, 'utf8');
-    const minified = await minify(html, options);
-    fs.writeFileSync(outputPath, minified);
-    console.log(`✓ Minified ${path.basename(inputPath)} -> ${path.basename(outputPath)}`);
-  } catch (err) {
-    console.error(`✗ Error processing ${inputPath}:`, err);
-    process.exit(1);
-  }
-}
-
-// 主函数
-async function main() {
-  const staticDir = path.join(__dirname, '..', 'static');
-  const files = [
-    ['tokeninfo.html', 'tokeninfo.min.html'],
-  ];
-
-  for (const [input, output] of files) {
-    await minifyFile(
-      path.join(staticDir, input),
-      path.join(staticDir, output)
-    );
-  }
-}
-
-main();

scripts/minify.js

@@ -0,0 +1,157 @@
+#!/usr/bin/env node
+
+const { minify: minifyHtml } = require('html-minifier-terser');
+const { minify: minifyJs } = require('terser');
+const CleanCSS = require('clean-css');
+const MarkdownIt = require('markdown-it');
+const fs = require('fs');
+const path = require('path');
+
+// 配置选项
+const options = {
+  collapseWhitespace: true,
+  removeComments: true,
+  removeEmptyAttributes: true,
+  removeOptionalTags: true,
+  removeRedundantAttributes: true,
+  removeScriptTypeAttributes: true,
+  removeStyleLinkTypeAttributes: true,
+  minifyCSS: true,
+  minifyJS: true,
+  processScripts: ['application/json'],
+};
+
+// CSS 压缩选项
+const cssOptions = {
+  level: 2
+};
+
+// 处理文件
+async function minifyFile(inputPath, outputPath) {
+  try {
+    let ext = path.extname(inputPath).toLowerCase();
+    if (ext === '.md') ext = '.html';
+    const filename = path.basename(inputPath);
+    let content = fs.readFileSync(inputPath, 'utf8');
+    let minified;
+
+    // 特殊处理 readme.html
+    if (filename.toLowerCase() === 'readme.md') {
+      const md = new MarkdownIt({
+        html: true,
+        linkify: true,
+        typographer: true
+      });
+      const readmeMdPath = path.join(__dirname, '..', 'README.md');
+      const markdownContent = fs.readFileSync(readmeMdPath, 'utf8');
+      // 添加基本的 markdown 样式
+      const htmlContent = `
+<!DOCTYPE html>
+<html>
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>README</title>
+    <style>
+        body {
+            max-width: 800px;
+            margin: 0 auto;
+            padding: 20px;
+            font-family: -apple-system, BlinkMacSystemFont, "Segoe UI", Helvetica, Arial, sans-serif;
+            line-height: 1.6;
+        }
+        pre {
+            background-color: #f6f8fa;
+            padding: 16px;
+            border-radius: 6px;
+            overflow: auto;
+        }
+        code {
+            background-color: #f6f8fa;
+            padding: 0.2em 0.4em;
+            border-radius: 3px;
+        }
+        img {
+            max-width: 100%;
+        }
+        table {
+            border-collapse: collapse;
+            width: 100%;
+        }
+        table td, table th {
+            border: 1px solid #dfe2e5;
+            padding: 6px 13px;
+        }
+        blockquote {
+            border-left: 4px solid #dfe2e5;
+            margin: 0;
+            padding: 0 1em;
+            color: #6a737d;
+        }
+    </style>
+</head>
+<body>
+    ${md.render(markdownContent)}
+</body>
+</html>
+      `;
+      content = htmlContent;
+    }
+
+    switch (ext) {
+      case '.html':
+        minified = await minifyHtml(content, options);
+        break;
+      case '.js':
+        const result = await minifyJs(content);
+        minified = result.code;
+        break;
+      case '.css':
+        minified = new CleanCSS(cssOptions).minify(content).styles;
+        break;
+      default:
+        throw new Error(`Unsupported file type: ${ext}`);
+    }
+
+    fs.writeFileSync(outputPath, minified);
+    console.log(`✓ Minified ${path.basename(inputPath)} -> ${path.basename(outputPath)}`);
+  } catch (err) {
+    console.error(`✗ Error processing ${inputPath}:`, err);
+    process.exit(1);
+  }
+}
+
+// 主函数
+async function main() {
+  // 获取命令行参数，跳过前两个参数（node和脚本路径）
+  const files = process.argv.slice(2);
+
+  if (files.length === 0) {
+    console.error('No input files specified');
+    process.exit(1);
+  }
+
+  const staticDir = path.join(__dirname, '..', 'static');
+
+  for (const file of files) {
+    // 特殊处理 README.md 的输入路径
+    let inputPath;
+    let outputPath;
+
+    if (file.toLowerCase() === 'readme.md') {
+      inputPath = path.join(__dirname, '..', 'README.md');
+      outputPath = path.join(staticDir, 'readme.min.html');
+    } else {
+      inputPath = path.join(staticDir, file);
+      const ext = path.extname(file);
+      outputPath = path.join(
+        staticDir,
+        file.replace(ext, `.min${ext}`)
+      );
+    }
+
+    await minifyFile(inputPath, outputPath);
+  }
+}
+
+main();

scripts/package-lock.json

@@ -8,7 +8,10 @@
       "name": "html-minifier-scripts",
       "version": "1.0.0",
       "dependencies": {
-        "html-minifier-terser": "^7.2.0"
+        "clean-css": "^5.3.3",
+        "html-minifier-terser": "^7.2.0",
+        "markdown-it": "^14.1.0",
+        "terser": "^5.37.0"
       },
       "engines": {
         "node": ">=14.0.0"
@@ -84,6 +87,12 @@
         "node": ">=0.4.0"
       }
     },
+    "node_modules/argparse": {
+      "version": "2.0.1",
+      "resolved": "https://registry.npmjs.org/argparse/-/argparse-2.0.1.tgz",
+      "integrity": "sha512-8+9WqebbFzpX9OR+Wa6O29asIogeRMzcGtAINdpMHHyAg10f05aSFVBbcEqGf/PXw1EjAZ+q2/bEBg3DvurK3Q==",
+      "license": "Python-2.0"
+    },
     "node_modules/buffer-from": {
       "version": "1.1.2",
       "resolved": "https://registry.npmjs.org/buffer-from/-/buffer-from-1.1.2.tgz",
@@ -164,6 +173,15 @@
         "node": "^14.13.1 || >=16.0.0"
       }
     },
+    "node_modules/linkify-it": {
+      "version": "5.0.0",
+      "resolved": "https://registry.npmjs.org/linkify-it/-/linkify-it-5.0.0.tgz",
+      "integrity": "sha512-5aHCbzQRADcdP+ATqnDuhhJ/MRIqDkZX5pyjFHRRysS8vZ5AbqGEoFIb6pYHPZ+L/OC2Lc+xT8uHVVR5CAK/wQ==",
+      "license": "MIT",
+      "dependencies": {
+        "uc.micro": "^2.0.0"
+      }
+    },
     "node_modules/lower-case": {
       "version": "2.0.2",
       "resolved": "https://registry.npmjs.org/lower-case/-/lower-case-2.0.2.tgz",
@@ -173,6 +191,29 @@
         "tslib": "^2.0.3"
       }
     },
+    "node_modules/markdown-it": {
+      "version": "14.1.0",
+      "resolved": "https://registry.npmjs.org/markdown-it/-/markdown-it-14.1.0.tgz",
+      "integrity": "sha512-a54IwgWPaeBCAAsv13YgmALOF1elABB08FxO9i+r4VFk5Vl4pKokRPeX8u5TCgSsPi6ec1otfLjdOpVcgbpshg==",
+      "license": "MIT",
+      "dependencies": {
+        "argparse": "^2.0.1",
+        "entities": "^4.4.0",
+        "linkify-it": "^5.0.0",
+        "mdurl": "^2.0.0",
+        "punycode.js": "^2.3.1",
+        "uc.micro": "^2.1.0"
+      },
+      "bin": {
+        "markdown-it": "bin/markdown-it.mjs"
+      }
+    },
+    "node_modules/mdurl": {
+      "version": "2.0.0",
+      "resolved": "https://registry.npmjs.org/mdurl/-/mdurl-2.0.0.tgz",
+      "integrity": "sha512-Lf+9+2r+Tdp5wXDXC4PcIBjTDtq4UKjCPMQhKIuzpJNW0b96kVqSwW0bT7FhRSfmAiFYgP+SCRvdrDozfh0U5w==",
+      "license": "MIT"
+    },
     "node_modules/no-case": {
       "version": "3.0.4",
       "resolved": "https://registry.npmjs.org/no-case/-/no-case-3.0.4.tgz",
@@ -203,6 +244,15 @@
         "tslib": "^2.0.3"
       }
     },
+    "node_modules/punycode.js": {
+      "version": "2.3.1",
+      "resolved": "https://registry.npmjs.org/punycode.js/-/punycode.js-2.3.1.tgz",
+      "integrity": "sha512-uxFIHU0YlHYhDQtV4R9J6a52SLx28BCjT+4ieh7IGbgwVJWO+km431c4yRlREUAsAmt/uMjQUyQHNEPf0M39CA==",
+      "license": "MIT",
+      "engines": {
+        "node": ">=6"
+      }
+    },
     "node_modules/relateurl": {
       "version": "0.2.7",
       "resolved": "https://registry.npmjs.org/relateurl/-/relateurl-0.2.7.tgz",
@@ -260,6 +310,12 @@
       "resolved": "https://registry.npmjs.org/tslib/-/tslib-2.8.1.tgz",
       "integrity": "sha512-oJFu94HQb+KVduSUQL7wnpmqnfmLsOA/nAh6b6EH0wCEoK0/mPeXU6c3wKDV83MkOuHPRHtSXKKU99IBazS/2w==",
       "license": "0BSD"
+    },
+    "node_modules/uc.micro": {
+      "version": "2.1.0",
+      "resolved": "https://registry.npmjs.org/uc.micro/-/uc.micro-2.1.0.tgz",
+      "integrity": "sha512-ARDJmphmdvUk6Glw7y9DQ2bFkKBHwQHLi2lsaH6PPmz/Ka9sFOBsBluozhDltWmnv9u/cF6Rt87znRTPV+yp/A==",
+      "license": "MIT"
     }
   }
 }

scripts/package.json

@@ -6,6 +6,9 @@
     "node": ">=14.0.0"
   },
   "dependencies": {
-    "html-minifier-terser": "^7.2.0"
+    "clean-css": "^5.3.3",
+    "html-minifier-terser": "^7.2.0",
+    "markdown-it": "^14.1.0",
+    "terser": "^5.37.0"
   }
-}
+}

scripts/setup-windows.ps1

@@ -1,31 +0,0 @@
-# ���� PowerShell ����Ϊ UTF-8
-[Console]::OutputEncoding = [System.Text.Encoding]::UTF8
-$OutputEncoding = [System.Text.Encoding]::UTF8
-
-# ����Ƿ��Թ���ԱȨ������
-if (-NOT ([Security.Principal.WindowsPrincipal][Security.Principal.WindowsIdentity]::GetCurrent()).IsInRole([Security.Principal.WindowsBuiltInRole] "Administrator")) {
-    Write-Warning "���Թ���ԱȨ�����д˽ű�"
-    exit 1
-}
-
-# ��鲢��װ Chocolatey
-if (!(Get-Command choco -ErrorAction SilentlyContinue)) {
-    Write-Output "���ڰ�װ Chocolatey..."
-    Set-ExecutionPolicy Bypass -Scope Process -Force
-    [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072
-    iex ((New-Object System.Net.WebClient).DownloadString('https://chocolatey.org/install.ps1'))
-}
-
-# ��װ��Ҫ�Ĺ���
-Write-Output "���ڰ�װ��Ҫ�Ĺ���..."
-choco install -y mingw
-choco install -y protoc
-choco install -y git
-
-# ��װ Rust ����
-Write-Output "���ڰ�װ Rust ����..."
-rustup target add x86_64-pc-windows-msvc
-rustup target add x86_64-unknown-linux-gnu
-cargo install cross
-
-Write-Output "��װ��ɣ�"

scripts/setup.ps1

@@ -0,0 +1,179 @@
+# ���ô���ʱִֹͣ��
+$ErrorActionPreference = "Stop"
+$ProgressPreference = "SilentlyContinue"  # �ӿ������ٶ�
+
+# ��ɫ�������
+function Write-Info  { param($Message) Write-Host "[INFO] $Message" -ForegroundColor Blue }
+function Write-Warn  { param($Message) Write-Host "[WARN] $Message" -ForegroundColor Yellow }
+function Write-Success { param($Message) Write-Host "[SUCCESS] $Message" -ForegroundColor Green }
+function Write-Error { param($Message) Write-Host "[ERROR] $Message" -ForegroundColor Red; exit 1 }
+
+# ������ԱȨ��
+function Test-Administrator {
+    $user = [Security.Principal.WindowsIdentity]::GetCurrent()
+    $principal = New-Object Security.Principal.WindowsPrincipal $user
+    return $principal.IsInRole([Security.Principal.WindowsBuiltInRole]::Administrator)
+}
+
+if (-not (Test-Administrator)) {
+    Write-Error "���Թ���ԱȨ�����д˽ű�"
+}
+
+# ������Ϣ
+function Show-Help {
+    Write-Host @"
+�÷�: $(Split-Path $MyInvocation.ScriptName -Leaf) [ѡ��]
+
+ѡ��:
+  -NoVS           ����װ Visual Studio Build Tools
+  -NoRust         ����װ Rust
+  -NoNode         ����װ Node.js
+  -Help           ��ʾ�˰�����Ϣ
+
+ʾ��:
+  .\setup.ps1
+  .\setup.ps1 -NoVS
+  .\setup.ps1 -NoRust -NoNode
+"@
+}
+
+# ��������
+param(
+    [switch]$NoVS,
+    [switch]$NoRust,
+    [switch]$NoNode,
+    [switch]$Help
+)
+
+if ($Help) {
+    Show-Help
+    exit 0
+}
+
+# ��鲢��װ Chocolatey
+function Install-Chocolatey {
+    Write-Info "��� Chocolatey..."
+    if (-not (Get-Command choco -ErrorAction SilentlyContinue)) {
+        Write-Info "��װ Chocolatey..."
+        Set-ExecutionPolicy Bypass -Scope Process -Force
+        [System.Net.ServicePointManager]::SecurityProtocol = [System.Net.ServicePointManager]::SecurityProtocol -bor 3072
+        try {
+            Invoke-Expression ((New-Object System.Net.WebClient).DownloadString('https://chocolatey.org/install.ps1'))
+        }
+        catch {
+            Write-Error "��װ Chocolatey ʧ��: $_"
+        }
+        # ˢ�»�������
+        $env:Path = [System.Environment]::GetEnvironmentVariable("Path","Machine") + ";" + [System.Environment]::GetEnvironmentVariable("Path","User")
+    }
+}
+
+# ��װ Visual Studio Build Tools
+function Install-VSBuildTools {
+    if ($NoVS) {
+        Write-Info "���� Visual Studio Build Tools ��װ"
+        return
+    }
+
+    Write-Info "��� Visual Studio Build Tools..."
+    $vsPath = "${env:ProgramFiles(x86)}\Microsoft Visual Studio\Installer\vswhere.exe"
+    if (-not (Test-Path $vsPath)) {
+        Write-Info "��װ Visual Studio Build Tools..."
+        try {
+            # ���ذ�װ����
+            $vsInstallerUrl = "https://aka.ms/vs/17/release/vs_BuildTools.exe"
+            $vsInstallerPath = "$env:TEMP\vs_BuildTools.exe"
+            Invoke-WebRequest -Uri $vsInstallerUrl -OutFile $vsInstallerPath
+
+            # ��װ
+            $process = Start-Process -FilePath $vsInstallerPath -ArgumentList `
+                "--quiet", "--wait", "--norestart", "--nocache", `
+                "--installPath", "${env:ProgramFiles(x86)}\Microsoft Visual Studio\2022\BuildTools", `
+                "--add", "Microsoft.VisualStudio.Workload.VCTools" `
+                -NoNewWindow -Wait -PassThru
+
+            if ($process.ExitCode -ne 0) {
+                Write-Error "Visual Studio Build Tools ��װʧ��"
+            }
+
+            Remove-Item $vsInstallerPath -Force
+        }
+        catch {
+            Write-Error "��װ Visual Studio Build Tools ʧ��: $_"
+        }
+    }
+    else {
+        Write-Info "Visual Studio Build Tools �Ѱ�װ"
+    }
+}
+
+# ��װ Rust
+function Install-Rust {
+    if ($NoRust) {
+        Write-Info "���� Rust ��װ"
+        return
+    }
+
+    Write-Info "��� Rust..."
+    if (-not (Get-Command rustc -ErrorAction SilentlyContinue)) {
+        Write-Info "��װ Rust..."
+        try {
+            $rustupInit = "$env:TEMP\rustup-init.exe"
+            Invoke-WebRequest -Uri "https://win.rustup.rs" -OutFile $rustupInit
+            Start-Process -FilePath $rustupInit -ArgumentList "-y" -Wait
+            Remove-Item $rustupInit -Force
+
+            # ˢ�»�������
+            $env:Path = [System.Environment]::GetEnvironmentVariable("Path","Machine") + ";" + [System.Environment]::GetEnvironmentVariable("Path","User")
+        }
+        catch {
+            Write-Error "��װ Rust ʧ��: $_"
+        }
+    }
+
+    # ����Ŀ��ƽ̨
+    Write-Info "���� Rust Ŀ��ƽ̨..."
+    $arch = if ([Environment]::Is64BitOperatingSystem) { "x86_64" } else { "i686" }
+    rustup target add "$arch-pc-windows-msvc"
+}
+
+# ��װ��������
+function Install-Tools {
+    Write-Info "��װ��Ҫ����..."
+    
+    # ��װ protoc
+    if (-not (Get-Command protoc -ErrorAction SilentlyContinue)) {
+        Write-Info "��װ Protocol Buffers..."
+        choco install -y protoc
+    }
+
+    # ��װ Git
+    if (-not (Get-Command git -ErrorAction SilentlyContinue)) {
+        Write-Info "��װ Git..."
+        choco install -y git
+    }
+
+    # ��װ Node.js
+    if (-not $NoNode -and -not (Get-Command node -ErrorAction SilentlyContinue)) {
+        Write-Info "��װ Node.js..."
+        choco install -y nodejs
+    }
+
+    # ˢ�»�������
+    $env:Path = [System.Environment]::GetEnvironmentVariable("Path","Machine") + ";" + [System.Environment]::GetEnvironmentVariable("Path","User")
+}
+
+# ������
+try {
+    Write-Info "��ʼ��װ��Ҫ���..."
+    
+    Install-Chocolatey
+    Install-VSBuildTools
+    Install-Rust
+    Install-Tools
+
+    Write-Success "��װ��ɣ�"
+}
+catch {
+    Write-Error "��װ�����г��ִ���: $_"
+}

scripts/setup.sh

@@ -0,0 +1,157 @@
+#!/bin/bash
+
+# 设置错误时退出
+set -e
+
+# 颜色输出
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+BLUE='\033[0;34m'
+NC='\033[0m' # No Color
+
+info() {
+    echo -e "${BLUE}[INFO] $1${NC}"
+}
+
+error() {
+    echo -e "${RED}[ERROR] $1${NC}"
+    exit 1
+}
+
+# 检查是否为 root 用户（FreeBSD 和 Linux）
+if [ "$(uname)" != "Darwin" ] && [ "$EUID" -ne 0 ]; then
+    error "请使用 root 权限运行此脚本 (sudo ./setup.sh)"
+fi
+
+# 检测包管理器
+if command -v brew &> /dev/null; then
+    PKG_MANAGER="brew"
+    info "检测到 macOS/Homebrew 系统"
+elif command -v pkg &> /dev/null; then
+    PKG_MANAGER="pkg"
+    info "检测到 FreeBSD 系统"
+elif command -v apt-get &> /dev/null; then
+    PKG_MANAGER="apt-get"
+    info "检测到 Debian/Ubuntu 系统"
+elif command -v dnf &> /dev/null; then
+    PKG_MANAGER="dnf"
+    info "检测到 Fedora/RHEL 系统"
+elif command -v yum &> /dev/null; then
+    PKG_MANAGER="yum"
+    info "检测到 CentOS 系统"
+else
+    error "未检测到支持的包管理器"
+fi
+
+# 更新包管理器缓存
+info "更新包管理器缓存..."
+case $PKG_MANAGER in
+    "brew")
+        brew update
+        ;;
+    "pkg")
+        pkg update
+        ;;
+    *)
+        $PKG_MANAGER update -y
+        ;;
+esac
+
+# 安装基础构建工具
+info "安装基础构建工具..."
+case $PKG_MANAGER in
+    "brew")
+        brew install \
+            protobuf \
+            pkg-config \
+            openssl \
+            curl \
+            git \
+            node
+        ;;
+    "pkg")
+        pkg install -y \
+            gmake \
+            protobuf \
+            pkgconf \
+            openssl \
+            curl \
+            git \
+            node
+        ;;
+    "apt-get")
+        $PKG_MANAGER install -y --no-install-recommends \
+            build-essential \
+            protobuf-compiler \
+            pkg-config \
+            libssl-dev \
+            ca-certificates \
+            curl \
+            tzdata \
+            git
+        ;;
+    *)
+        $PKG_MANAGER install -y \
+            gcc \
+            gcc-c++ \
+            make \
+            protobuf-compiler \
+            pkg-config \
+            openssl-devel \
+            ca-certificates \
+            curl \
+            tzdata \
+            git
+        ;;
+esac
+
+# 安装 Node.js 和 npm（如果还没有通过包管理器安装）
+if ! command -v node &> /dev/null && [ "$PKG_MANAGER" != "brew" ] && [ "$PKG_MANAGER" != "pkg" ]; then
+    info "安装 Node.js 和 npm..."
+    if [ "$PKG_MANAGER" = "apt-get" ]; then
+        curl -fsSL https://deb.nodesource.com/setup_lts.x | bash -
+        $PKG_MANAGER install -y nodejs
+    else
+        curl -fsSL https://rpm.nodesource.com/setup_lts.x | bash -
+        $PKG_MANAGER install -y nodejs
+    fi
+fi
+
+# 安装 Rust（如果未安装）
+if ! command -v rustc &> /dev/null; then
+    info "安装 Rust..."
+    curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
+    . "$HOME/.cargo/env"
+fi
+
+# 添加目标平台
+info "添加 Rust 目标平台..."
+case "$(uname)" in
+    "FreeBSD")
+        rustup target add x86_64-unknown-freebsd
+        ;;
+    "Darwin")
+        rustup target add x86_64-apple-darwin aarch64-apple-darwin
+        ;;
+    *)
+        rustup target add x86_64-unknown-linux-gnu
+        ;;
+esac
+
+# 清理包管理器缓存
+case $PKG_MANAGER in
+    "apt-get")
+        rm -rf /var/lib/apt/lists/*
+        ;;
+    "pkg")
+        pkg clean -y
+        ;;
+esac
+
+# 设置时区（除了 macOS）
+if [ "$(uname)" != "Darwin" ]; then
+    info "设置时区为 Asia/Shanghai..."
+    ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime
+fi
+
+echo -e "${GREEN}安装完成！${NC}"

serve.ts

@@ -0,0 +1 @@
+Deno.serve(async(r:Request)=>{const rs=(s:number,m:string)=>new Response(m,{status:s,headers:{"Access-Control-Allow-Origin":"*"}});const h=r.headers.get("x-co");if(!h)return rs(400,"Missing header");const a=["api2.cursor.sh","www.cursor.com"];if(!a.includes(h))return rs(403,"Host denied");const u=new URL(r.url),p=["/aiserver.v1.AiService/StreamChat","/aiserver.v1.AiService/StreamChatWeb","/auth/full_stripe_profile","/api/usage","/api/auth/me"];if(!p.includes(u.pathname))return rs(404,"Path invalid");const hd=new Headers(r.headers);hd.delete("x-co");hd.set("Host",h);try{const f=await fetch(`https://${h}${u.pathname}${u.search}`,{method:r.method,headers:hd,body:r.body});const fh=new Headers(f.headers);fh.set("Access-Control-Allow-Origin","*");return new Response(f.body,{status:f.status,headers:fh})}catch(e){return rs(500,"Server error")}});

src/app.rs

@@ -0,0 +1,4 @@
+pub mod config;
+pub mod constant;
+pub mod model;
+pub mod lazy;

src/app/config.rs

@@ -0,0 +1,159 @@
+use super::{constant::AUTHORIZATION_BEARER_PREFIX, lazy::AUTH_TOKEN, model::AppConfig};
+use crate::common::model::{
+    config::{ConfigData, ConfigUpdateRequest},
+    ApiStatus, ErrorResponse, NormalResponse,
+};
+use axum::{
+    http::{header::AUTHORIZATION, HeaderMap, StatusCode},
+    Json,
+};
+
+// 定义处理更新操作的宏
+macro_rules! handle_updates {
+    ($request:expr, $($field:ident => $update_fn:expr),* $(,)?) => {
+        $(
+            if let Some(value) = $request.$field {
+                $update_fn(value);
+            }
+        )*
+    };
+}
+
+// 定义处理重置操作的宏
+macro_rules! handle_resets {
+    ($request:expr, $($field:ident => $reset_fn:expr),* $(,)?) => {
+        $(
+            if $request.$field.is_some() {
+                $reset_fn();
+            }
+        )*
+    };
+}
+
+pub async fn handle_config_update(
+    headers: HeaderMap,
+    Json(request): Json<ConfigUpdateRequest>,
+) -> Result<Json<NormalResponse<ConfigData>>, (StatusCode, Json<ErrorResponse>)> {
+    let auth_header = headers
+        .get(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 != AUTH_TOKEN.as_str() {
+        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),
+                vision_ability: AppConfig::get_vision_ability(),
+                enable_slow_pool: AppConfig::get_slow_pool(),
+                enable_all_claude: AppConfig::get_allow_claude(),
+                usage_check_models: AppConfig::get_usage_check(),
+                enable_dynamic_key: AppConfig::get_dynamic_key(),
+                share_token: AppConfig::get_share_token(),
+                proxies: AppConfig::get_proxies(),
+                include_web_references: AppConfig::get_web_refs(),
+            }),
+            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,
+                        }),
+                    ));
+                }
+            }
+
+            handle_updates!(request,
+                vision_ability => AppConfig::update_vision_ability,
+                enable_slow_pool => AppConfig::update_slow_pool,
+                enable_all_claude => AppConfig::update_allow_claude,
+                usage_check_models => AppConfig::update_usage_check,
+                enable_dynamic_key => AppConfig::update_dynamic_key,
+                share_token => AppConfig::update_share_token,
+                proxies => AppConfig::update_proxies,
+                include_web_references => AppConfig::update_web_refs,
+            );
+
+            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,
+                        }),
+                    ));
+                }
+            }
+
+            handle_resets!(request,
+                vision_ability => AppConfig::reset_vision_ability,
+                enable_slow_pool => AppConfig::reset_slow_pool,
+                enable_all_claude => AppConfig::reset_allow_claude,
+                usage_check_models => AppConfig::reset_usage_check,
+                enable_dynamic_key => AppConfig::reset_dynamic_key,
+                share_token => AppConfig::reset_share_token,
+                proxies => AppConfig::reset_proxies,
+                include_web_references => AppConfig::reset_web_refs,
+            );
+
+            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

@@ -0,0 +1,83 @@
+macro_rules! def_pub_const {
+    ($name:ident, $value:expr) => {
+        pub const $name: &'static str = $value;
+    };
+}
+
+pub const COMMA: char = ',';
+
+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"));
+
+def_pub_const!(EMPTY_STRING, "");
+
+def_pub_const!(COMMA_STRING, ",");
+
+def_pub_const!(ROUTE_ROOT_PATH, "/");
+def_pub_const!(ROUTE_HEALTH_PATH, "/health");
+def_pub_const!(ROUTE_GET_HASH, "/get-hash");
+def_pub_const!(ROUTE_GET_CHECKSUM, "/get-checksum");
+def_pub_const!(ROUTE_GET_TIMESTAMP_HEADER, "/get-tsheader");
+def_pub_const!(ROUTE_USER_INFO_PATH, "/userinfo");
+def_pub_const!(ROUTE_API_PATH, "/api");
+def_pub_const!(ROUTE_LOGS_PATH, "/logs");
+def_pub_const!(ROUTE_CONFIG_PATH, "/config");
+def_pub_const!(ROUTE_TOKENS_PATH, "/tokens");
+def_pub_const!(ROUTE_TOKENS_GET_PATH, "/tokens/get");
+def_pub_const!(ROUTE_TOKENS_RELOAD_PATH, "/tokens/reload");
+def_pub_const!(ROUTE_TOKENS_UPDATE_PATH, "/tokens/update");
+def_pub_const!(ROUTE_TOKENS_ADD_PATH, "/tokens/add");
+def_pub_const!(ROUTE_TOKENS_DELETE_PATH, "/tokens/delete");
+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");
+def_pub_const!(ROUTE_BASIC_CALIBRATION_PATH, "/basic-calibration");
+def_pub_const!(ROUTE_BUILD_KEY_PATH, "/build-key");
+
+def_pub_const!(DEFAULT_TOKEN_LIST_FILE_NAME, ".tokens");
+
+def_pub_const!(STATUS_PENDING, "pending");
+def_pub_const!(STATUS_SUCCESS, "success");
+def_pub_const!(STATUS_FAILED, "failed");
+
+def_pub_const!(HEADER_NAME_GHOST_MODE, "x-ghost-mode");
+
+def_pub_const!(TRUE, "true");
+def_pub_const!(FALSE, "false");
+
+// 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"
+);
+
+def_pub_const!(AUTHORIZATION_BEARER_PREFIX, "Bearer ");
+
+def_pub_const!(CURSOR_API2_HOST, "api2.cursor.sh");
+def_pub_const!(CURSOR_HOST, "www.cursor.com");
+def_pub_const!(CURSOR_SETTINGS_URL, "https://www.cursor.com/settings");
+
+def_pub_const!(OBJECT_CHAT_COMPLETION, "chat.completion");
+def_pub_const!(OBJECT_CHAT_COMPLETION_CHUNK, "chat.completion.chunk");
+
+// def_pub_const!(CURSOR_API2_STREAM_CHAT, "StreamChat");
+// def_pub_const!(CURSOR_API2_GET_USER_INFO, "GetUserInfo");
+
+def_pub_const!(FINISH_REASON_STOP, "stop");
+
+def_pub_const!(ERR_INVALID_PATH, "无效的路径");
+
+// def_pub_const!(ERR_CHECKSUM_NO_GOOD, "checksum no good");

src/app/lazy.rs

@@ -0,0 +1,189 @@
+use super::constant::{
+    COMMA, CURSOR_API2_HOST, CURSOR_HOST, DEFAULT_TOKEN_LIST_FILE_NAME, EMPTY_STRING,
+};
+use crate::common::utils::{
+    parse_ascii_char_from_env, parse_bool_from_env, parse_string_from_env, parse_usize_from_env,
+};
+use std::sync::LazyLock;
+use tokio::sync::{Mutex, OnceCell};
+
+macro_rules! def_pub_static {
+    // 基础版本：直接存储 String
+    ($name:ident, $value:expr) => {
+        pub static $name: LazyLock<String> = LazyLock::new(|| $value);
+    };
+
+    // 环境变量版本
+    ($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());
+    };
+}
+
+// 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_LIST_FILE, env: "TOKEN_LIST_FILE", default: DEFAULT_TOKEN_LIST_FILE_NAME);
+def_pub_static!(ROUTE_MODELS_PATH, format!("{}/v1/models", *ROUTE_PREFIX));
+def_pub_static!(
+    ROUTE_CHAT_PATH,
+    format!("{}/v1/chat/completions", *ROUTE_PREFIX)
+);
+
+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
+}
+
+def_pub_static!(DEFAULT_INSTRUCTIONS, env: "DEFAULT_INSTRUCTIONS", default: "Respond in Chinese by default");
+
+def_pub_static!(REVERSE_PROXY_HOST, env: "REVERSE_PROXY_HOST", default: EMPTY_STRING);
+
+const DEFAULT_KEY_PREFIX: &str = "sk-";
+
+pub static KEY_PREFIX: LazyLock<String> = LazyLock::new(|| {
+    let value = parse_string_from_env("KEY_PREFIX", DEFAULT_KEY_PREFIX)
+        .trim()
+        .to_string();
+    if value.is_empty() {
+        DEFAULT_KEY_PREFIX.to_string()
+    } else {
+        value
+    }
+});
+
+pub static KEY_PREFIX_LEN: LazyLock<usize> = LazyLock::new(|| KEY_PREFIX.len());
+
+pub static TOKEN_DELIMITER: LazyLock<char> = LazyLock::new(|| {
+    let delimiter = parse_ascii_char_from_env("TOKEN_DELIMITER", COMMA);
+    if delimiter.is_ascii_alphabetic()
+        || delimiter.is_ascii_digit()
+        || delimiter == '+'
+        || delimiter == '/'
+    {
+        COMMA
+    } else {
+        delimiter
+    }
+});
+
+pub static USE_COMMA_DELIMITER: LazyLock<bool> = LazyLock::new(|| {
+    let enable = parse_bool_from_env("USE_COMMA_DELIMITER", true);
+    if enable && *TOKEN_DELIMITER == COMMA {
+        false
+    } else {
+        enable
+    }
+});
+
+pub static USE_REVERSE_PROXY: LazyLock<bool> = LazyLock::new(|| !REVERSE_PROXY_HOST.is_empty());
+
+macro_rules! def_cursor_api_url {
+    ($name:ident, $api_host:expr, $path:expr) => {
+        pub static $name: LazyLock<String> = LazyLock::new(|| {
+            let host = if *USE_REVERSE_PROXY {
+                &*REVERSE_PROXY_HOST
+            } else {
+                $api_host
+            };
+            format!("https://{}{}", host, $path)
+        });
+    };
+}
+
+def_cursor_api_url!(
+    CURSOR_API2_CHAT_URL,
+    CURSOR_API2_HOST,
+    "/aiserver.v1.AiService/StreamChat"
+);
+
+def_cursor_api_url!(
+    CURSOR_API2_CHAT_WEB_URL,
+    CURSOR_API2_HOST,
+    "/aiserver.v1.AiService/StreamChatWeb"
+);
+
+def_cursor_api_url!(
+    CURSOR_API2_STRIPE_URL,
+    CURSOR_API2_HOST,
+    "/auth/full_stripe_profile"
+);
+
+def_cursor_api_url!(CURSOR_USAGE_API_URL, CURSOR_HOST, "/api/usage");
+
+def_cursor_api_url!(CURSOR_USER_API_URL, CURSOR_HOST, "/api/auth/me");
+
+pub(super) static LOGS_FILE_PATH: LazyLock<String> =
+    LazyLock::new(|| parse_string_from_env("LOGS_FILE_PATH", "logs.bin"));
+
+pub(super) static PAGES_FILE_PATH: LazyLock<String> =
+    LazyLock::new(|| parse_string_from_env("PAGES_FILE_PATH", "pages.bin"));
+
+pub static DEBUG: LazyLock<bool> = LazyLock::new(|| parse_bool_from_env("DEBUG", false));
+
+// 使用环境变量 "DEBUG_LOG_FILE" 来指定日志文件路径，默认值为 "debug.log"
+static DEBUG_LOG_FILE: LazyLock<String> =
+    LazyLock::new(|| parse_string_from_env("DEBUG_LOG_FILE", "debug.log"));
+
+// 使用 OnceCell 结合 Mutex 来异步初始化 LOG_FILE
+static LOG_FILE: OnceCell<Mutex<tokio::fs::File>> = OnceCell::const_new();
+
+pub(crate) async fn get_log_file() -> &'static Mutex<tokio::fs::File> {
+    LOG_FILE
+        .get_or_init(|| async {
+            Mutex::new(
+                tokio::fs::OpenOptions::new()
+                    .create(true)
+                    .append(true)
+                    .open(&*DEBUG_LOG_FILE)
+                    .await
+                    .expect("无法打开日志文件"),
+            )
+        })
+        .await
+}
+
+#[macro_export]
+macro_rules! debug_println {
+    ($($arg:tt)*) => {
+        if *crate::app::lazy::DEBUG {
+            let time = chrono::Local::now().format("%Y-%m-%d %H:%M:%S").to_string();
+            let log_message = format!("{} - {}", time, format!($($arg)*));
+            use tokio::io::AsyncWriteExt as _;
+
+            // 使用 tokio 的 spawn 在后台异步写入日志
+            tokio::spawn(async move {
+                let log_file = crate::app::lazy::get_log_file().await;
+                // 使用 MutexGuard 获取可变引用
+                let mut file = log_file.lock().await;
+                if let Err(err) = file.write_all(log_message.as_bytes()).await {
+                    eprintln!("写入日志文件失败: {}", err);
+                }
+                if let Err(err) = file.write_all(b"\n").await {
+                    eprintln!("写入换行符失败: {}", err);
+                }
+                // 可以选择在写入失败时 panic，或者忽略
+                // panic!("写入日志文件失败: {}", err);
+            });
+        }
+    };
+}
+
+pub static REQUEST_LOGS_LIMIT: LazyLock<usize> =
+    LazyLock::new(|| std::cmp::min(parse_usize_from_env("REQUEST_LOGS_LIMIT", 100), 2000));
+
+pub static SERVICE_TIMEOUT: LazyLock<u64> = LazyLock::new(|| {
+    let timeout = parse_usize_from_env("SERVICE_TIMEOUT", 30);
+    u64::try_from(timeout).map(|t| t.min(600)).unwrap_or(30)
+});

src/app/model.rs

@@ -0,0 +1,486 @@
+use crate::{
+    app::constant::{
+        EMPTY_STRING, ERR_INVALID_PATH, ROUTE_ABOUT_PATH, ROUTE_API_PATH, ROUTE_BUILD_KEY_PATH,
+        ROUTE_CONFIG_PATH, ROUTE_LOGS_PATH, ROUTE_README_PATH, ROUTE_ROOT_PATH,
+        ROUTE_SHARED_JS_PATH, ROUTE_SHARED_STYLES_PATH, ROUTE_TOKENS_PATH,
+    },
+    chat::model::Message,
+    common::{
+        client::rebuild_http_client,
+        model::{userinfo::TokenProfile, ApiStatus},
+        utils::{generate_checksum_with_repair, parse_bool_from_env, parse_string_from_env},
+    },
+};
+use parking_lot::RwLock;
+use rkyv::{Archive, Deserialize as RkyvDeserialize, Serialize as RkyvSerialize};
+use serde::{Deserialize, Serialize};
+use std::sync::LazyLock;
+
+mod usage_check;
+pub use usage_check::UsageCheck;
+mod config;
+mod proxies;
+pub use proxies::Proxies;
+mod build_key;
+pub use build_key::*;
+
+use super::constant::{STATUS_FAILED, STATUS_PENDING, STATUS_SUCCESS};
+
+// 页面内容类型枚举
+#[derive(Clone, Serialize, Deserialize, Archive, RkyvDeserialize, RkyvSerialize)]
+#[serde(tag = "type", content = "content")]
+pub enum PageContent {
+    #[serde(rename = "default")]
+    Default, // 默认行为
+    #[serde(rename = "text")]
+    Text(String), // 纯文本
+    #[serde(rename = "html")]
+    Html(String), // HTML 内容
+}
+
+impl Default for PageContent {
+    fn default() -> Self {
+        Self::Default
+    }
+}
+
+// 静态配置
+#[derive(Default, Clone)]
+pub struct AppConfig {
+    vision_ability: VisionAbility,
+    slow_pool: bool,
+    allow_claude: bool,
+    pages: Pages,
+    usage_check: UsageCheck,
+    dynamic_key: bool,
+    share_token: String,
+    is_share: bool,
+    proxies: Proxies,
+    web_refs: bool,
+}
+
+#[derive(Serialize, Deserialize, Clone, Copy, PartialEq)]
+pub enum VisionAbility {
+    #[serde(rename = "none", alias = "disabled")]
+    None,
+    #[serde(rename = "base64", alias = "base64-only")]
+    Base64,
+    #[serde(rename = "all", alias = "base64-http")]
+    All,
+}
+
+impl VisionAbility {
+    pub fn from_str(s: &str) -> Self {
+        match s.to_lowercase().as_str() {
+            "none" | "disabled" => Self::None,
+            "base64" | "base64-only" => Self::Base64,
+            "all" | "base64-http" => Self::All,
+            _ => Self::default(),
+        }
+    }
+
+    pub fn is_none(&self) -> bool {
+        matches!(self, VisionAbility::None)
+    }
+}
+
+impl Default for VisionAbility {
+    fn default() -> Self {
+        Self::Base64
+    }
+}
+
+#[derive(Clone, Default, Archive, RkyvDeserialize, RkyvSerialize)]
+pub struct Pages {
+    pub root_content: PageContent,
+    pub logs_content: PageContent,
+    pub config_content: PageContent,
+    pub tokeninfo_content: PageContent,
+    pub shared_styles_content: PageContent,
+    pub shared_js_content: PageContent,
+    pub about_content: PageContent,
+    pub readme_content: PageContent,
+    pub api_content: PageContent,
+    pub build_key_content: PageContent,
+}
+
+// 运行时状态
+pub struct AppState {
+    pub total_requests: u64,
+    pub active_requests: u64,
+    pub error_requests: u64,
+    pub request_logs: Vec<RequestLog>,
+    pub token_infos: Vec<TokenInfo>,
+}
+
+// 全局配置实例
+pub static APP_CONFIG: LazyLock<RwLock<AppConfig>> =
+    LazyLock::new(|| RwLock::new(AppConfig::default()));
+
+macro_rules! config_methods {
+    ($($field:ident: $type:ty, $default:expr;)*) => {
+        $(
+            paste::paste! {
+                pub fn [<get_ $field>]() -> $type
+                where
+                    $type: Copy + PartialEq,
+                {
+                    APP_CONFIG.read().$field
+                }
+
+                pub fn [<update_ $field>](value: $type)
+                where
+                    $type: Copy + PartialEq,
+                {
+                    let current = Self::[<get_ $field>]();
+                    if current != value {
+                        APP_CONFIG.write().$field = value;
+                    }
+                }
+
+                pub fn [<reset_ $field>]()
+                where
+                    $type: Copy + PartialEq,
+                {
+                    let default_value = $default;
+                    let current = Self::[<get_ $field>]();
+                    if current != default_value {
+                        APP_CONFIG.write().$field = default_value;
+                    }
+                }
+            }
+        )*
+    };
+}
+
+macro_rules! config_methods_clone {
+    ($($field:ident: $type:ty, $default:expr;)*) => {
+        $(
+            paste::paste! {
+                pub fn [<get_ $field>]() -> $type
+                where
+                    $type: Clone + PartialEq,
+                {
+                    APP_CONFIG.read().$field.clone()
+                }
+
+                pub fn [<update_ $field>](value: $type)
+                where
+                    $type: Clone + PartialEq,
+                {
+                    let current = Self::[<get_ $field>]();
+                    if current != value {
+                        APP_CONFIG.write().$field = value;
+                    }
+                }
+
+                pub fn [<reset_ $field>]()
+                where
+                    $type: Clone + PartialEq,
+                {
+                    let default_value = $default;
+                    let current = Self::[<get_ $field>]();
+                    if current != default_value {
+                        APP_CONFIG.write().$field = default_value;
+                    }
+                }
+            }
+        )*
+    };
+}
+
+impl AppConfig {
+    pub fn init() {
+        let mut config = APP_CONFIG.write();
+        config.vision_ability =
+            VisionAbility::from_str(&parse_string_from_env("VISION_ABILITY", EMPTY_STRING));
+        config.slow_pool = parse_bool_from_env("ENABLE_SLOW_POOL", false);
+        config.allow_claude = parse_bool_from_env("PASS_ANY_CLAUDE", false);
+        config.usage_check =
+            UsageCheck::from_str(&parse_string_from_env("USAGE_CHECK", EMPTY_STRING));
+        config.dynamic_key = parse_bool_from_env("DYNAMIC_KEY", false);
+        config.share_token = parse_string_from_env("SHARED_TOKEN", EMPTY_STRING);
+        config.is_share = !config.share_token.is_empty();
+        config.proxies = match std::env::var("PROXIES") {
+            Ok(proxies) => Proxies::from_str(proxies.as_str()),
+            Err(_) => Proxies::default(),
+        };
+        config.web_refs = parse_bool_from_env("INCLUDE_WEB_REFERENCES", false)
+    }
+
+    config_methods! {
+        slow_pool: bool, false;
+        allow_claude: bool, false;
+        dynamic_key: bool, false;
+        web_refs: bool, false;
+    }
+
+    config_methods_clone! {
+        vision_ability: VisionAbility, VisionAbility::default();
+        usage_check: UsageCheck, UsageCheck::default();
+    }
+
+    pub fn get_share_token() -> String {
+        APP_CONFIG.read().share_token.clone()
+    }
+
+    pub fn update_share_token(value: String) {
+        let current = Self::get_share_token();
+        if current != value {
+            let mut config = APP_CONFIG.write();
+            config.share_token = value;
+            config.is_share = !config.share_token.is_empty();
+        }
+    }
+
+    pub fn reset_share_token() {
+        let current = Self::get_share_token();
+        if !current.is_empty() {
+            let mut config = APP_CONFIG.write();
+            config.share_token = String::new();
+            config.is_share = false;
+        }
+    }
+
+    pub fn get_proxies() -> Proxies {
+        APP_CONFIG.read().proxies.clone()
+    }
+
+    pub fn update_proxies(value: Proxies) {
+        let current = Self::get_proxies();
+        if current != value {
+            let mut config = APP_CONFIG.write();
+            config.proxies = value;
+            rebuild_http_client();
+        }
+    }
+
+    pub fn reset_proxies() {
+        let default_value = Proxies::default();
+        let current = Self::get_proxies();
+        if current != default_value {
+            let mut config = APP_CONFIG.write();
+            config.proxies = default_value;
+            rebuild_http_client();
+        }
+    }
+
+    pub fn get_page_content(path: &str) -> Option<PageContent> {
+        match path {
+            ROUTE_ROOT_PATH => Some(APP_CONFIG.read().pages.root_content.clone()),
+            ROUTE_LOGS_PATH => Some(APP_CONFIG.read().pages.logs_content.clone()),
+            ROUTE_CONFIG_PATH => Some(APP_CONFIG.read().pages.config_content.clone()),
+            ROUTE_TOKENS_PATH => Some(APP_CONFIG.read().pages.tokeninfo_content.clone()),
+            ROUTE_SHARED_STYLES_PATH => Some(APP_CONFIG.read().pages.shared_styles_content.clone()),
+            ROUTE_SHARED_JS_PATH => Some(APP_CONFIG.read().pages.shared_js_content.clone()),
+            ROUTE_ABOUT_PATH => Some(APP_CONFIG.read().pages.about_content.clone()),
+            ROUTE_README_PATH => Some(APP_CONFIG.read().pages.readme_content.clone()),
+            ROUTE_API_PATH => Some(APP_CONFIG.read().pages.api_content.clone()),
+            ROUTE_BUILD_KEY_PATH => Some(APP_CONFIG.read().pages.build_key_content.clone()),
+            _ => None,
+        }
+    }
+
+    pub fn update_page_content(path: &str, content: PageContent) -> Result<(), &'static str> {
+        let mut config = APP_CONFIG.write();
+        match path {
+            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_TOKENS_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,
+            ROUTE_API_PATH => config.pages.api_content = content,
+            ROUTE_BUILD_KEY_PATH => config.pages.build_key_content = content,
+            _ => return Err(ERR_INVALID_PATH),
+        }
+        Ok(())
+    }
+
+    pub fn reset_page_content(path: &str) -> Result<(), &'static str> {
+        let mut config = APP_CONFIG.write();
+        match path {
+            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_TOKENS_PATH => config.pages.tokeninfo_content = PageContent::default(),
+            ROUTE_SHARED_STYLES_PATH => config.pages.shared_styles_content = PageContent::default(),
+            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(),
+            ROUTE_API_PATH => config.pages.api_content = PageContent::default(),
+            ROUTE_BUILD_KEY_PATH => config.pages.build_key_content = PageContent::default(),
+            _ => return Err(ERR_INVALID_PATH),
+        }
+        Ok(())
+    }
+
+    pub fn is_share() -> bool {
+        APP_CONFIG.read().is_share
+    }
+}
+
+impl AppState {
+    pub fn new(token_infos: Vec<TokenInfo>) -> Self {
+        // 尝试加载保存的日志
+        let request_logs = tokio::task::block_in_place(|| {
+            tokio::runtime::Handle::current()
+                .block_on(async { Self::load_saved_logs().await.unwrap_or_default() })
+        });
+
+        Self {
+            total_requests: request_logs.len() as u64,
+            active_requests: 0,
+            error_requests: request_logs
+                .iter()
+                .filter(|log| matches!(log.status, LogStatus::Failed))
+                .count() as u64,
+            request_logs,
+            token_infos,
+        }
+    }
+
+    pub fn update_checksum(&mut self) {
+        for token_info in self.token_infos.iter_mut() {
+            token_info.checksum = generate_checksum_with_repair(&token_info.checksum);
+        }
+    }
+}
+
+#[derive(Clone, Archive, RkyvDeserialize, RkyvSerialize)]
+pub enum LogStatus {
+    Pending,
+    Success,
+    Failed,
+}
+
+impl Serialize for LogStatus {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: serde::Serializer,
+    {
+        serializer.serialize_str(self.as_str_name())
+    }
+}
+
+impl LogStatus {
+    pub fn as_str_name(&self) -> &'static str {
+        match self {
+            Self::Pending => STATUS_PENDING,
+            Self::Success => STATUS_SUCCESS,
+            Self::Failed => STATUS_FAILED,
+        }
+    }
+
+    pub fn from_str_name(s: &str) -> Option<Self> {
+        match s {
+            STATUS_PENDING => Some(Self::Pending),
+            STATUS_SUCCESS => Some(Self::Success),
+            STATUS_FAILED => Some(Self::Failed),
+            _ => None,
+        }
+    }
+}
+
+// 请求日志
+#[derive(Serialize, Clone, Archive, RkyvDeserialize, RkyvSerialize)]
+pub struct RequestLog {
+    pub id: u64,
+    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 timing: TimingInfo,
+    pub stream: bool,
+    pub status: LogStatus,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub error: Option<String>,
+}
+
+#[derive(Serialize, Clone, Archive, RkyvDeserialize, RkyvSerialize)]
+pub struct TimingInfo {
+    pub total: f64, // 总用时(秒)
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub first: Option<f64>, // 首字时间(秒)
+}
+
+// 聊天请求
+#[derive(Deserialize)]
+pub struct ChatRequest {
+    pub model: String,
+    pub messages: Vec<Message>,
+    #[serde(default)]
+    pub stream: bool,
+}
+
+// 用于存储 token 信息
+#[derive(Serialize, Clone, Archive, RkyvDeserialize, RkyvSerialize)]
+pub struct TokenInfo {
+    pub token: String,
+    pub checksum: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub profile: Option<TokenProfile>,
+}
+
+// TokenUpdateRequest 结构体
+#[derive(Deserialize)]
+pub struct TokenUpdateRequest {
+    pub tokens: String,
+}
+
+#[derive(Deserialize)]
+pub struct TokenAddRequestTokenInfo {
+    pub token: String,
+    #[serde(default)]
+    pub checksum: Option<String>,
+}
+
+// TokensDeleteRequest 结构体
+#[derive(Deserialize)]
+pub struct TokensDeleteRequest {
+    #[serde(default)]
+    pub tokens: Vec<String>,
+    #[serde(default)]
+    pub expectation: TokensDeleteResponseExpectation,
+}
+
+#[derive(Deserialize, Default)]
+#[serde(rename_all = "snake_case")]
+pub enum TokensDeleteResponseExpectation {
+    #[default]
+    Simple,
+    UpdatedTokens,
+    FailedTokens,
+    Detailed,
+}
+
+impl TokensDeleteResponseExpectation {
+    pub fn needs_updated_tokens(&self) -> bool {
+        matches!(
+            self,
+            TokensDeleteResponseExpectation::UpdatedTokens
+                | TokensDeleteResponseExpectation::Detailed
+        )
+    }
+
+    pub fn needs_failed_tokens(&self) -> bool {
+        matches!(
+            self,
+            TokensDeleteResponseExpectation::FailedTokens
+                | TokensDeleteResponseExpectation::Detailed
+        )
+    }
+}
+
+// TokensDeleteResponse 结构体
+#[derive(Serialize)]
+pub struct TokensDeleteResponse {
+    pub status: ApiStatus,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub updated_tokens: Option<Vec<String>>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub failed_tokens: Option<Vec<String>>,
+}

src/app/model/build_key.rs

@@ -0,0 +1,74 @@
+use serde::{Deserialize, Serialize};
+
+use crate::{app::constant::COMMA, chat::constant::AVAILABLE_MODELS};
+
+#[derive(Deserialize)]
+pub struct BuildKeyRequest {
+    pub auth_token: String,
+    #[serde(default)]
+    pub disable_vision: Option<bool>,
+    #[serde(default)]
+    pub enable_slow_pool: Option<bool>,
+    #[serde(default)]
+    pub usage_check_models: Option<UsageCheckModelConfig>,
+    #[serde(default)]
+    pub include_web_references: Option<bool>,
+}
+pub struct UsageCheckModelConfig {
+    pub model_type: UsageCheckModelType,
+    pub model_ids: Vec<&'static str>,
+}
+
+impl<'de> Deserialize<'de> for UsageCheckModelConfig {
+    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
+    where
+        D: serde::Deserializer<'de>,
+    {
+        #[derive(Deserialize)]
+        struct Helper {
+            #[serde(rename = "type")]
+            model_type: UsageCheckModelType,
+            #[serde(default)]
+            model_ids: String,
+        }
+
+        let helper = Helper::deserialize(deserializer)?;
+
+        let model_ids = if helper.model_ids.is_empty() {
+            Vec::new()
+        } else {
+            helper
+                .model_ids
+                .split(COMMA)
+                .filter_map(|model| {
+                    let model = model.trim();
+                    AVAILABLE_MODELS
+                        .iter()
+                        .find(|m| m.id == model)
+                        .map(|m| m.id)
+                })
+                .collect()
+        };
+
+        Ok(UsageCheckModelConfig {
+            model_type: helper.model_type,
+            model_ids,
+        })
+    }
+}
+
+#[derive(Deserialize)]
+#[serde(rename_all = "lowercase")]
+pub enum UsageCheckModelType {
+    Default,
+    Disabled,
+    All,
+    Custom,
+}
+
+#[derive(Serialize)]
+#[serde(rename_all = "lowercase")]
+pub enum BuildKeyResponse {
+    Key(String),
+    Error(String),
+}

src/app/model/config.rs

@@ -0,0 +1,114 @@
+use memmap2::{MmapMut, MmapOptions};
+use rkyv::{archived_root, Deserialize as _};
+use std::fs::OpenOptions;
+
+use crate::app::lazy::{LOGS_FILE_PATH, PAGES_FILE_PATH};
+
+use super::{AppConfig, AppState, Pages, RequestLog, APP_CONFIG};
+
+impl AppState {
+    // 保存日志的方法
+    pub(crate) async fn save_logs(&self) -> Result<(), Box<dyn std::error::Error>> {
+        // 序列化日志
+        let bytes = rkyv::to_bytes::<_, 256>(&self.request_logs)?;
+
+        // 创建或打开文件
+        let file = OpenOptions::new()
+            .read(true)
+            .write(true)
+            .create(true)
+            .open(LOGS_FILE_PATH.as_str())?;
+
+        // 添加大小检查
+        if bytes.len() > usize::MAX / 2 {
+            return Err("日志数据过大".into());
+        }
+
+        // 设置文件大小
+        file.set_len(bytes.len() as u64)?;
+
+        // 创建可写入的内存映射
+        let mut mmap = unsafe { MmapMut::map_mut(&file)? };
+
+        // 写入数据
+        mmap.copy_from_slice(&bytes);
+
+        // 同步到磁盘
+        mmap.flush()?;
+
+        Ok(())
+    }
+
+    // 加载日志的方法
+    pub(super) async fn load_saved_logs() -> Result<Vec<RequestLog>, Box<dyn std::error::Error>> {
+        let file = match OpenOptions::new().read(true).open(LOGS_FILE_PATH.as_str()) {
+            Ok(file) => file,
+            Err(e) if e.kind() == std::io::ErrorKind::NotFound => {
+                return Ok(Vec::new());
+            }
+            Err(e) => return Err(Box::new(e)),
+        };
+
+        // 添加文件大小检查
+        if file.metadata()?.len() > usize::MAX as u64 {
+            return Err("日志文件过大".into());
+        }
+
+        // 创建只读内存映射
+        let mmap = unsafe { MmapOptions::new().map(&file)? };
+
+        // 验证并反序列化数据
+        let archived = unsafe { archived_root::<Vec<RequestLog>>(&mmap) };
+        Ok(archived.deserialize(&mut rkyv::Infallible)?)
+    }
+}
+
+impl AppConfig {
+    pub fn save_config() -> Result<(), Box<dyn std::error::Error>> {
+        let pages = APP_CONFIG.read().pages.clone();
+        let bytes = rkyv::to_bytes::<_, 256>(&pages)?;
+
+        let file = OpenOptions::new()
+            .read(true)
+            .write(true)
+            .create(true)
+            .open(PAGES_FILE_PATH.as_str())?;
+
+        // 添加大小检查
+        if bytes.len() > usize::MAX / 2 {
+            return Err("配置数据过大".into());
+        }
+
+        file.set_len(bytes.len() as u64)?;
+
+        let mut mmap = unsafe { MmapMut::map_mut(&file)? };
+        mmap.copy_from_slice(&bytes);
+        mmap.flush()?;
+
+        Ok(())
+    }
+
+    pub fn load_saved_config() -> Result<(), Box<dyn std::error::Error>> {
+        let file = match OpenOptions::new().read(true).open(PAGES_FILE_PATH.as_str()) {
+            Ok(file) => file,
+            Err(e) if e.kind() == std::io::ErrorKind::NotFound => {
+                return Ok(());
+            }
+            Err(e) => return Err(Box::new(e)),
+        };
+
+        // 添加文件大小检查
+        if file.metadata()?.len() > usize::MAX as u64 {
+            return Err("配置文件过大".into());
+        }
+
+        let mmap = unsafe { MmapOptions::new().map(&file)? };
+
+        let archived = unsafe { archived_root::<Pages>(&mmap) };
+        let pages = archived.deserialize(&mut rkyv::Infallible)?;
+        let mut config = APP_CONFIG.write();
+        config.pages = pages;
+
+        Ok(())
+    }
+}

src/app/model/proxies.rs

@@ -0,0 +1,81 @@
+use reqwest::{Client, Proxy};
+use serde::{Serialize, Serializer};
+use serde::{Deserialize, Deserializer};
+// use rkyv::{Archive, Deserialize as RkyvDeserialize, Serialize as RkyvSerialize};
+
+use crate::app::constant::COMMA_STRING;
+
+#[derive(Clone, Default, PartialEq)]
+pub enum Proxies {
+    No,
+    #[default]
+    System,
+    List(Vec<String>),
+}
+
+impl Serialize for Proxies {
+    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
+    where
+        S: Serializer,
+    {
+        match self {
+            Proxies::No => serializer.serialize_str(""),
+            Proxies::System => serializer.serialize_str("system"),
+            Proxies::List(urls) => serializer.serialize_str(&urls.join(COMMA_STRING)),
+        }
+    }
+}
+
+impl<'de> Deserialize<'de> for Proxies {
+    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
+    where
+        D: Deserializer<'de>,
+    {
+        let s = <String as serde::Deserialize>::deserialize(deserializer)?;
+        Ok(Proxies::from_str(&s))
+    }
+}
+
+impl Proxies {
+    /// 从字符串创建 Proxies
+    ///
+    /// # Arguments
+    /// * `s` - 代理字符串:
+    ///   - "" 或 "no": 不使用代理
+    ///   - "system": 使用系统代理
+    ///   - 其他: 尝试解析为代理列表，无效则返回 System
+    pub fn from_str(s: &str) -> Self {
+        match s.trim() {
+            "" | "no" => Self::No,
+            "system" => Self::System,
+            urls => {
+                let valid_proxies: Vec<String> = urls
+                    .split(',')
+                    .filter_map(|url| {
+                        let trimmed = url.trim();
+                        (!trimmed.is_empty() && Proxy::all(trimmed).is_ok())
+                            .then(|| trimmed.to_string())
+                    })
+                    .collect();
+
+                if valid_proxies.is_empty() {
+                    Self::default()
+                } else {
+                    Self::List(valid_proxies)
+                }
+            }
+        }
+    }
+
+    pub fn get_client(&self) -> Client {
+        match self {
+            Proxies::No => Client::builder().no_proxy().build().unwrap(),
+            Proxies::System => Client::new(),
+            Proxies::List(list) => {
+                // 使用第一个代理（已经确保是有效的）
+                let proxy = Proxy::all(list[0].clone()).unwrap();
+                Client::builder().proxy(proxy).build().unwrap()
+            }
+        }
+    }
+}

src/app/model/usage_check.rs

@@ -0,0 +1,172 @@
+use crate::{
+    app::constant::{COMMA, COMMA_STRING},
+    chat::{config::key_config, constant::AVAILABLE_MODELS},
+};
+use serde::{Deserialize, Serialize};
+// use rkyv::{Archive, Deserialize as RkyvDeserialize, Serialize as RkyvSerialize};
+
+#[derive(Clone, PartialEq)]
+pub enum UsageCheck {
+    None,
+    Default,
+    All,
+    Custom(Vec<&'static str>),
+}
+
+impl UsageCheck {
+    pub fn from_proto(model: Option<&key_config::UsageCheckModel>) -> Option<Self> {
+        model.map(|model| {
+            use key_config::usage_check_model::Type;
+            match Type::try_from(model.r#type).unwrap_or(Type::Default) {
+                Type::Default | Type::Disabled => Self::None,
+                Type::All => Self::All,
+                Type::Custom => {
+                    let models: Vec<&'static str> = model
+                        .model_ids
+                        .iter()
+                        .filter_map(|id| AVAILABLE_MODELS.iter().find(|m| m.id == id).map(|m| m.id))
+                        .collect();
+                    if models.is_empty() {
+                        Self::None
+                    } else {
+                        Self::Custom(models)
+                    }
+                }
+            }
+        })
+    }
+
+    // pub fn to_proto(&self) -> key_config::UsageCheckModel {
+    //     use key_config::usage_check_model::Type;
+    //     match self {
+    //         Self::None => key_config::UsageCheckModel {
+    //             r#type: Type::Disabled.into(),
+    //             model_ids: vec![],
+    //         },
+    //         Self::Default => key_config::UsageCheckModel {
+    //             r#type: Type::Default.into(),
+    //             model_ids: vec![],
+    //         },
+    //         Self::All => key_config::UsageCheckModel {
+    //             r#type: Type::All.into(),
+    //             model_ids: vec![],
+    //         },
+    //         Self::Custom(models) => key_config::UsageCheckModel {
+    //             r#type: Type::Custom.into(),
+    //             model_ids: models.iter().map(|&s| s.to_string()).collect(),
+    //         },
+    //     }
+    // }
+}
+
+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(COMMA_STRING))?;
+            }
+        }
+        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 as serde::Deserialize>::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(COMMA)
+                    .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)
+                }
+            }
+        })
+    }
+}
+
+impl UsageCheck {
+    pub fn from_str(s: &str) -> Self {
+        match s.trim().to_lowercase().as_str() {
+            "none" | "disabled" => Self::None,
+            "default" => Self::Default,
+            "all" | "everything" => Self::All,
+            list => {
+                if list.is_empty() {
+                    return Self::default();
+                }
+                let models: Vec<&'static str> = list
+                    .split(COMMA)
+                    .filter_map(|model| {
+                        let model = model.trim();
+                        AVAILABLE_MODELS
+                            .iter()
+                            .find(|m| m.id == model)
+                            .map(|m| m.id)
+                    })
+                    .collect();
+
+                if models.is_empty() {
+                    Self::default()
+                } else {
+                    Self::Custom(models)
+                }
+            }
+        }
+    }
+}

src/chat.rs

@@ -0,0 +1,10 @@
+pub mod adapter;
+pub mod aiserver;
+pub mod config;
+pub mod constant;
+pub mod error;
+// pub mod middleware;
+pub mod model;
+pub mod route;
+pub mod service;
+pub mod stream;

src/chat/adapter.rs

@@ -0,0 +1,473 @@
+use base64::{engine::general_purpose::STANDARD as BASE64, Engine as _};
+use image::guess_format;
+use prost::Message as _;
+use reqwest::Client;
+use uuid::Uuid;
+
+use crate::{
+    app::{
+        constant::EMPTY_STRING,
+        lazy::DEFAULT_INSTRUCTIONS,
+        model::{AppConfig, VisionAbility},
+    },
+    common::client::HTTP_CLIENT,
+};
+
+use super::{
+    aiserver::v1::{
+        conversation_message, image_proto, AzureState, ChatExternalLink, ConversationMessage, ExplicitContext, GetChatRequest, ImageProto, ModelDetails
+    },
+    constant::{ERR_UNSUPPORTED_GIF, ERR_UNSUPPORTED_IMAGE_FORMAT, LONG_CONTEXT_MODELS},
+    model::{Message, MessageContent, Role},
+};
+
+async fn process_chat_inputs(
+    inputs: Vec<Message>,
+    disable_vision: bool,
+) -> (String, Vec<ConversationMessage>, Vec<String>) {
+    // 收集 system 指令
+    let instructions = inputs
+        .iter()
+        .filter(|input| input.role == Role::System)
+        .map(|input| match &input.content {
+            MessageContent::Text(text) => text.clone(),
+            MessageContent::Vision(contents) => contents
+                .iter()
+                .filter_map(|content| {
+                    if content.content_type == "text" {
+                        content.text.clone()
+                    } else {
+                        None
+                    }
+                })
+                .collect::<Vec<String>>()
+                .join("\n"),
+        })
+        .collect::<Vec<String>>()
+        .join("\n\n");
+
+    // 使用默认指令或收集到的指令
+    let instructions = if instructions.is_empty() {
+        DEFAULT_INSTRUCTIONS.clone()
+    } else {
+        instructions
+    };
+
+    // 过滤出 user 和 assistant 对话
+    let mut chat_inputs: Vec<Message> = inputs
+        .into_iter()
+        .filter(|input| input.role == Role::User || input.role == Role::Assistant)
+        .collect();
+
+    // 处理空对话情况
+    if chat_inputs.is_empty() {
+        return (
+            instructions,
+            vec![ConversationMessage {
+                text: EMPTY_STRING.into(),
+                r#type: conversation_message::MessageType::Human as i32,
+                attached_code_chunks: vec![],
+                codebase_context_chunks: vec![],
+                commits: vec![],
+                pull_requests: vec![],
+                git_diffs: vec![],
+                assistant_suggested_diffs: vec![],
+                interpreter_results: vec![],
+                images: vec![],
+                attached_folders: vec![],
+                approximate_lint_errors: vec![],
+                bubble_id: Uuid::new_v4().to_string(),
+                server_bubble_id: None,
+                attached_folders_new: vec![],
+                lints: vec![],
+                user_responses_to_suggested_code_blocks: vec![],
+                relevant_files: vec![],
+                tool_results: vec![],
+                notepads: vec![],
+                is_capability_iteration: Some(false),
+                capabilities: vec![],
+                edit_trail_contexts: vec![],
+                suggested_code_blocks: vec![],
+                diffs_for_compressing_files: vec![],
+                multi_file_linter_errors: vec![],
+                diff_histories: vec![],
+                recently_viewed_files: vec![],
+                recent_locations_history: vec![],
+                is_agentic: false,
+                file_diff_trajectories: vec![],
+                conversation_summary: None,
+            }],
+            vec![],
+        );
+    }
+
+    // 处理 WebReferences 开头的 assistant 消息
+    chat_inputs = chat_inputs
+        .into_iter()
+        .map(|mut input| {
+            if let (Role::Assistant, MessageContent::Text(text)) = (&input.role, &input.content) {
+                if text.starts_with("WebReferences:") {
+                    if let Some(pos) = text.find("\n\n") {
+                        input.content = MessageContent::Text(text[pos + 2..].to_owned());
+                    }
+                }
+            }
+            input
+        })
+        .collect();
+
+    // 如果第一条是 assistant，插入空的 user 消息
+    if chat_inputs
+        .first()
+        .map_or(false, |input| input.role == Role::Assistant)
+    {
+        chat_inputs.insert(
+            0,
+            Message {
+                role: Role::User,
+                content: MessageContent::Text(EMPTY_STRING.into()),
+            },
+        );
+    }
+
+    // 处理连续相同角色的情况
+    let mut i = 1;
+    while i < chat_inputs.len() {
+        if chat_inputs[i].role == chat_inputs[i - 1].role {
+            let insert_role = if chat_inputs[i].role == Role::User {
+                Role::Assistant
+            } else {
+                Role::User
+            };
+            chat_inputs.insert(
+                i,
+                Message {
+                    role: insert_role,
+                    content: MessageContent::Text(EMPTY_STRING.into()),
+                },
+            );
+        }
+        i += 1;
+    }
+
+    // 确保最后一条是 user
+    if chat_inputs
+        .last()
+        .map_or(false, |input| input.role == Role::Assistant)
+    {
+        chat_inputs.push(Message {
+            role: Role::User,
+            content: MessageContent::Text(EMPTY_STRING.into()),
+        });
+    }
+
+    // 转换为 proto messages
+    let mut messages = Vec::new();
+    for input in chat_inputs {
+        let (text, images) = match input.content {
+            MessageContent::Text(text) => (text, vec![]),
+            MessageContent::Vision(contents) => {
+                let mut text_parts = Vec::new();
+                let mut images = Vec::new();
+
+                for content in contents {
+                    match content.content_type.as_str() {
+                        "text" => {
+                            if let Some(text) = content.text {
+                                text_parts.push(text);
+                            }
+                        }
+                        "image_url" => {
+                            if !disable_vision {
+                                if let Some(image_url) = &content.image_url {
+                                    let url = image_url.url.clone();
+                                    let client = HTTP_CLIENT.read().clone();
+                                    let result = tokio::spawn(async move {
+                                        fetch_image_data(&url, client).await
+                                    });
+                                    if let Ok(Ok((image_data, dimensions))) = result.await {
+                                        images.push(ImageProto {
+                                            data: image_data,
+                                            dimension: dimensions,
+                                        });
+                                    }
+                                }
+                            }
+                        }
+                        _ => {}
+                    }
+                }
+                (text_parts.join("\n"), images)
+            }
+        };
+
+        messages.push(ConversationMessage {
+            text,
+            r#type: if input.role == Role::User {
+                conversation_message::MessageType::Human as i32
+            } else {
+                conversation_message::MessageType::Ai as i32
+            },
+            attached_code_chunks: vec![],
+            codebase_context_chunks: vec![],
+            commits: vec![],
+            pull_requests: vec![],
+            git_diffs: vec![],
+            assistant_suggested_diffs: vec![],
+            interpreter_results: vec![],
+            images,
+            attached_folders: vec![],
+            approximate_lint_errors: vec![],
+            bubble_id: Uuid::new_v4().to_string(),
+            server_bubble_id: None,
+            attached_folders_new: vec![],
+            lints: vec![],
+            user_responses_to_suggested_code_blocks: vec![],
+            relevant_files: vec![],
+            tool_results: vec![],
+            notepads: vec![],
+            is_capability_iteration: None,
+            capabilities: vec![],
+            edit_trail_contexts: vec![],
+            suggested_code_blocks: vec![],
+            diffs_for_compressing_files: vec![],
+            multi_file_linter_errors: vec![],
+            diff_histories: vec![],
+            recently_viewed_files: vec![],
+            recent_locations_history: vec![],
+            is_agentic: false,
+            file_diff_trajectories: vec![],
+            conversation_summary: None,
+        });
+    }
+
+    let mut urls = Vec::new();
+    if let Some(last_msg) = messages.last() {
+        if last_msg.r#type == conversation_message::MessageType::Human as i32 {
+            let text = &last_msg.text;
+            let mut chars = text.chars().peekable();
+            
+            while let Some(c) = chars.next() {
+                if c == '@' {
+                    let mut url = String::new();
+                    while let Some(&next_char) = chars.peek() {
+                        if next_char.is_whitespace() {
+                            break;
+                        }
+                        url.push(chars.next().unwrap());
+                    }
+                    if let Ok(parsed_url) = url::Url::parse(&url) {
+                        if parsed_url.scheme() == "http" || parsed_url.scheme() == "https" {
+                            urls.push(url);
+                        }
+                    }
+                }
+            }
+        }
+    }
+
+    (instructions, messages, urls)
+}
+
+async fn fetch_image_data(
+    url: &str,
+    client: Client,
+) -> Result<(Vec<u8>, Option<image_proto::Dimension>), Box<dyn std::error::Error + Send + Sync>> {
+    // 在进入异步操作前获取并释放锁
+    let vision_ability = AppConfig::get_vision_ability();
+
+    match vision_ability {
+        VisionAbility::None => Err("图片功能已禁用".into()),
+
+        VisionAbility::Base64 => {
+            if !url.starts_with("data:image/") {
+                return Err("仅支持 base64 编码的图片".into());
+            }
+            process_base64_image(url)
+        }
+
+        VisionAbility::All => {
+            if url.starts_with("data:image/") {
+                process_base64_image(url)
+            } else {
+                process_http_image(url, client).await
+            }
+        }
+    }
+}
+
+// 处理 base64 编码的图片
+fn process_base64_image(
+    url: &str,
+) -> Result<(Vec<u8>, Option<image_proto::Dimension>), Box<dyn std::error::Error + Send + Sync>> {
+    let parts: Vec<&str> = url.split("base64,").collect();
+    if parts.len() != 2 {
+        return Err("无效的 base64 图片格式".into());
+    }
+
+    // 检查图片格式
+    let format = parts[0].to_lowercase();
+    if !format.contains("png")
+        && !format.contains("jpeg")
+        && !format.contains("jpg")
+        && !format.contains("webp")
+        && !format.contains("gif")
+    {
+        return Err(ERR_UNSUPPORTED_IMAGE_FORMAT.into());
+    }
+
+    let image_data = BASE64.decode(parts[1])?;
+
+    // 检查是否为动态 GIF
+    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(ERR_UNSUPPORTED_GIF.into());
+            }
+        }
+    }
+
+    // 获取图片尺寸
+    let dimensions = if let Ok(img) = image::load_from_memory(&image_data) {
+        Some(image_proto::Dimension {
+            width: img.width() as i32,
+            height: img.height() as i32,
+        })
+    } else {
+        None
+    };
+
+    Ok((image_data, dimensions))
+}
+
+// 处理 HTTP 图片 URL
+async fn process_http_image(
+    url: &str,
+    client: Client,
+) -> Result<(Vec<u8>, Option<image_proto::Dimension>), Box<dyn std::error::Error + Send + Sync>> {
+    let response = client.get(url).send().await?;
+    let image_data = response.bytes().await?.to_vec();
+    let format = guess_format(&image_data)?;
+
+    // 检查图片格式
+    match format {
+        image::ImageFormat::Png | image::ImageFormat::Jpeg | image::ImageFormat::WebP => {
+            // 这些格式都支持
+        }
+        image::ImageFormat::Gif => {
+            if let Ok(frames) =
+                gif::DecodeOptions::new().read_info(std::io::Cursor::new(&image_data))
+            {
+                if frames.into_iter().count() > 1 {
+                    return Err(ERR_UNSUPPORTED_GIF.into());
+                }
+            }
+        }
+        _ => return Err(ERR_UNSUPPORTED_IMAGE_FORMAT.into()),
+    }
+
+    // 获取图片尺寸
+    let dimensions = if let Ok(img) = image::load_from_memory_with_format(&image_data, format) {
+        Some(image_proto::Dimension {
+            width: img.width() as i32,
+            height: img.height() as i32,
+        })
+    } else {
+        None
+    };
+
+    Ok((image_data, dimensions))
+}
+
+pub async fn encode_chat_message(
+    inputs: Vec<Message>,
+    model_name: &str,
+    disable_vision: bool,
+    enable_slow_pool: bool,
+    is_search: bool,
+) -> Result<Vec<u8>, Box<dyn std::error::Error + Send + Sync>> {
+    // 在进入异步操作前获取并释放锁
+    let enable_slow_pool = {
+        if enable_slow_pool {
+            Some(true)
+        } else {
+            None
+        }
+    };
+
+    let (instructions, messages, urls) = process_chat_inputs(inputs, disable_vision).await;
+
+    let explicit_context = if !instructions.trim().is_empty() {
+        Some(ExplicitContext {
+            context: instructions,
+            repo_context: None,
+        })
+    } else {
+        None
+    };
+
+    let base_uuid = rand::random::<u16>();
+    let external_links = urls.into_iter().enumerate().map(|(i, url)| {
+        let uuid = base_uuid.wrapping_add(i as u16);
+        ChatExternalLink {
+            url,
+            uuid: uuid.to_string(),
+        }
+    }).collect();
+
+    let chat = GetChatRequest {
+        current_file: None,
+        conversation: messages,
+        repositories: vec![],
+        explicit_context,
+        workspace_root_path: None,
+        code_blocks: vec![],
+        model_details: Some(ModelDetails {
+            model_name: Some(model_name.to_string()),
+            api_key: None,
+            enable_ghost_mode: None,
+            azure_state: Some(AzureState {
+                api_key: String::new(),
+                base_url: String::new(),
+                deployment: String::new(),
+                use_azure: false,
+            }),
+            enable_slow_pool,
+            openai_api_base_url: None,
+        }),
+        documentation_identifiers: vec![],
+        request_id: Uuid::new_v4().to_string(),
+        linter_errors: None,
+        summary: None,
+        summary_up_until_index: None,
+        allow_long_file_scan: Some(false),
+        is_bash: Some(false),
+        conversation_id: Uuid::new_v4().to_string(),
+        can_handle_filenames_after_language_ids: Some(true),
+        use_web: if is_search {
+            Some("full_search".to_string())
+        } else {
+            None
+        },
+        quotes: vec![],
+        debug_info: None,
+        workspace_id: None,
+        external_links,
+        commit_notes: vec![],
+        long_context_mode: Some(LONG_CONTEXT_MODELS.contains(&model_name)),
+        is_eval: Some(false),
+        desired_max_tokens: None,
+        context_ast: None,
+        is_composer: None,
+        runnable_code_blocks: Some(false),
+        should_cache: Some(false),
+    };
+
+    let mut encoded = Vec::new();
+    chat.encode(&mut encoded)?;
+
+    let len_prefix = format!("{:010x}", encoded.len()).to_uppercase();
+    let content = hex::encode_upper(&encoded);
+
+    Ok(hex::decode(len_prefix + &content)?)
+}

src/chat/aiserver.rs

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

src/chat/aiserver/v1.rs

@@ -0,0 +1,57 @@
+include!(concat!(env!("OUT_DIR"), "/aiserver.v1.rs"));
+use error_details::Error;
+
+impl ErrorDetails {
+    pub fn status_code(&self) -> u16 {
+        match Error::try_from(self.error) {
+            Ok(error) => match error {
+                Error::Unspecified => 500,
+                Error::BadApiKey
+                | Error::InvalidAuthId
+                | Error::AuthTokenNotFound
+                | Error::AuthTokenExpired
+                | Error::Unauthorized => 401,
+                Error::NotLoggedIn
+                | Error::NotHighEnoughPermissions
+                | Error::AgentRequiresLogin
+                | Error::ProUserOnly
+                | Error::TaskNoPermissions => 403,
+                Error::NotFound
+                | Error::UserNotFound
+                | Error::TaskUuidNotFound
+                | Error::AgentEngineNotFound
+                | Error::GitgraphNotFound
+                | Error::FileNotFound => 404,
+                Error::FreeUserRateLimitExceeded
+                | Error::ProUserRateLimitExceeded
+                | Error::OpenaiRateLimitExceeded
+                | Error::OpenaiAccountLimitExceeded
+                | Error::GenericRateLimitExceeded
+                | Error::Gpt4VisionPreviewRateLimit
+                | Error::ApiKeyRateLimit => 429,
+                Error::BadRequest
+                | Error::BadModelName
+                | Error::SlashEditFileTooLong
+                | Error::FileUnsupported
+                | Error::ClaudeImageTooLarge => 400,
+                Error::Deprecated
+                | Error::FreeUserUsageLimit
+                | Error::ProUserUsageLimit
+                | Error::ResourceExhausted
+                | Error::Openai
+                | Error::MaxTokens
+                | Error::ApiKeyNotSupported
+                | Error::UserAbortedRequest
+                | Error::CustomMessage
+                | Error::OutdatedClient
+                | Error::Debounced
+                | Error::RepositoryServiceRepositoryIsNotInitialized => 500,
+            },
+            Err(_) => 500,
+        }
+    }
+
+    // pub fn is_expected(&self) -> bool {
+    //     self.is_expected.unwrap_or_default()
+    // }
+}

src/chat/aiserver/v1/lite.proto

@@ -0,0 +1,1156 @@
+syntax = "proto3";
+package aiserver.v1;
+enum ClientSideToolV2 { // aiserver.v1.ClientSideToolV2
+	CLIENT_SIDE_TOOL_V2_UNSPECIFIED = 0;
+	CLIENT_SIDE_TOOL_V2_READ_SEMSEARCH_FILES = 1;
+	CLIENT_SIDE_TOOL_V2_READ_FILE_FOR_IMPORTS = 2;
+	CLIENT_SIDE_TOOL_V2_RIPGREP_SEARCH = 3;
+	CLIENT_SIDE_TOOL_V2_RUN_TERMINAL_COMMAND = 4;
+	CLIENT_SIDE_TOOL_V2_READ_FILE = 5;
+	CLIENT_SIDE_TOOL_V2_LIST_DIR = 6;
+	CLIENT_SIDE_TOOL_V2_EDIT_FILE = 7;
+	CLIENT_SIDE_TOOL_V2_FILE_SEARCH = 8;
+	CLIENT_SIDE_TOOL_V2_SEMANTIC_SEARCH_FULL = 9;
+	CLIENT_SIDE_TOOL_V2_CREATE_FILE = 10;
+	CLIENT_SIDE_TOOL_V2_DELETE_FILE = 11;
+}
+enum EmbeddingModel { // aiserver.v1.EmbeddingModel
+	EMBEDDING_MODEL_UNSPECIFIED = 0;
+	EMBEDDING_MODEL_VOYAGE_CODE_2 = 1;
+	EMBEDDING_MODEL_TEXT_EMBEDDINGS_LARGE_3 = 2;
+	EMBEDDING_MODEL_QWEN_1_5B_CUSTOM = 3;
+}
+enum ChunkType { // aiserver.v1.ChunkType
+	CHUNK_TYPE_UNSPECIFIED = 0;
+	CHUNK_TYPE_CODEBASE = 1;
+	CHUNK_TYPE_LONG_FILE = 2;
+	CHUNK_TYPE_DOCS = 3;
+}
+enum FastApplySource { // aiserver.v1.FastApplySource
+	FAST_APPLY_SOURCE_UNSPECIFIED = 0;
+	FAST_APPLY_SOURCE_COMPOSER = 1;
+	FAST_APPLY_SOURCE_CLICKED_APPLY = 2;
+	FAST_APPLY_SOURCE_CACHED_APPLY = 3;
+}
+enum BuiltinTool { // aiserver.v1.BuiltinTool
+	BUILTIN_TOOL_UNSPECIFIED = 0;
+	BUILTIN_TOOL_SEARCH = 1;
+	BUILTIN_TOOL_READ_CHUNK = 2;
+	BUILTIN_TOOL_GOTODEF = 3;
+	BUILTIN_TOOL_EDIT = 4;
+	BUILTIN_TOOL_UNDO_EDIT = 5;
+	BUILTIN_TOOL_END = 6;
+	BUILTIN_TOOL_NEW_FILE = 7;
+	BUILTIN_TOOL_ADD_TEST = 8;
+	BUILTIN_TOOL_RUN_TEST = 9;
+	BUILTIN_TOOL_DELETE_TEST = 10;
+	BUILTIN_TOOL_SAVE_FILE = 11;
+	BUILTIN_TOOL_GET_TESTS = 12;
+	BUILTIN_TOOL_GET_SYMBOLS = 13;
+	BUILTIN_TOOL_SEMANTIC_SEARCH = 14;
+	BUILTIN_TOOL_GET_PROJECT_STRUCTURE = 15;
+	BUILTIN_TOOL_CREATE_RM_FILES = 16;
+	BUILTIN_TOOL_RUN_TERMINAL_COMMANDS = 17;
+	BUILTIN_TOOL_NEW_EDIT = 18;
+	BUILTIN_TOOL_READ_WITH_LINTER = 19;
+}
+enum FeatureType { // aiserver.v1.FeatureType
+	FEATURE_TYPE_UNSPECIFIED = 0;
+	FEATURE_TYPE_EDIT = 1;
+	FEATURE_TYPE_GENERATE = 2;
+	FEATURE_TYPE_INLINE_LONG_COMPLETION = 3;
+}
+enum TaskStatus { // aiserver.v1.TaskStatus
+	TASK_STATUS_UNSPECIFIED = 0;
+	TASK_STATUS_RUNNING = 1;
+	TASK_STATUS_PAUSED = 2;
+	TASK_STATUS_DONE = 3;
+	TASK_STATUS_NOT_STARTED = 4;
+}
+enum RerankerAlgorithm { // aiserver.v1.RerankerAlgorithm
+	RERANKER_ALGORITHM_UNSPECIFIED = 0;
+	RERANKER_ALGORITHM_LULEA = 1;
+	RERANKER_ALGORITHM_UMEA = 2;
+	RERANKER_ALGORITHM_NONE = 3;
+	RERANKER_ALGORITHM_LLAMA = 4;
+	RERANKER_ALGORITHM_STARCODER_V1 = 5;
+	RERANKER_ALGORITHM_GPT_3_5_LOGPROBS = 6;
+	RERANKER_ALGORITHM_LULEA_HAIKU = 7;
+	RERANKER_ALGORITHM_COHERE = 8;
+	RERANKER_ALGORITHM_VOYAGE = 9;
+	RERANKER_ALGORITHM_VOYAGE_EMBEDS = 10;
+	RERANKER_ALGORITHM_IDENTITY = 11;
+	RERANKER_ALGORITHM_ADA_EMBEDS = 12;
+}
+enum RechunkerChoice { // aiserver.v1.RechunkerChoice
+	RECHUNKER_CHOICE_UNSPECIFIED = 0;
+	RECHUNKER_CHOICE_IDENTITY = 1;
+	RECHUNKER_CHOICE_600_TOKS = 2;
+	RECHUNKER_CHOICE_2400_TOKS = 3;
+	RECHUNKER_CHOICE_4000_TOKS = 4;
+}
+enum LintGenerator { // aiserver.v1.LintGenerator
+	LINT_GENERATOR_UNSPECIFIED = 0;
+	LINT_GENERATOR_NAIVE = 1;
+	LINT_GENERATOR_COMMENT_PIPELINE = 2;
+	LINT_GENERATOR_SIMPLE_BUG = 3;
+	LINT_GENERATOR_SIMPLE_LINT_RULES = 4;
+}
+enum LintDiscriminator { // aiserver.v1.LintDiscriminator
+	LINT_DISCRIMINATOR_UNSPECIFIED = 0;
+	LINT_DISCRIMINATOR_SPECIFIC_RULES = 1;
+	LINT_DISCRIMINATOR_COMPILE_ERRORS = 2;
+	LINT_DISCRIMINATOR_CHANGE_BEHAVIOR = 3;
+	LINT_DISCRIMINATOR_RELEVANCE = 4;
+	LINT_DISCRIMINATOR_USER_AWARENESS = 5;
+	LINT_DISCRIMINATOR_CORRECTNESS = 6;
+	LINT_DISCRIMINATOR_CHUNKING = 7;
+	LINT_DISCRIMINATOR_TYPO = 8;
+	LINT_DISCRIMINATOR_CONFIDENCE = 9;
+	LINT_DISCRIMINATOR_DISMISSED_BUGS = 10;
+}
+enum CppSource { // aiserver.v1.CppSource
+	CPP_SOURCE_UNSPECIFIED = 0;
+	CPP_SOURCE_LINE_CHANGE = 1;
+	CPP_SOURCE_TYPING = 2;
+	CPP_SOURCE_OPTION_HOLD = 3;
+	CPP_SOURCE_LINTER_ERRORS = 4;
+	CPP_SOURCE_PARAMETER_HINTS = 5;
+	CPP_SOURCE_CURSOR_PREDICTION = 6;
+	CPP_SOURCE_MANUAL_TRIGGER = 7;
+	CPP_SOURCE_EDITOR_CHANGE = 8;
+}
+enum ChunkingStrategy { // aiserver.v1.ChunkingStrategy
+	CHUNKING_STRATEGY_UNSPECIFIED = 0;
+	CHUNKING_STRATEGY_DEFAULT = 1;
+}
+message ErrorDetails { // aiserver.v1.ErrorDetails
+		enum Error { // aiserver.v1.ErrorDetails.Error
+			ERROR_UNSPECIFIED = 0;
+			ERROR_BAD_API_KEY = 1;
+			ERROR_NOT_LOGGED_IN = 2;
+			ERROR_INVALID_AUTH_ID = 3;
+			ERROR_NOT_HIGH_ENOUGH_PERMISSIONS = 4;
+			ERROR_BAD_MODEL_NAME = 5;
+			ERROR_USER_NOT_FOUND = 6;
+			ERROR_FREE_USER_RATE_LIMIT_EXCEEDED = 7;
+			ERROR_PRO_USER_RATE_LIMIT_EXCEEDED = 8;
+			ERROR_FREE_USER_USAGE_LIMIT = 9;
+			ERROR_PRO_USER_USAGE_LIMIT = 10;
+			ERROR_AUTH_TOKEN_NOT_FOUND = 11;
+			ERROR_AUTH_TOKEN_EXPIRED = 12;
+			ERROR_OPENAI = 13;
+			ERROR_OPENAI_RATE_LIMIT_EXCEEDED = 14;
+			ERROR_OPENAI_ACCOUNT_LIMIT_EXCEEDED = 15;
+			ERROR_TASK_UUID_NOT_FOUND = 16;
+			ERROR_TASK_NO_PERMISSIONS = 17;
+			ERROR_AGENT_REQUIRES_LOGIN = 18;
+			ERROR_AGENT_ENGINE_NOT_FOUND = 19;
+			ERROR_MAX_TOKENS = 20;
+			ERROR_USER_ABORTED_REQUEST = 21;
+			ERROR_GENERIC_RATE_LIMIT_EXCEEDED = 22;
+			ERROR_PRO_USER_ONLY = 23;
+			ERROR_API_KEY_NOT_SUPPORTED = 24;
+			ERROR_SLASH_EDIT_FILE_TOO_LONG = 26;
+			ERROR_FILE_UNSUPPORTED = 27;
+			ERROR_GPT_4_VISION_PREVIEW_RATE_LIMIT = 28;
+			ERROR_CUSTOM_MESSAGE = 29;
+			ERROR_OUTDATED_CLIENT = 30;
+			ERROR_CLAUDE_IMAGE_TOO_LARGE = 31;
+			ERROR_GITGRAPH_NOT_FOUND = 32;
+			ERROR_FILE_NOT_FOUND = 33;
+			ERROR_API_KEY_RATE_LIMIT = 34;
+			ERROR_DEBOUNCED = 35;
+			ERROR_BAD_REQUEST = 36;
+			ERROR_REPOSITORY_SERVICE_REPOSITORY_IS_NOT_INITIALIZED = 37;
+			ERROR_UNAUTHORIZED = 38;
+			ERROR_NOT_FOUND = 39;
+			ERROR_DEPRECATED = 40;
+			ERROR_RESOURCE_EXHAUSTED = 41;
+		}
+	Error error = 1;
+	CustomErrorDetails details = 2;
+	optional bool is_expected = 3;
+}
+
+message CustomErrorDetails { // aiserver.v1.CustomErrorDetails
+	string title = 1;
+	string detail = 2;
+	optional bool allow_command_links_potentially_unsafe_please_only_use_for_handwritten_trusted_markdown = 3;
+	optional bool is_retryable = 4;
+	optional bool show_request_id = 5;
+}
+
+message GetChatRequest { // aiserver.v1.GetChatRequest
+	CurrentFileInfo current_file = 1;
+	repeated ConversationMessage conversation = 2;
+	repeated RepositoryInfo repositories = 3;
+	ExplicitContext explicit_context = 4;
+	optional string workspace_root_path = 5;
+	repeated CodeBlock code_blocks = 6;
+	ModelDetails model_details = 7;
+	repeated string documentation_identifiers = 8;
+	string request_id = 9;
+	LinterErrors linter_errors = 10;
+	optional string summary = 11;
+	optional int32 summary_up_until_index = 12;
+	optional bool allow_long_file_scan = 13;
+	optional bool is_bash = 14;
+	string conversation_id = 15;
+	optional bool can_handle_filenames_after_language_ids = 16;
+	optional string use_web = 17;
+	repeated ChatQuote quotes = 18;
+	optional DebugInfo debug_info = 19;
+	optional string workspace_id = 20;
+	repeated ChatExternalLink external_links = 21;
+	repeated CommitNote commit_notes = 23;
+	optional bool long_context_mode = 22;
+	optional bool is_eval = 24;
+	optional int32 desired_max_tokens = 26;
+	ContextAST context_ast = 25;
+	optional bool is_composer = 27;
+	optional bool runnable_code_blocks = 28;
+	optional bool should_cache = 29;
+}
+message CurrentFileInfo { // aiserver.v1.CurrentFileInfo
+		message NotebookCell { // aiserver.v1.CurrentFileInfo.NotebookCell
+		}
+	string relative_workspace_path = 1;
+	string contents = 2;
+	bool rely_on_filesync = 18;
+	optional string sha_256_hash = 17;
+	repeated NotebookCell cells = 16;
+	repeated BM25Chunk top_chunks = 10;
+	int32 contents_start_at_line = 9;
+	CursorPosition cursor_position = 3;
+	repeated DataframeInfo dataframes = 4;
+	int32 total_number_of_lines = 8;
+	string language_id = 5;
+	CursorRange selection = 6;
+	optional int32 alternative_version_id = 11;
+	repeated Diagnostic diagnostics = 7;
+	optional int32 file_version = 14;
+	repeated int32 cell_start_lines = 15;
+	string workspace_root_path = 19;
+}
+message BM25Chunk { // aiserver.v1.BM25Chunk
+	string content = 1;
+	SimplestRange range = 2;
+	int32 score = 3;
+	string relative_path = 4;
+}
+message SimplestRange { // aiserver.v1.SimplestRange
+	int32 start_line = 1;
+	int32 end_line_inclusive = 2;
+}
+message CursorPosition { // aiserver.v1.CursorPosition
+	int32 line = 1;
+	int32 column = 2;
+}
+message DataframeInfo { // aiserver.v1.DataframeInfo
+		message Column { // aiserver.v1.DataframeInfo.Column
+			string key = 1;
+			string type = 2;
+		}
+	string name = 1;
+	string shape = 2;
+	int32 data_dimensionality = 3;
+	repeated Column columns = 6;
+	int32 row_count = 7;
+	string index_column = 8;
+}
+message CursorRange { // aiserver.v1.CursorRange
+	CursorPosition start_position = 1;
+	CursorPosition end_position = 2;
+}
+message Diagnostic { // aiserver.v1.Diagnostic
+		enum DiagnosticSeverity { // aiserver.v1.Diagnostic.DiagnosticSeverity
+			DIAGNOSTIC_SEVERITY_UNSPECIFIED = 0;
+			DIAGNOSTIC_SEVERITY_ERROR = 1;
+			DIAGNOSTIC_SEVERITY_WARNING = 2;
+			DIAGNOSTIC_SEVERITY_INFORMATION = 3;
+			DIAGNOSTIC_SEVERITY_HINT = 4;
+		}
+		message RelatedInformation { // aiserver.v1.Diagnostic.RelatedInformation
+			string message = 1;
+			CursorRange range = 2;
+		}
+	string message = 1;
+	CursorRange range = 2;
+	DiagnosticSeverity severity = 3;
+	repeated RelatedInformation related_information = 4;
+}
+message ConversationMessage { // aiserver.v1.ConversationMessage
+		enum MessageType { // aiserver.v1.ConversationMessage.MessageType
+			MESSAGE_TYPE_UNSPECIFIED = 0;
+			MESSAGE_TYPE_HUMAN = 1;
+			MESSAGE_TYPE_AI = 2;
+		}
+		message CodeChunk { // aiserver.v1.ConversationMessage.CodeChunk
+				enum SummarizationStrategy { // aiserver.v1.ConversationMessage.CodeChunk.SummarizationStrategy
+					SUMMARIZATION_STRATEGY_NONE_UNSPECIFIED = 0;
+					SUMMARIZATION_STRATEGY_SUMMARIZED = 1;
+					SUMMARIZATION_STRATEGY_EMBEDDED = 2;
+				}
+				enum Intent { // aiserver.v1.ConversationMessage.CodeChunk.Intent
+					INTENT_UNSPECIFIED = 0;
+					INTENT_COMPOSER_FILE = 1;
+					INTENT_COMPRESSED_COMPOSER_FILE = 2;
+					INTENT_RECENTLY_VIEWED_FILE = 3;
+					INTENT_OUTLINE = 4;
+					INTENT_MENTIONED_FILE = 5;
+				}
+			string relative_workspace_path = 1;
+			int32 start_line_number = 2;
+			repeated string lines = 3;
+			optional SummarizationStrategy summarization_strategy = 4;
+			string language_identifier = 5;
+			optional Intent intent = 6;
+			optional bool is_final_version = 7;
+			optional bool is_first_version = 8;
+			optional bool contents_are_missing = 9;
+		}
+		message ApproximateLintError { // aiserver.v1.ConversationMessage.ApproximateLintError
+			string message = 1;
+			string value = 2;
+			int32 start_line = 3;
+			int32 end_line = 4;
+			int32 start_column = 5;
+			int32 end_column = 6;
+		}
+		message Lints { // aiserver.v1.ConversationMessage.Lints
+			GetLintsForChangeResponse lints = 1;
+			string chat_codeblock_model_value = 2;
+		}
+		message ToolResult { // aiserver.v1.ConversationMessage.ToolResult
+				message CodeChunk { // aiserver.v1.ConversationMessage.CodeChunk
+						enum SummarizationStrategy { // aiserver.v1.ConversationMessage.CodeChunk.SummarizationStrategy
+							SUMMARIZATION_STRATEGY_NONE_UNSPECIFIED = 0;
+							SUMMARIZATION_STRATEGY_SUMMARIZED = 1;
+							SUMMARIZATION_STRATEGY_EMBEDDED = 2;
+						}
+						enum Intent { // aiserver.v1.ConversationMessage.CodeChunk.Intent
+							INTENT_UNSPECIFIED = 0;
+							INTENT_COMPOSER_FILE = 1;
+							INTENT_COMPRESSED_COMPOSER_FILE = 2;
+							INTENT_RECENTLY_VIEWED_FILE = 3;
+							INTENT_OUTLINE = 4;
+							INTENT_MENTIONED_FILE = 5;
+						}
+					string relative_workspace_path = 1;
+					int32 start_line_number = 2;
+					repeated string lines = 3;
+					optional SummarizationStrategy summarization_strategy = 4;
+					string language_identifier = 5;
+					optional Intent intent = 6;
+					optional bool is_final_version = 7;
+					optional bool is_first_version = 8;
+					optional bool contents_are_missing = 9;
+				}
+			string tool_call_id = 1;
+			string tool_name = 2;
+			uint32 tool_index = 3;
+			string args = 4;
+			string raw_args = 5;
+			repeated CodeChunk attached_code_chunks = 6;
+			optional string content = 7;
+			ClientSideToolV2Result result = 8;
+			optional ToolResultError error = 9;
+		}
+		message NotepadContext { // aiserver.v1.ConversationMessage.NotepadContext
+				message CodeChunk { // aiserver.v1.ConversationMessage.CodeChunk
+						enum SummarizationStrategy { // aiserver.v1.ConversationMessage.CodeChunk.SummarizationStrategy
+							SUMMARIZATION_STRATEGY_NONE_UNSPECIFIED = 0;
+							SUMMARIZATION_STRATEGY_SUMMARIZED = 1;
+							SUMMARIZATION_STRATEGY_EMBEDDED = 2;
+						}
+						enum Intent { // aiserver.v1.ConversationMessage.CodeChunk.Intent
+							INTENT_UNSPECIFIED = 0;
+							INTENT_COMPOSER_FILE = 1;
+							INTENT_COMPRESSED_COMPOSER_FILE = 2;
+							INTENT_RECENTLY_VIEWED_FILE = 3;
+							INTENT_OUTLINE = 4;
+							INTENT_MENTIONED_FILE = 5;
+						}
+					string relative_workspace_path = 1;
+					int32 start_line_number = 2;
+					repeated string lines = 3;
+					optional SummarizationStrategy summarization_strategy = 4;
+					string language_identifier = 5;
+					optional Intent intent = 6;
+					optional bool is_final_version = 7;
+					optional bool is_first_version = 8;
+					optional bool contents_are_missing = 9;
+				}
+			string name = 1;
+			string text = 2;
+			repeated CodeChunk attached_code_chunks = 3;
+			repeated string attached_folders = 4;
+			repeated Commit commits = 5;
+			repeated PullRequest pull_requests = 6;
+			repeated GitDiff git_diffs = 7;
+			repeated ImageProto images = 8;
+		}
+		message EditTrailContext { // aiserver.v1.ConversationMessage.EditTrailContext
+				message EditLocation { // aiserver.v1.ConversationMessage.EditLocation
+					string relative_workspace_path = 1;
+					SimplestRange range = 3;
+					SimplestRange initial_range = 4;
+					string context_lines = 5;
+					string text = 6;
+					SimplestRange text_range = 7;
+				}
+			string unique_id = 1;
+			repeated EditLocation edit_trail_sorted = 2;
+		}
+		message RecentLocation { // aiserver.v1.ConversationMessage.RecentLocation
+			string relative_workspace_path = 1;
+			int32 line_number = 2;
+		}
+	string text = 1;
+	MessageType type = 2;
+	repeated CodeChunk attached_code_chunks = 3;
+	repeated CodeBlock codebase_context_chunks = 4;
+	repeated Commit commits = 5;
+	repeated PullRequest pull_requests = 6;
+	repeated GitDiff git_diffs = 7;
+	repeated SimpleFileDiff assistant_suggested_diffs = 8;
+	repeated InterpreterResult interpreter_results = 9;
+	repeated ImageProto images = 10;
+	repeated string attached_folders = 11;
+	repeated ApproximateLintError approximate_lint_errors = 12;
+	string bubble_id = 13;
+	optional string server_bubble_id = 32;
+	repeated FolderInfo attached_folders_new = 14;
+	repeated Lints lints = 15;
+	repeated UserResponseToSuggestedCodeBlock user_responses_to_suggested_code_blocks = 16;
+	repeated string relevant_files = 17;
+	repeated ToolResult tool_results = 18;
+	repeated NotepadContext notepads = 19;
+	optional bool is_capability_iteration = 20;
+	repeated ComposerCapabilityRequest capabilities = 21;
+	repeated EditTrailContext edit_trail_contexts = 22;
+	repeated SuggestedCodeBlock suggested_code_blocks = 23;
+	repeated RedDiff diffs_for_compressing_files = 24;
+	repeated LinterErrorsWithoutFileContents multi_file_linter_errors = 25;
+	repeated DiffHistoryData diff_histories = 26;
+	repeated CodeChunk recently_viewed_files = 27;
+	repeated RecentLocation recent_locations_history = 28;
+	bool is_agentic = 29;
+	repeated ComposerFileDiffHistory file_diff_trajectories = 30;
+	optional ConversationSummary conversation_summary = 31;
+}
+message CodeBlock { // aiserver.v1.CodeBlock
+		message Signatures { // aiserver.v1.CodeBlock.Signatures
+			repeated CursorRange ranges = 1;
+		}
+	string relative_workspace_path = 1;
+	optional string file_contents = 2;
+	CursorRange range = 3;
+	string contents = 4;
+	Signatures signatures = 5;
+	optional string override_contents = 6;
+	optional string original_contents = 7;
+	repeated DetailedLine detailed_lines = 8;
+}
+message DetailedLine { // aiserver.v1.DetailedLine
+	string text = 1;
+	float line_number = 2;
+	bool is_signature = 3;
+}
+message Commit { // aiserver.v1.Commit
+	string sha = 1;
+	string message = 2;
+	string description = 3;
+	repeated FileDiff diff = 4;
+	string author = 5;
+	string date = 6;
+}
+message FileDiff { // aiserver.v1.FileDiff
+		message Chunk { // aiserver.v1.FileDiff.Chunk
+			string content = 1;
+			repeated string lines = 2;
+			int32 old_start = 3;
+			int32 old_lines = 4;
+			int32 new_start = 5;
+			int32 new_lines = 6;
+		}
+	string from = 1;
+	string to = 2;
+	repeated Chunk chunks = 3;
+}
+message PullRequest { // aiserver.v1.PullRequest
+	string title = 1;
+	string body = 2;
+	repeated FileDiff diff = 3;
+}
+message GitDiff { // aiserver.v1.GitDiff
+		enum DiffType { // aiserver.v1.GitDiff.DiffType
+			DIFF_TYPE_UNSPECIFIED = 0;
+			DIFF_TYPE_DIFF_TO_HEAD = 1;
+			DIFF_TYPE_DIFF_FROM_BRANCH_TO_MAIN = 2;
+		}
+	repeated FileDiff diffs = 1;
+	DiffType diff_type = 2;
+}
+message SimpleFileDiff { // aiserver.v1.SimpleFileDiff
+		message Chunk { // aiserver.v1.SimpleFileDiff.Chunk
+			repeated string old_lines = 1;
+			repeated string new_lines = 2;
+			LineRange old_range = 3;
+			LineRange new_range = 4;
+		}
+	string relative_workspace_path = 1;
+	repeated Chunk chunks = 3;
+}
+message LineRange { // aiserver.v1.LineRange
+	int32 start_line_number = 1;
+	int32 end_line_number_inclusive = 2;
+}
+message InterpreterResult { // aiserver.v1.InterpreterResult
+	string output = 1;
+	bool success = 2;
+}
+message ImageProto { // aiserver.v1.ImageProto
+		message Dimension { // aiserver.v1.ImageProto.Dimension
+			int32 width = 1;
+			int32 height = 2;
+		}
+	bytes data = 1;
+	Dimension dimension = 2;
+}
+message FolderInfo { // aiserver.v1.FolderInfo
+	string relative_path = 1;
+	repeated FolderFileInfo files = 2;
+}
+message FolderFileInfo { // aiserver.v1.FolderFileInfo
+	string relative_path = 1;
+	string content = 2;
+	bool truncated = 3;
+	float score = 4;
+}
+message GetLintsForChangeResponse { // aiserver.v1.GetLintsForChangeResponse
+		message Lint { // aiserver.v1.GetLintsForChangeResponse.Lint
+				message QuickFix { // aiserver.v1.GetLintsForChangeResponse.Lint.QuickFix
+						message Edit { // aiserver.v1.GetLintsForChangeResponse.Lint.QuickFix.Edit
+							string relative_workspace_path = 1;
+							string text = 2;
+							int32 start_line_number_one_indexed = 3;
+							int32 start_column_one_indexed = 4;
+							int32 end_line_number_inclusive_one_indexed = 5;
+							int32 end_column_one_indexed = 6;
+						}
+					string message = 1;
+					string kind = 2;
+					bool is_preferred = 3;
+					repeated Edit edits = 4;
+				}
+			string message = 1;
+			string severity = 2;
+			string relative_workspace_path = 3;
+			int32 start_line_number_one_indexed = 4;
+			int32 start_column_one_indexed = 5;
+			int32 end_line_number_inclusive_one_indexed = 6;
+			int32 end_column_one_indexed = 7;
+			repeated QuickFix quick_fixes = 9;
+		}
+	repeated Lint lints = 1;
+}
+message UserResponseToSuggestedCodeBlock { // aiserver.v1.UserResponseToSuggestedCodeBlock
+		enum UserResponseType { // aiserver.v1.UserResponseToSuggestedCodeBlock.UserResponseType
+			USER_RESPONSE_TYPE_UNSPECIFIED = 0;
+			USER_RESPONSE_TYPE_ACCEPT = 1;
+			USER_RESPONSE_TYPE_REJECT = 2;
+			USER_RESPONSE_TYPE_MODIFY = 3;
+		}
+	UserResponseType user_response_type = 1;
+	string file_path = 2;
+	optional FileDiff user_modifications_to_suggested_code_blocks = 3;
+}
+message ClientSideToolV2Result { // aiserver.v1.ClientSideToolV2Result
+	ClientSideToolV2 tool = 1;
+	ReadSemsearchFilesResult read_semsearch_files_result = 2;
+	ReadFileForImportsResult read_file_for_imports_result = 3;
+	RipgrepSearchResult ripgrep_search_result = 4;
+	RunTerminalCommandResult run_terminal_command_result = 5;
+	ReadFileResult read_file_result = 6;
+	ListDirResult list_dir_result = 9;
+	EditFileResult edit_file_result = 10;
+	ToolCallFileSearchResult file_search_result = 11;
+	SemanticSearchFullResult semantic_search_full_result = 18;
+	CreateFileResult create_file_result = 19;
+	DeleteFileResult delete_file_result = 20;
+	optional ToolResultError error = 8;
+}
+message ReadSemsearchFilesResult { // aiserver.v1.ReadSemsearchFilesResult
+	repeated CodeResult code_results = 1;
+}
+message CodeResult { // aiserver.v1.CodeResult
+	CodeBlock code_block = 1;
+	float score = 2;
+}
+message ReadFileForImportsResult { // aiserver.v1.ReadFileForImportsResult
+	string contents = 1;
+}
+message RipgrepSearchResult { // aiserver.v1.RipgrepSearchResult
+	RipgrepSearchResultInternal internal = 1;
+}
+message RipgrepSearchResultInternal { // aiserver.v1.RipgrepSearchResultInternal
+		message IFileMatch { // aiserver.v1.RipgrepSearchResultInternal.IFileMatch
+				message ITextSearchResult { // aiserver.v1.RipgrepSearchResultInternal.ITextSearchResult
+						message ITextSearchMatch { // aiserver.v1.RipgrepSearchResultInternal.ITextSearchMatch
+								message ISearchRangeSetPairing { // aiserver.v1.RipgrepSearchResultInternal.ISearchRangeSetPairing
+										message ISearchRange { // aiserver.v1.RipgrepSearchResultInternal.ISearchRange
+											int32 start_line_number = 1;
+											int32 start_column = 2;
+											int32 end_line_number = 3;
+											int32 end_column = 4;
+										}
+									ISearchRange source = 1;
+									ISearchRange preview = 2;
+								}
+							optional string uri = 1;
+							repeated ISearchRangeSetPairing range_locations = 2;
+							string preview_text = 3;
+							optional int32 webview_index = 4;
+							optional string cell_fragment = 5;
+						}
+						message ITextSearchContext { // aiserver.v1.RipgrepSearchResultInternal.ITextSearchContext
+							optional string uri = 1;
+							string text = 2;
+							int32 line_number = 3;
+						}
+					ITextSearchMatch match = 1;
+					ITextSearchContext context = 2;
+				}
+			string resource = 1;
+			repeated ITextSearchResult results = 2;
+		}
+		enum SearchCompletionExitCode { // aiserver.v1.RipgrepSearchResultInternal.SearchCompletionExitCode
+			SEARCH_COMPLETION_EXIT_CODE_UNSPECIFIED = 0;
+			SEARCH_COMPLETION_EXIT_CODE_NORMAL = 1;
+			SEARCH_COMPLETION_EXIT_CODE_NEW_SEARCH_STARTED = 2;
+		}
+		message ITextSearchCompleteMessage { // aiserver.v1.RipgrepSearchResultInternal.ITextSearchCompleteMessage
+				enum TextSearchCompleteMessageType { // aiserver.v1.RipgrepSearchResultInternal.TextSearchCompleteMessageType
+					TEXT_SEARCH_COMPLETE_MESSAGE_TYPE_UNSPECIFIED = 0;
+					TEXT_SEARCH_COMPLETE_MESSAGE_TYPE_INFORMATION = 1;
+					TEXT_SEARCH_COMPLETE_MESSAGE_TYPE_WARNING = 2;
+				}
+			string text = 1;
+			TextSearchCompleteMessageType type = 2;
+			optional bool trusted = 3;
+		}
+		message IFileSearchStats { // aiserver.v1.RipgrepSearchResultInternal.IFileSearchStats
+				message ISearchEngineStats { // aiserver.v1.RipgrepSearchResultInternal.ISearchEngineStats
+					int32 file_walk_time = 1;
+					int32 directories_walked = 2;
+					int32 files_walked = 3;
+					int32 cmd_time = 4;
+					optional int32 cmd_result_count = 5;
+				}
+				message ICachedSearchStats { // aiserver.v1.RipgrepSearchResultInternal.ICachedSearchStats
+					bool cache_was_resolved = 1;
+					int32 cache_lookup_time = 2;
+					int32 cache_filter_time = 3;
+					int32 cache_entry_count = 4;
+				}
+				message IFileSearchProviderStats { // aiserver.v1.RipgrepSearchResultInternal.IFileSearchProviderStats
+					int32 provider_time = 1;
+					int32 post_process_time = 2;
+				}
+				enum FileSearchProviderType { // aiserver.v1.RipgrepSearchResultInternal.IFileSearchStats.FileSearchProviderType
+					FILE_SEARCH_PROVIDER_TYPE_UNSPECIFIED = 0;
+					FILE_SEARCH_PROVIDER_TYPE_FILE_SEARCH_PROVIDER = 1;
+					FILE_SEARCH_PROVIDER_TYPE_SEARCH_PROCESS = 2;
+				}
+			bool from_cache = 1;
+			ISearchEngineStats search_engine_stats = 2;
+			ICachedSearchStats cached_search_stats = 3;
+			IFileSearchProviderStats file_search_provider_stats = 4;
+			int32 result_count = 5;
+			FileSearchProviderType type = 6;
+			optional int32 sorting_time = 7;
+		}
+		message ITextSearchStats { // aiserver.v1.RipgrepSearchResultInternal.ITextSearchStats
+				enum TextSearchProviderType { // aiserver.v1.RipgrepSearchResultInternal.ITextSearchStats.TextSearchProviderType
+					TEXT_SEARCH_PROVIDER_TYPE_UNSPECIFIED = 0;
+					TEXT_SEARCH_PROVIDER_TYPE_TEXT_SEARCH_PROVIDER = 1;
+					TEXT_SEARCH_PROVIDER_TYPE_SEARCH_PROCESS = 2;
+					TEXT_SEARCH_PROVIDER_TYPE_AI_TEXT_SEARCH_PROVIDER = 3;
+				}
+			TextSearchProviderType type = 1;
+		}
+	repeated IFileMatch results = 1;
+	optional SearchCompletionExitCode exit = 2;
+	optional bool limit_hit = 3;
+	repeated ITextSearchCompleteMessage messages = 4;
+	IFileSearchStats file_search_stats = 5;
+	ITextSearchStats text_search_stats = 6;
+}
+message RunTerminalCommandResult { // aiserver.v1.RunTerminalCommandResult
+	string output = 1;
+	int32 exit_code = 2;
+	optional bool rejected = 3;
+	bool popped_out_into_background = 4;
+}
+message ReadFileResult { // aiserver.v1.ReadFileResult
+	string contents = 1;
+	bool did_downgrade_to_line_range = 2;
+	bool did_shorten_line_range = 3;
+	bool did_set_default_line_range = 4;
+	optional string full_file_contents = 5;
+	optional string outline = 6;
+	optional int32 start_line_one_indexed = 7;
+	optional int32 end_line_one_indexed_inclusive = 8;
+	string relative_workspace_path = 9;
+	bool did_shorten_char_range = 10;
+}
+message ListDirResult { // aiserver.v1.ListDirResult
+		message File { // aiserver.v1.ListDirResult.File
+				message Timestamp { // google.protobuf.Timestamp
+					int64 seconds = 1;
+					int32 nanos = 2;
+				}
+			string name = 1;
+			bool is_directory = 2;
+			optional int64 size = 3;
+			optional Timestamp last_modified = 4;
+			optional int32 num_children = 5;
+			optional int32 num_lines = 6;
+		}
+	repeated File files = 1;
+	string directory_relative_workspace_path = 2;
+}
+message EditFileResult { // aiserver.v1.EditFileResult
+		message FileDiff { // aiserver.v1.EditFileResult.FileDiff
+				message ChunkDiff { // aiserver.v1.EditFileResult.FileDiff.ChunkDiff
+					string diff_string = 1;
+					int32 old_start = 2;
+					int32 new_start = 3;
+					int32 old_lines = 4;
+					int32 new_lines = 5;
+					int32 lines_removed = 6;
+					int32 lines_added = 7;
+				}
+				enum Editor { // aiserver.v1.EditFileResult.FileDiff.Editor
+					EDITOR_UNSPECIFIED = 0;
+					EDITOR_AI = 1;
+					EDITOR_HUMAN = 2;
+				}
+			repeated ChunkDiff chunks = 1;
+			Editor editor = 2;
+			bool hit_timeout = 3;
+		}
+	FileDiff diff = 1;
+	bool is_applied = 2;
+	bool apply_failed = 3;
+}
+message ToolCallFileSearchResult { // aiserver.v1.ToolCallFileSearchResult
+		message File { // aiserver.v1.ToolCallFileSearchResult.File
+			string uri = 1;
+		}
+	repeated File files = 1;
+	optional bool limit_hit = 2;
+	int32 num_results = 3;
+}
+message SemanticSearchFullResult { // aiserver.v1.SemanticSearchFullResult
+	repeated CodeResult code_results = 1;
+}
+message CreateFileResult { // aiserver.v1.CreateFileResult
+	bool file_created_successfully = 1;
+	bool file_already_exists = 2;
+}
+message DeleteFileResult { // aiserver.v1.DeleteFileResult
+	bool rejected = 1;
+	bool file_non_existent = 2;
+	bool file_deleted_successfully = 3;
+}
+message ToolResultError { // aiserver.v1.ToolResultError
+	string client_visible_error_message = 1;
+	string model_visible_error_message = 2;
+}
+message ComposerCapabilityRequest { // aiserver.v1.ComposerCapabilityRequest
+		enum ComposerCapabilityType { // aiserver.v1.ComposerCapabilityRequest.ComposerCapabilityType
+			COMPOSER_CAPABILITY_TYPE_UNSPECIFIED = 0;
+			COMPOSER_CAPABILITY_TYPE_LOOP_ON_LINTS = 1;
+			COMPOSER_CAPABILITY_TYPE_LOOP_ON_TESTS = 2;
+			COMPOSER_CAPABILITY_TYPE_MEGA_PLANNER = 3;
+			COMPOSER_CAPABILITY_TYPE_LOOP_ON_COMMAND = 4;
+			COMPOSER_CAPABILITY_TYPE_TOOL_CALL = 5;
+			COMPOSER_CAPABILITY_TYPE_DIFF_REVIEW = 6;
+			COMPOSER_CAPABILITY_TYPE_CONTEXT_PICKING = 7;
+			COMPOSER_CAPABILITY_TYPE_EDIT_TRAIL = 8;
+			COMPOSER_CAPABILITY_TYPE_AUTO_CONTEXT = 9;
+			COMPOSER_CAPABILITY_TYPE_CONTEXT_PLANNER = 10;
+			COMPOSER_CAPABILITY_TYPE_DIFF_HISTORY = 11;
+			COMPOSER_CAPABILITY_TYPE_REMEMBER_THIS = 12;
+			COMPOSER_CAPABILITY_TYPE_DECOMPOSER = 13;
+			COMPOSER_CAPABILITY_TYPE_USES_CODEBASE = 14;
+			COMPOSER_CAPABILITY_TYPE_TOOL_FORMER = 15;
+		}
+		message LoopOnLintsCapability { // aiserver.v1.ComposerCapabilityRequest.LoopOnLintsCapability
+			repeated LinterErrors linter_errors = 1;
+			optional string custom_instructions = 2;
+		}
+		message LoopOnTestsCapability { // aiserver.v1.ComposerCapabilityRequest.LoopOnTestsCapability
+			repeated string test_names = 1;
+			optional string custom_instructions = 2;
+		}
+		message MegaPlannerCapability { // aiserver.v1.ComposerCapabilityRequest.MegaPlannerCapability
+			optional string custom_instructions = 1;
+		}
+		message LoopOnCommandCapability { // aiserver.v1.ComposerCapabilityRequest.LoopOnCommandCapability
+			string command = 1;
+			optional string custom_instructions = 2;
+			optional string output = 3;
+			optional int32 exit_code = 4;
+		}
+		message ToolCallCapability { // aiserver.v1.ComposerCapabilityRequest.ToolCallCapability
+				message ToolSchema { // aiserver.v1.ComposerCapabilityRequest.ToolSchema
+						enum ToolType { // aiserver.v1.ComposerCapabilityRequest.ToolType
+							TOOL_TYPE_UNSPECIFIED = 0;
+							TOOL_TYPE_ADD_FILE_TO_CONTEXT = 1;
+							TOOL_TYPE_RUN_TERMINAL_COMMAND = 2;
+							TOOL_TYPE_ITERATE = 3;
+							TOOL_TYPE_REMOVE_FILE_FROM_CONTEXT = 4;
+							TOOL_TYPE_SEMANTIC_SEARCH_CODEBASE = 5;
+						}
+					ToolType type = 1;
+					string name = 2;
+					repeated string required = 4;
+				}
+			optional string custom_instructions = 1;
+			repeated ToolSchema tool_schemas = 2;
+			repeated string relevant_files = 3;
+			repeated string files_in_context = 4;
+			repeated string semantic_search_files = 5;
+		}
+		message DiffReviewCapability { // aiserver.v1.ComposerCapabilityRequest.DiffReviewCapability
+				message SimpleFileDiff { // aiserver.v1.ComposerCapabilityRequest.DiffReviewCapability.SimpleFileDiff
+						message Chunk { // aiserver.v1.ComposerCapabilityRequest.DiffReviewCapability.SimpleFileDiff.Chunk
+							repeated string old_lines = 1;
+							repeated string new_lines = 2;
+							LineRange old_range = 3;
+							LineRange new_range = 4;
+						}
+					string relative_workspace_path = 1;
+					repeated Chunk chunks = 3;
+				}
+			optional string custom_instructions = 1;
+			repeated SimpleFileDiff diffs = 2;
+		}
+		message ContextPickingCapability { // aiserver.v1.ComposerCapabilityRequest.ContextPickingCapability
+			optional string custom_instructions = 1;
+			repeated string potential_context_files = 2;
+			repeated CodeChunk potential_context_code_chunks = 3;
+			repeated string files_in_context = 4;
+		}
+		message EditTrailCapability { // aiserver.v1.ComposerCapabilityRequest.EditTrailCapability
+			optional string custom_instructions = 1;
+		}
+		message AutoContextCapability { // aiserver.v1.ComposerCapabilityRequest.AutoContextCapability
+			optional string custom_instructions = 1;
+			repeated string additional_files = 2;
+		}
+		message ContextPlannerCapability { // aiserver.v1.ComposerCapabilityRequest.ContextPlannerCapability
+			optional string custom_instructions = 1;
+			repeated CodeChunk attached_code_chunks = 2;
+		}
+		message RememberThisCapability { // aiserver.v1.ComposerCapabilityRequest.RememberThisCapability
+			optional string custom_instructions = 1;
+			string memory = 2;
+		}
+		message DecomposerCapability { // aiserver.v1.ComposerCapabilityRequest.DecomposerCapability
+			optional string custom_instructions = 1;
+		}
+	ComposerCapabilityType type = 1;
+	LoopOnLintsCapability loop_on_lints = 2;
+	LoopOnTestsCapability loop_on_tests = 3;
+	MegaPlannerCapability mega_planner = 4;
+	LoopOnCommandCapability loop_on_command = 5;
+	ToolCallCapability tool_call = 6;
+	DiffReviewCapability diff_review = 7;
+	ContextPickingCapability context_picking = 8;
+	EditTrailCapability edit_trail = 9;
+	AutoContextCapability auto_context = 10;
+	ContextPlannerCapability context_planner = 11;
+	RememberThisCapability remember_this = 12;
+	DecomposerCapability decomposer = 13;
+}
+message LinterErrors { // aiserver.v1.LinterErrors
+	string relative_workspace_path = 1;
+	repeated LinterError errors = 2;
+	string file_contents = 3;
+}
+message LinterError { // aiserver.v1.LinterError
+		message RelatedInformation { // aiserver.v1.Diagnostic.RelatedInformation
+			string message = 1;
+			CursorRange range = 2;
+		}
+		enum DiagnosticSeverity { // aiserver.v1.Diagnostic.DiagnosticSeverity
+			DIAGNOSTIC_SEVERITY_UNSPECIFIED = 0;
+			DIAGNOSTIC_SEVERITY_ERROR = 1;
+			DIAGNOSTIC_SEVERITY_WARNING = 2;
+			DIAGNOSTIC_SEVERITY_INFORMATION = 3;
+			DIAGNOSTIC_SEVERITY_HINT = 4;
+		}
+	string message = 1;
+	CursorRange range = 2;
+	optional string source = 3;
+	repeated RelatedInformation related_information = 4;
+	optional DiagnosticSeverity severity = 5;
+}
+message CodeChunk { // aiserver.v1.CodeChunk
+		enum SummarizationStrategy { // aiserver.v1.CodeChunk.SummarizationStrategy
+			SUMMARIZATION_STRATEGY_NONE_UNSPECIFIED = 0;
+			SUMMARIZATION_STRATEGY_SUMMARIZED = 1;
+			SUMMARIZATION_STRATEGY_EMBEDDED = 2;
+		}
+		enum Intent { // aiserver.v1.CodeChunk.Intent
+			INTENT_UNSPECIFIED = 0;
+			INTENT_COMPOSER_FILE = 1;
+			INTENT_COMPRESSED_COMPOSER_FILE = 2;
+		}
+	string relative_workspace_path = 1;
+	int32 start_line_number = 2;
+	repeated string lines = 3;
+	optional SummarizationStrategy summarization_strategy = 4;
+	string language_identifier = 5;
+	optional Intent intent = 6;
+	optional bool is_final_version = 7;
+	optional bool is_first_version = 8;
+}
+message SuggestedCodeBlock { // aiserver.v1.SuggestedCodeBlock
+	string relative_workspace_path = 1;
+}
+message RedDiff { // aiserver.v1.RedDiff
+	string relative_workspace_path = 1;
+	repeated SimplestRange red_ranges = 2;
+	repeated SimplestRange red_ranges_reversed = 3;
+	string start_hash = 4;
+	string end_hash = 5;
+}
+message LinterErrorsWithoutFileContents { // aiserver.v1.LinterErrorsWithoutFileContents
+	string relative_workspace_path = 1;
+	repeated LinterError errors = 2;
+}
+message DiffHistoryData { // aiserver.v1.DiffHistoryData
+	string relative_workspace_path = 1;
+	repeated ComposerFileDiff diffs = 2;
+	double timestamp = 3;
+	string unique_id = 4;
+	ComposerFileDiff start_to_end_diff = 5;
+}
+message ComposerFileDiff { // aiserver.v1.ComposerFileDiff
+		message ChunkDiff { // aiserver.v1.ComposerFileDiff.ChunkDiff
+			string diff_string = 1;
+			int32 old_start = 2;
+			int32 new_start = 3;
+			int32 old_lines = 4;
+			int32 new_lines = 5;
+			int32 lines_removed = 6;
+			int32 lines_added = 7;
+		}
+		enum Editor { // aiserver.v1.ComposerFileDiff.Editor
+			EDITOR_UNSPECIFIED = 0;
+			EDITOR_AI = 1;
+			EDITOR_HUMAN = 2;
+		}
+	repeated ChunkDiff chunks = 1;
+	Editor editor = 2;
+	bool hit_timeout = 3;
+}
+message ComposerFileDiffHistory { // aiserver.v1.ComposerFileDiffHistory
+	string file_name = 1;
+	repeated string diff_history = 2;
+	repeated double diff_history_timestamps = 3;
+}
+message ConversationSummary { // aiserver.v1.ConversationSummary
+	string summary = 1;
+	string truncation_last_bubble_id_inclusive = 2;
+	string client_should_start_sending_from_inclusive_bubble_id = 3;
+}
+message RepositoryInfo { // aiserver.v1.RepositoryInfo
+	string relative_workspace_path = 1;
+	repeated string remote_urls = 2;
+	repeated string remote_names = 3;
+	string repo_name = 4;
+	string repo_owner = 5;
+	bool is_tracked = 6;
+	bool is_local = 7;
+	optional int32 num_files = 8;
+	optional double orthogonal_transform_seed = 9;
+	optional EmbeddingModel preferred_embedding_model = 10;
+}
+message ExplicitContext { // aiserver.v1.ExplicitContext
+	string context = 1;
+	optional string repo_context = 2;
+}
+message ModelDetails { // aiserver.v1.ModelDetails
+	optional string model_name = 1;
+	optional string api_key = 2;
+	optional bool enable_ghost_mode = 3;
+	optional AzureState azure_state = 4;
+	optional bool enable_slow_pool = 5;
+	optional string openai_api_base_url = 6;
+}
+message AzureState { // aiserver.v1.AzureState
+	string api_key = 1;
+	string base_url = 2;
+	string deployment = 3;
+	bool use_azure = 4;
+}
+message ChatQuote { // aiserver.v1.ChatQuote
+	string markdown = 1;
+	string bubble_id = 2;
+	int32 section_index = 3;
+}
+message DebugInfo { // aiserver.v1.DebugInfo
+		message Breakpoint { // aiserver.v1.DebugInfo.Breakpoint
+			string relative_workspace_path = 1;
+			int32 line_number = 2;
+			repeated string lines_before_breakpoint = 3;
+			repeated string lines_after_breakpoint = 4;
+			optional string exception_info = 5;
+		}
+		message CallStackFrame { // aiserver.v1.DebugInfo.CallStackFrame
+				message Scope { // aiserver.v1.DebugInfo.Scope
+						message Variable { // aiserver.v1.DebugInfo.Variable
+							string name = 1;
+							string value = 2;
+							optional string type = 3;
+						}
+					string name = 1;
+					repeated Variable variables = 2;
+				}
+			string relative_workspace_path = 1;
+			int32 line_number = 2;
+			string function_name = 3;
+			repeated Scope scopes = 4;
+		}
+	Breakpoint breakpoint = 1;
+	repeated CallStackFrame call_stack = 2;
+	repeated CodeBlock history = 3;
+}
+message ChatExternalLink { // aiserver.v1.ChatExternalLink
+	string url = 1;
+	string uuid = 2;
+}
+message CommitNote { // aiserver.v1.CommitNote
+	string note = 1;
+	string commit_hash = 2;
+}
+message ContextAST { // aiserver.v1.ContextAST
+	repeated ContainerTree files = 1;
+}
+message ContainerTree { // aiserver.v1.ContainerTree
+	string relative_workspace_path = 1;
+	repeated ContainerTreeNode nodes = 2;
+}
+message ContainerTreeNode { // aiserver.v1.ContainerTreeNode
+		message Container { // aiserver.v1.ContainerTreeNode.Container
+				message Reference { // aiserver.v1.ContainerTreeNode.Reference
+					string value = 1;
+					string relative_workspace_path = 2;
+				}
+			string doc_string = 1;
+			string header = 2;
+			string trailer = 3;
+			repeated ContainerTreeNode children = 5;
+			repeated Reference references = 6;
+			double score = 7;
+		}
+		message Blob { // aiserver.v1.ContainerTreeNode.Blob
+			optional string value = 1;
+		}
+		message Symbol { // aiserver.v1.ContainerTreeNode.Symbol
+				message Reference { // aiserver.v1.ContainerTreeNode.Reference
+					string value = 1;
+					string relative_workspace_path = 2;
+				}
+			string doc_string = 1;
+			string value = 2;
+			repeated Reference references = 6;
+			double score = 7;
+		}
+	Container container = 1;
+	Blob blob = 2;
+	Symbol symbol = 3;
+}
+message StreamChatResponse { // aiserver.v1.StreamChatResponse
+		message ChunkIdentity { // aiserver.v1.StreamChatResponse.ChunkIdentity
+			string file_name = 1;
+			int32 start_line = 2;
+			int32 end_line = 3;
+			string text = 4;
+			ChunkType chunk_type = 5;
+		}
+	string text = 1;
+	optional string server_bubble_id = 22;
+	optional string debugging_only_chat_prompt = 2;
+	optional int32 debugging_only_token_count = 3;
+	DocumentationCitation document_citation = 4;
+	optional string filled_prompt = 5;
+	optional bool is_big_file = 6;
+	optional string intermediate_text = 7;
+	optional bool is_using_slow_request = 10;
+	optional ChunkIdentity chunk_identity = 8;
+	optional DocsReference docs_reference = 9;
+	optional WebCitation web_citation = 11;
+	optional StatusUpdates status_updates = 12;
+	optional ServerTimingInfo timing_info = 13;
+	optional SymbolLink symbol_link = 14;
+	optional FileLink file_link = 15;
+	optional ConversationSummary conversation_summary = 16;
+	optional ServiceStatusUpdate service_status_update = 17;
+}
+message DocumentationCitation { // aiserver.v1.DocumentationCitation
+	repeated DocumentationChunk chunks = 1;
+}
+message DocumentationChunk { // aiserver.v1.DocumentationChunk
+	string doc_name = 1;
+	string page_url = 2;
+	string documentation_chunk = 3;
+	float score = 4;
+	string page_title = 5;
+}
+message DocsReference { // aiserver.v1.DocsReference
+	string title = 1;
+	string url = 2;
+}
+message WebCitation { // aiserver.v1.WebCitation
+	repeated WebReference references = 1;
+}
+message WebReference { // aiserver.v1.WebReference
+	string title = 2;
+	string url = 1;
+}
+message StatusUpdates { // aiserver.v1.StatusUpdates
+	repeated StatusUpdate updates = 1;
+}
+message StatusUpdate { // aiserver.v1.StatusUpdate
+	string message = 1;
+	optional string metadata = 2;
+}
+message ServerTimingInfo { // aiserver.v1.ServerTimingInfo
+	double server_start_time = 1;
+	double server_first_token_time = 2;
+	double server_request_sent_time = 3;
+	double server_end_time = 4;
+}
+message SymbolLink { // aiserver.v1.SymbolLink
+	string symbol_name = 1;
+	string symbol_search_string = 2;
+	string relative_workspace_path = 3;
+	int32 rough_line_number = 4;
+}
+message FileLink { // aiserver.v1.FileLink
+	string display_name = 1;
+	string relative_workspace_path = 2;
+}
+message ServiceStatusUpdate { // aiserver.v1.ServiceStatusUpdate
+	string message = 1;
+	string codicon = 2;
+	optional bool allow_command_links_potentially_unsafe_please_only_use_for_handwritten_trusted_markdown = 3;
+}

src/chat/config.rs

@@ -0,0 +1,30 @@
+use crate::AppConfig;
+
+include!(concat!(env!("OUT_DIR"), "/key.rs"));
+
+impl KeyConfig {
+    pub fn new_with_global() -> Self {
+        Self {
+            auth_token: None,
+            disable_vision: Some(AppConfig::get_vision_ability().is_none()),
+            enable_slow_pool: Some(AppConfig::get_slow_pool()),
+            usage_check_models: None,
+            include_web_references: Some(AppConfig::get_web_refs()),
+        }
+    }
+
+    pub fn copy_without_auth_token(&self, config: &mut Self) {
+        if self.disable_vision.is_some() {
+            config.disable_vision = self.disable_vision;
+        }
+        if self.enable_slow_pool.is_some() {
+            config.enable_slow_pool = self.enable_slow_pool;
+        }
+        if self.usage_check_models.is_some() {
+            config.usage_check_models = self.usage_check_models.clone();
+        }
+        if self.include_web_references.is_some() {
+            config.include_web_references = self.include_web_references;
+        }
+    }
+}

src/chat/config/key.proto

@@ -0,0 +1,46 @@
+syntax = "proto3";
+
+package key;
+
+// 动态配置的 API KEY
+message KeyConfig {
+  // 认证令牌信息
+  message TokenInfo {
+    string sub = 1;        // 用户标识符
+    int64 exp = 2;         // 过期时间（Unix 时间戳）
+    string randomness = 3; // 随机字符串
+    string signature = 4;  // 签名
+    bytes machine_id = 5;  // 机器ID的SHA256哈希值
+    bytes mac_id = 6;      // MAC地址的SHA256哈希值
+  }
+
+  // 认证令牌（必需）
+  TokenInfo auth_token = 1;
+
+  // 是否禁用图片处理能力
+  optional bool disable_vision = 4;
+
+  // 是否启用慢速池
+  optional bool enable_slow_pool = 5;
+
+  // 使用量检查模型规则
+  message UsageCheckModel {
+    // 检查类型
+    enum Type {
+      TYPE_DEFAULT = 0;     // 未指定
+      TYPE_DISABLED = 1;    // 禁用
+      TYPE_ALL = 2;         // 全部
+      TYPE_CUSTOM = 3;      // 自定义列表
+    }
+    Type type = 1;                  // 检查类型
+    repeated string model_ids = 2; // 模型 ID 列表，当 type 为 TYPE_CUSTOM 时生效
+  }
+  // 使用量检查模型规则
+  optional UsageCheckModel usage_check_models = 6;
+
+  // 包含网络引用
+  optional bool include_web_references = 7;
+
+  // 密码SHA256哈希值
+  // bytes secret = 2;
+}

src/chat/constant.rs

@@ -0,0 +1,204 @@
+use super::model::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"
+);
+def_pub_const!(ERR_NODATA, "No data");
+
+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!(DEEPSEEK, "deepseek");
+
+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");
+def_pub_const!(DEEPSEEK_V3, "deepseek-v3");
+def_pub_const!(DEEPSEEK_R1, "deepseek-r1");
+
+// #[derive(Clone, PartialEq, rkyv::Archive, rkyv::Deserialize, rkyv::Serialize)]
+// pub enum ModelType {
+//     Claude35Sonnet,
+//     Gpt4,
+//     Gpt4o,
+//     Claude3Opus,
+//     CursorFast,
+//     CursorSmall,
+//     Gpt35Turbo,
+//     Gpt4Turbo202404,
+//     Gpt4o128k,
+//     Gemini15Flash500k,
+//     Claude3Haiku200k,
+//     Claude35Sonnet200k,
+//     Claude35Sonnet20241022,
+//     Gpt4oMini,
+//     O1Mini,
+//     O1Preview,
+//     O1,
+//     Claude35Haiku,
+//     GeminiExp1206,
+//     Gemini20FlashThinkingExp,
+//     Gemini20FlashExp,
+//     DeepseekV3,
+//     DeepseekR1,
+// }
+
+macro_rules! create_model {
+    ($($id:expr, $owner:expr),* $(,)?) => {
+        pub const AVAILABLE_MODELS: [Model; count!($( ($id, $owner) )*)] = [
+            $(
+                Model {
+                    id: $id,
+                    created: CREATED,
+                    object: MODEL_OBJECT,
+                    owned_by: $owner,
+                },
+            )*
+        ];
+    };
+}
+
+macro_rules! count {
+    () => (0);
+    (($id:expr, $owner:expr) $( ($id2:expr, $owner2:expr) )*) => (1 + count!($( ($id2, $owner2) )*));
+}
+
+// impl ModelType {
+//     pub fn as_str_name(&self) -> &'static str {
+//         match self {
+//             ModelType::Claude35Sonnet => CLAUDE_3_5_SONNET,
+//             ModelType::Gpt4 => GPT_4,
+//             ModelType::Gpt4o => GPT_4O,
+//             ModelType::Claude3Opus => CLAUDE_3_OPUS,
+//             ModelType::CursorFast => CURSOR_FAST,
+//             ModelType::CursorSmall => CURSOR_SMALL,
+//             ModelType::Gpt35Turbo => GPT_3_5_TURBO,
+//             ModelType::Gpt4Turbo202404 => GPT_4_TURBO_2024_04_09,
+//             ModelType::Gpt4o128k => GPT_4O_128K,
+//             ModelType::Gemini15Flash500k => GEMINI_1_5_FLASH_500K,
+//             ModelType::Claude3Haiku200k => CLAUDE_3_HAIKU_200K,
+//             ModelType::Claude35Sonnet200k => CLAUDE_3_5_SONNET_200K,
+//             ModelType::Claude35Sonnet20241022 => CLAUDE_3_5_SONNET_20241022,
+//             ModelType::Gpt4oMini => GPT_4O_MINI,
+//             ModelType::O1Mini => O1_MINI,
+//             ModelType::O1Preview => O1_PREVIEW,
+//             ModelType::O1 => O1,
+//             ModelType::Claude35Haiku => CLAUDE_3_5_HAIKU,
+//             ModelType::GeminiExp1206 => GEMINI_EXP_1206,
+//             ModelType::Gemini20FlashThinkingExp => GEMINI_2_0_FLASH_THINKING_EXP,
+//             ModelType::Gemini20FlashExp => GEMINI_2_0_FLASH_EXP,
+//             ModelType::DeepseekV3 => DEEPSEEK_V3,
+//             ModelType::DeepseekR1 => DEEPSEEK_R1,
+//         }
+//     }
+
+//     pub fn from_str_name(id :&str) -> Option<ModelType> {
+//         match id {
+//             CLAUDE_3_5_SONNET => Some(ModelType::Claude35Sonnet),
+//             GPT_4 => Some(ModelType::Gpt4),
+//             GPT_4O => Some(ModelType::Gpt4o),
+//             CLAUDE_3_OPUS => Some(ModelType::Claude3Opus),
+//             CURSOR_FAST => Some(ModelType::CursorFast),
+//             CURSOR_SMALL => Some(ModelType::CursorSmall),
+//             GPT_3_5_TURBO => Some(ModelType::Gpt35Turbo),
+//             GPT_4_TURBO_2024_04_09 => Some(ModelType::Gpt4Turbo202404),
+//             GPT_4O_128K => Some(ModelType::Gpt4o128k),
+//             GEMINI_1_5_FLASH_500K => Some(ModelType::Gemini15Flash500k),
+//             CLAUDE_3_HAIKU_200K => Some(ModelType::Claude3Haiku200k),
+//             CLAUDE_3_5_SONNET_200K => Some(ModelType::Claude35Sonnet200k),
+//             CLAUDE_3_5_SONNET_20241022 => Some(ModelType::Claude35Sonnet20241022),
+//             GPT_4O_MINI => Some(ModelType::Gpt4oMini),
+//             O1_MINI => Some(ModelType::O1Mini),
+//             O1_PREVIEW => Some(ModelType::O1Preview),
+//             O1 => Some(ModelType::O1),
+//             CLAUDE_3_5_HAIKU => Some(ModelType::Claude35Haiku),
+//             GEMINI_EXP_1206 => Some(ModelType::GeminiExp1206),
+//             GEMINI_2_0_FLASH_THINKING_EXP => Some(ModelType::Gemini20FlashThinkingExp),
+//             GEMINI_2_0_FLASH_EXP => Some(ModelType::Gemini20FlashExp),
+//             DEEPSEEK_V3 => Some(ModelType::DeepseekV3),
+//             DEEPSEEK_R1 => Some(ModelType::DeepseekR1),
+//             _ => None,
+//         }
+//     }
+// }
+
+create_model!(
+    CLAUDE_3_5_SONNET, ANTHROPIC,
+    GPT_4, OPENAI,
+    GPT_4O, OPENAI,
+    CLAUDE_3_OPUS, ANTHROPIC,
+    CURSOR_FAST, CURSOR,
+    CURSOR_SMALL, CURSOR,
+    GPT_3_5_TURBO, OPENAI,
+    GPT_4_TURBO_2024_04_09, OPENAI,
+    GPT_4O_128K, OPENAI,
+    GEMINI_1_5_FLASH_500K, GOOGLE,
+    CLAUDE_3_HAIKU_200K, ANTHROPIC,
+    CLAUDE_3_5_SONNET_200K, ANTHROPIC,
+    CLAUDE_3_5_SONNET_20241022, ANTHROPIC,
+    GPT_4O_MINI, OPENAI,
+    O1_MINI, OPENAI,
+    O1_PREVIEW, OPENAI,
+    O1, OPENAI,
+    CLAUDE_3_5_HAIKU, ANTHROPIC,
+    GEMINI_EXP_1206, GOOGLE,
+    GEMINI_2_0_FLASH_THINKING_EXP, GOOGLE,
+    GEMINI_2_0_FLASH_EXP, GOOGLE,
+    DEEPSEEK_V3, DEEPSEEK,
+    DEEPSEEK_R1, DEEPSEEK,
+);
+
+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,
+];
+
+// include!("constant/models.rs");

src/chat/constant/models.rs

@@ -0,0 +1,118 @@
+pub struct DefaultModel {
+    pub default_on: bool,
+    pub is_long_context_only: Option<bool>,
+    pub name: &'static str,
+}
+
+pub const AVAILABLE_MODELS2: [DefaultModel; 22] = [
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(false),
+        name: CLAUDE_3_5_SONNET,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: GPT_4,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: None,
+        name: GPT_4O,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: CLAUDE_3_OPUS,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: CURSOR_FAST,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: CURSOR_SMALL,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: GPT_3_5_TURBO,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: GPT_4_TURBO_2024_04_09,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(true),
+        name: GPT_4O_128K,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(true),
+        name: GEMINI_1_5_FLASH_500K,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(true),
+        name: CLAUDE_3_HAIKU_200K,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(true),
+        name: CLAUDE_3_5_SONNET_200K,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: Some(false),
+        name: CLAUDE_3_5_SONNET_20241022,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(false),
+        name: GPT_4O_MINI,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(false),
+        name: O1_MINI,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(false),
+        name: O1_PREVIEW,
+    },
+    DefaultModel {
+        default_on: true,
+        is_long_context_only: Some(false),
+        name: O1,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: Some(false),
+        name: CLAUDE_3_5_HAIKU,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: GEMINI_EXP_1206,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: GEMINI_2_0_FLASH_THINKING_EXP,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: GEMINI_2_0_FLASH_EXP,
+    },
+    DefaultModel {
+        default_on: false,
+        is_long_context_only: None,
+        name: DEEPSEEK_V3,
+    },
+];

src/chat/error.rs

@@ -0,0 +1,139 @@
+use super::aiserver::v1::ErrorDetails;
+use crate::common::model::{ApiStatus, ErrorResponse as CommonErrorResponse};
+use base64::{engine::general_purpose::STANDARD_NO_PAD, Engine as _};
+use prost::Message as _;
+use reqwest::StatusCode;
+use serde::{Deserialize, Serialize};
+
+#[derive(Deserialize)]
+pub struct ChatError {
+    error: ErrorBody,
+}
+
+#[derive(Deserialize)]
+pub struct ErrorBody {
+    code: String,
+    // message: String, always: Error
+    details: Vec<ErrorDetail>,
+}
+
+#[derive(Deserialize)]
+pub struct ErrorDetail {
+    // #[serde(rename = "type")]
+    // error_type: String, always: aiserver.v1.ErrorDetails
+    // debug: ErrorDebug,
+    value: String,
+}
+
+// #[derive(Deserialize)]
+// pub struct ErrorDebug {
+//     error: String,
+//     details: ErrorDetails,
+//     // #[serde(rename = "isExpected")]
+//     // is_expected: Option<bool>,
+// }
+
+// #[derive(Deserialize)]
+// pub struct ErrorDetails {
+//     title: String,
+//     detail: String,
+//     // #[serde(rename = "isRetryable")]
+//     // is_retryable: Option<bool>,
+// }
+
+impl ChatError {
+    pub fn to_error_response(self) -> ErrorResponse {
+        if self.error.details.is_empty() {
+            return ErrorResponse {
+                status: 500,
+                code: "unknown".to_string(),
+                error: None,
+            };
+        }
+
+        let error_details = self.error.details.first().and_then(|detail| {
+            STANDARD_NO_PAD
+                .decode(&detail.value)
+                .ok()
+                .map(bytes::Bytes::from)
+                .and_then(|buf| ErrorDetails::decode(buf).ok())
+        });
+
+        let status = error_details
+            .as_ref()
+            .map(|details| details.status_code())
+            .unwrap_or(500);
+
+        ErrorResponse {
+            status,
+            code: self.error.code,
+            error: error_details
+                .and_then(|details| details.details)
+                .map(|custom_details| Error {
+                    message: custom_details.title,
+                    details: custom_details.detail,
+                }),
+        }
+    }
+}
+
+#[derive(Serialize)]
+pub struct ErrorResponse {
+    pub status: u16,
+    pub code: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub error: Option<Error>,
+}
+
+#[derive(Serialize)]
+pub struct Error {
+    pub message: String,
+    pub details: String,
+    // pub value: String,
+}
+
+impl ErrorResponse {
+    // pub fn to_json(&self) -> serde_json::Value {
+    //     serde_json::to_value(self).unwrap()
+    // }
+
+    pub fn status_code(&self) -> StatusCode {
+        StatusCode::from_u16(self.status).unwrap()
+    }
+
+    pub fn native_code(&self) -> String {
+        self.error.as_ref().map_or_else(
+            || self.code.replace("_", " "),
+            |error| error.message.clone(),
+        )
+    }
+
+    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 {
+    ChatError(ChatError),
+    DataLengthLessThan5,
+    EmptyStream,
+}
+
+impl std::fmt::Display for StreamError {
+    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
+        match self {
+            StreamError::ChatError(error) => write!(f, "{}", error.error.code),
+            StreamError::DataLengthLessThan5 => write!(f, "data length less than 5"),
+            StreamError::EmptyStream => write!(f, "empty stream"),
+        }
+    }
+}

src/chat/middleware.rs

@@ -0,0 +1,2 @@
+mod auth;
+pub use auth::*;

src/chat/middleware/auth.rs

@@ -0,0 +1,23 @@
+use crate::app::{constant::AUTHORIZATION_BEARER_PREFIX, lazy::AUTH_TOKEN};
+use axum::{
+    body::Body,
+    http::{header::AUTHORIZATION, Request, StatusCode},
+    middleware::Next,
+    response::Response,
+};
+
+// 认证中间件函数
+pub async fn auth_middleware(request: Request<Body>, next: Next) -> Result<Response, StatusCode> {
+    let auth_header = request
+        .headers()
+        .get(AUTHORIZATION)
+        .and_then(|h| h.to_str().ok())
+        .and_then(|h| h.strip_prefix(AUTHORIZATION_BEARER_PREFIX))
+        .ok_or(StatusCode::UNAUTHORIZED)?;
+
+    if auth_header != AUTH_TOKEN.as_str() {
+        return Err(StatusCode::UNAUTHORIZED);
+    }
+
+    Ok(next.run(request).await)
+}

src/chat/model.rs

@@ -0,0 +1,107 @@
+use serde::{Deserialize, Serialize};
+
+#[derive(Serialize, Deserialize)]
+#[serde(untagged)]
+pub enum MessageContent {
+    Text(String),
+    Vision(Vec<VisionMessageContent>),
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct VisionMessageContent {
+    #[serde(rename = "type")]
+    pub content_type: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub text: Option<String>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub image_url: Option<ImageUrl>,
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct ImageUrl {
+    pub url: String,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub detail: Option<String>,
+}
+
+#[derive(Serialize, Deserialize)]
+pub struct Message {
+    pub role: Role,
+    pub content: MessageContent,
+}
+
+#[derive(Serialize, Deserialize, PartialEq)]
+pub enum Role {
+    #[serde(rename = "system", alias = "developer")]
+    System,
+    #[serde(rename = "user", alias = "human")]
+    User,
+    #[serde(rename = "assistant", alias = "ai")]
+    Assistant,
+}
+
+#[derive(Serialize)]
+pub struct ChatResponse {
+    pub id: String,
+    pub object: String,
+    pub created: i64,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub model: Option<String>,
+    pub choices: Vec<Choice>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub usage: Option<Usage>,
+}
+
+#[derive(Serialize)]
+pub struct Choice {
+    pub index: i32,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub message: Option<Message>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub delta: Option<Delta>,
+    pub finish_reason: Option<String>,
+}
+
+#[derive(Serialize)]
+pub struct Delta {
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub role: Option<Role>,
+    #[serde(skip_serializing_if = "Option::is_none")]
+    pub content: Option<String>,
+}
+
+#[derive(Serialize)]
+pub struct Usage {
+    pub prompt_tokens: u32,
+    pub completion_tokens: u32,
+    pub total_tokens: u32,
+}
+
+// 模型定义
+#[derive(Serialize, Clone)]
+pub struct Model {
+    pub id: &'static str,
+    pub created: &'static i64,
+    pub object: &'static str,
+    pub owned_by: &'static str,
+}
+
+use super::constant::USAGE_CHECK_MODELS;
+use crate::app::model::{AppConfig, UsageCheck};
+
+impl Model {
+    pub fn is_usage_check(&self, usage_check: Option<UsageCheck>) -> bool {
+        match usage_check.unwrap_or(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/route.rs

@@ -0,0 +1,19 @@
+mod logs;
+pub use logs::{handle_logs, handle_logs_post};
+mod health;
+pub use health::{handle_health, handle_root};
+mod tokens;
+pub use tokens::{
+    handle_add_tokens, handle_basic_calibration, handle_delete_tokens, handle_get_checksum,
+    handle_get_hash, handle_get_timestamp_header, handle_get_tokens, handle_reload_tokens,
+    handle_tokens_page, handle_update_tokens,
+};
+mod profile;
+pub use profile::handle_user_info;
+mod config;
+pub use config::{
+    handle_about, handle_build_key, handle_build_key_page, handle_config_page, handle_env_example,
+    handle_readme, handle_static,
+};
+mod api;
+pub use api::handle_api_page;

src/chat/route/api.rs

@@ -0,0 +1,26 @@
+use axum::response::{IntoResponse, Response};
+use reqwest::header::CONTENT_TYPE;
+
+use crate::{
+    app::constant::{
+        CONTENT_TYPE_TEXT_HTML_WITH_UTF8, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8, ROUTE_API_PATH,
+    },
+    AppConfig, PageContent,
+};
+
+pub async fn handle_api_page() -> impl IntoResponse {
+    match AppConfig::get_page_content(ROUTE_API_PATH).unwrap_or_default() {
+        PageContent::Default => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(include_str!("../../../static/api.min.html").to_string())
+            .unwrap(),
+        PageContent::Text(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8)
+            .body(content.clone())
+            .unwrap(),
+        PageContent::Html(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(content.clone())
+            .unwrap(),
+    }
+}

src/chat/route/config.rs

@@ -0,0 +1,204 @@
+use crate::{
+    app::{
+        constant::{
+            AUTHORIZATION_BEARER_PREFIX, CONTENT_TYPE_TEXT_CSS_WITH_UTF8, CONTENT_TYPE_TEXT_HTML_WITH_UTF8, CONTENT_TYPE_TEXT_JS_WITH_UTF8, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8, ROUTE_ABOUT_PATH, ROUTE_BUILD_KEY_PATH, ROUTE_CONFIG_PATH, ROUTE_README_PATH, ROUTE_SHARED_JS_PATH, ROUTE_SHARED_STYLES_PATH
+        },
+        lazy::{AUTH_TOKEN, KEY_PREFIX},
+        model::{AppConfig, BuildKeyRequest, BuildKeyResponse, PageContent, UsageCheckModelType},
+    },
+    chat::config::{key_config, KeyConfig},
+    common::utils::{to_base64, token_to_tokeninfo},
+};
+use axum::{
+    body::Body,
+    extract::Path,
+    http::{
+        header::{AUTHORIZATION, CONTENT_TYPE, LOCATION},
+        HeaderMap, StatusCode,
+    },
+    response::{IntoResponse, Response},
+    Json,
+};
+use prost::Message as _;
+
+pub async fn handle_env_example() -> impl IntoResponse {
+    Response::builder()
+        .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8)
+        .body(include_str!("../../../.env.example").to_string())
+        .unwrap()
+}
+
+// 配置页面处理函数
+pub async fn handle_config_page() -> impl IntoResponse {
+    match AppConfig::get_page_content(ROUTE_CONFIG_PATH).unwrap_or_default() {
+        PageContent::Default => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(include_str!("../../../static/config.min.html").to_string())
+            .unwrap(),
+        PageContent::Text(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8)
+            .body(content.clone())
+            .unwrap(),
+        PageContent::Html(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(content.clone())
+            .unwrap(),
+    }
+}
+
+pub async fn handle_static(Path(path): Path<String>) -> impl IntoResponse {
+    match path.as_str() {
+        "shared-styles.css" => {
+            match AppConfig::get_page_content(ROUTE_SHARED_STYLES_PATH).unwrap_or_default() {
+                PageContent::Default => Response::builder()
+                    .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_CSS_WITH_UTF8)
+                    .body(include_str!("../../../static/shared-styles.min.css").to_string())
+                    .unwrap(),
+                PageContent::Text(content) | PageContent::Html(content) => Response::builder()
+                    .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_CSS_WITH_UTF8)
+                    .body(content.clone())
+                    .unwrap(),
+            }
+        }
+        "shared.js" => {
+            match AppConfig::get_page_content(ROUTE_SHARED_JS_PATH).unwrap_or_default() {
+                PageContent::Default => Response::builder()
+                    .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_JS_WITH_UTF8)
+                    .body(include_str!("../../../static/shared.min.js").to_string())
+                    .unwrap(),
+                PageContent::Text(content) | PageContent::Html(content) => Response::builder()
+                    .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_JS_WITH_UTF8)
+                    .body(content.clone())
+                    .unwrap(),
+            }
+        }
+        _ => Response::builder()
+            .status(StatusCode::NOT_FOUND)
+            .body("Not found".to_string())
+            .unwrap(),
+    }
+}
+
+pub async fn handle_readme() -> impl IntoResponse {
+    match AppConfig::get_page_content(ROUTE_README_PATH).unwrap_or_default() {
+        PageContent::Default => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(include_str!("../../../static/readme.min.html").to_string())
+            .unwrap(),
+        PageContent::Text(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8)
+            .body(content.clone())
+            .unwrap(),
+        PageContent::Html(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(content.clone())
+            .unwrap(),
+    }
+}
+
+pub async fn handle_about() -> impl IntoResponse {
+    match AppConfig::get_page_content(ROUTE_ABOUT_PATH).unwrap_or_default() {
+        PageContent::Default => Response::builder()
+            .status(StatusCode::TEMPORARY_REDIRECT)
+            .header(LOCATION, ROUTE_README_PATH)
+            .body(Body::empty())
+            .unwrap(),
+        PageContent::Text(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8)
+            .body(Body::from(content.clone()))
+            .unwrap(),
+        PageContent::Html(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(Body::from(content.clone()))
+            .unwrap(),
+    }
+}
+
+pub async fn handle_build_key_page() -> impl IntoResponse {
+    match AppConfig::get_page_content(ROUTE_BUILD_KEY_PATH).unwrap_or_default() {
+        PageContent::Default => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(include_str!("../../../static/build_key.min.html").to_string())
+            .unwrap(),
+        PageContent::Text(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8)
+            .body(content.clone())
+            .unwrap(),
+        PageContent::Html(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(content.clone())
+            .unwrap(),
+    }
+}
+
+pub async fn handle_build_key(
+    headers: HeaderMap,
+    Json(request): Json<BuildKeyRequest>,
+) -> (StatusCode, Json<BuildKeyResponse>) {
+    // 验证认证令牌
+    if AppConfig::is_share() {
+        let auth_header = headers
+            .get(AUTHORIZATION)
+            .and_then(|h| h.to_str().ok())
+            .and_then(|h| h.strip_prefix(AUTHORIZATION_BEARER_PREFIX));
+
+        if auth_header.map_or(true, |h| h != AppConfig::get_share_token().as_str() && h != AUTH_TOKEN.as_str()) {
+            return (
+                StatusCode::UNAUTHORIZED,
+                Json(BuildKeyResponse::Error("Unauthorized".to_owned())),
+            );
+        }
+    }
+
+    // 验证并解析 auth_token
+    let token_info = match token_to_tokeninfo(&request.auth_token) {
+        Some(info) => info,
+        None => {
+            return (
+                StatusCode::BAD_REQUEST,
+                Json(BuildKeyResponse::Error("Invalid auth token".to_owned())),
+            )
+        }
+    };
+
+    // 构建 proto 消息
+    let mut key_config = KeyConfig {
+        auth_token: Some(token_info),
+        disable_vision: request.disable_vision,
+        enable_slow_pool: request.enable_slow_pool,
+        usage_check_models: None,
+        include_web_references: request.include_web_references,
+    };
+
+    if let Some(usage_check_models) = request.usage_check_models {
+        let usage_check = key_config::UsageCheckModel {
+            r#type: match usage_check_models.model_type {
+                UsageCheckModelType::Default => {
+                    key_config::usage_check_model::Type::Default as i32
+                }
+                UsageCheckModelType::Disabled => {
+                    key_config::usage_check_model::Type::Disabled as i32
+                }
+                UsageCheckModelType::All => key_config::usage_check_model::Type::All as i32,
+                UsageCheckModelType::Custom => key_config::usage_check_model::Type::Custom as i32,
+            },
+            model_ids: if matches!(usage_check_models.model_type, UsageCheckModelType::Custom) {
+                usage_check_models
+                    .model_ids
+                    .iter()
+                    .map(|s| s.to_string())
+                    .collect()
+            } else {
+                Vec::new()
+            },
+        };
+        key_config.usage_check_models = Some(usage_check);
+    }
+
+    // 序列化
+    let encoded = key_config.encode_to_vec();
+
+    let key = format!("{}{}", *KEY_PREFIX, to_base64(&encoded));
+
+    (StatusCode::OK, Json(BuildKeyResponse::Key(key)))
+}

src/chat/route/health.rs

@@ -0,0 +1,140 @@
+use crate::{
+    app::{
+        constant::{
+            AUTHORIZATION_BEARER_PREFIX, CONTENT_TYPE_TEXT_HTML_WITH_UTF8,
+            CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8, PKG_VERSION, ROUTE_ABOUT_PATH, ROUTE_API_PATH,
+            ROUTE_BASIC_CALIBRATION_PATH, ROUTE_BUILD_KEY_PATH, ROUTE_CONFIG_PATH,
+            ROUTE_ENV_EXAMPLE_PATH, ROUTE_GET_CHECKSUM, ROUTE_GET_HASH, ROUTE_GET_TIMESTAMP_HEADER,
+            ROUTE_HEALTH_PATH, ROUTE_LOGS_PATH, ROUTE_README_PATH, ROUTE_ROOT_PATH,
+            ROUTE_STATIC_PATH, ROUTE_TOKENS_ADD_PATH, ROUTE_TOKENS_DELETE_PATH,
+            ROUTE_TOKENS_GET_PATH, ROUTE_TOKENS_PATH, ROUTE_TOKENS_UPDATE_PATH,
+            ROUTE_USER_INFO_PATH,
+        },
+        lazy::{get_start_time, AUTH_TOKEN, ROUTE_CHAT_PATH, ROUTE_MODELS_PATH},
+        model::{AppConfig, AppState, PageContent},
+    },
+    chat::constant::AVAILABLE_MODELS,
+    common::model::{
+        health::{CpuInfo, HealthCheckResponse, MemoryInfo, SystemInfo, SystemStats},
+        ApiStatus,
+    },
+};
+use axum::{
+    body::Body,
+    extract::State,
+    http::{
+        header::{CONTENT_TYPE, LOCATION},
+        HeaderMap, StatusCode,
+    },
+    response::{IntoResponse, Response},
+    Json,
+};
+use chrono::Local;
+use reqwest::header::AUTHORIZATION;
+use std::sync::Arc;
+use sysinfo::{CpuRefreshKind, MemoryRefreshKind, RefreshKind, System};
+use tokio::sync::Mutex;
+
+pub async fn handle_root() -> impl IntoResponse {
+    match AppConfig::get_page_content(ROUTE_ROOT_PATH).unwrap_or_default() {
+        PageContent::Default => Response::builder()
+            .status(StatusCode::TEMPORARY_REDIRECT)
+            .header(LOCATION, ROUTE_HEALTH_PATH)
+            .body(Body::empty())
+            .unwrap(),
+        PageContent::Text(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_PLAIN_WITH_UTF8)
+            .body(Body::from(content.clone()))
+            .unwrap(),
+        PageContent::Html(content) => Response::builder()
+            .header(CONTENT_TYPE, CONTENT_TYPE_TEXT_HTML_WITH_UTF8)
+            .body(Body::from(content.clone()))
+            .unwrap(),
+    }
+}
+
+pub async fn handle_health(
+    State(state): State<Arc<Mutex<AppState>>>,
+    headers: HeaderMap,
+) -> Json<HealthCheckResponse> {
+    let start_time = get_start_time();
+    let uptime = (Local::now() - start_time).num_seconds();
+
+    // 先检查 headers 是否包含有效的认证信息
+    let stats = if headers
+        .get(AUTHORIZATION)
+        .and_then(|h| h.to_str().ok())
+        .and_then(|h| h.strip_prefix(AUTHORIZATION_BEARER_PREFIX))
+        .map_or(false, |token| token == AUTH_TOKEN.as_str())
+    {
+        // 只有在需要系统信息时才创建实例
+        let mut sys = System::new_with_specifics(
+            RefreshKind::nothing()
+                .with_memory(MemoryRefreshKind::everything())
+                .with_cpu(CpuRefreshKind::everything()),
+        );
+
+        std::thread::sleep(sysinfo::MINIMUM_CPU_UPDATE_INTERVAL);
+
+        // 刷新 CPU 和内存信息
+        sys.refresh_memory();
+        sys.refresh_cpu_usage();
+
+        let pid = std::process::id() as usize;
+        let process = sys.process(pid.into());
+
+        // 获取内存信息
+        let memory = process.map(|p| p.memory()).unwrap_or(0);
+
+        // 获取 CPU 使用率
+        let cpu_usage = sys.global_cpu_usage();
+
+        let state = state.lock().await;
+
+        Some(SystemStats {
+            started: start_time.to_string(),
+            total_requests: state.total_requests,
+            active_requests: state.active_requests,
+            system: SystemInfo {
+                memory: MemoryInfo {
+                    rss: memory, // 物理内存使用量(字节)
+                },
+                cpu: CpuInfo {
+                    usage: cpu_usage, // CPU 使用率(百分比)
+                },
+            },
+        })
+    } else {
+        None
+    };
+
+    Json(HealthCheckResponse {
+        status: ApiStatus::Healthy,
+        version: PKG_VERSION,
+        uptime,
+        stats,
+        models: AVAILABLE_MODELS.iter().map(|m| m.id).collect::<Vec<_>>(),
+        endpoints: vec![
+            ROUTE_CHAT_PATH.as_str(),
+            ROUTE_MODELS_PATH.as_str(),
+            ROUTE_TOKENS_PATH,
+            ROUTE_TOKENS_GET_PATH,
+            ROUTE_TOKENS_UPDATE_PATH,
+            ROUTE_TOKENS_ADD_PATH,
+            ROUTE_TOKENS_DELETE_PATH,
+            ROUTE_LOGS_PATH,
+            ROUTE_ENV_EXAMPLE_PATH,
+            ROUTE_CONFIG_PATH,
+            ROUTE_STATIC_PATH,
+            ROUTE_ABOUT_PATH,
+            ROUTE_README_PATH,
+            ROUTE_API_PATH,
+            ROUTE_GET_HASH,
+            ROUTE_GET_CHECKSUM,
+            ROUTE_GET_TIMESTAMP_HEADER,
+            ROUTE_BASIC_CALIBRATION_PATH,
+            ROUTE_USER_INFO_PATH,
+            ROUTE_BUILD_KEY_PATH,
+        ],
+    })
+}