JianFeeeee
0e754617a4
Go 单二进制网关 + React 前端 + opencode 桥接插件。部署产物是 「一个二进制加一个 .db 文件」：前端经 go:embed 打进二进制， 数据库默认内置 SQLite，systemd 托管。 核心设计 - 三维寻址 name@path.session，按最后一个 . 切分；session 位三态： 省略=默认会话 / new=强制新建 / 具体别名=必须已存在（否则 404 无法送达） - 会话别名默认复用 Agent 平台自己的命名机制（opencode 的 slug 与模型生成的 标题），不在本侧另造一套；人显式定过的别名不被平台同步覆盖 - 对话树不建 tree_nodes 表：parent_mail_id 已完整编码树结构， 再维护一张表就是第二份真相。用递归 CTE 查，按方向分块加载 - 附件内容存磁盘、按 sha256 内容寻址，数据库只存元数据；天然去重， 且路径与用户 filename 无关，杜绝 ../ 穿越 - 配额约束的是模型的自主发信，不是 harness 的转发：插件代劳的权限询问与 最终总结走免配额通道，靠上游消息 id 做幂等键而非计数 - 往返预算下沉到会话（写信时给、对话页里改）+ Agent 全局配额，两层都要过 后端 gateway/ - models/repo/handler/middleware/sse/blob 分层；两方言（SQLite/PostgreSQL） 共用一份 repo 层 SQL，差异集中在 internal/db - 多用户认证（bcrypt cost12、登录限速、会话隔离、权限边界） - 密钥体系：Agent 密钥与用户密钥分表，三种生命周期；登记式密钥让全文 只从客户端流向服务器一次 - 所有「判断 + 自增」都在同一条 UPDATE 里（配额、预算、one_time 密钥、 附件挂载），并发下不会刷穿 前端 web/ - 三栏布局、三段式地址补全、权限卡片、密钥面板、配额面板、对话树、附件 - 全站纯 SVG 图标，不使用 emoji - api/ 即可复用的客户端 SDK：基地址与凭证集中在 api/config.ts 插件 plugins/opencode-mail-bridge/ - 六个工具 + 两类自动转发（permission.ask 钩子接管平台原生权限询问、 session.idle 时转发本轮总结）
298 lines
13 KiB
SQL
298 lines
13 KiB
SQL
-- AgentMail MVP Schema
|
||
-- PostgreSQL 14+
|
||
|
||
-- Users table（人类多用户；username 与 agents.agent_name 共用命名空间）
|
||
CREATE TABLE IF NOT EXISTS users (
|
||
user_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
username VARCHAR(64) NOT NULL UNIQUE,
|
||
display_name VARCHAR(128) NOT NULL DEFAULT '',
|
||
password_hash VARCHAR(255) NOT NULL,
|
||
role VARCHAR(16) NOT NULL DEFAULT 'user',
|
||
status VARCHAR(16) NOT NULL DEFAULT 'active',
|
||
created_at TIMESTAMPTZ DEFAULT NOW(),
|
||
last_login TIMESTAMPTZ
|
||
);
|
||
|
||
CREATE TABLE IF NOT EXISTS user_sessions (
|
||
token VARCHAR(64) PRIMARY KEY,
|
||
user_id UUID NOT NULL REFERENCES users(user_id) ON DELETE CASCADE,
|
||
created_at TIMESTAMPTZ DEFAULT NOW(),
|
||
expires_at TIMESTAMPTZ NOT NULL,
|
||
user_agent VARCHAR(256) DEFAULT ''
|
||
);
|
||
|
||
CREATE INDEX IF NOT EXISTS idx_user_sessions_user ON user_sessions(user_id);
|
||
CREATE INDEX IF NOT EXISTS idx_user_sessions_exp ON user_sessions(expires_at);
|
||
|
||
-- 用户权限边界：可调用的 Agent 与可访问的工作区目录
|
||
-- allowed_agents：["deepseekharness","pi"]，空数组 = 不限（继承系统默认）
|
||
-- allowed_paths ：["/program","/home/x"]，空数组 = 不限；按前缀匹配
|
||
-- agent_aliases ：{"大龙":"deepseekharness","小派":"pi"}，发信时自动解析真实名
|
||
ALTER TABLE users ADD COLUMN IF NOT EXISTS allowed_agents JSONB NOT NULL DEFAULT '[]';
|
||
ALTER TABLE users ADD COLUMN IF NOT EXISTS allowed_paths JSONB NOT NULL DEFAULT '[]';
|
||
ALTER TABLE users ADD COLUMN IF NOT EXISTS agent_aliases JSONB NOT NULL DEFAULT '{}';
|
||
|
||
-- Agents table
|
||
CREATE TABLE IF NOT EXISTS agents (
|
||
agent_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
agent_name VARCHAR(64) NOT NULL UNIQUE,
|
||
secret VARCHAR(128) NOT NULL,
|
||
host_url VARCHAR(256) NOT NULL DEFAULT '',
|
||
workspaces JSONB NOT NULL DEFAULT '[]',
|
||
platform VARCHAR(32) NOT NULL DEFAULT 'pi',
|
||
status VARCHAR(16) NOT NULL DEFAULT 'offline',
|
||
max_rounds INT NOT NULL DEFAULT 10,
|
||
used_rounds INT NOT NULL DEFAULT 0,
|
||
last_seen TIMESTAMPTZ,
|
||
created_at TIMESTAMPTZ DEFAULT NOW()
|
||
);
|
||
|
||
-- Sessions table
|
||
CREATE TABLE IF NOT EXISTS sessions (
|
||
session_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
session_alias VARCHAR(128),
|
||
from_agent VARCHAR(64) NOT NULL,
|
||
subject VARCHAR(512) NOT NULL,
|
||
status VARCHAR(32) NOT NULL DEFAULT 'active',
|
||
owner_user_id UUID REFERENCES users(user_id),
|
||
created_at TIMESTAMPTZ DEFAULT NOW(),
|
||
updated_at TIMESTAMPTZ DEFAULT NOW()
|
||
);
|
||
|
||
CREATE INDEX IF NOT EXISTS idx_sessions_alias ON sessions(session_alias);
|
||
CREATE INDEX IF NOT EXISTS idx_sessions_status ON sessions(status);
|
||
|
||
-- 会话别名负责寻址（name@path.<alias>），必须全局唯一。
|
||
-- 部分唯一索引：未命名会话（NULL）不受约束，可以有任意多个。
|
||
CREATE UNIQUE INDEX IF NOT EXISTS idx_sessions_alias_uniq
|
||
ON sessions(session_alias) WHERE session_alias IS NOT NULL;
|
||
|
||
-- 已存在的库补列（必须先于依赖该列的索引）
|
||
ALTER TABLE sessions ADD COLUMN IF NOT EXISTS owner_user_id UUID REFERENCES users(user_id);
|
||
|
||
-- 用户驳回过的改名提议。记下来才能让提示条不再反复弹同一个建议。
|
||
ALTER TABLE sessions ADD COLUMN IF NOT EXISTS rename_dismissed TEXT;
|
||
|
||
-- 别名是谁定的：'platform'（Agent 平台自动同步，可被后续同步覆盖）
|
||
-- 或 'manual'（人显式指定，平台同步不得覆盖）。
|
||
-- 没有这个标记，平台的下一次 session.updated 会把人刚接受的名字冲掉。
|
||
ALTER TABLE sessions ADD COLUMN IF NOT EXISTS alias_source TEXT NOT NULL DEFAULT 'platform';
|
||
|
||
-- 本次任务的往返预算（0 = 本会话不限，仅受 Agent 全局配额约束）。
|
||
--
|
||
-- 配额的真实语义是「这件事值得多少个来回」，那是任务的属性而不是 Agent 的属性：
|
||
-- 只有 agents.max_rounds 一个全局计数器时，两个并行任务会互相抢额度，
|
||
-- 且 used_rounds 单调递增，一旦跑满就得管理员手工重置才能再干活。
|
||
-- 因此预算下沉到会话，由人在写信时给、在对话页里随时调。
|
||
--
|
||
-- Agent 全局配额仍然生效（两者都要过）：否则 Agent 自己 .new 开一串会话，
|
||
-- 每条都是全新预算，全局上限就形同虚设。
|
||
ALTER TABLE sessions ADD COLUMN IF NOT EXISTS max_rounds INTEGER NOT NULL DEFAULT 0;
|
||
ALTER TABLE sessions ADD COLUMN IF NOT EXISTS used_rounds INTEGER NOT NULL DEFAULT 0;
|
||
|
||
CREATE INDEX IF NOT EXISTS idx_sessions_owner ON sessions(owner_user_id);
|
||
|
||
-- Mails table
|
||
CREATE TABLE IF NOT EXISTS mails (
|
||
mail_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
session_id UUID NOT NULL REFERENCES sessions(session_id),
|
||
parent_mail_id UUID REFERENCES mails(mail_id),
|
||
|
||
from_name VARCHAR(64) NOT NULL,
|
||
from_workspace VARCHAR(128) DEFAULT '',
|
||
to_name VARCHAR(64) NOT NULL,
|
||
to_workspace VARCHAR(128) DEFAULT '',
|
||
|
||
subject VARCHAR(512) NOT NULL,
|
||
body TEXT NOT NULL,
|
||
|
||
-- 拄送列表：[{"name":"pi","path":"root","session":"new","raw":"pi@root.new"}]
|
||
cc_list JSONB NOT NULL DEFAULT '[]',
|
||
|
||
mail_type VARCHAR(32) NOT NULL DEFAULT 'normal',
|
||
permission_options JSONB,
|
||
permission_result VARCHAR(32),
|
||
|
||
status VARCHAR(16) NOT NULL DEFAULT 'unread',
|
||
created_at TIMESTAMPTZ DEFAULT NOW(),
|
||
|
||
hop_limit INT DEFAULT 5
|
||
);
|
||
|
||
CREATE INDEX IF NOT EXISTS idx_mails_session ON mails(session_id);
|
||
CREATE INDEX IF NOT EXISTS idx_mails_to ON mails(to_name, status);
|
||
CREATE INDEX IF NOT EXISTS idx_mails_parent ON mails(parent_mail_id);
|
||
|
||
-- 已存在的库补列（重复运行安全）
|
||
ALTER TABLE mails ADD COLUMN IF NOT EXISTS cc_list JSONB NOT NULL DEFAULT '[]';
|
||
|
||
-- Agent 在正文里提议改会话别名（<!-- agentmail:rename-session … -->）。
|
||
-- 存在邮件上而非会话上：邮件是不可篡改的历史记录，「谁在哪一封里提了什么」应当留痕。
|
||
ALTER TABLE mails ADD COLUMN IF NOT EXISTS rename_alias TEXT;
|
||
ALTER TABLE mails ADD COLUMN IF NOT EXISTS rename_reason TEXT;
|
||
|
||
-- 抄送检索：cc_list @> '[{"name":"pi"}]' 走 GIN
|
||
CREATE INDEX IF NOT EXISTS idx_mails_cc_list ON mails USING GIN (cc_list jsonb_path_ops);
|
||
|
||
-- 历史数据规整：早期写入过首字母大写的键（Raw/Name/Path/Session），统一成小写
|
||
UPDATE mails
|
||
SET cc_list = (
|
||
SELECT COALESCE(jsonb_agg(
|
||
jsonb_strip_nulls(jsonb_build_object(
|
||
'name', COALESCE(e->>'name', e->>'Name'),
|
||
'path', COALESCE(e->>'path', e->>'Path'),
|
||
'session', COALESCE(e->>'session', e->>'Session'),
|
||
'raw', COALESCE(e->>'raw', e->>'Raw')
|
||
))
|
||
), '[]'::jsonb)
|
||
FROM jsonb_array_elements(cc_list) AS e
|
||
)
|
||
WHERE cc_list @? '$[*].Name';
|
||
|
||
-- Permission requests table
|
||
CREATE TABLE IF NOT EXISTS permission_requests (
|
||
request_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
mail_id UUID NOT NULL REFERENCES mails(mail_id),
|
||
session_id UUID NOT NULL REFERENCES sessions(session_id),
|
||
agent_name VARCHAR(64) NOT NULL,
|
||
question TEXT NOT NULL,
|
||
options JSONB NOT NULL DEFAULT '["同意", "拒绝"]',
|
||
context TEXT DEFAULT '',
|
||
result VARCHAR(32),
|
||
decided_at TIMESTAMPTZ,
|
||
created_at TIMESTAMPTZ DEFAULT NOW()
|
||
);
|
||
|
||
CREATE INDEX IF NOT EXISTS idx_perm_agent ON permission_requests(agent_name, result);
|
||
CREATE INDEX IF NOT EXISTS idx_perm_pending ON permission_requests(result) WHERE result IS NULL;
|
||
|
||
-- 历史邮件里的字面量 'human' 迁移到默认管理员账号
|
||
-- （管理员由 Go 侧 EnsureAdminUser 首启创建，此处只做数据重写）
|
||
DO $$
|
||
DECLARE
|
||
admin_name TEXT;
|
||
BEGIN
|
||
SELECT username INTO admin_name
|
||
FROM users WHERE role = 'admin' AND status = 'active'
|
||
ORDER BY created_at ASC LIMIT 1;
|
||
|
||
IF admin_name IS NULL THEN
|
||
RETURN; -- 还没有管理员，下次迁移再试
|
||
END IF;
|
||
|
||
UPDATE mails SET from_name = admin_name WHERE from_name = 'human';
|
||
UPDATE mails SET to_name = admin_name WHERE to_name = 'human';
|
||
UPDATE sessions SET from_agent = admin_name WHERE from_agent = 'human';
|
||
|
||
-- 拄送列表里的 human 一并重写
|
||
UPDATE mails
|
||
SET cc_list = (
|
||
SELECT COALESCE(jsonb_agg(
|
||
CASE WHEN e->>'name' = 'human'
|
||
THEN jsonb_set(
|
||
jsonb_set(e, '{name}', to_jsonb(admin_name)),
|
||
'{raw}',
|
||
to_jsonb(admin_name || '@' || COALESCE(e->>'path','') ||
|
||
CASE WHEN COALESCE(e->>'session','') = '' THEN ''
|
||
ELSE '.' || (e->>'session') END))
|
||
ELSE e END
|
||
), '[]'::jsonb)
|
||
FROM jsonb_array_elements(cc_list) AS e
|
||
)
|
||
WHERE cc_list @> '[{"name":"human"}]';
|
||
|
||
-- 人类发起的会话补上 owner
|
||
UPDATE sessions s
|
||
SET owner_user_id = u.user_id
|
||
FROM users u
|
||
WHERE u.username = admin_name
|
||
AND s.owner_user_id IS NULL
|
||
AND s.from_agent = admin_name;
|
||
END $$;
|
||
|
||
-- ---------- 密钥认证体系 ----------
|
||
--
|
||
-- 两类密钥，共享一个全局唯一的 token 命名空间（验证时先查 agent_keys 再查 user_keys）：
|
||
-- agent_keys：管理员签发，用于 Agent 注册/心跳/SSE
|
||
-- user_keys ：用户自助签发，仅用于 /me/* 人类邮箱接口，不可注册 Agent
|
||
--
|
||
-- key_type：
|
||
-- permanent — 永不过期，可重复使用
|
||
-- one_time — 首次验证后写 used_at，再用即拒
|
||
-- timed — expires_at 之后失效
|
||
|
||
CREATE TABLE IF NOT EXISTS agent_keys (
|
||
key_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
key_token VARCHAR(128) NOT NULL UNIQUE,
|
||
agent_name VARCHAR(64), -- NULL = 待绑定
|
||
key_type VARCHAR(16) NOT NULL DEFAULT 'permanent',
|
||
label VARCHAR(128) NOT NULL DEFAULT '',
|
||
expires_at TIMESTAMPTZ,
|
||
used_at TIMESTAMPTZ,
|
||
created_by UUID REFERENCES users(user_id),
|
||
created_at TIMESTAMPTZ DEFAULT NOW()
|
||
);
|
||
|
||
CREATE INDEX IF NOT EXISTS idx_agent_keys_token ON agent_keys(key_token);
|
||
CREATE INDEX IF NOT EXISTS idx_agent_keys_agent ON agent_keys(agent_name);
|
||
|
||
CREATE TABLE IF NOT EXISTS user_keys (
|
||
key_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
key_token VARCHAR(128) NOT NULL UNIQUE,
|
||
user_id UUID NOT NULL REFERENCES users(user_id) ON DELETE CASCADE,
|
||
label VARCHAR(128) NOT NULL DEFAULT '',
|
||
key_type VARCHAR(16) NOT NULL DEFAULT 'permanent',
|
||
expires_at TIMESTAMPTZ,
|
||
used_at TIMESTAMPTZ,
|
||
created_at TIMESTAMPTZ DEFAULT NOW()
|
||
);
|
||
|
||
CREATE INDEX IF NOT EXISTS idx_user_keys_user ON user_keys(user_id);
|
||
CREATE INDEX IF NOT EXISTS idx_user_keys_token ON user_keys(key_token);
|
||
|
||
-- ---------- 附件 ----------
|
||
--
|
||
-- 文件内容存磁盘（内容寻址：路径由 sha256 派生），数据库只存元数据。
|
||
--
|
||
-- mail_id 为 NULL 表示「已上传但还没挂到邮件上」的待用附件：
|
||
-- 上传与发信是两步（Agent 工具是 JSON 接口，没法在发信时带 multipart），
|
||
-- 中间态必须允许存在；超时未挂载的由 GC 清掉。
|
||
|
||
-- 插件自动转发的邮件登记表。
|
||
--
|
||
-- **配额约束的是模型的自主发信，不是 harness 的转发**（基本原则）：
|
||
-- 配额存在的意义是防止 Agent 无限自我循环。而「把平台原生的权限询问转给人」
|
||
-- 与「把本轮的最终总结转给人」都是插件代劳的搬运，不是模型自己决定要发的信 ——
|
||
-- 对它们收费会导致配额用尽时 Agent 连交代都做不了。
|
||
--
|
||
-- relay_key 是上游那条消息的稳定标识（opencode 的 permission id / assistant message id）。
|
||
-- 唯一约束把「同一条上游消息只转一次」变成一条 INSERT 的成败：
|
||
-- * 插件重试、SSE 重连后重放都不会产生第二封
|
||
-- * 也顺带给免配额通道加了结构性上限 —— 想多转就得拿出不同的上游消息 id
|
||
CREATE TABLE IF NOT EXISTS relayed_mails (
|
||
agent_name VARCHAR(64) NOT NULL,
|
||
relay_key VARCHAR(160) NOT NULL,
|
||
mail_id UUID REFERENCES mails(mail_id),
|
||
kind VARCHAR(32) NOT NULL DEFAULT '',
|
||
created_at TIMESTAMPTZ DEFAULT NOW(),
|
||
PRIMARY KEY (agent_name, relay_key)
|
||
);
|
||
|
||
-- 人类决策后要按 mail_id 反查上游 permission id
|
||
CREATE INDEX IF NOT EXISTS idx_relayed_mail ON relayed_mails(mail_id);
|
||
|
||
CREATE TABLE IF NOT EXISTS attachments (
|
||
attachment_id UUID PRIMARY KEY DEFAULT gen_random_uuid(),
|
||
mail_id UUID REFERENCES mails(mail_id) ON DELETE CASCADE,
|
||
uploader VARCHAR(64) NOT NULL,
|
||
filename VARCHAR(512) NOT NULL,
|
||
content_type VARCHAR(128) NOT NULL DEFAULT 'application/octet-stream',
|
||
size_bytes BIGINT NOT NULL,
|
||
sha256 CHAR(64) NOT NULL,
|
||
created_at TIMESTAMPTZ DEFAULT NOW()
|
||
);
|
||
|
||
CREATE INDEX IF NOT EXISTS idx_attachments_mail ON attachments(mail_id);
|
||
CREATE INDEX IF NOT EXISTS idx_attachments_sha ON attachments(sha256);
|
||
CREATE INDEX IF NOT EXISTS idx_attachments_orphan ON attachments(created_at) WHERE mail_id IS NULL;
|