File size: 2,556 Bytes
bbb1195 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 | use serde_json::Value;
/// 使用 Antigravity 的 loadCodeAssist API 获取 project_id
/// 这是获取 cloudaicompanionProject 的正确方式
pub async fn fetch_project_id(access_token: &str) -> Result<String, String> {
let url = "https://cloudcode-pa.googleapis.com/v1internal:loadCodeAssist";
let request_body = serde_json::json!({
"metadata": {
"ideType": "ANTIGRAVITY"
}
});
let client = crate::utils::http::create_client(30);
let response = client
.post(url)
.bearer_auth(access_token)
.header("Host", "cloudcode-pa.googleapis.com")
.header("User-Agent", "antigravity/1.11.9 windows/amd64")
.header("Content-Type", "application/json")
.json(&request_body)
.send()
.await
.map_err(|e| format!("loadCodeAssist 请求失败: {}", e))?;
if !response.status().is_success() {
let status = response.status();
let body = response.text().await.unwrap_or_default();
return Err(format!("loadCodeAssist 返回错误 {}: {}", status, body));
}
let data: Value = response.json()
.await
.map_err(|e| format!("解析响应失败: {}", e))?;
// 提取 cloudaicompanionProject
if let Some(project_id) = data.get("cloudaicompanionProject")
.and_then(|v| v.as_str()) {
return Ok(project_id.to_string());
}
// 如果没有返回 project_id，说明账号无资格，使用内置随机生成逻辑作为兜底
let mock_id = generate_mock_project_id();
tracing::warn!("账号无资格获取官方 cloudaicompanionProject，将使用随机生成的 Project ID 作为兜底: {}", mock_id);
Ok(mock_id)
}
/// 生成随机 project_id（当无法从 API 获取时使用）
/// 格式：{形容词}-{名词}-{5位随机字符}
pub fn generate_mock_project_id() -> String {
use rand::Rng;
let adjectives = ["useful", "bright", "swift", "calm", "bold"];
let nouns = ["fuze", "wave", "spark", "flow", "core"];
let mut rng = rand::thread_rng();
let adj = adjectives[rng.gen_range(0..adjectives.len())];
let noun = nouns[rng.gen_range(0..nouns.len())];
// 生成5位随机字符（base36）
let random_num: String = (0..5)
.map(|_| {
let chars = "abcdefghijklmnopqrstuvwxyz0123456789";
let idx = rng.gen_range(0..chars.len());
chars.chars().nth(idx).unwrap()
})
.collect();
format!("{}-{}-{}", adj, noun, random_num)
}
|