feat(mcp): add Rust module and Tauri commands for Claude MCP
- Manage ~/.claude/settings.local.json and ~/.claude/mcp.json - Atomic JSON read/write with preservation of unknown fields - Commands: get_claude_mcp_status, read_claude_mcp_config, set_claude_mcp_enable_all_projects, upsert_claude_mcp_server, delete_claude_mcp_server, validate_mcp_command - Register commands in Tauri invoke handler - PATH-based command availability check (non-executing)
This commit is contained in:
218
src-tauri/src/claude_mcp.rs
Normal file
218
src-tauri/src/claude_mcp.rs
Normal file
@@ -0,0 +1,218 @@
|
||||
use serde::{Deserialize, Serialize};
|
||||
use serde_json::Value;
|
||||
use std::env;
|
||||
use std::fs;
|
||||
use std::path::{Path, PathBuf};
|
||||
|
||||
use crate::config::{atomic_write, get_claude_config_dir};
|
||||
|
||||
#[derive(Debug, Clone, Serialize, Deserialize)]
|
||||
#[serde(rename_all = "camelCase")]
|
||||
pub struct McpStatus {
|
||||
pub settings_local_path: String,
|
||||
pub settings_local_exists: bool,
|
||||
pub enable_all_project_mcp_servers: bool,
|
||||
pub mcp_json_path: String,
|
||||
pub mcp_json_exists: bool,
|
||||
pub server_count: usize,
|
||||
}
|
||||
|
||||
fn claude_dir() -> PathBuf {
|
||||
get_claude_config_dir()
|
||||
}
|
||||
|
||||
fn settings_local_path() -> PathBuf {
|
||||
claude_dir().join("settings.local.json")
|
||||
}
|
||||
|
||||
fn mcp_json_path() -> PathBuf {
|
||||
claude_dir().join("mcp.json")
|
||||
}
|
||||
|
||||
fn read_json_value(path: &Path) -> Result<Value, String> {
|
||||
if !path.exists() {
|
||||
return Ok(serde_json::json!({}));
|
||||
}
|
||||
let content =
|
||||
fs::read_to_string(path).map_err(|e| format!("读取文件失败: {}: {}", path.display(), e))?;
|
||||
let value: Value =
|
||||
serde_json::from_str(&content).map_err(|e| format!("解析 JSON 失败: {}: {}", path.display(), e))?;
|
||||
Ok(value)
|
||||
}
|
||||
|
||||
fn write_json_value(path: &Path, value: &Value) -> Result<(), String> {
|
||||
if let Some(parent) = path.parent() {
|
||||
fs::create_dir_all(parent)
|
||||
.map_err(|e| format!("创建目录失败: {}: {}", parent.display(), e))?;
|
||||
}
|
||||
let json = serde_json::to_string_pretty(value).map_err(|e| format!("序列化 JSON 失败: {}", e))?;
|
||||
atomic_write(path, json.as_bytes())
|
||||
}
|
||||
|
||||
pub fn get_mcp_status() -> Result<McpStatus, String> {
|
||||
let settings_local = settings_local_path();
|
||||
let mcp_path = mcp_json_path();
|
||||
|
||||
let mut enable = false;
|
||||
if settings_local.exists() {
|
||||
let v = read_json_value(&settings_local)?;
|
||||
enable = v
|
||||
.get("enableAllProjectMcpServers")
|
||||
.and_then(|x| x.as_bool())
|
||||
.unwrap_or(false);
|
||||
}
|
||||
|
||||
let (exists, count) = if mcp_path.exists() {
|
||||
let v = read_json_value(&mcp_path)?;
|
||||
let servers = v.get("mcpServers").and_then(|x| x.as_object());
|
||||
(true, servers.map(|m| m.len()).unwrap_or(0))
|
||||
} else {
|
||||
(false, 0)
|
||||
};
|
||||
|
||||
Ok(McpStatus {
|
||||
settings_local_path: settings_local.to_string_lossy().to_string(),
|
||||
settings_local_exists: settings_local.exists(),
|
||||
enable_all_project_mcp_servers: enable,
|
||||
mcp_json_path: mcp_path.to_string_lossy().to_string(),
|
||||
mcp_json_exists: exists,
|
||||
server_count: count,
|
||||
})
|
||||
}
|
||||
|
||||
pub fn read_mcp_json() -> Result<Option<String>, String> {
|
||||
let path = mcp_json_path();
|
||||
if !path.exists() {
|
||||
return Ok(None);
|
||||
}
|
||||
let content =
|
||||
fs::read_to_string(&path).map_err(|e| format!("读取 MCP 配置失败: {}", e))?;
|
||||
Ok(Some(content))
|
||||
}
|
||||
|
||||
pub fn set_enable_all_projects(enable: bool) -> Result<bool, String> {
|
||||
let path = settings_local_path();
|
||||
let mut v = if path.exists() { read_json_value(&path)? } else { serde_json::json!({}) };
|
||||
|
||||
let current = v
|
||||
.get("enableAllProjectMcpServers")
|
||||
.and_then(|x| x.as_bool())
|
||||
.unwrap_or(false);
|
||||
if current == enable && path.exists() {
|
||||
return Ok(false);
|
||||
}
|
||||
|
||||
if let Some(obj) = v.as_object_mut() {
|
||||
obj.insert(
|
||||
"enableAllProjectMcpServers".to_string(),
|
||||
Value::Bool(enable),
|
||||
);
|
||||
}
|
||||
write_json_value(&path, &v)?;
|
||||
Ok(true)
|
||||
}
|
||||
|
||||
pub fn upsert_mcp_server(id: &str, spec: Value) -> Result<bool, String> {
|
||||
if id.trim().is_empty() {
|
||||
return Err("MCP 服务器 ID 不能为空".into());
|
||||
}
|
||||
// 基础字段校验(尽量宽松)
|
||||
if !spec.is_object() {
|
||||
return Err("MCP 服务器定义必须为 JSON 对象".into());
|
||||
}
|
||||
let t = spec
|
||||
.get("type")
|
||||
.and_then(|x| x.as_str())
|
||||
.unwrap_or("");
|
||||
if t != "stdio" && t != "sse" {
|
||||
return Err("MCP 服务器 type 必须是 'stdio' 或 'sse'".into());
|
||||
}
|
||||
let cmd = spec.get("command").and_then(|x| x.as_str()).unwrap_or("");
|
||||
if cmd.is_empty() {
|
||||
return Err("MCP 服务器缺少 command".into());
|
||||
}
|
||||
|
||||
let path = mcp_json_path();
|
||||
let mut root = if path.exists() { read_json_value(&path)? } else { serde_json::json!({}) };
|
||||
|
||||
// 确保 mcpServers 对象存在
|
||||
{
|
||||
let obj = root.as_object_mut().ok_or_else(|| "mcp.json 根必须是对象".to_string())?;
|
||||
if !obj.contains_key("mcpServers") {
|
||||
obj.insert("mcpServers".into(), serde_json::json!({}));
|
||||
}
|
||||
}
|
||||
|
||||
let before = root.clone();
|
||||
if let Some(servers) = root
|
||||
.get_mut("mcpServers")
|
||||
.and_then(|v| v.as_object_mut())
|
||||
{
|
||||
servers.insert(id.to_string(), spec);
|
||||
}
|
||||
|
||||
if before == root && path.exists() {
|
||||
return Ok(false);
|
||||
}
|
||||
|
||||
write_json_value(&path, &root)?;
|
||||
Ok(true)
|
||||
}
|
||||
|
||||
pub fn delete_mcp_server(id: &str) -> Result<bool, String> {
|
||||
if id.trim().is_empty() {
|
||||
return Err("MCP 服务器 ID 不能为空".into());
|
||||
}
|
||||
let path = mcp_json_path();
|
||||
if !path.exists() {
|
||||
return Ok(false);
|
||||
}
|
||||
let mut root = read_json_value(&path)?;
|
||||
let Some(servers) = root.get_mut("mcpServers").and_then(|v| v.as_object_mut()) else {
|
||||
return Ok(false);
|
||||
};
|
||||
let existed = servers.remove(id).is_some();
|
||||
if !existed {
|
||||
return Ok(false);
|
||||
}
|
||||
write_json_value(&path, &root)?;
|
||||
Ok(true)
|
||||
}
|
||||
|
||||
pub fn validate_command_in_path(cmd: &str) -> Result<bool, String> {
|
||||
if cmd.trim().is_empty() {
|
||||
return Ok(false);
|
||||
}
|
||||
// 如果包含路径分隔符,直接判断是否存在可执行文件
|
||||
if cmd.contains('/') || cmd.contains('\\') {
|
||||
return Ok(Path::new(cmd).exists());
|
||||
}
|
||||
|
||||
let path_var = env::var_os("PATH").unwrap_or_default();
|
||||
let paths = env::split_paths(&path_var);
|
||||
|
||||
#[cfg(windows)]
|
||||
let exts: Vec<String> = env::var("PATHEXT")
|
||||
.unwrap_or(".COM;.EXE;.BAT;.CMD".into())
|
||||
.split(';')
|
||||
.map(|s| s.trim().to_uppercase())
|
||||
.collect();
|
||||
|
||||
for p in paths {
|
||||
let candidate = p.join(cmd);
|
||||
if candidate.is_file() {
|
||||
return Ok(true);
|
||||
}
|
||||
#[cfg(windows)]
|
||||
{
|
||||
for ext in &exts {
|
||||
let cand = p.join(format!("{}{}", cmd, ext));
|
||||
if cand.is_file() {
|
||||
return Ok(true);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Ok(false)
|
||||
}
|
||||
|
||||
@@ -7,6 +7,7 @@ use tauri_plugin_opener::OpenerExt;
|
||||
|
||||
use crate::app_config::AppType;
|
||||
use crate::claude_plugin;
|
||||
use crate::claude_mcp;
|
||||
use crate::codex_config;
|
||||
use crate::config::{self, get_claude_settings_path, ConfigStatus};
|
||||
use crate::provider::{Provider, ProviderMeta};
|
||||
@@ -659,6 +660,46 @@ pub async fn open_app_config_folder(handle: tauri::AppHandle) -> Result<bool, St
|
||||
Ok(true)
|
||||
}
|
||||
|
||||
// =====================
|
||||
// Claude MCP 管理命令
|
||||
// =====================
|
||||
|
||||
/// 获取 Claude MCP 状态(settings.local.json 与 mcp.json)
|
||||
#[tauri::command]
|
||||
pub async fn get_claude_mcp_status() -> Result<crate::claude_mcp::McpStatus, String> {
|
||||
claude_mcp::get_mcp_status()
|
||||
}
|
||||
|
||||
/// 读取 mcp.json 文本内容(不存在则返回 Ok(None))
|
||||
#[tauri::command]
|
||||
pub async fn read_claude_mcp_config() -> Result<Option<String>, String> {
|
||||
claude_mcp::read_mcp_json()
|
||||
}
|
||||
|
||||
/// 设置 enableAllProjectMcpServers 开关
|
||||
#[tauri::command]
|
||||
pub async fn set_claude_mcp_enable_all_projects(enable: bool) -> Result<bool, String> {
|
||||
claude_mcp::set_enable_all_projects(enable)
|
||||
}
|
||||
|
||||
/// 新增或更新一个 MCP 服务器条目
|
||||
#[tauri::command]
|
||||
pub async fn upsert_claude_mcp_server(id: String, spec: serde_json::Value) -> Result<bool, String> {
|
||||
claude_mcp::upsert_mcp_server(&id, spec)
|
||||
}
|
||||
|
||||
/// 删除一个 MCP 服务器条目
|
||||
#[tauri::command]
|
||||
pub async fn delete_claude_mcp_server(id: String) -> Result<bool, String> {
|
||||
claude_mcp::delete_mcp_server(&id)
|
||||
}
|
||||
|
||||
/// 校验命令是否在 PATH 中可用(不执行)
|
||||
#[tauri::command]
|
||||
pub async fn validate_mcp_command(cmd: String) -> Result<bool, String> {
|
||||
claude_mcp::validate_command_in_path(&cmd)
|
||||
}
|
||||
|
||||
/// 获取设置
|
||||
#[tauri::command]
|
||||
pub async fn get_settings() -> Result<crate::settings::AppSettings, String> {
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
mod app_config;
|
||||
mod claude_plugin;
|
||||
mod claude_mcp;
|
||||
mod codex_config;
|
||||
mod commands;
|
||||
mod config;
|
||||
@@ -421,6 +422,13 @@ pub fn run() {
|
||||
commands::read_claude_plugin_config,
|
||||
commands::apply_claude_plugin_config,
|
||||
commands::is_claude_plugin_applied,
|
||||
// Claude MCP management
|
||||
commands::get_claude_mcp_status,
|
||||
commands::read_claude_mcp_config,
|
||||
commands::set_claude_mcp_enable_all_projects,
|
||||
commands::upsert_claude_mcp_server,
|
||||
commands::delete_claude_mcp_server,
|
||||
commands::validate_mcp_command,
|
||||
// ours: endpoint speed test + custom endpoint management
|
||||
commands::test_api_endpoints,
|
||||
commands::get_custom_endpoints,
|
||||
|
||||
Reference in New Issue
Block a user