feat(claude-code): 支持通过 /api/oauth/usage 查询账号 5H/周额度并在号池展示

2026-10-08 18:37:46 +08:00 · 2026-09-29 22:40:44 +08:00
parent 00a315e3bf
commit 8093899b5a
15 changed files with 653 additions and 33 deletions
@@ -0,0 +1,79 @@
+use std::collections::BTreeMap;
+
+use aether_data_contracts::repository::provider_catalog::StoredProviderCatalogEndpoint;
+
+use crate::capability::ProviderPoolCapabilities;
+use crate::provider::{
+    provider_pool_endpoint_format_matches, provider_pool_matching_endpoint, ProviderPoolAdapter,
+};
+use crate::quota_refresh::ProviderPoolQuotaRequestSpec;
+
+pub const CLAUDE_CODE_OAUTH_USAGE_URL: &str = "https://api.anthropic.com/api/oauth/usage";
+pub const CLAUDE_CODE_OAUTH_BETA: &str = "oauth-2025-04-20";
+pub const CLAUDE_CODE_USAGE_USER_AGENT: &str = "claude-code/2.1.284";
+
+#[derive(Debug, Clone, Default)]
+pub struct ClaudeCodeProviderPoolAdapter;
+
+impl ProviderPoolAdapter for ClaudeCodeProviderPoolAdapter {
+    fn provider_type(&self) -> &'static str {
+        "claude_code"
+    }
+
+    fn capabilities(&self) -> ProviderPoolCapabilities {
+        ProviderPoolCapabilities {
+            quota_refresh: true,
+            ..ProviderPoolCapabilities::default()
+        }
+    }
+
+    fn quota_refresh_endpoint(
+        &self,
+        endpoints: &[StoredProviderCatalogEndpoint],
+        include_inactive: bool,
+    ) -> Option<StoredProviderCatalogEndpoint> {
+        provider_pool_matching_endpoint(endpoints, include_inactive, |endpoint| {
+            provider_pool_endpoint_format_matches(endpoint, "claude:messages")
+        })
+    }
+
+    fn quota_refresh_missing_endpoint_message(&self) -> String {
+        "找不到有效的 claude:messages 端点".to_string()
+    }
+}
+
+/// Builds the `GET /api/oauth/usage` request that reports the account's 5h / 7d
+/// utilization windows. The URL is fixed (the origin allowlist only accepts
+/// `api.anthropic.com`), independent of the inference endpoint's base URL.
+pub fn build_claude_code_pool_quota_request(
+    key_id: &str,
+    authorization: (String, String),
+) -> ProviderPoolQuotaRequestSpec {
+    let headers = BTreeMap::from([
+        ("authorization".to_string(), authorization.1),
+        ("accept".to_string(), "application/json".to_string()),
+        ("content-type".to_string(), "application/json".to_string()),
+        (
+            "anthropic-beta".to_string(),
+            CLAUDE_CODE_OAUTH_BETA.to_string(),
+        ),
+        (
+            "user-agent".to_string(),
+            CLAUDE_CODE_USAGE_USER_AGENT.to_string(),
+        ),
+    ]);
+
+    ProviderPoolQuotaRequestSpec {
+        request_id: format!("claude-code-quota:{key_id}"),
+        provider_name: "claude_code".to_string(),
+        quota_kind: "claude_code".to_string(),
+        method: "GET".to_string(),
+        url: CLAUDE_CODE_OAUTH_USAGE_URL.to_string(),
+        headers,
+        content_type: None,
+        json_body: None,
+        client_api_format: "claude:messages".to_string(),
+        provider_api_format: "claude_code:oauth_usage".to_string(),
+        model_name: Some("oauth_usage".to_string()),
+    }
+}