Compare commits
2 Commits
secrets-mc
...
secrets-mc
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
1e597559a2 | ||
|
|
e3ca43ca3f |
@@ -112,7 +112,7 @@ oauth_accounts (
|
|||||||
- 错误:业务层 `anyhow::Result`,避免生产路径 `unwrap()`。
|
- 错误:业务层 `anyhow::Result`,避免生产路径 `unwrap()`。
|
||||||
- 异步:`tokio` + `sqlx` async。
|
- 异步:`tokio` + `sqlx` async。
|
||||||
- SQL:`sqlx::query` / `query_as` 参数绑定;动态 WHERE 仍须用占位符绑定。
|
- SQL:`sqlx::query` / `query_as` 参数绑定;动态 WHERE 仍须用占位符绑定。
|
||||||
- 日志:运维用 `tracing`;面向用户的 Web 响应走 axum handler。
|
- 日志:运维用 `tracing`;面向用户的 Web 响应走 axum handler。tracing 字段风格:变量名即字段名时用简写(`%var`、`?var`、`var`),否则用显式形式(`field = %expr`)。
|
||||||
- 审计:写操作成功后尽量 `audit::log_tx`;失败可 `warn`,不掩盖主错误。
|
- 审计:写操作成功后尽量 `audit::log_tx`;失败可 `warn`,不掩盖主错误。
|
||||||
- 加密:密钥由用户密码短语通过 **PBKDF2-SHA256(600k 次)** 在客户端派生,服务端只存 `key_salt`/`key_check`/`key_params`,不持有原始密钥。Web 客户端在浏览器本地完成加解密;MCP 客户端通过 `X-Encryption-Key` 请求头传递密钥,服务端临时解密后返回明文。
|
- 加密:密钥由用户密码短语通过 **PBKDF2-SHA256(600k 次)** 在客户端派生,服务端只存 `key_salt`/`key_check`/`key_params`,不持有原始密钥。Web 客户端在浏览器本地完成加解密;MCP 客户端通过 `X-Encryption-Key` 请求头传递密钥,服务端临时解密后返回明文。
|
||||||
- MCP:tools 参数与 JSON Schema(`schemars`)保持同步,鉴权以请求扩展中的用户上下文为准。
|
- MCP:tools 参数与 JSON Schema(`schemars`)保持同步,鉴权以请求扩展中的用户上下文为准。
|
||||||
@@ -154,7 +154,7 @@ git tag -l 'secrets-mcp-*'
|
|||||||
|------|------|
|
|------|------|
|
||||||
| `SECRETS_DATABASE_URL` | **必填**。PostgreSQL URL。 |
|
| `SECRETS_DATABASE_URL` | **必填**。PostgreSQL URL。 |
|
||||||
| `BASE_URL` | 对外基址;OAuth 回调 `${BASE_URL}/auth/google/callback`。 |
|
| `BASE_URL` | 对外基址;OAuth 回调 `${BASE_URL}/auth/google/callback`。 |
|
||||||
| `SECRETS_MCP_BIND` | 监听地址,默认 `0.0.0.0:9315`。 |
|
| `SECRETS_MCP_BIND` | 监听地址,默认 `127.0.0.1:9315`(容器/远程直接暴露时需改为 `0.0.0.0:9315`)。 |
|
||||||
| `GOOGLE_CLIENT_ID` / `GOOGLE_CLIENT_SECRET` | 可选;仅运行时配置。 |
|
| `GOOGLE_CLIENT_ID` / `GOOGLE_CLIENT_SECRET` | 可选;仅运行时配置。 |
|
||||||
| `RUST_LOG` | 如 `secrets_mcp=debug`。 |
|
| `RUST_LOG` | 如 `secrets_mcp=debug`。 |
|
||||||
|
|
||||||
|
|||||||
39
Cargo.lock
generated
39
Cargo.lock
generated
@@ -1809,6 +1809,25 @@ dependencies = [
|
|||||||
"syn",
|
"syn",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rmp"
|
||||||
|
version = "0.8.15"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "4ba8be72d372b2c9b35542551678538b562e7cf86c3315773cae48dfbfe7790c"
|
||||||
|
dependencies = [
|
||||||
|
"num-traits",
|
||||||
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "rmp-serde"
|
||||||
|
version = "1.3.1"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "72f81bee8c8ef9b577d1681a70ebbc962c232461e397b22c208c43c04b67a155"
|
||||||
|
dependencies = [
|
||||||
|
"rmp",
|
||||||
|
"serde",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "rsa"
|
name = "rsa"
|
||||||
version = "0.9.10"
|
version = "0.9.10"
|
||||||
@@ -1949,7 +1968,7 @@ dependencies = [
|
|||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "secrets-mcp"
|
name = "secrets-mcp"
|
||||||
version = "0.1.11"
|
version = "0.2.1"
|
||||||
dependencies = [
|
dependencies = [
|
||||||
"anyhow",
|
"anyhow",
|
||||||
"askama",
|
"askama",
|
||||||
@@ -1967,10 +1986,12 @@ dependencies = [
|
|||||||
"serde_json",
|
"serde_json",
|
||||||
"sha2",
|
"sha2",
|
||||||
"sqlx",
|
"sqlx",
|
||||||
|
"time",
|
||||||
"tokio",
|
"tokio",
|
||||||
"tower",
|
"tower",
|
||||||
"tower-http",
|
"tower-http",
|
||||||
"tower-sessions",
|
"tower-sessions",
|
||||||
|
"tower-sessions-sqlx-store-chrono",
|
||||||
"tracing",
|
"tracing",
|
||||||
"tracing-subscriber",
|
"tracing-subscriber",
|
||||||
"urlencoding",
|
"urlencoding",
|
||||||
@@ -2766,6 +2787,22 @@ dependencies = [
|
|||||||
"tower-sessions-core",
|
"tower-sessions-core",
|
||||||
]
|
]
|
||||||
|
|
||||||
|
[[package]]
|
||||||
|
name = "tower-sessions-sqlx-store-chrono"
|
||||||
|
version = "0.14.2"
|
||||||
|
source = "registry+https://github.com/rust-lang/crates.io-index"
|
||||||
|
checksum = "b295c8fc08db03246e92773c5e10119b72db6bc4240112135bebb0e49670804f"
|
||||||
|
dependencies = [
|
||||||
|
"async-trait",
|
||||||
|
"axum",
|
||||||
|
"chrono",
|
||||||
|
"rmp-serde",
|
||||||
|
"sqlx",
|
||||||
|
"thiserror",
|
||||||
|
"time",
|
||||||
|
"tower-sessions-core",
|
||||||
|
]
|
||||||
|
|
||||||
[[package]]
|
[[package]]
|
||||||
name = "tracing"
|
name = "tracing"
|
||||||
version = "0.1.44"
|
version = "0.1.44"
|
||||||
|
|||||||
@@ -19,7 +19,7 @@ cargo build --release -p secrets-mcp
|
|||||||
|------|------|
|
|------|------|
|
||||||
| `SECRETS_DATABASE_URL` | **必填**。PostgreSQL 连接串(建议专用库,如 `secrets-mcp`)。 |
|
| `SECRETS_DATABASE_URL` | **必填**。PostgreSQL 连接串(建议专用库,如 `secrets-mcp`)。 |
|
||||||
| `BASE_URL` | 对外访问基址;OAuth 回调为 `{BASE_URL}/auth/google/callback`。默认 `http://localhost:9315`。 |
|
| `BASE_URL` | 对外访问基址;OAuth 回调为 `{BASE_URL}/auth/google/callback`。默认 `http://localhost:9315`。 |
|
||||||
| `SECRETS_MCP_BIND` | 监听地址,默认 `0.0.0.0:9315`。反代时常为 `127.0.0.1:9315`。 |
|
| `SECRETS_MCP_BIND` | 监听地址,默认 `127.0.0.1:9315`。容器内或直接对外暴露端口时请改为 `0.0.0.0:9315`;反代时常为 `127.0.0.1:9315`。 |
|
||||||
| `GOOGLE_CLIENT_ID` / `GOOGLE_CLIENT_SECRET` | 可选;不配置则无 Google 登录入口。运行时从环境读取,勿写入 CI、勿打入二进制。 |
|
| `GOOGLE_CLIENT_ID` / `GOOGLE_CLIENT_SECRET` | 可选;不配置则无 Google 登录入口。运行时从环境读取,勿写入 CI、勿打入二进制。 |
|
||||||
|
|
||||||
```bash
|
```bash
|
||||||
|
|||||||
@@ -55,35 +55,6 @@ pub fn decrypt_json(master_key: &[u8; 32], data: &[u8]) -> Result<Value> {
|
|||||||
serde_json::from_slice(&bytes).context("deserialize decrypted JSON")
|
serde_json::from_slice(&bytes).context("deserialize decrypted JSON")
|
||||||
}
|
}
|
||||||
|
|
||||||
// ─── Per-user key management (DEPRECATED — kept only for migration) ───────────
|
|
||||||
|
|
||||||
/// Generate a new random 32-byte per-user encryption key.
|
|
||||||
#[allow(dead_code)]
|
|
||||||
pub fn generate_user_key() -> [u8; 32] {
|
|
||||||
use aes_gcm::aead::rand_core::RngCore;
|
|
||||||
let mut key = [0u8; 32];
|
|
||||||
OsRng.fill_bytes(&mut key);
|
|
||||||
key
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Wrap a per-user key with the server master key using AES-256-GCM.
|
|
||||||
#[allow(dead_code)]
|
|
||||||
pub fn wrap_user_key(server_master_key: &[u8; 32], user_key: &[u8; 32]) -> Result<Vec<u8>> {
|
|
||||||
encrypt(server_master_key, user_key.as_ref())
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Unwrap a per-user key using the server master key.
|
|
||||||
#[allow(dead_code)]
|
|
||||||
pub fn unwrap_user_key(server_master_key: &[u8; 32], wrapped: &[u8]) -> Result<[u8; 32]> {
|
|
||||||
let bytes = decrypt(server_master_key, wrapped)?;
|
|
||||||
if bytes.len() != 32 {
|
|
||||||
bail!("unwrapped user key has unexpected length {}", bytes.len());
|
|
||||||
}
|
|
||||||
let mut key = [0u8; 32];
|
|
||||||
key.copy_from_slice(&bytes);
|
|
||||||
Ok(key)
|
|
||||||
}
|
|
||||||
|
|
||||||
// ─── Client-supplied key extraction ──────────────────────────────────────────
|
// ─── Client-supplied key extraction ──────────────────────────────────────────
|
||||||
|
|
||||||
/// Parse a 64-char hex string (from X-Encryption-Key header) into a 32-byte key.
|
/// Parse a 64-char hex string (from X-Encryption-Key header) into a 32-byte key.
|
||||||
@@ -100,33 +71,6 @@ pub fn extract_key_from_hex(hex_str: &str) -> Result<[u8; 32]> {
|
|||||||
Ok(key)
|
Ok(key)
|
||||||
}
|
}
|
||||||
|
|
||||||
// ─── Server master key ────────────────────────────────────────────────────────
|
|
||||||
|
|
||||||
/// Load the server master key from `SERVER_MASTER_KEY` environment variable (64 hex chars).
|
|
||||||
pub fn load_master_key_auto() -> Result<[u8; 32]> {
|
|
||||||
let hex_str = std::env::var("SERVER_MASTER_KEY").map_err(|_| {
|
|
||||||
anyhow::anyhow!(
|
|
||||||
"SERVER_MASTER_KEY is not set. \
|
|
||||||
Generate one with: openssl rand -hex 32"
|
|
||||||
)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
if hex_str.is_empty() {
|
|
||||||
bail!("SERVER_MASTER_KEY is set but empty");
|
|
||||||
}
|
|
||||||
|
|
||||||
let bytes = hex::decode_hex(hex_str.trim())?;
|
|
||||||
if bytes.len() != 32 {
|
|
||||||
bail!(
|
|
||||||
"SERVER_MASTER_KEY must be 64 hex chars (32 bytes), got {} bytes",
|
|
||||||
bytes.len()
|
|
||||||
);
|
|
||||||
}
|
|
||||||
let mut key = [0u8; 32];
|
|
||||||
key.copy_from_slice(&bytes);
|
|
||||||
Ok(key)
|
|
||||||
}
|
|
||||||
|
|
||||||
// ─── Public hex helpers ───────────────────────────────────────────────────────
|
// ─── Public hex helpers ───────────────────────────────────────────────────────
|
||||||
|
|
||||||
pub mod hex {
|
pub mod hex {
|
||||||
@@ -186,22 +130,4 @@ mod tests {
|
|||||||
let dec = decrypt_json(&key, &enc).unwrap();
|
let dec = decrypt_json(&key, &enc).unwrap();
|
||||||
assert_eq!(dec, value);
|
assert_eq!(dec, value);
|
||||||
}
|
}
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn user_key_wrap_unwrap_roundtrip() {
|
|
||||||
let server_key = [0xABu8; 32];
|
|
||||||
let user_key = [0xCDu8; 32];
|
|
||||||
let wrapped = wrap_user_key(&server_key, &user_key).unwrap();
|
|
||||||
let unwrapped = unwrap_user_key(&server_key, &wrapped).unwrap();
|
|
||||||
assert_eq!(unwrapped, user_key);
|
|
||||||
}
|
|
||||||
|
|
||||||
#[test]
|
|
||||||
fn user_key_wrap_wrong_server_key_fails() {
|
|
||||||
let server_key1 = [0xABu8; 32];
|
|
||||||
let server_key2 = [0xEFu8; 32];
|
|
||||||
let user_key = [0xCDu8; 32];
|
|
||||||
let wrapped = wrap_user_key(&server_key1, &user_key).unwrap();
|
|
||||||
assert!(unwrap_user_key(&server_key2, &wrapped).is_err());
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -156,6 +156,37 @@ pub async fn migrate(pool: &PgPool) -> Result<()> {
|
|||||||
CREATE INDEX IF NOT EXISTS idx_oauth_accounts_user ON oauth_accounts(user_id);
|
CREATE INDEX IF NOT EXISTS idx_oauth_accounts_user ON oauth_accounts(user_id);
|
||||||
CREATE UNIQUE INDEX IF NOT EXISTS idx_oauth_accounts_user_provider
|
CREATE UNIQUE INDEX IF NOT EXISTS idx_oauth_accounts_user_provider
|
||||||
ON oauth_accounts(user_id, provider);
|
ON oauth_accounts(user_id, provider);
|
||||||
|
|
||||||
|
-- FK: user_id columns -> users(id) (nullable = legacy rows; ON DELETE SET NULL)
|
||||||
|
DO $$ BEGIN
|
||||||
|
IF NOT EXISTS (
|
||||||
|
SELECT 1 FROM pg_constraint WHERE conname = 'fk_entries_user_id'
|
||||||
|
) THEN
|
||||||
|
ALTER TABLE entries
|
||||||
|
ADD CONSTRAINT fk_entries_user_id
|
||||||
|
FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE SET NULL;
|
||||||
|
END IF;
|
||||||
|
END $$;
|
||||||
|
|
||||||
|
DO $$ BEGIN
|
||||||
|
IF NOT EXISTS (
|
||||||
|
SELECT 1 FROM pg_constraint WHERE conname = 'fk_entries_history_user_id'
|
||||||
|
) THEN
|
||||||
|
ALTER TABLE entries_history
|
||||||
|
ADD CONSTRAINT fk_entries_history_user_id
|
||||||
|
FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE SET NULL;
|
||||||
|
END IF;
|
||||||
|
END $$;
|
||||||
|
|
||||||
|
DO $$ BEGIN
|
||||||
|
IF NOT EXISTS (
|
||||||
|
SELECT 1 FROM pg_constraint WHERE conname = 'fk_audit_log_user_id'
|
||||||
|
) THEN
|
||||||
|
ALTER TABLE audit_log
|
||||||
|
ADD CONSTRAINT fk_audit_log_user_id
|
||||||
|
FOREIGN KEY (user_id) REFERENCES users(id) ON DELETE SET NULL;
|
||||||
|
END IF;
|
||||||
|
END $$;
|
||||||
"#,
|
"#,
|
||||||
)
|
)
|
||||||
.execute(pool)
|
.execute(pool)
|
||||||
|
|||||||
@@ -1,6 +1,6 @@
|
|||||||
[package]
|
[package]
|
||||||
name = "secrets-mcp"
|
name = "secrets-mcp"
|
||||||
version = "0.1.11"
|
version = "0.2.1"
|
||||||
edition.workspace = true
|
edition.workspace = true
|
||||||
|
|
||||||
[[bin]]
|
[[bin]]
|
||||||
@@ -19,6 +19,8 @@ axum-extra = { version = "0.10", features = ["typed-header"] }
|
|||||||
tower = "0.5"
|
tower = "0.5"
|
||||||
tower-http = { version = "0.6", features = ["cors", "trace"] }
|
tower-http = { version = "0.6", features = ["cors", "trace"] }
|
||||||
tower-sessions = "0.14"
|
tower-sessions = "0.14"
|
||||||
|
tower-sessions-sqlx-store-chrono = { version = "0.14", features = ["postgres"] }
|
||||||
|
time = "0.3"
|
||||||
|
|
||||||
# OAuth (manual token exchange via reqwest)
|
# OAuth (manual token exchange via reqwest)
|
||||||
reqwest.workspace = true
|
reqwest.workspace = true
|
||||||
|
|||||||
@@ -5,11 +5,11 @@ use axum::{
|
|||||||
body::{Body, Bytes, to_bytes},
|
body::{Body, Bytes, to_bytes},
|
||||||
extract::{ConnectInfo, Request},
|
extract::{ConnectInfo, Request},
|
||||||
http::{
|
http::{
|
||||||
HeaderMap, Method,
|
HeaderMap, Method, StatusCode,
|
||||||
header::{CONTENT_LENGTH, CONTENT_TYPE, USER_AGENT},
|
header::{CONTENT_LENGTH, CONTENT_TYPE, USER_AGENT},
|
||||||
},
|
},
|
||||||
middleware::Next,
|
middleware::Next,
|
||||||
response::Response,
|
response::{IntoResponse, Response},
|
||||||
};
|
};
|
||||||
|
|
||||||
/// Axum middleware that logs structured info for every HTTP request.
|
/// Axum middleware that logs structured info for every HTTP request.
|
||||||
@@ -68,10 +68,23 @@ pub async fn request_logging_middleware(req: Request, next: Next) -> Response {
|
|||||||
}
|
}
|
||||||
Err(e) => {
|
Err(e) => {
|
||||||
tracing::warn!(path, error = %e, "failed to buffer MCP request body for logging");
|
tracing::warn!(path, error = %e, "failed to buffer MCP request body for logging");
|
||||||
// Reconstruct with empty body; request was consumed — return 500.
|
let elapsed = start.elapsed().as_millis();
|
||||||
// This branch is highly unlikely in practice.
|
tracing::info!(
|
||||||
let resp = next.run(Request::from_parts(parts, Body::empty())).await;
|
method = method.as_str(),
|
||||||
return resp;
|
path,
|
||||||
|
status = StatusCode::INTERNAL_SERVER_ERROR.as_u16(),
|
||||||
|
elapsed_ms = elapsed,
|
||||||
|
client_ip = ip.as_deref(),
|
||||||
|
ua = ua.as_deref(),
|
||||||
|
content_length = content_len,
|
||||||
|
mcp_session = mcp_session.as_deref(),
|
||||||
|
"mcp request",
|
||||||
|
);
|
||||||
|
return (
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR,
|
||||||
|
"failed to read request body",
|
||||||
|
)
|
||||||
|
.into_response();
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -15,8 +15,11 @@ use rmcp::transport::streamable_http_server::{
|
|||||||
use sqlx::PgPool;
|
use sqlx::PgPool;
|
||||||
use tower_http::cors::{Any, CorsLayer};
|
use tower_http::cors::{Any, CorsLayer};
|
||||||
use tower_sessions::cookie::SameSite;
|
use tower_sessions::cookie::SameSite;
|
||||||
use tower_sessions::{MemoryStore, SessionManagerLayer};
|
use tower_sessions::session_store::ExpiredDeletion;
|
||||||
|
use tower_sessions::{Expiry, SessionManagerLayer};
|
||||||
|
use tower_sessions_sqlx_store_chrono::PostgresStore;
|
||||||
use tracing_subscriber::EnvFilter;
|
use tracing_subscriber::EnvFilter;
|
||||||
|
use tracing_subscriber::fmt::time::FormatTime;
|
||||||
|
|
||||||
use secrets_core::config::resolve_db_url;
|
use secrets_core::config::resolve_db_url;
|
||||||
use secrets_core::db::{create_pool, migrate};
|
use secrets_core::db::{create_pool, migrate};
|
||||||
@@ -47,12 +50,27 @@ fn load_oauth_config(prefix: &str, base_url: &str, path: &str) -> Option<OAuthCo
|
|||||||
})
|
})
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/// Log line timestamps in the process local timezone (honors `TZ` / system zone).
|
||||||
|
#[derive(Clone, Copy, Default)]
|
||||||
|
struct LocalRfc3339Time;
|
||||||
|
|
||||||
|
impl FormatTime for LocalRfc3339Time {
|
||||||
|
fn format_time(&self, w: &mut tracing_subscriber::fmt::format::Writer<'_>) -> std::fmt::Result {
|
||||||
|
write!(
|
||||||
|
w,
|
||||||
|
"{}",
|
||||||
|
chrono::Local::now().to_rfc3339_opts(chrono::SecondsFormat::Millis, false)
|
||||||
|
)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
#[tokio::main]
|
#[tokio::main]
|
||||||
async fn main() -> Result<()> {
|
async fn main() -> Result<()> {
|
||||||
// Load .env if present
|
// Load .env if present
|
||||||
let _ = dotenvy::dotenv();
|
let _ = dotenvy::dotenv();
|
||||||
|
|
||||||
tracing_subscriber::fmt()
|
tracing_subscriber::fmt()
|
||||||
|
.with_timer(LocalRfc3339Time)
|
||||||
.with_env_filter(
|
.with_env_filter(
|
||||||
EnvFilter::try_from_default_env()
|
EnvFilter::try_from_default_env()
|
||||||
.unwrap_or_else(|_| "secrets_mcp=info,tower_http=info".into()),
|
.unwrap_or_else(|_| "secrets_mcp=info,tower_http=info".into()),
|
||||||
@@ -72,7 +90,8 @@ async fn main() -> Result<()> {
|
|||||||
|
|
||||||
// ── Configuration ─────────────────────────────────────────────────────────
|
// ── Configuration ─────────────────────────────────────────────────────────
|
||||||
let base_url = load_env_var("BASE_URL").unwrap_or_else(|| "http://localhost:9315".to_string());
|
let base_url = load_env_var("BASE_URL").unwrap_or_else(|| "http://localhost:9315".to_string());
|
||||||
let bind_addr = load_env_var("SECRETS_MCP_BIND").unwrap_or_else(|| "0.0.0.0:9315".to_string());
|
let bind_addr =
|
||||||
|
load_env_var("SECRETS_MCP_BIND").unwrap_or_else(|| "127.0.0.1:9315".to_string());
|
||||||
|
|
||||||
// ── OAuth providers ───────────────────────────────────────────────────────
|
// ── OAuth providers ───────────────────────────────────────────────────────
|
||||||
let google_config = load_oauth_config("GOOGLE", &base_url, "/auth/google/callback");
|
let google_config = load_oauth_config("GOOGLE", &base_url, "/auth/google/callback");
|
||||||
@@ -83,12 +102,23 @@ async fn main() -> Result<()> {
|
|||||||
);
|
);
|
||||||
}
|
}
|
||||||
|
|
||||||
// ── Session store ─────────────────────────────────────────────────────────
|
// ── Session store (PostgreSQL-backed) ─────────────────────────────────────
|
||||||
let session_store = MemoryStore::default();
|
let session_store = PostgresStore::new(pool.clone());
|
||||||
|
session_store
|
||||||
|
.migrate()
|
||||||
|
.await
|
||||||
|
.context("failed to run session table migration")?;
|
||||||
|
// Prune expired rows every hour; task is aborted when the server shuts down.
|
||||||
|
let session_cleanup = tokio::spawn(
|
||||||
|
session_store
|
||||||
|
.clone()
|
||||||
|
.continuously_delete_expired(tokio::time::Duration::from_secs(3600)),
|
||||||
|
);
|
||||||
// Strict would drop the session cookie on redirect from Google → our origin (cross-site nav).
|
// Strict would drop the session cookie on redirect from Google → our origin (cross-site nav).
|
||||||
let session_layer = SessionManagerLayer::new(session_store)
|
let session_layer = SessionManagerLayer::new(session_store)
|
||||||
.with_secure(base_url.starts_with("https://"))
|
.with_secure(base_url.starts_with("https://"))
|
||||||
.with_same_site(SameSite::Lax);
|
.with_same_site(SameSite::Lax)
|
||||||
|
.with_expiry(Expiry::OnInactivity(time::Duration::days(14)));
|
||||||
|
|
||||||
// ── App state ─────────────────────────────────────────────────────────────
|
// ── App state ─────────────────────────────────────────────────────────────
|
||||||
let app_state = AppState {
|
let app_state = AppState {
|
||||||
@@ -149,6 +179,7 @@ async fn main() -> Result<()> {
|
|||||||
.await
|
.await
|
||||||
.context("server error")?;
|
.context("server error")?;
|
||||||
|
|
||||||
|
session_cleanup.abort();
|
||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ use serde::Deserialize;
|
|||||||
use sqlx::PgPool;
|
use sqlx::PgPool;
|
||||||
use uuid::Uuid;
|
use uuid::Uuid;
|
||||||
|
|
||||||
|
use secrets_core::models::ExportFormat;
|
||||||
use secrets_core::service::{
|
use secrets_core::service::{
|
||||||
add::{AddParams, run as svc_add},
|
add::{AddParams, run as svc_add},
|
||||||
delete::{DeleteParams, run as svc_delete},
|
delete::{DeleteParams, run as svc_delete},
|
||||||
@@ -30,6 +31,32 @@ use secrets_core::service::{
|
|||||||
|
|
||||||
use crate::auth::AuthUser;
|
use crate::auth::AuthUser;
|
||||||
|
|
||||||
|
// ── MCP client-facing errors (no internal details) ───────────────────────────
|
||||||
|
|
||||||
|
fn mcp_err_missing_http_parts() -> rmcp::ErrorData {
|
||||||
|
rmcp::ErrorData::internal_error("Invalid MCP request context.", None)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn mcp_err_internal_logged(
|
||||||
|
tool: &'static str,
|
||||||
|
user_id: Option<Uuid>,
|
||||||
|
err: impl std::fmt::Display,
|
||||||
|
) -> rmcp::ErrorData {
|
||||||
|
tracing::warn!(tool, ?user_id, error = %err, "tool call failed");
|
||||||
|
rmcp::ErrorData::internal_error(
|
||||||
|
"Request failed due to a server error. Check service logs if you need details.",
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
fn mcp_err_invalid_encryption_key_logged(err: impl std::fmt::Display) -> rmcp::ErrorData {
|
||||||
|
tracing::warn!(error = %err, "invalid X-Encryption-Key");
|
||||||
|
rmcp::ErrorData::invalid_request(
|
||||||
|
"Invalid X-Encryption-Key: must be exactly 64 hexadecimal characters (32-byte key).",
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
// ── Shared state ──────────────────────────────────────────────────────────────
|
// ── Shared state ──────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
#[derive(Clone)]
|
#[derive(Clone)]
|
||||||
@@ -51,7 +78,7 @@ impl SecretsService {
|
|||||||
let parts = ctx
|
let parts = ctx
|
||||||
.extensions
|
.extensions
|
||||||
.get::<http::request::Parts>()
|
.get::<http::request::Parts>()
|
||||||
.ok_or_else(|| rmcp::ErrorData::internal_error("Missing HTTP parts", None))?;
|
.ok_or_else(mcp_err_missing_http_parts)?;
|
||||||
Ok(parts.extensions.get::<AuthUser>().map(|a| a.user_id))
|
Ok(parts.extensions.get::<AuthUser>().map(|a| a.user_id))
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -60,7 +87,7 @@ impl SecretsService {
|
|||||||
let parts = ctx
|
let parts = ctx
|
||||||
.extensions
|
.extensions
|
||||||
.get::<http::request::Parts>()
|
.get::<http::request::Parts>()
|
||||||
.ok_or_else(|| rmcp::ErrorData::internal_error("Missing HTTP parts", None))?;
|
.ok_or_else(mcp_err_missing_http_parts)?;
|
||||||
parts
|
parts
|
||||||
.extensions
|
.extensions
|
||||||
.get::<AuthUser>()
|
.get::<AuthUser>()
|
||||||
@@ -74,7 +101,7 @@ impl SecretsService {
|
|||||||
let parts = ctx
|
let parts = ctx
|
||||||
.extensions
|
.extensions
|
||||||
.get::<http::request::Parts>()
|
.get::<http::request::Parts>()
|
||||||
.ok_or_else(|| rmcp::ErrorData::internal_error("Missing HTTP parts", None))?;
|
.ok_or_else(mcp_err_missing_http_parts)?;
|
||||||
let hex_str = parts
|
let hex_str = parts
|
||||||
.headers
|
.headers
|
||||||
.get("x-encryption-key")
|
.get("x-encryption-key")
|
||||||
@@ -89,8 +116,29 @@ impl SecretsService {
|
|||||||
.map_err(|_| {
|
.map_err(|_| {
|
||||||
rmcp::ErrorData::invalid_request("Invalid X-Encryption-Key header value", None)
|
rmcp::ErrorData::invalid_request("Invalid X-Encryption-Key header value", None)
|
||||||
})?;
|
})?;
|
||||||
|
let trimmed = hex_str.trim();
|
||||||
|
if trimmed.len() != 64 {
|
||||||
|
tracing::warn!(
|
||||||
|
got_len = trimmed.len(),
|
||||||
|
"X-Encryption-Key has wrong length after trim"
|
||||||
|
);
|
||||||
|
return Err(rmcp::ErrorData::invalid_request(
|
||||||
|
format!(
|
||||||
|
"X-Encryption-Key must be exactly 64 hex characters (32-byte key), got {} characters.",
|
||||||
|
trimmed.len()
|
||||||
|
),
|
||||||
|
None,
|
||||||
|
));
|
||||||
|
}
|
||||||
|
if !trimmed.chars().all(|c| c.is_ascii_hexdigit()) {
|
||||||
|
tracing::warn!("X-Encryption-Key contains non-hexadecimal characters");
|
||||||
|
return Err(rmcp::ErrorData::invalid_request(
|
||||||
|
"X-Encryption-Key contains non-hexadecimal characters.",
|
||||||
|
None,
|
||||||
|
));
|
||||||
|
}
|
||||||
secrets_core::crypto::extract_key_from_hex(hex_str)
|
secrets_core::crypto::extract_key_from_hex(hex_str)
|
||||||
.map_err(|e| rmcp::ErrorData::invalid_request(e.to_string(), None))
|
.map_err(mcp_err_invalid_encryption_key_logged)
|
||||||
}
|
}
|
||||||
|
|
||||||
/// Require both user_id and encryption key.
|
/// Require both user_id and encryption key.
|
||||||
@@ -250,8 +298,13 @@ struct EnvMapInput {
|
|||||||
#[tool_router]
|
#[tool_router]
|
||||||
impl SecretsService {
|
impl SecretsService {
|
||||||
#[tool(
|
#[tool(
|
||||||
description = "Search entries in the secrets store. Returns entries with metadata and \
|
description = "Search entries in the secrets store. Requires Bearer API key. Returns \
|
||||||
secret field names (not values). Use secrets_get to decrypt secret values."
|
entries with metadata and secret field names (not values). Use secrets_get to decrypt secret values.",
|
||||||
|
annotations(
|
||||||
|
title = "Search Secrets",
|
||||||
|
read_only_hint = true,
|
||||||
|
idempotent_hint = true
|
||||||
|
)
|
||||||
)]
|
)]
|
||||||
async fn secrets_search(
|
async fn secrets_search(
|
||||||
&self,
|
&self,
|
||||||
@@ -259,7 +312,7 @@ impl SecretsService {
|
|||||||
ctx: RequestContext<RoleServer>,
|
ctx: RequestContext<RoleServer>,
|
||||||
) -> Result<CallToolResult, rmcp::ErrorData> {
|
) -> Result<CallToolResult, rmcp::ErrorData> {
|
||||||
let t = Instant::now();
|
let t = Instant::now();
|
||||||
let user_id = Self::user_id_from_ctx(&ctx)?;
|
let user_id = Self::require_user_id(&ctx)?;
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
tool = "secrets_search",
|
tool = "secrets_search",
|
||||||
?user_id,
|
?user_id,
|
||||||
@@ -281,14 +334,11 @@ impl SecretsService {
|
|||||||
sort: input.sort.as_deref().unwrap_or("name"),
|
sort: input.sort.as_deref().unwrap_or("name"),
|
||||||
limit: input.limit.unwrap_or(20),
|
limit: input.limit.unwrap_or(20),
|
||||||
offset: input.offset.unwrap_or(0),
|
offset: input.offset.unwrap_or(0),
|
||||||
user_id,
|
user_id: Some(user_id),
|
||||||
},
|
},
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_search", Some(user_id), e))?;
|
||||||
tracing::warn!(tool = "secrets_search", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
let summary = input.summary.unwrap_or(false);
|
let summary = input.summary.unwrap_or(false);
|
||||||
let entries: Vec<serde_json::Value> = result
|
let entries: Vec<serde_json::Value> = result
|
||||||
@@ -341,7 +391,12 @@ impl SecretsService {
|
|||||||
#[tool(
|
#[tool(
|
||||||
description = "Get decrypted secret field values for an entry. Requires your \
|
description = "Get decrypted secret field values for an entry. Requires your \
|
||||||
encryption key via X-Encryption-Key header (64 hex chars, PBKDF2-derived). \
|
encryption key via X-Encryption-Key header (64 hex chars, PBKDF2-derived). \
|
||||||
Returns all fields, or a specific field if 'field' is provided."
|
Returns all fields, or a specific field if 'field' is provided.",
|
||||||
|
annotations(
|
||||||
|
title = "Get Secret Values",
|
||||||
|
read_only_hint = true,
|
||||||
|
idempotent_hint = true
|
||||||
|
)
|
||||||
)]
|
)]
|
||||||
async fn secrets_get(
|
async fn secrets_get(
|
||||||
&self,
|
&self,
|
||||||
@@ -371,10 +426,7 @@ impl SecretsService {
|
|||||||
Some(user_id),
|
Some(user_id),
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_get", Some(user_id), e))?;
|
||||||
tracing::warn!(tool = "secrets_get", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
tool = "secrets_get",
|
tool = "secrets_get",
|
||||||
@@ -395,10 +447,7 @@ impl SecretsService {
|
|||||||
Some(user_id),
|
Some(user_id),
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_get", Some(user_id), e))?;
|
||||||
tracing::warn!(tool = "secrets_get", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
let count = secrets.len();
|
let count = secrets.len();
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
@@ -416,7 +465,8 @@ impl SecretsService {
|
|||||||
#[tool(
|
#[tool(
|
||||||
description = "Add or upsert an entry with metadata and encrypted secret fields. \
|
description = "Add or upsert an entry with metadata and encrypted secret fields. \
|
||||||
Requires X-Encryption-Key header. \
|
Requires X-Encryption-Key header. \
|
||||||
Meta and secret values use 'key=value', 'key=@file', or 'key:=<json>' format."
|
Meta and secret values use 'key=value', 'key=@file', or 'key:=<json>' format.",
|
||||||
|
annotations(title = "Add Secret Entry")
|
||||||
)]
|
)]
|
||||||
async fn secrets_add(
|
async fn secrets_add(
|
||||||
&self,
|
&self,
|
||||||
@@ -452,10 +502,7 @@ impl SecretsService {
|
|||||||
&user_key,
|
&user_key,
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_add", Some(user_id), e))?;
|
||||||
tracing::warn!(tool = "secrets_add", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
tool = "secrets_add",
|
tool = "secrets_add",
|
||||||
@@ -472,7 +519,8 @@ impl SecretsService {
|
|||||||
|
|
||||||
#[tool(
|
#[tool(
|
||||||
description = "Incrementally update an existing entry. Requires X-Encryption-Key header. \
|
description = "Incrementally update an existing entry. Requires X-Encryption-Key header. \
|
||||||
Only the fields you specify are changed; everything else is preserved."
|
Only the fields you specify are changed; everything else is preserved.",
|
||||||
|
annotations(title = "Update Secret Entry")
|
||||||
)]
|
)]
|
||||||
async fn secrets_update(
|
async fn secrets_update(
|
||||||
&self,
|
&self,
|
||||||
@@ -514,10 +562,7 @@ impl SecretsService {
|
|||||||
&user_key,
|
&user_key,
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_update", Some(user_id), e))?;
|
||||||
tracing::warn!(tool = "secrets_update", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
tool = "secrets_update",
|
tool = "secrets_update",
|
||||||
@@ -534,7 +579,8 @@ impl SecretsService {
|
|||||||
|
|
||||||
#[tool(
|
#[tool(
|
||||||
description = "Delete one entry (specify namespace+kind+name) or bulk delete all \
|
description = "Delete one entry (specify namespace+kind+name) or bulk delete all \
|
||||||
entries matching namespace+kind. Use dry_run=true to preview."
|
entries matching namespace+kind. Use dry_run=true to preview.",
|
||||||
|
annotations(title = "Delete Secret Entry", destructive_hint = true)
|
||||||
)]
|
)]
|
||||||
async fn secrets_delete(
|
async fn secrets_delete(
|
||||||
&self,
|
&self,
|
||||||
@@ -564,10 +610,7 @@ impl SecretsService {
|
|||||||
},
|
},
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_delete", user_id, e))?;
|
||||||
tracing::warn!(tool = "secrets_delete", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
tool = "secrets_delete",
|
tool = "secrets_delete",
|
||||||
@@ -582,7 +625,12 @@ impl SecretsService {
|
|||||||
|
|
||||||
#[tool(
|
#[tool(
|
||||||
description = "View change history for an entry. Returns a list of versions with \
|
description = "View change history for an entry. Returns a list of versions with \
|
||||||
actions and timestamps."
|
actions and timestamps.",
|
||||||
|
annotations(
|
||||||
|
title = "View Secret History",
|
||||||
|
read_only_hint = true,
|
||||||
|
idempotent_hint = true
|
||||||
|
)
|
||||||
)]
|
)]
|
||||||
async fn secrets_history(
|
async fn secrets_history(
|
||||||
&self,
|
&self,
|
||||||
@@ -609,10 +657,7 @@ impl SecretsService {
|
|||||||
user_id,
|
user_id,
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_history", user_id, e))?;
|
||||||
tracing::warn!(tool = "secrets_history", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
tool = "secrets_history",
|
tool = "secrets_history",
|
||||||
@@ -626,7 +671,8 @@ impl SecretsService {
|
|||||||
|
|
||||||
#[tool(
|
#[tool(
|
||||||
description = "Rollback an entry to a previous version. Requires X-Encryption-Key header. \
|
description = "Rollback an entry to a previous version. Requires X-Encryption-Key header. \
|
||||||
Omit to_version to restore the most recent snapshot."
|
Omit to_version to restore the most recent snapshot.",
|
||||||
|
annotations(title = "Rollback Secret Entry", destructive_hint = true)
|
||||||
)]
|
)]
|
||||||
async fn secrets_rollback(
|
async fn secrets_rollback(
|
||||||
&self,
|
&self,
|
||||||
@@ -655,10 +701,7 @@ impl SecretsService {
|
|||||||
Some(user_id),
|
Some(user_id),
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_rollback", Some(user_id), e))?;
|
||||||
tracing::warn!(tool = "secrets_rollback", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
tool = "secrets_rollback",
|
tool = "secrets_rollback",
|
||||||
@@ -672,7 +715,12 @@ impl SecretsService {
|
|||||||
|
|
||||||
#[tool(
|
#[tool(
|
||||||
description = "Export matching entries with decrypted secrets as JSON/TOML/YAML string. \
|
description = "Export matching entries with decrypted secrets as JSON/TOML/YAML string. \
|
||||||
Requires X-Encryption-Key header. Useful for backup or data migration."
|
Requires X-Encryption-Key header. Useful for backup or data migration.",
|
||||||
|
annotations(
|
||||||
|
title = "Export Secrets",
|
||||||
|
read_only_hint = true,
|
||||||
|
idempotent_hint = true
|
||||||
|
)
|
||||||
)]
|
)]
|
||||||
async fn secrets_export(
|
async fn secrets_export(
|
||||||
&self,
|
&self,
|
||||||
@@ -706,15 +754,23 @@ impl SecretsService {
|
|||||||
Some(&user_key),
|
Some(&user_key),
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_export", Some(user_id), e))?;
|
||||||
tracing::warn!(tool = "secrets_export", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
let serialized = format
|
let fmt = format.parse::<ExportFormat>().map_err(|e| {
|
||||||
.parse::<secrets_core::models::ExportFormat>()
|
tracing::warn!(
|
||||||
.and_then(|fmt| fmt.serialize(&data))
|
tool = "secrets_export",
|
||||||
.map_err(|e| rmcp::ErrorData::internal_error(e.to_string(), None))?;
|
?user_id,
|
||||||
|
error = %e,
|
||||||
|
"invalid export format"
|
||||||
|
);
|
||||||
|
rmcp::ErrorData::invalid_request(
|
||||||
|
"Invalid export format. Use json, toml, or yaml.",
|
||||||
|
None,
|
||||||
|
)
|
||||||
|
})?;
|
||||||
|
let serialized = fmt
|
||||||
|
.serialize(&data)
|
||||||
|
.map_err(|e| mcp_err_internal_logged("secrets_export", Some(user_id), e))?;
|
||||||
|
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
tool = "secrets_export",
|
tool = "secrets_export",
|
||||||
@@ -729,7 +785,8 @@ impl SecretsService {
|
|||||||
#[tool(
|
#[tool(
|
||||||
description = "Build the environment variable map from entry secrets with decrypted \
|
description = "Build the environment variable map from entry secrets with decrypted \
|
||||||
plaintext values. Requires X-Encryption-Key header. \
|
plaintext values. Requires X-Encryption-Key header. \
|
||||||
Returns a JSON object of VAR_NAME -> plaintext_value ready for injection."
|
Returns a JSON object of VAR_NAME -> plaintext_value ready for injection.",
|
||||||
|
annotations(title = "Build Env Map", read_only_hint = true, idempotent_hint = true)
|
||||||
)]
|
)]
|
||||||
async fn secrets_env_map(
|
async fn secrets_env_map(
|
||||||
&self,
|
&self,
|
||||||
@@ -761,10 +818,7 @@ impl SecretsService {
|
|||||||
Some(user_id),
|
Some(user_id),
|
||||||
)
|
)
|
||||||
.await
|
.await
|
||||||
.map_err(|e| {
|
.map_err(|e| mcp_err_internal_logged("secrets_env_map", Some(user_id), e))?;
|
||||||
tracing::warn!(tool = "secrets_env_map", ?user_id, error = %e, "tool call failed");
|
|
||||||
rmcp::ErrorData::internal_error(e.to_string(), None)
|
|
||||||
})?;
|
|
||||||
|
|
||||||
let entry_count = env_map.len();
|
let entry_count = env_map.len();
|
||||||
tracing::info!(
|
tracing::info!(
|
||||||
@@ -785,13 +839,17 @@ impl SecretsService {
|
|||||||
impl ServerHandler for SecretsService {
|
impl ServerHandler for SecretsService {
|
||||||
fn get_info(&self) -> InitializeResult {
|
fn get_info(&self) -> InitializeResult {
|
||||||
let mut info = InitializeResult::new(ServerCapabilities::builder().enable_tools().build());
|
let mut info = InitializeResult::new(ServerCapabilities::builder().enable_tools().build());
|
||||||
info.server_info = Implementation::new("secrets-mcp", env!("CARGO_PKG_VERSION"));
|
info.server_info = Implementation::new("secrets-mcp", env!("CARGO_PKG_VERSION"))
|
||||||
info.protocol_version = ProtocolVersion::V_2025_03_26;
|
.with_title("Secrets MCP")
|
||||||
|
.with_description(
|
||||||
|
"Secure cross-device secrets and configuration management with encrypted secret fields.",
|
||||||
|
);
|
||||||
|
info.protocol_version = ProtocolVersion::V_2025_06_18;
|
||||||
info.instructions = Some(
|
info.instructions = Some(
|
||||||
"Manage cross-device secrets and configuration securely. \
|
"Manage cross-device secrets and configuration securely. \
|
||||||
Data is encrypted with your passphrase-derived key. \
|
Data is encrypted with your passphrase-derived key. \
|
||||||
Include your 64-char hex key in the X-Encryption-Key header for all read/write operations. \
|
Include your 64-char hex key in the X-Encryption-Key header for all read/write operations. \
|
||||||
Use secrets_search to discover entries (no key needed), \
|
Use secrets_search to discover entries (Bearer token required; encryption key not needed), \
|
||||||
secrets_get to decrypt secret values, \
|
secrets_get to decrypt secret values, \
|
||||||
and secrets_add/secrets_update to write encrypted secrets."
|
and secrets_add/secrets_update to write encrypted secrets."
|
||||||
.to_string(),
|
.to_string(),
|
||||||
|
|||||||
@@ -76,12 +76,22 @@ fn google_cfg(state: &AppState) -> Option<&OAuthConfig> {
|
|||||||
}
|
}
|
||||||
|
|
||||||
async fn current_user_id(session: &Session) -> Option<Uuid> {
|
async fn current_user_id(session: &Session) -> Option<Uuid> {
|
||||||
session
|
match session.get::<String>(SESSION_USER_ID).await {
|
||||||
.get::<String>(SESSION_USER_ID)
|
Ok(opt) => match opt {
|
||||||
.await
|
Some(s) => match Uuid::parse_str(&s) {
|
||||||
.ok()
|
Ok(id) => Some(id),
|
||||||
.flatten()
|
Err(e) => {
|
||||||
.and_then(|s| Uuid::parse_str(&s).ok())
|
tracing::warn!(error = %e, user_id_str = %s, "invalid user_id UUID in session");
|
||||||
|
None
|
||||||
|
}
|
||||||
|
},
|
||||||
|
None => None,
|
||||||
|
},
|
||||||
|
Err(e) => {
|
||||||
|
tracing::warn!(error = %e, "failed to read user_id from session");
|
||||||
|
None
|
||||||
|
}
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
fn request_client_ip(headers: &HeaderMap, connect_info: ConnectInfo<SocketAddr>) -> Option<String> {
|
fn request_client_ip(headers: &HeaderMap, connect_info: ConnectInfo<SocketAddr>) -> Option<String> {
|
||||||
@@ -112,6 +122,9 @@ fn request_user_agent(headers: &HeaderMap) -> Option<String> {
|
|||||||
|
|
||||||
pub fn web_router() -> Router<AppState> {
|
pub fn web_router() -> Router<AppState> {
|
||||||
Router::new()
|
Router::new()
|
||||||
|
.route("/robots.txt", get(robots_txt))
|
||||||
|
.route("/llms.txt", get(llms_txt))
|
||||||
|
.route("/ai.txt", get(ai_txt))
|
||||||
.route("/favicon.svg", get(favicon_svg))
|
.route("/favicon.svg", get(favicon_svg))
|
||||||
.route(
|
.route(
|
||||||
"/favicon.ico",
|
"/favicon.ico",
|
||||||
@@ -139,6 +152,33 @@ pub fn web_router() -> Router<AppState> {
|
|||||||
.route("/api/apikey/regenerate", post(api_apikey_regenerate))
|
.route("/api/apikey/regenerate", post(api_apikey_regenerate))
|
||||||
}
|
}
|
||||||
|
|
||||||
|
fn text_asset_response(content: &'static str, content_type: &'static str) -> Response {
|
||||||
|
Response::builder()
|
||||||
|
.status(StatusCode::OK)
|
||||||
|
.header(header::CONTENT_TYPE, content_type)
|
||||||
|
.header(header::CACHE_CONTROL, "public, max-age=86400")
|
||||||
|
.body(Body::from(content))
|
||||||
|
.expect("text asset response")
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn robots_txt() -> Response {
|
||||||
|
text_asset_response(
|
||||||
|
include_str!("../static/robots.txt"),
|
||||||
|
"text/plain; charset=utf-8",
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn llms_txt() -> Response {
|
||||||
|
text_asset_response(
|
||||||
|
include_str!("../static/llms.txt"),
|
||||||
|
"text/markdown; charset=utf-8",
|
||||||
|
)
|
||||||
|
}
|
||||||
|
|
||||||
|
async fn ai_txt() -> Response {
|
||||||
|
llms_txt().await
|
||||||
|
}
|
||||||
|
|
||||||
async fn favicon_svg() -> Response {
|
async fn favicon_svg() -> Response {
|
||||||
Response::builder()
|
Response::builder()
|
||||||
.status(StatusCode::OK)
|
.status(StatusCode::OK)
|
||||||
@@ -177,7 +217,10 @@ async fn auth_google(
|
|||||||
session
|
session
|
||||||
.insert(SESSION_OAUTH_STATE, &oauth_state)
|
.insert(SESSION_OAUTH_STATE, &oauth_state)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
|
.map_err(|e| {
|
||||||
|
tracing::error!(error = %e, "failed to insert oauth_state into session");
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?;
|
||||||
|
|
||||||
let url = google_auth_url(config, &oauth_state);
|
let url = google_auth_url(config, &oauth_state);
|
||||||
Ok(Redirect::to(&url).into_response())
|
Ok(Redirect::to(&url).into_response())
|
||||||
@@ -251,10 +294,10 @@ where
|
|||||||
return Ok(Redirect::to("/?error=oauth_missing_state").into_response());
|
return Ok(Redirect::to("/?error=oauth_missing_state").into_response());
|
||||||
};
|
};
|
||||||
|
|
||||||
let expected_state: Option<String> = session
|
let expected_state: Option<String> = session.get(SESSION_OAUTH_STATE).await.map_err(|e| {
|
||||||
.get(SESSION_OAUTH_STATE)
|
tracing::error!(provider, error = %e, "failed to read oauth_state from session");
|
||||||
.await
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
|
})?;
|
||||||
if expected_state.as_deref() != Some(returned_state) {
|
if expected_state.as_deref() != Some(returned_state) {
|
||||||
tracing::warn!(
|
tracing::warn!(
|
||||||
provider,
|
provider,
|
||||||
@@ -263,7 +306,9 @@ where
|
|||||||
);
|
);
|
||||||
return Ok(Redirect::to("/?error=oauth_state").into_response());
|
return Ok(Redirect::to("/?error=oauth_state").into_response());
|
||||||
}
|
}
|
||||||
session.remove::<String>(SESSION_OAUTH_STATE).await.ok();
|
if let Err(e) = session.remove::<String>(SESSION_OAUTH_STATE).await {
|
||||||
|
tracing::warn!(provider, error = %e, "failed to remove oauth_state from session");
|
||||||
|
}
|
||||||
|
|
||||||
let config = match provider {
|
let config = match provider {
|
||||||
"google" => state
|
"google" => state
|
||||||
@@ -280,17 +325,25 @@ where
|
|||||||
StatusCode::INTERNAL_SERVER_ERROR
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
})?;
|
})?;
|
||||||
|
|
||||||
let bind_mode: bool = session
|
let bind_mode: bool = match session.get::<bool>(SESSION_OAUTH_BIND_MODE).await {
|
||||||
.get(SESSION_OAUTH_BIND_MODE)
|
Ok(v) => v.unwrap_or(false),
|
||||||
.await
|
Err(e) => {
|
||||||
.unwrap_or(None)
|
tracing::error!(
|
||||||
.unwrap_or(false);
|
provider,
|
||||||
|
error = %e,
|
||||||
|
"failed to read oauth_bind_mode from session"
|
||||||
|
);
|
||||||
|
return Err(StatusCode::INTERNAL_SERVER_ERROR);
|
||||||
|
}
|
||||||
|
};
|
||||||
|
|
||||||
if bind_mode {
|
if bind_mode {
|
||||||
let user_id = current_user_id(session)
|
let user_id = current_user_id(session)
|
||||||
.await
|
.await
|
||||||
.ok_or(StatusCode::UNAUTHORIZED)?;
|
.ok_or(StatusCode::UNAUTHORIZED)?;
|
||||||
session.remove::<bool>(SESSION_OAUTH_BIND_MODE).await.ok();
|
if let Err(e) = session.remove::<bool>(SESSION_OAUTH_BIND_MODE).await {
|
||||||
|
tracing::warn!(provider, error = %e, "failed to remove oauth_bind_mode from session after bind");
|
||||||
|
}
|
||||||
|
|
||||||
let profile = OAuthProfile {
|
let profile = OAuthProfile {
|
||||||
provider: user_info.provider,
|
provider: user_info.provider,
|
||||||
@@ -328,11 +381,25 @@ where
|
|||||||
session
|
session
|
||||||
.insert(SESSION_USER_ID, user.id.to_string())
|
.insert(SESSION_USER_ID, user.id.to_string())
|
||||||
.await
|
.await
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
|
.map_err(|e| {
|
||||||
|
tracing::error!(
|
||||||
|
error = %e,
|
||||||
|
user_id = %user.id,
|
||||||
|
"failed to insert user_id into session after OAuth"
|
||||||
|
);
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?;
|
||||||
session
|
session
|
||||||
.insert(SESSION_LOGIN_PROVIDER, &provider)
|
.insert(SESSION_LOGIN_PROVIDER, &provider)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
|
.map_err(|e| {
|
||||||
|
tracing::error!(
|
||||||
|
provider,
|
||||||
|
error = %e,
|
||||||
|
"failed to insert login_provider into session after OAuth"
|
||||||
|
);
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?;
|
||||||
|
|
||||||
log_login(
|
log_login(
|
||||||
&state.pool,
|
&state.pool,
|
||||||
@@ -350,7 +417,9 @@ where
|
|||||||
// ── Logout ────────────────────────────────────────────────────────────────────
|
// ── Logout ────────────────────────────────────────────────────────────────────
|
||||||
|
|
||||||
async fn auth_logout(session: Session) -> impl IntoResponse {
|
async fn auth_logout(session: Session) -> impl IntoResponse {
|
||||||
session.flush().await.ok();
|
if let Err(e) = session.flush().await {
|
||||||
|
tracing::warn!(error = %e, "failed to flush session on logout");
|
||||||
|
}
|
||||||
Redirect::to("/")
|
Redirect::to("/")
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -364,10 +433,10 @@ async fn dashboard(
|
|||||||
return Ok(Redirect::to("/").into_response());
|
return Ok(Redirect::to("/").into_response());
|
||||||
};
|
};
|
||||||
|
|
||||||
let user = match get_user_by_id(&state.pool, user_id)
|
let user = match get_user_by_id(&state.pool, user_id).await.map_err(|e| {
|
||||||
.await
|
tracing::error!(error = %e, %user_id, "failed to load user for dashboard");
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
{
|
})? {
|
||||||
Some(u) => u,
|
Some(u) => u,
|
||||||
None => return Ok(Redirect::to("/").into_response()),
|
None => return Ok(Redirect::to("/").into_response()),
|
||||||
};
|
};
|
||||||
@@ -391,10 +460,10 @@ async fn audit_page(
|
|||||||
return Ok(Redirect::to("/").into_response());
|
return Ok(Redirect::to("/").into_response());
|
||||||
};
|
};
|
||||||
|
|
||||||
let user = match get_user_by_id(&state.pool, user_id)
|
let user = match get_user_by_id(&state.pool, user_id).await.map_err(|e| {
|
||||||
.await
|
tracing::error!(error = %e, %user_id, "failed to load user for audit page");
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
{
|
})? {
|
||||||
Some(u) => u,
|
Some(u) => u,
|
||||||
None => return Ok(Redirect::to("/").into_response()),
|
None => return Ok(Redirect::to("/").into_response()),
|
||||||
};
|
};
|
||||||
@@ -439,7 +508,10 @@ async fn account_bind_google(
|
|||||||
session
|
session
|
||||||
.insert(SESSION_OAUTH_BIND_MODE, true)
|
.insert(SESSION_OAUTH_BIND_MODE, true)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
|
.map_err(|e| {
|
||||||
|
tracing::error!(error = %e, "failed to insert oauth_bind_mode into session");
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?;
|
||||||
|
|
||||||
let redirect_uri = format!("{}/account/bind/google/callback", state.base_url);
|
let redirect_uri = format!("{}/account/bind/google/callback", state.base_url);
|
||||||
let mut cfg = state
|
let mut cfg = state
|
||||||
@@ -448,7 +520,13 @@ async fn account_bind_google(
|
|||||||
.ok_or(StatusCode::SERVICE_UNAVAILABLE)?;
|
.ok_or(StatusCode::SERVICE_UNAVAILABLE)?;
|
||||||
cfg.redirect_uri = redirect_uri;
|
cfg.redirect_uri = redirect_uri;
|
||||||
let st = random_state();
|
let st = random_state();
|
||||||
session.insert(SESSION_OAUTH_STATE, &st).await.ok();
|
if let Err(e) = session.insert(SESSION_OAUTH_STATE, &st).await {
|
||||||
|
tracing::error!(error = %e, "failed to insert oauth_state for account bind flow");
|
||||||
|
if let Err(rm) = session.remove::<bool>(SESSION_OAUTH_BIND_MODE).await {
|
||||||
|
tracing::warn!(error = %rm, "failed to roll back oauth_bind_mode after oauth_state insert failure");
|
||||||
|
}
|
||||||
|
return Err(StatusCode::INTERNAL_SERVER_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
Ok(Redirect::to(&google_auth_url(&cfg, &st)).into_response())
|
Ok(Redirect::to(&google_auth_url(&cfg, &st)).into_response())
|
||||||
}
|
}
|
||||||
@@ -492,7 +570,10 @@ async fn account_unbind(
|
|||||||
let current_login_provider = session
|
let current_login_provider = session
|
||||||
.get::<String>(SESSION_LOGIN_PROVIDER)
|
.get::<String>(SESSION_LOGIN_PROVIDER)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
|
.map_err(|e| {
|
||||||
|
tracing::error!(error = %e, "failed to read login_provider from session");
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?;
|
||||||
|
|
||||||
unbind_oauth_account(
|
unbind_oauth_account(
|
||||||
&state.pool,
|
&state.pool,
|
||||||
@@ -532,7 +613,10 @@ async fn api_key_salt(
|
|||||||
|
|
||||||
let user = get_user_by_id(&state.pool, user_id)
|
let user = get_user_by_id(&state.pool, user_id)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?
|
.map_err(|e| {
|
||||||
|
tracing::error!(error = %e, %user_id, "failed to load user for key-salt API");
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?
|
||||||
.ok_or(StatusCode::UNAUTHORIZED)?;
|
.ok_or(StatusCode::UNAUTHORIZED)?;
|
||||||
|
|
||||||
if user.key_salt.is_none() {
|
if user.key_salt.is_none() {
|
||||||
@@ -576,10 +660,17 @@ async fn api_key_setup(
|
|||||||
.await
|
.await
|
||||||
.ok_or(StatusCode::UNAUTHORIZED)?;
|
.ok_or(StatusCode::UNAUTHORIZED)?;
|
||||||
|
|
||||||
let salt = hex::decode_hex(&body.salt).map_err(|_| StatusCode::BAD_REQUEST)?;
|
let salt = hex::decode_hex(&body.salt).map_err(|e| {
|
||||||
let key_check = hex::decode_hex(&body.key_check).map_err(|_| StatusCode::BAD_REQUEST)?;
|
tracing::warn!(error = %e, "invalid hex in key-setup salt");
|
||||||
|
StatusCode::BAD_REQUEST
|
||||||
|
})?;
|
||||||
|
let key_check = hex::decode_hex(&body.key_check).map_err(|e| {
|
||||||
|
tracing::warn!(error = %e, "invalid hex in key-setup key_check");
|
||||||
|
StatusCode::BAD_REQUEST
|
||||||
|
})?;
|
||||||
|
|
||||||
if salt.len() != 32 {
|
if salt.len() != 32 {
|
||||||
|
tracing::warn!(salt_len = salt.len(), "key-setup salt must be 32 bytes");
|
||||||
return Err(StatusCode::BAD_REQUEST);
|
return Err(StatusCode::BAD_REQUEST);
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -608,9 +699,10 @@ async fn api_apikey_get(
|
|||||||
.await
|
.await
|
||||||
.ok_or(StatusCode::UNAUTHORIZED)?;
|
.ok_or(StatusCode::UNAUTHORIZED)?;
|
||||||
|
|
||||||
let api_key = ensure_api_key(&state.pool, user_id)
|
let api_key = ensure_api_key(&state.pool, user_id).await.map_err(|e| {
|
||||||
.await
|
tracing::error!(error = %e, %user_id, "ensure_api_key failed");
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?;
|
||||||
|
|
||||||
Ok(Json(ApiKeyResponse { api_key }))
|
Ok(Json(ApiKeyResponse { api_key }))
|
||||||
}
|
}
|
||||||
@@ -625,7 +717,10 @@ async fn api_apikey_regenerate(
|
|||||||
|
|
||||||
let api_key = regenerate_api_key(&state.pool, user_id)
|
let api_key = regenerate_api_key(&state.pool, user_id)
|
||||||
.await
|
.await
|
||||||
.map_err(|_| StatusCode::INTERNAL_SERVER_ERROR)?;
|
.map_err(|e| {
|
||||||
|
tracing::error!(error = %e, %user_id, "regenerate_api_key failed");
|
||||||
|
StatusCode::INTERNAL_SERVER_ERROR
|
||||||
|
})?;
|
||||||
|
|
||||||
Ok(Json(ApiKeyResponse { api_key }))
|
Ok(Json(ApiKeyResponse { api_key }))
|
||||||
}
|
}
|
||||||
|
|||||||
23
crates/secrets-mcp/static/llms.txt
Normal file
23
crates/secrets-mcp/static/llms.txt
Normal file
@@ -0,0 +1,23 @@
|
|||||||
|
# Secrets MCP
|
||||||
|
|
||||||
|
> 给 AI 与自动化工具的简要说明。本站是 **secrets-mcp**:Streamable HTTP **MCP**(Model Context Protocol)与 **Web 控制台** 的组合,用于在多租户场景下存储条目元数据与加密后的秘密字段;持久化在 PostgreSQL。用户通过 OAuth(如已配置)登录 Web;MCP 调用使用 API Key 与加密相关请求头。
|
||||||
|
|
||||||
|
## 不应抓取或索引的内容
|
||||||
|
|
||||||
|
- **`/mcp`**:MCP 流式 HTTP 端点(JSON-RPC 等),**不是** HTML 文档,也不适合作为公开知识库来源。
|
||||||
|
- **`/api/*`**:会话或 API Key 相关的 HTTP API。
|
||||||
|
- **`/dashboard`、`/audit`、`/auth/*`、`/account/*`**:需浏览器会话,属于用户私有界面与 OAuth 流程。
|
||||||
|
|
||||||
|
## 给 AI 助手的实用提示
|
||||||
|
|
||||||
|
- 向用户说明连接方式时:MCP 基址为 `{BASE_URL}/mcp`(`BASE_URL` 由部署方设置),通常需要 `Authorization: Bearer <api_key>`;读写加密秘密时还需按部署文档传递 `X-Encryption-Key` 等头(与客户端模式有关)。
|
||||||
|
- **不要编造**本实例的数据库 URL、OAuth 密钥、回调地址或任何凭据;一律以用户环境变量与运维文档为准。
|
||||||
|
- Web 端在浏览器内用密码短语派生密钥完成端到端加密;MCP 路径下服务端可能在请求周期内临时使用客户端提供的密钥处理密文(架构细节见项目 README「加密架构」)。
|
||||||
|
|
||||||
|
## 延伸阅读
|
||||||
|
|
||||||
|
- 开源仓库中的 `README.md`、`AGENTS.md`(若可访问)包含环境变量、表结构与运维约定。
|
||||||
|
|
||||||
|
## 关于本文件
|
||||||
|
|
||||||
|
- 遵循常见的 **`/llms.txt`** 约定,便于人类与 LLM 快速了解站点性质与抓取边界;同文可在 **`/ai.txt`** 获取。
|
||||||
27
crates/secrets-mcp/static/robots.txt
Normal file
27
crates/secrets-mcp/static/robots.txt
Normal file
@@ -0,0 +1,27 @@
|
|||||||
|
# Secrets MCP — robots.txt
|
||||||
|
# 本站为需登录的私密控制台与 MCP API;以下路径请勿抓取,以免浪费配额并避免误索引敏感端点。
|
||||||
|
# This host serves an authenticated dashboard and machine APIs; please skip crawling the paths below.
|
||||||
|
|
||||||
|
User-agent: *
|
||||||
|
Disallow: /mcp
|
||||||
|
Disallow: /api/
|
||||||
|
Disallow: /dashboard
|
||||||
|
Disallow: /audit
|
||||||
|
Disallow: /auth/
|
||||||
|
Disallow: /account/
|
||||||
|
|
||||||
|
# 面向 AI / LLM 的机器可读站点说明(Markdown):/llms.txt
|
||||||
|
# Human & AI-readable site summary: /llms.txt (also /ai.txt)
|
||||||
|
|
||||||
|
User-agent: GPTBot
|
||||||
|
User-agent: Google-Extended
|
||||||
|
User-agent: anthropic-ai
|
||||||
|
User-agent: Claude-Web
|
||||||
|
User-agent: PerplexityBot
|
||||||
|
User-agent: Bytespider
|
||||||
|
Disallow: /mcp
|
||||||
|
Disallow: /api/
|
||||||
|
Disallow: /dashboard
|
||||||
|
Disallow: /audit
|
||||||
|
Disallow: /auth/
|
||||||
|
Disallow: /account/
|
||||||
@@ -2,6 +2,7 @@
|
|||||||
# 复制此文件为 .env 并填写真实值
|
# 复制此文件为 .env 并填写真实值
|
||||||
|
|
||||||
# ─── 数据库 ───────────────────────────────────────────────────────────
|
# ─── 数据库 ───────────────────────────────────────────────────────────
|
||||||
|
# Web 会话(tower-sessions)与业务数据共用此库;启动时会自动 migrate 会话表,无需额外环境变量。
|
||||||
SECRETS_DATABASE_URL=postgres://postgres:PASSWORD@HOST:PORT/secrets-mcp
|
SECRETS_DATABASE_URL=postgres://postgres:PASSWORD@HOST:PORT/secrets-mcp
|
||||||
|
|
||||||
# ─── 服务地址 ─────────────────────────────────────────────────────────
|
# ─── 服务地址 ─────────────────────────────────────────────────────────
|
||||||
|
|||||||
22
scripts/cleanup-orphan-user-ids.sql
Normal file
22
scripts/cleanup-orphan-user-ids.sql
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
-- Run against prod BEFORE deploying secrets-mcp with FK migration.
|
||||||
|
-- Requires: write access to SECRETS_DATABASE_URL.
|
||||||
|
-- Example: psql "$SECRETS_DATABASE_URL" -v ON_ERROR_STOP=1 -f scripts/cleanup-orphan-user-ids.sql
|
||||||
|
|
||||||
|
BEGIN;
|
||||||
|
|
||||||
|
UPDATE entries
|
||||||
|
SET user_id = NULL
|
||||||
|
WHERE user_id IS NOT NULL
|
||||||
|
AND NOT EXISTS (SELECT 1 FROM users u WHERE u.id = entries.user_id);
|
||||||
|
|
||||||
|
UPDATE entries_history
|
||||||
|
SET user_id = NULL
|
||||||
|
WHERE user_id IS NOT NULL
|
||||||
|
AND NOT EXISTS (SELECT 1 FROM users u WHERE u.id = entries_history.user_id);
|
||||||
|
|
||||||
|
UPDATE audit_log
|
||||||
|
SET user_id = NULL
|
||||||
|
WHERE user_id IS NOT NULL
|
||||||
|
AND NOT EXISTS (SELECT 1 FROM users u WHERE u.id = audit_log.user_id);
|
||||||
|
|
||||||
|
COMMIT;
|
||||||
Reference in New Issue
Block a user