MailUI4Agents/gateway/internal/models/models.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"`
}

// 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"`
}

// 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"`
	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"`
}

// 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"`
}