JianFeeeee
a44fd6949b
L2 核心改动：sessions 表补 permission_mode / permission_enforcement 两列 （sqlite + pg 同步），三桥 lib/permission-mode.js 翻译档位到平台原生配置， homeagent advisory 模式提示词告知模型实际强制力。四桥全部携带 from_session_id 供 relay 去重与会话回溯。 FromHuman / ToHuman 判据已加入心跳 payload 与 notify/mail.go。
306 lines
12 KiB
Go
306 lines
12 KiB
Go
package models
|
||
|
||
import (
|
||
"strings"
|
||
"time"
|
||
|
||
"github.com/google/uuid"
|
||
)
|
||
|
||
// Agent 代表一个已注册的 Agent 实例
|
||
type Agent struct {
|
||
ID uuid.UUID `json:"agent_id"`
|
||
Name string `json:"agent_name"`
|
||
Secret string `json:"-"`
|
||
HostURL string `json:"host_url"`
|
||
Workspaces []Workspace `json:"workspaces"`
|
||
Platform string `json:"platform"`
|
||
Status string `json:"status"`
|
||
// DefaultRounds 是派给该 Agent 的新任务默认多少个来回（0 = 不限）。
|
||
// 真正的额度在每条会话上（sessions.max_rounds），这里只是默认值。
|
||
DefaultRounds int `json:"default_rounds"`
|
||
// UsedRounds 是累计发信数，纯统计，不拦请求。
|
||
// 它原本是「终身额度」——那种额度跑满要人工重置才能再干活，
|
||
// 而 Agent 是长期在线的，所以已降级为观测数据。
|
||
UsedRounds int `json:"used_rounds"`
|
||
LastSeen *time.Time `json:"last_seen"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
|
||
// ModeEnforcement 是该平台插件自报的权限档位强制能力（native / advisory），
|
||
// 随心跳上报（与模型目录同一条通道，见 I-1：平台自己说的才算）。
|
||
//
|
||
// 为什么要存：发件人在派活前得知道 plan 档在对方那儿到底算不算。
|
||
// homeagent 的核心没有工具调用拦截点，档位只能写进提示词 ——
|
||
// 把这个事实藏起来比做不到本身更危险。
|
||
ModeEnforcement string `json:"mode_enforcement"`
|
||
}
|
||
|
||
// Workspace 是 Agent 管理的项目工作区
|
||
type Workspace struct {
|
||
Name string `json:"name"`
|
||
Path string `json:"path"`
|
||
}
|
||
|
||
// Session 是有明确边界的任务会话
|
||
type Session struct {
|
||
ID uuid.UUID `json:"session_id"`
|
||
Alias *string `json:"session_alias"`
|
||
FromAgent string `json:"from_agent"`
|
||
Subject string `json:"subject"`
|
||
Status string `json:"status"`
|
||
OwnerUserID *uuid.UUID `json:"owner_user_id"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
UpdatedAt time.Time `json:"updated_at"`
|
||
MailCount int `json:"mail_count,omitempty"`
|
||
|
||
// RenameDismissed 是用户驳回过的改名提议。
|
||
// 记下来才能让提示条不再反复弹同一个建议。
|
||
RenameDismissed string `json:"rename_dismissed,omitempty"`
|
||
|
||
// AliasSource 记录别名是谁定的：
|
||
// platform = Agent 平台自动同步来的，后续同步可以覆盖
|
||
// manual = 人显式指定（手工改名或接受了 Agent 的提议），平台同步不得覆盖
|
||
// 没有这个区分，平台下一次 session.updated 会把人刚定的名字冲掉。
|
||
AliasSource string `json:"alias_source,omitempty"`
|
||
|
||
// MaxRounds/UsedRounds 是本任务的往返预算（0 = 本会话不限）。
|
||
// 配额的语义是「这件事值得多少个来回」，那是任务的属性而非 Agent 的属性，
|
||
// 所以在写信时给、在对话页里随时调。
|
||
MaxRounds int `json:"max_rounds"`
|
||
UsedRounds int `json:"used_rounds"`
|
||
|
||
// PermissionMode 声明本任务允许 Agent 动手到什么程度：
|
||
// plan / workspace / full。空值按 DefaultPermissionMode 处理。
|
||
PermissionMode string `json:"permission_mode"`
|
||
|
||
// PermissionEnforcement 记录接收平台是否真正强制了权限档位：
|
||
// native / advisory。它描述执行事实，不与 PermissionMode 混为一谈。
|
||
PermissionEnforcement string `json:"permission_enforcement"`
|
||
}
|
||
|
||
// User 是人类用户（多用户账号体系）
|
||
type User struct {
|
||
ID uuid.UUID `json:"user_id"`
|
||
Username string `json:"username"`
|
||
DisplayName string `json:"display_name"`
|
||
PasswordHash string `json:"-"`
|
||
Role string `json:"role"` // admin / user
|
||
Status string `json:"status"` // active / disabled
|
||
CreatedAt time.Time `json:"created_at"`
|
||
LastLogin *time.Time `json:"last_login"`
|
||
|
||
// 权限边界：空切片 = 不限
|
||
AllowedAgents []string `json:"allowed_agents"`
|
||
AllowedPaths []string `json:"allowed_paths"`
|
||
}
|
||
|
||
// IsAdmin 判断是否管理员
|
||
func (u User) IsAdmin() bool { return u.Role == "admin" }
|
||
|
||
// CanUseAgent 判断用户是否可向指定 Agent 发信
|
||
// 空白名单（或为空） = 不限；管理员不受限；收件方是人类用户时不走此限制
|
||
func (u User) CanUseAgent(agentName string) bool {
|
||
if u.IsAdmin() || len(u.AllowedAgents) == 0 {
|
||
return true
|
||
}
|
||
for _, a := range u.AllowedAgents {
|
||
if a == agentName {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// CanUsePath 判断用户是否可访问指定工作区。
|
||
// 空白名单 = 不限；管理员不受限；空 path（人类地址）总是允许。
|
||
// 匹配规则：完全相等，或白名单项作为目录前缀（/program 允许 /program/sub）。
|
||
func (u User) CanUsePath(path string) bool {
|
||
if u.IsAdmin() || len(u.AllowedPaths) == 0 || path == "" {
|
||
return true
|
||
}
|
||
for _, p := range u.AllowedPaths {
|
||
if p == "" {
|
||
continue
|
||
}
|
||
if path == p {
|
||
return true
|
||
}
|
||
prefix := p
|
||
if !strings.HasSuffix(prefix, "/") {
|
||
prefix += "/"
|
||
}
|
||
if strings.HasPrefix(path, prefix) {
|
||
return true
|
||
}
|
||
}
|
||
return false
|
||
}
|
||
|
||
// Mail 是会话中的一封邮件
|
||
type Mail struct {
|
||
ID uuid.UUID `json:"mail_id"`
|
||
SessionID uuid.UUID `json:"session_id"`
|
||
ParentMailID *uuid.UUID `json:"parent_mail_id"`
|
||
FromName string `json:"from_name"`
|
||
FromWorkspace string `json:"from_workspace"`
|
||
ToName string `json:"to_name"`
|
||
ToWorkspace string `json:"to_workspace"`
|
||
CCList []Address `json:"cc_list"`
|
||
Subject string `json:"subject"`
|
||
Body string `json:"body"`
|
||
MailType string `json:"mail_type"`
|
||
PermOptions []string `json:"permission_options,omitempty"`
|
||
PermResult string `json:"permission_result,omitempty"`
|
||
Status string `json:"status"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
HopLimit int `json:"hop_limit"`
|
||
|
||
SessionAlias string `json:"session_alias,omitempty"`
|
||
|
||
// SessionWorkspace 是**这条会话**的工作目录（sessions.workspace）。
|
||
//
|
||
// 为什么不能用 FromWorkspace / ToWorkspace 代替：
|
||
// - 人 → Agent：to_workspace 是真路径，from_workspace 为空（人没有工作目录）
|
||
// - Agent → 人：to_workspace 为空，而 **from_workspace 存的是 Agent 名
|
||
// 而不是路径**（历史遗留，见 db/migrate.go 的 sessions.workspace 注释）
|
||
//
|
||
// 于是「Agent 发来的这封信，那个 Agent 在哪个目录干活」只能从会话上取 ——
|
||
// 前端要靠它拼出 `name@path.alias` 这个可投递地址（界面上曾显示成
|
||
// `dsh@dsh`，就是拿 from_workspace 当路径拼出来的）。
|
||
SessionWorkspace string `json:"session_workspace,omitempty"`
|
||
|
||
BodyPreview string `json:"body_preview,omitempty"`
|
||
|
||
// Attachments 仅在读取单封邮件/会话线程时填充；列表接口为省带宽留空
|
||
Attachments []Attachment `json:"attachments,omitempty"`
|
||
|
||
// RenameAlias / RenameReason 是 Agent 在本封正文里提议的新会话别名。
|
||
// 存在邮件上而非会话上：邮件是不可篡改的历史记录，
|
||
// 「谁在哪一封里提了什么」应当留痕。
|
||
RenameAlias string `json:"rename_alias,omitempty"`
|
||
RenameReason string `json:"rename_reason,omitempty"`
|
||
|
||
// FromHuman 表示发件方是人类用户而不是 Agent。
|
||
//
|
||
// 插件靠它判「要不要自动转发本轮结论」：Agent 之间不自动回，
|
||
// 否则两边都以为对方的插件会代它开口，持续互相唤醒（生产实测 6 轮）。
|
||
//
|
||
// **补拉路径必须有它**：SSE 事件里叫 `from_human`，而插件重启后走
|
||
// `GET /mail/inbox` 补投 —— 那条路径上没有这个字段的话，补投的邮件会被
|
||
// 保守当成 Agent 来信，于是人发的那封失去自动回信。
|
||
FromHuman bool `json:"from_human"`
|
||
|
||
// ToHuman 表示收件方是人类用户而不是 Agent（判据与 FromHuman 同源：
|
||
// to_name 是否存在于 users 表）。
|
||
//
|
||
// 前端拼地址时靠它决定「要不要带 path 与会话位」：人只写名字，
|
||
// Agent 才拼 `name@path.session`。此前靠 `to_workspace` 是否为空的启发式 ——
|
||
// 但对 Agent 而言 to_workspace 存的是 Agent 名而不是路径（历史遗留），
|
||
// 那条启发式在「Agent 名恰好为空」时会猜错。显式布尔胜过猜。
|
||
ToHuman bool `json:"to_human"`
|
||
|
||
// PermissionMode / PermissionEnforcement 是所属会话的权限档位与实际强制力。
|
||
//
|
||
// **补拉路径必须有它们**（与 FromHuman 同一个理由）：SSE 事件里叫
|
||
// `permission_mode` / `permission_enforcement`，而插件重启后走
|
||
// `GET /mail/inbox` 补投 —— 那条路径上没有这两个字段的话，补投的邮件
|
||
// 会拿不到档位，插件只能回落默认档 —— 于是一条 plan 档的任务在重启后
|
||
// 惄惄变成了 workspace 档。
|
||
PermissionMode string `json:"permission_mode"`
|
||
PermissionEnforcement string `json:"permission_enforcement"`
|
||
}
|
||
|
||
// PermissionRequest 是 Agent 向人类发起的权限请求
|
||
type PermissionRequest struct {
|
||
ID uuid.UUID `json:"request_id"`
|
||
MailID uuid.UUID `json:"mail_id"`
|
||
SessionID uuid.UUID `json:"session_id"`
|
||
AgentName string `json:"agent_name"`
|
||
Question string `json:"question"`
|
||
Options []string `json:"options"`
|
||
Context string `json:"context"`
|
||
Result *string `json:"result"`
|
||
DecidedAt *time.Time `json:"decided_at"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
}
|
||
|
||
// SSE 事件类型
|
||
const (
|
||
EventNewMail = "new_mail"
|
||
EventPermissionDecision = "permission_decision"
|
||
EventSessionUpdate = "session_update"
|
||
EventAgentOnline = "agent_online"
|
||
)
|
||
|
||
// ---------- 密钥认证 ----------
|
||
|
||
// 密钥类型：签发时决定其生命周期
|
||
const (
|
||
// KeyPermanent 永不过期，可重复使用（正式部署的 Agent 用这个）
|
||
KeyPermanent = "permanent"
|
||
// KeyOneTime 首次验证后即失效（用于把 Agent 首次接入的窗口压到最小）
|
||
KeyOneTime = "one_time"
|
||
// KeyTimed 到 ExpiresAt 之后失效
|
||
KeyTimed = "timed"
|
||
)
|
||
|
||
// ValidKeyType 判断密钥类型是否受支持
|
||
func ValidKeyType(t string) bool {
|
||
return t == KeyPermanent || t == KeyOneTime || t == KeyTimed
|
||
}
|
||
|
||
// AgentKey 是管理员签发的 Agent 接入密钥。
|
||
// AgentName 为空表示「待绑定」——密钥有效但还没指定属于哪个 Agent，
|
||
// 首次注册时由注册请求里的 name 落定。
|
||
type AgentKey struct {
|
||
ID uuid.UUID `json:"key_id"`
|
||
Token string `json:"key_token,omitempty"` // 仅创建时回显一次
|
||
TokenHint string `json:"token_hint"` // 前 8 位 + 省略号，用于列表展示
|
||
AgentName *string `json:"agent_name"`
|
||
KeyType string `json:"key_type"`
|
||
Label string `json:"label"`
|
||
ExpiresAt *time.Time `json:"expires_at"`
|
||
UsedAt *time.Time `json:"used_at"`
|
||
CreatedBy *uuid.UUID `json:"created_by"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
}
|
||
|
||
// UserKey 是用户自助签发的客户端连接密钥，只能用于 /me/* 人类邮箱接口。
|
||
type UserKey struct {
|
||
ID uuid.UUID `json:"key_id"`
|
||
Token string `json:"key_token,omitempty"` // 仅创建时回显一次
|
||
TokenHint string `json:"token_hint"`
|
||
UserID uuid.UUID `json:"user_id"`
|
||
Label string `json:"label"`
|
||
KeyType string `json:"key_type"`
|
||
ExpiresAt *time.Time `json:"expires_at"`
|
||
UsedAt *time.Time `json:"used_at"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
}
|
||
|
||
// TokenHint 返回密钥的展示形式：只露前 8 位。
|
||
// 密钥全文仅在创建响应里出现一次，之后任何列表接口都只给 hint。
|
||
func TokenHint(token string) string {
|
||
if len(token) <= 8 {
|
||
return token
|
||
}
|
||
return token[:8] + "…"
|
||
}
|
||
|
||
// ---------- 附件 ----------
|
||
|
||
// Attachment 是一封邮件的附件元数据。文件内容存磁盘，按 sha256 内容寻址。
|
||
//
|
||
// MailID 为空表示「已上传、尚未挂到邮件上」：上传与发信是两步操作
|
||
// （Agent 侧工具走 JSON，无法在发信请求里带 multipart），中间态必须允许存在。
|
||
type Attachment struct {
|
||
ID uuid.UUID `json:"attachment_id"`
|
||
MailID *uuid.UUID `json:"mail_id"`
|
||
Uploader string `json:"uploader"`
|
||
Filename string `json:"filename"`
|
||
ContentType string `json:"content_type"`
|
||
SizeBytes int64 `json:"size_bytes"`
|
||
SHA256 string `json:"sha256"`
|
||
CreatedAt time.Time `json:"created_at"`
|
||
}
|