xfy
cc88a7e90a
三处错误（据 omp 客户端实际报错）： 1. 字段名 transport→type，值 streamable-http→http：omp 不识别 transport， 遇未知字段退化为 stdio 因缺 command 报错丢弃 2. 路径 ~/.pi/agent/mcp.json→.mcp.json / ~/.omp/agent/mcp.json / ~/.mcp.json 3. omp 与 Claude Code 的 JSON 体现完全相同，差异仅在配置文件路径
360 lines
14 KiB
Rust
360 lines
14 KiB
Rust
//! MCP 访问令牌管理：Dioxus server functions。
|
||
//!
|
||
//! 管理员在后台 `/admin/mcp` 签发/查看/撤销为 AI 客户端（Claude Code / Cursor /
|
||
//! Cline）准备的 bearer 令牌。明文 token 仅在签发与「重新查看」时返回给管理员，
|
||
//! 数据库只存 AES-GCM 密文（`token_enc`，可解密重查）+ SHA-256 哈希（`token_hash`，
|
||
//! 每请求 O(1) 常量查找，见 `src/mcp/auth.rs`）。
|
||
//!
|
||
//! 鉴权走 cookie session（`get_current_admin_user`），与其它后台 server-fn 一致；
|
||
//! MCP 工具路径（bearer）无法调用这些 server-fn——那是 `src/mcp/tools/*` 的职责。
|
||
|
||
use dioxus::prelude::*;
|
||
|
||
#[cfg(feature = "server")]
|
||
use crate::models::mcp_token::McpToken;
|
||
use crate::models::mcp_token::{CreateTokenResponse, McpTokenSummary, TokenScope};
|
||
|
||
/// 令牌有效期预设：管理员在 UI 上从下拉菜单选择。
|
||
///
|
||
/// 序列化形式供前端选择回传（`days1` / `days7` / `days30` / `days90` / `never`）。
|
||
/// `Never` 对应 `expires_at = NULL`（长期令牌）；其余按当前时间 + N 天计算。
|
||
#[derive(Debug, Clone, Copy, serde::Deserialize, serde::Serialize, PartialEq, Eq)]
|
||
#[serde(rename_all = "lowercase")]
|
||
pub enum TokenLifetime {
|
||
/// 1 天（默认推荐：最小权限、轮换友好）。
|
||
Days1,
|
||
/// 7 天。
|
||
Days7,
|
||
/// 30 天。
|
||
Days30,
|
||
/// 90 天。
|
||
Days90,
|
||
/// 永不过期（`expires_at = NULL`）。仅用于可信长期客户端。
|
||
Never,
|
||
}
|
||
|
||
impl TokenLifetime {
|
||
/// 计算签发时刻对应的过期时间戳（UTC）。`Never` 返回 `None`。
|
||
#[cfg(feature = "server")]
|
||
fn expires_at(self) -> Option<chrono::DateTime<chrono::Utc>> {
|
||
let now = chrono::Utc::now();
|
||
match self {
|
||
TokenLifetime::Days1 => Some(now + chrono::Duration::days(1)),
|
||
TokenLifetime::Days7 => Some(now + chrono::Duration::days(7)),
|
||
TokenLifetime::Days30 => Some(now + chrono::Duration::days(30)),
|
||
TokenLifetime::Days90 => Some(now + chrono::Duration::days(90)),
|
||
TokenLifetime::Never => None,
|
||
}
|
||
}
|
||
}
|
||
|
||
/// 签发新的 MCP 令牌。
|
||
///
|
||
/// 生成明文 `ygg_<32 hex>`，AES-GCM 加密后存密文 + SHA-256 哈希；明文随响应一次性
|
||
/// 返回给管理员（后续可经 [`reveal_mcp_token`] 重新查看）。仅 admin。
|
||
#[server]
|
||
pub async fn create_mcp_token(
|
||
name: String,
|
||
scope: TokenScope,
|
||
lifetime: TokenLifetime,
|
||
) -> Result<CreateTokenResponse, ServerFnError> {
|
||
#[cfg(feature = "server")]
|
||
{
|
||
use crate::api::auth::get_current_admin_user;
|
||
use crate::api::error::AppError;
|
||
use crate::db::pool::get_conn;
|
||
use crate::mcp::auth::{hash_token, TOKEN_PREFIX};
|
||
use crate::mcp::crypto::encrypt_token;
|
||
|
||
let admin = get_current_admin_user().await?;
|
||
|
||
// 名称规范化与校验：去空白后非空，限制长度。
|
||
let name = name.trim().to_string();
|
||
if name.is_empty() {
|
||
return Err(AppError::BadRequest("令牌名称不能为空".to_string()).into());
|
||
}
|
||
if name.chars().count() > 64 {
|
||
return Err(AppError::BadRequest("令牌名称过长（上限 64 字符）".to_string()).into());
|
||
}
|
||
|
||
// 加密主密钥必须已配置，否则无法安全存储明文。
|
||
if crate::mcp::crypto::mcp_enc_key().is_none() {
|
||
return Err(AppError::Internal("MCP_TOKEN_ENC_KEY 未设置").into());
|
||
}
|
||
|
||
// 明文 token：`ygg_` + 32 字节随机数 hex（64 hex 字符）。
|
||
let mut bytes = [0u8; 32];
|
||
rand::RngCore::fill_bytes(&mut rand::thread_rng(), &mut bytes);
|
||
let plaintext = format!("{TOKEN_PREFIX}{}", hex::encode(bytes));
|
||
let hash = hash_token(&plaintext);
|
||
let enc = encrypt_token(&plaintext)
|
||
.ok_or(AppError::Internal("MCP_TOKEN_ENC_KEY 未设置"))?;
|
||
let id = uuid::Uuid::new_v4();
|
||
let expires_at = lifetime.expires_at();
|
||
let scope_str = scope.as_str();
|
||
|
||
let client = get_conn().await.map_err(AppError::db_conn)?;
|
||
|
||
let row = client
|
||
.query_one(
|
||
"INSERT INTO mcp_tokens \
|
||
(id, user_id, name, scope, token_enc, token_hash, expires_at) \
|
||
VALUES ($1::uuid, $2, $3, $4, $5, $6, $7) \
|
||
RETURNING id::text, user_id, name, scope, created_at, expires_at, \
|
||
last_used_at, revoked_at",
|
||
&[
|
||
&id,
|
||
&admin.id,
|
||
&name,
|
||
&scope_str,
|
||
&enc,
|
||
&hash,
|
||
&expires_at,
|
||
],
|
||
)
|
||
.await
|
||
.map_err(AppError::query)?;
|
||
|
||
let token = row_to_mcp_token_meta(&row);
|
||
Ok(CreateTokenResponse {
|
||
summary: token.into(),
|
||
plaintext,
|
||
})
|
||
}
|
||
#[cfg(not(feature = "server"))]
|
||
unreachable!()
|
||
}
|
||
|
||
/// 列出当前管理员名下的全部令牌（不含任何密钥材料，仅展示用元数据）。
|
||
///
|
||
/// 按 `created_at DESC` 排序，最近签发的在前。仅 admin。
|
||
#[server]
|
||
pub async fn list_mcp_tokens() -> Result<Vec<McpTokenSummary>, ServerFnError> {
|
||
#[cfg(feature = "server")]
|
||
{
|
||
use crate::api::auth::get_current_admin_user;
|
||
use crate::api::error::AppError;
|
||
use crate::db::pool::get_conn;
|
||
|
||
let admin = get_current_admin_user().await?;
|
||
let client = get_conn().await.map_err(AppError::db_conn)?;
|
||
|
||
let rows = client
|
||
.query(
|
||
"SELECT id::text, user_id, name, scope, created_at, expires_at, \
|
||
last_used_at, revoked_at \
|
||
FROM mcp_tokens \
|
||
WHERE user_id = $1 \
|
||
ORDER BY created_at DESC",
|
||
&[&admin.id],
|
||
)
|
||
.await
|
||
.map_err(AppError::query)?;
|
||
|
||
Ok(rows
|
||
.iter()
|
||
.map(row_to_mcp_token_meta)
|
||
.map(McpTokenSummary::from)
|
||
.collect())
|
||
}
|
||
#[cfg(not(feature = "server"))]
|
||
unreachable!()
|
||
}
|
||
|
||
/// 重新查看令牌明文（可多次调用：明文以密文形式落库，可解密还原）。
|
||
///
|
||
/// 找不到令牌、或令牌不属于当前管理员 → 返回 `None`（不区分原因，避免探测）。
|
||
/// 仅 admin。
|
||
#[server]
|
||
pub async fn reveal_mcp_token(id: String) -> Result<Option<String>, ServerFnError> {
|
||
#[cfg(feature = "server")]
|
||
{
|
||
use crate::api::auth::get_current_admin_user;
|
||
use crate::api::error::AppError;
|
||
use crate::db::pool::get_conn;
|
||
use crate::mcp::crypto::decrypt_token;
|
||
|
||
let admin = get_current_admin_user().await?;
|
||
let client = get_conn().await.map_err(AppError::db_conn)?;
|
||
|
||
// id 由前端以字符串传入（表 id 列是 uuid）：解析失败视作令牌不存在。
|
||
let id = match uuid::Uuid::parse_str(&id) {
|
||
Ok(u) => u,
|
||
Err(_) => return Ok(None),
|
||
};
|
||
|
||
// 仅取属于当前管理员的令牌的密文，避免越权解密他人令牌。
|
||
let row = client
|
||
.query_opt(
|
||
"SELECT token_enc FROM mcp_tokens WHERE id = $1::uuid AND user_id = $2",
|
||
&[&id, &admin.id],
|
||
)
|
||
.await
|
||
.map_err(AppError::query)?;
|
||
|
||
// 解密失败（密钥缺失/密文被篡改）也归一到 None：调用方无法区分，
|
||
// 按「该令牌不可解密」处理（等同于失效）。
|
||
Ok(row
|
||
.map(|r| r.get::<_, String>("token_enc"))
|
||
.and_then(|enc| decrypt_token(&enc)))
|
||
}
|
||
#[cfg(not(feature = "server"))]
|
||
unreachable!()
|
||
}
|
||
|
||
/// 撤销令牌（软删除：置 `revoked_at = now()`，行保留以备审计）。
|
||
///
|
||
/// 找不到或非本人令牌 → 静默无操作（不报错，避免探测）。仅 admin。
|
||
#[server]
|
||
pub async fn revoke_mcp_token(id: String) -> Result<(), ServerFnError> {
|
||
#[cfg(feature = "server")]
|
||
{
|
||
use crate::api::auth::get_current_admin_user;
|
||
use crate::api::error::AppError;
|
||
use crate::db::pool::get_conn;
|
||
|
||
let admin = get_current_admin_user().await?;
|
||
let client = get_conn().await.map_err(AppError::db_conn)?;
|
||
|
||
// id 解析失败视作令牌不存在（静默无操作，避免探测）。
|
||
let id = match uuid::Uuid::parse_str(&id) {
|
||
Ok(u) => u,
|
||
Err(_) => return Ok(()),
|
||
};
|
||
|
||
client
|
||
.execute(
|
||
"UPDATE mcp_tokens SET revoked_at = NOW() \
|
||
WHERE id = $1::uuid AND user_id = $2 AND revoked_at IS NULL",
|
||
&[&id, &admin.id],
|
||
)
|
||
.await
|
||
.map_err(AppError::query)?;
|
||
|
||
Ok(())
|
||
}
|
||
#[cfg(not(feature = "server"))]
|
||
unreachable!()
|
||
}
|
||
|
||
/// 把数据库行解析为令牌元数据（不含明文；密文/哈希已 `#[serde(skip)]`，这里置空）。
|
||
///
|
||
/// `scope` 列存的是字符串；非法值（理论不可能，除非手工改库）按 read 兜底并记日志，
|
||
/// 不 panic。
|
||
#[cfg(feature = "server")]
|
||
fn row_to_mcp_token_meta(row: &tokio_postgres::Row) -> McpToken {
|
||
let scope_str: String = row.get("scope");
|
||
let scope = TokenScope::from_db(&scope_str).unwrap_or_else(|| {
|
||
tracing::warn!(scope = %scope_str, "mcp_tokens.scope 非法值，兜底为 read");
|
||
TokenScope::Read
|
||
});
|
||
McpToken {
|
||
id: row.get("id"),
|
||
user_id: row.get("user_id"),
|
||
name: row.get("name"),
|
||
scope,
|
||
token_enc: String::new(),
|
||
token_hash: String::new(),
|
||
created_at: row.get("created_at"),
|
||
expires_at: row.get("expires_at"),
|
||
last_used_at: row.get("last_used_at"),
|
||
revoked_at: row.get("revoked_at"),
|
||
}
|
||
}
|
||
|
||
/// 单个客户端配置片段：标题 + 原始文本（供复制）+ syntect 高亮 HTML（供展示）。
|
||
///
|
||
/// 高亮 HTML 由 `crate::highlight::server::highlight_code` 生成（spaced CSS class 风格，
|
||
/// 配合 `public/highlight.css`）；前端需将其置于 `.md-content pre code` 作用域下，
|
||
/// 否则高亮 CSS 选择器不匹配（见 `src/bin/generate_highlight_css.rs` 的 base 重写）。
|
||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||
pub struct McpConfigSnippet {
|
||
/// 显示标题（含客户端名与目标文件路径）。
|
||
pub title: String,
|
||
/// 原始配置文本（供「复制」按钮复制，未高亮）。
|
||
pub content: String,
|
||
/// syntect 高亮后的 HTML（`<span>` 序列，无 `<pre>/<code>` 外壳）。
|
||
pub content_html: String,
|
||
}
|
||
|
||
/// 各客户端 MCP 配置片段集合。
|
||
///
|
||
/// 由 `get_mcp_client_configs` server fn 返回。`ClientConfigs`（在 `src/mcp/config.rs`）
|
||
/// 是 server-only（`mcp` 模块整体 `#[cfg(feature = "server")]` 门控）；这里把每个片段的
|
||
/// 原始文本与高亮 HTML 打包为可两端共享的 DTO，让 WASM 前端单次请求即可渲染带高亮的配置块。
|
||
#[derive(Debug, Clone, PartialEq, serde::Serialize, serde::Deserialize)]
|
||
pub struct McpClientConfigs {
|
||
/// 有序配置片段（标题、原始文本、高亮 HTML）。
|
||
pub snippets: Vec<McpConfigSnippet>,
|
||
}
|
||
|
||
/// 根据明文令牌生成各客户端配置片段（含 syntect 高亮 HTML）。
|
||
///
|
||
/// 配置生成与高亮均在服务端完成（`crate::mcp::config` / `crate::highlight` 均为
|
||
/// server-only），返回给前端展示。`APP_BASE_URL` 环境变量也只在服务端读取。仅 admin。
|
||
#[server]
|
||
pub async fn get_mcp_client_configs(token: String) -> Result<McpClientConfigs, ServerFnError> {
|
||
#[cfg(feature = "server")]
|
||
{
|
||
use crate::api::auth::get_current_admin_user;
|
||
use crate::highlight::server::highlight_code;
|
||
|
||
let _admin = get_current_admin_user().await?;
|
||
let c = crate::mcp::config::generate_client_configs(
|
||
&crate::mcp::config::base_url_from_env(),
|
||
&token,
|
||
);
|
||
// (标题, 内容, 语言)：JSON 配置用 json 语法高亮，CLI 一行命令用 bash。
|
||
let entries: [(&str, String, &str); 7] = [
|
||
("Oh-My-Pi（项目根 .mcp.json / ~/.omp/agent/mcp.json 或 ~/.mcp.json）", c.omp_json, "json"),
|
||
("OpenCode（~/.config/opencode/opencode.json 或项目根 opencode.json）", c.opencode_json, "json"),
|
||
("Claude Code（.mcp.json / ~/.claude.json）", c.claude_code_json, "json"),
|
||
("Cursor（~/.cursor/mcp.json）", c.cursor_json, "json"),
|
||
("Cline（cline_mcp_settings.json）", c.cline_json, "json"),
|
||
("通用（单 server entry）", c.generic_json, "json"),
|
||
("Claude Code CLI", c.claude_cli, "bash"),
|
||
];
|
||
let snippets = entries
|
||
.into_iter()
|
||
.map(|(title, content, lang)| McpConfigSnippet {
|
||
title: title.to_string(),
|
||
content_html: highlight_code(&content, Some(lang)),
|
||
content,
|
||
})
|
||
.collect();
|
||
Ok(McpClientConfigs { snippets })
|
||
}
|
||
#[cfg(not(feature = "server"))]
|
||
unreachable!()
|
||
}
|
||
|
||
#[cfg(all(test, feature = "server"))]
|
||
mod tests {
|
||
use super::*;
|
||
|
||
#[test]
|
||
fn lifetime_expires_at_days() {
|
||
let now = chrono::Utc::now();
|
||
let d1 = TokenLifetime::Days1.expires_at().unwrap();
|
||
let d7 = TokenLifetime::Days7.expires_at().unwrap();
|
||
assert!(d1 > now);
|
||
assert!(d7 > d1);
|
||
// 7 天与 1 天的差应≈6 天（容忍微量时钟漂移）。
|
||
let delta = (d7 - d1).num_seconds() as f64 / 86400.0;
|
||
assert!((5.9..6.1).contains(&delta));
|
||
}
|
||
|
||
#[test]
|
||
fn lifetime_never_is_none() {
|
||
assert!(TokenLifetime::Never.expires_at().is_none());
|
||
}
|
||
|
||
#[test]
|
||
fn lifetime_serde_roundtrip() {
|
||
let json = serde_json::to_string(&TokenLifetime::Days30).unwrap();
|
||
assert_eq!(json, "\"days30\"");
|
||
let back: TokenLifetime = serde_json::from_str(&json).unwrap();
|
||
assert_eq!(back, TokenLifetime::Days30);
|
||
}
|
||
}
|