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 时转发本轮总结）
88 lines
1.8 KiB
Go
88 lines
1.8 KiB
Go
package handler
|
|
|
|
import (
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
// 登录失败限速：同一用户名连续 N 次失败后锁定一段时间
|
|
const (
|
|
maxLoginFailures = 5
|
|
lockoutDuration = 5 * time.Minute
|
|
failureWindow = 15 * time.Minute
|
|
)
|
|
|
|
type failureRecord struct {
|
|
count int
|
|
firstSeen time.Time
|
|
lockedAt time.Time
|
|
}
|
|
|
|
type loginLimiter struct {
|
|
mu sync.Mutex
|
|
recs map[string]*failureRecord
|
|
}
|
|
|
|
var limiter = &loginLimiter{recs: make(map[string]*failureRecord)}
|
|
|
|
// Locked 返回该用户名是否处于锁定期，以及剩余秒数
|
|
func (l *loginLimiter) Locked(name string) (bool, int) {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
|
|
r, ok := l.recs[name]
|
|
if !ok || r.lockedAt.IsZero() {
|
|
return false, 0
|
|
}
|
|
elapsed := time.Since(r.lockedAt)
|
|
if elapsed >= lockoutDuration {
|
|
delete(l.recs, name)
|
|
return false, 0
|
|
}
|
|
return true, int((lockoutDuration - elapsed).Seconds())
|
|
}
|
|
|
|
// Fail 记录一次失败，达到阈值则锁定
|
|
func (l *loginLimiter) Fail(name string) {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
|
|
now := time.Now()
|
|
r, ok := l.recs[name]
|
|
if !ok || now.Sub(r.firstSeen) > failureWindow {
|
|
l.recs[name] = &failureRecord{count: 1, firstSeen: now}
|
|
return
|
|
}
|
|
r.count++
|
|
if r.count >= maxLoginFailures {
|
|
r.lockedAt = now
|
|
}
|
|
}
|
|
|
|
// Reset 登录成功后清除失败计数
|
|
func (l *loginLimiter) Reset(name string) {
|
|
l.mu.Lock()
|
|
defer l.mu.Unlock()
|
|
delete(l.recs, name)
|
|
}
|
|
|
|
// 定期清理过期记录，避免 map 无限增长
|
|
func init() {
|
|
go func() {
|
|
t := time.NewTicker(10 * time.Minute)
|
|
defer t.Stop()
|
|
for range t.C {
|
|
limiter.mu.Lock()
|
|
now := time.Now()
|
|
for k, r := range limiter.recs {
|
|
stale := now.Sub(r.firstSeen) > failureWindow &&
|
|
(r.lockedAt.IsZero() || now.Sub(r.lockedAt) > lockoutDuration)
|
|
if stale {
|
|
delete(limiter.recs, k)
|
|
}
|
|
}
|
|
limiter.mu.Unlock()
|
|
}
|
|
}()
|
|
}
|