服务端运行时keyring改为文件存储

This commit is contained in:
xintaofei
2026-03-29 19:08:00 +08:00
parent a7f628ee21
commit 85114272de
2 changed files with 66 additions and 6 deletions

View File

@@ -1,26 +1,30 @@
use keyring::Entry;
#[cfg(feature = "tauri-runtime")]
const SERVICE_NAME: &str = "codeg";
fn token_key(account_id: &str) -> String {
format!("github-token:{}", account_id)
}
// ── Tauri mode: OS keyring ──
#[cfg(feature = "tauri-runtime")]
pub fn set_token(account_id: &str, token: &str) -> Result<(), String> {
let entry = Entry::new(SERVICE_NAME, &token_key(account_id))
let entry = keyring::Entry::new(SERVICE_NAME, &token_key(account_id))
.map_err(|e| format!("keyring init error: {e}"))?;
entry
.set_password(token)
.map_err(|e| format!("keyring set error: {e}"))
}
#[cfg(feature = "tauri-runtime")]
pub fn get_token(account_id: &str) -> Option<String> {
let entry = Entry::new(SERVICE_NAME, &token_key(account_id)).ok()?;
let entry = keyring::Entry::new(SERVICE_NAME, &token_key(account_id)).ok()?;
entry.get_password().ok()
}
#[cfg(feature = "tauri-runtime")]
pub fn delete_token(account_id: &str) -> Result<(), String> {
let entry = Entry::new(SERVICE_NAME, &token_key(account_id))
let entry = keyring::Entry::new(SERVICE_NAME, &token_key(account_id))
.map_err(|e| format!("keyring init error: {e}"))?;
match entry.delete_credential() {
Ok(()) => Ok(()),
@@ -28,3 +32,58 @@ pub fn delete_token(account_id: &str) -> Result<(), String> {
Err(e) => Err(format!("keyring delete error: {e}")),
}
}
// ── Server mode: file-based token store ──
#[cfg(not(feature = "tauri-runtime"))]
fn tokens_file_path() -> std::path::PathBuf {
let dir = std::env::var("CODEG_DATA_DIR")
.map(std::path::PathBuf::from)
.unwrap_or_else(|_| {
dirs::data_dir()
.map(|d| d.join("codeg"))
.unwrap_or_else(|| std::path::PathBuf::from(".codeg-data"))
});
dir.join("tokens.json")
}
#[cfg(not(feature = "tauri-runtime"))]
fn read_tokens() -> std::collections::HashMap<String, String> {
let path = tokens_file_path();
std::fs::read_to_string(&path)
.ok()
.and_then(|s| serde_json::from_str(&s).ok())
.unwrap_or_default()
}
#[cfg(not(feature = "tauri-runtime"))]
fn write_tokens(tokens: &std::collections::HashMap<String, String>) -> Result<(), String> {
let path = tokens_file_path();
if let Some(parent) = path.parent() {
std::fs::create_dir_all(parent)
.map_err(|e| format!("failed to create token store directory: {e}"))?;
}
let json = serde_json::to_string_pretty(tokens)
.map_err(|e| format!("failed to serialize tokens: {e}"))?;
std::fs::write(&path, json)
.map_err(|e| format!("failed to write token store: {e}"))
}
#[cfg(not(feature = "tauri-runtime"))]
pub fn set_token(account_id: &str, token: &str) -> Result<(), String> {
let mut tokens = read_tokens();
tokens.insert(token_key(account_id), token.to_string());
write_tokens(&tokens)
}
#[cfg(not(feature = "tauri-runtime"))]
pub fn get_token(account_id: &str) -> Option<String> {
read_tokens().get(&token_key(account_id)).cloned()
}
#[cfg(not(feature = "tauri-runtime"))]
pub fn delete_token(account_id: &str) -> Result<(), String> {
let mut tokens = read_tokens();
tokens.remove(&token_key(account_id));
write_tokens(&tokens)
}