// agent-system:单二进制服务(HTTP/SSE + 内嵌前端 + 双档存储)。 // 部署约束(设计 4.1 原则 8):非容器化;拷贝即运行。 // 嵌入式档默认(SQLite + 本地对象目录,零外部服务);--dsn 切换 PostgreSQL 档 // (多用户/多实例,认证默认开启,apikey 子命令管理密钥)。 package main import ( "encoding/json" "flag" "fmt" "os" "strings" "time" "git-ssh.u1680832.nyat.app/weichen/Agent-System-Backend.git/internal/app" "git-ssh.u1680832.nyat.app/weichen/Agent-System-Backend.git/internal/config" "git-ssh.u1680832.nyat.app/weichen/Agent-System-Backend.git/internal/store" "git-ssh.u1680832.nyat.app/weichen/Agent-System-Backend.git/pkg/identity" ) var version = "0.1.0-dev" func main() { if len(os.Args) > 1 && !strings.HasPrefix(os.Args[1], "-") { switch os.Args[1] { case "version", "--version": if len(os.Args) > 2 && os.Args[2] == "--json" { if err := identity.WriteDocument(os.Stdout, "agent-system", version); err != nil { fmt.Fprintln(os.Stderr, "version:", err) os.Exit(1) } return } fmt.Println("agent-system", version) return case "serve": os.Args = append([]string{os.Args[0]}, os.Args[2:]...) case "apikey": os.Exit(runAPIKey(os.Args[2:])) case "config": os.Exit(runConfig(os.Args[2:])) case "user": os.Exit(runUser(os.Args[2:])) case "identity": os.Exit(runIdentity(os.Args[2:])) case "capabilities": os.Exit(runCapabilities(os.Args[2:])) default: fmt.Fprintf(os.Stderr, "agent-system: 未知子命令 %s\n", os.Args[1]) fmt.Fprintln(os.Stderr, "用法:agent-system [参数]") os.Exit(2) } } fs := flag.NewFlagSet("serve", flag.ExitOnError) cfgFile := fs.String("config", "", "显式配置文件(JSON,键见 config explain;无自动发现)") addr := fs.String("addr", "127.0.0.1:8787", "监听地址") data := fs.String("data", "./data", "数据目录(SQLite 与工件对象目录)") pluginsDir := fs.String("plugins", "./plugins", "插件目录(声明资产热加载)") mock := fs.Bool("mock", false, "演示模式(等同 runtime.mock=true,不调用模型)") dsn := fs.String("dsn", "", "PostgreSQL 档 DSN(空 = 嵌入式 SQLite 档)") artifacts := fs.String("artifacts", "", "工件对象目录(默认 /artifacts;PG 档多实例时指向共享卷)") auth := fs.String("auth", "auto", "API key 认证:auto(PG 档开、嵌入式关)| on | off") instance := fs.String("instance", "", "本进程标识(空 = 主机名:监听地址;启动对账只收口这个实例的 run)") insecure := fs.Bool("insecure-listen", false, "允许在非回环地址上不开启认证(每次启动都会警告)") allowedHosts := fs.String("allowed-hosts", "", "逗号分隔的额外主机名,供反向代理的 Host 与 Origin 校验") noExec := fs.Bool("disable-exec-tools", false, "硬关闭执行原语 run_shell/run_python/register_artifact(用户设置无法覆盖;多用户部署在执行隔离就位前建议开启)") execBackend := fs.String("exec-backend", "", "执行后端:direct(默认,现状直启)| sandbox(bwrap 隔离档:断网、只读系统根、仅会话目录可写、进程数限额)") execDepsDir := fs.String("exec-deps-dir", "", "sandbox 档只读依赖层目录(site-packages 等;空=不挂载)") execWorkspace := fs.String("exec-workspace", "", "sandbox 档工作区模式:空=直写会话目录 | overlay=冻结基底+私有 upper+CAS 回收(S0)") tier := fs.String("tier", "", "部署档:空=auto(DSN→PG,否则 SQLite)| local(SQLite,不允许 DSN)| team(PG fail-closed:缺 DSN/未认证拒绝启动,PG 故障拒绝就绪)") pgMaxOpen := fs.Int("pg-max-open", 0, "PG 最大打开连接(0=默认 16)") pgMaxIdle := fs.Int("pg-max-idle", 0, "PG 最大空闲连接(0=默认 8)") pgIdleSec := fs.Int("pg-conn-max-idle-sec", 0, "PG 空闲连接回收秒数(0=默认 300)") pgLifetimeSec := fs.Int("pg-conn-max-lifetime-sec", 0, "PG 连接最长生命周期秒数(0=不限)") pgQueryTimeoutSec := fs.Int("pg-query-timeout-sec", 0, "PG statement_timeout 秒数(0=不限;覆盖全部业务查询)") fs.Parse(os.Args[1:]) cliSet := visited(fs) resolved, err := config.Load(*cfgFile, os.LookupEnv, config.Config{ Addr: *addr, DataDir: *data, PluginsDir: *pluginsDir, Mock: *mock, StorageDSN: *dsn, ArtifactsDir: *artifacts, AuthMode: *auth, Instance: *instance, InsecureListen: *insecure, AllowedHostsRaw: *allowedHosts, DisableExecTools: *noExec, ExecBackend: *execBackend, ExecDepsDir: *execDepsDir, ExecWorkspace: *execWorkspace, Tier: *tier, PGMaxOpen: *pgMaxOpen, PGMaxIdle: *pgMaxIdle, PGConnMaxIdleSec: *pgIdleSec, PGConnMaxLifetimeSec: *pgLifetimeSec, PGQueryTimeoutSec: *pgQueryTimeoutSec, }, cliSet) if err != nil { fmt.Fprintln(os.Stderr, "agent-system:", err) os.Exit(1) } if msgs := resolved.TierProblems(); len(msgs) > 0 { for _, msg := range msgs { fmt.Fprintln(os.Stderr, "agent-system: tier guard:", msg) } os.Exit(1) } if err := app.Run(app.Options{ Addr: resolved.Addr, DataDir: resolved.DataDir, PluginsDir: resolved.PluginsDir, ForceMock: resolved.Mock, StorageDSN: resolved.StorageDSN, ArtifactsDir: resolved.ArtifactsDir, AuthMode: resolved.AuthMode, InstanceID: resolved.Instance, InsecureListen: resolved.InsecureListen, AllowedHosts: resolved.AllowedHosts(), DisableExecTools: resolved.DisableExecTools, ExecBackend: resolved.ExecBackend, ExecDepsDir: resolved.ExecDepsDir, ExecWorkspace: resolved.ExecWorkspace, Tier: resolved.Tier, PGPool: store.PoolOptions{ MaxOpenConn: resolved.PGMaxOpen, MaxIdleConn: resolved.PGMaxIdle, ConnMaxIdleTimeSec: resolved.PGConnMaxIdleSec, ConnMaxLifetimeSec: resolved.PGConnMaxLifetimeSec, QueryTimeoutSec: resolved.PGQueryTimeoutSec, }, }); err != nil { fmt.Fprintln(os.Stderr, "agent-system:", err) os.Exit(1) } } // visited 把连字符 flag 收成配置字段名,供 Load 的 cliSet 使用。 func visited(fs *flag.FlagSet) map[string]bool { set := map[string]bool{} fs.Visit(func(f *flag.Flag) { name := f.Name switch name { case "insecure-listen": name = "insecureListen" case "allowed-hosts": name = "allowedHosts" case "disable-exec-tools": name = "disableExecTools" case "pg-max-open": name = "pgMaxOpen" case "pg-max-idle": name = "pgMaxIdle" case "pg-conn-max-idle-sec": name = "pgConnMaxIdleSec" case "pg-conn-max-lifetime-sec": name = "pgConnMaxLifetimeSec" case "pg-query-timeout-sec": name = "pgQueryTimeoutSec" } set[name] = true }) return set } // runConfig:config explain——只读解析诊断(不建库、不写业务目录;DSN 脱敏)。 func runConfig(args []string) int { if len(args) == 0 || args[0] != "explain" { fmt.Fprintln(os.Stderr, "用法:agent-system config explain [--config f.json] [--addr …](其余 serve 同名 flag 用于演示覆盖)") return 2 } fs := flag.NewFlagSet("config explain", flag.ExitOnError) cfgFile := fs.String("config", "", "显式配置文件(JSON)") addr := fs.String("addr", "127.0.0.1:8787", "") data := fs.String("data", "./data", "") pluginsDir := fs.String("plugins", "./plugins", "") mock := fs.Bool("mock", false, "") dsn := fs.String("dsn", "", "") artifacts := fs.String("artifacts", "", "") auth := fs.String("auth", "auto", "") instance := fs.String("instance", "", "") insecure := fs.Bool("insecure-listen", false, "") allowedHosts := fs.String("allowed-hosts", "", "") noExec := fs.Bool("disable-exec-tools", false, "") execBackend := fs.String("exec-backend", "", "") execDepsDir := fs.String("exec-deps-dir", "", "") execWorkspace := fs.String("exec-workspace", "", "") tier := fs.String("tier", "", "") pgMaxOpen := fs.Int("pg-max-open", 0, "") pgMaxIdle := fs.Int("pg-max-idle", 0, "") pgIdleSec := fs.Int("pg-conn-max-idle-sec", 0, "") pgLifetimeSec := fs.Int("pg-conn-max-lifetime-sec", 0, "") pgQueryTimeoutSec := fs.Int("pg-query-timeout-sec", 0, "") fs.Parse(args[1:]) resolved, err := config.Load(*cfgFile, os.LookupEnv, config.Config{ Addr: *addr, DataDir: *data, PluginsDir: *pluginsDir, Mock: *mock, StorageDSN: *dsn, ArtifactsDir: *artifacts, AuthMode: *auth, Instance: *instance, InsecureListen: *insecure, AllowedHostsRaw: *allowedHosts, DisableExecTools: *noExec, ExecBackend: *execBackend, ExecDepsDir: *execDepsDir, ExecWorkspace: *execWorkspace, Tier: *tier, PGMaxOpen: *pgMaxOpen, PGMaxIdle: *pgMaxIdle, PGConnMaxIdleSec: *pgIdleSec, PGConnMaxLifetimeSec: *pgLifetimeSec, PGQueryTimeoutSec: *pgQueryTimeoutSec, }, visited(fs)) if err != nil { fmt.Fprintln(os.Stderr, "agent-system:", err) return 1 } resolved.Explain(os.Stdout, *cfgFile) return 0 } // runAPIKey:密钥管理子命令(create/list/revoke)。 func runAPIKey(args []string) int { // 子命令在前、flag 在后:先剥出子命令再解析(Go flag 遇位置参数即停)。 if len(args) == 0 { fmt.Fprintln(os.Stderr, "用法:agent-system apikey [--owner x --name y]") return 2 } sub := args[0] fs := flag.NewFlagSet("apikey "+sub, flag.ExitOnError) data := fs.String("data", "./data", "数据目录") dsn := fs.String("dsn", "", "PostgreSQL 档 DSN(空 = 嵌入式档)") dsnFile := fs.String("dsn-file", "", "0600权限的PostgreSQL DSN文件") owner := fs.String("owner", "", "租户名(owner)") name := fs.String("name", "default", "密钥名称") outFile := fs.String("out-file", "", "把明文密钥独占写入该文件(0600)") fs.Parse(args[1:]) if *owner == "" { fmt.Fprintln(os.Stderr, "apikey: --owner 必填") return 2 } if sub != "create" && *outFile != "" { fmt.Fprintln(os.Stderr, "apikey: 此子命令不生成密钥") return 2 } resolvedDSN, err := resolveAccountDSN(*dsn, *dsnFile) if err != nil { fmt.Fprintln(os.Stderr, "apikey:", err) return 1 } *dsn = resolvedDSN switch sub { case "create": delivery, err := reserveCredentialOutput(*outFile, "") if err != nil { fmt.Fprintln(os.Stderr, "apikey create:", err) return 1 } defer delivery.abort() key, err := app.CreateAPIKey(*dsn, *data, *owner, *name) if err != nil { fmt.Fprintln(os.Stderr, "apikey create:", err) return 1 } // 明文仅此一次输出;库中只存 SHA-256 摘要。 if err := delivery.deliver("api-key", *owner, key, nil); err != nil { fmt.Fprintln(os.Stderr, "apikey create:", err) return 1 } return 0 case "list": keys, err := app.ListAPIKeys(*dsn, *data, *owner) if err != nil { fmt.Fprintln(os.Stderr, "apikey list:", err) return 1 } out, _ := json.MarshalIndent(keys, "", " ") fmt.Println(string(out)) return 0 case "revoke": ok, err := app.RevokeAPIKey(*dsn, *data, *owner, *name) if err != nil { fmt.Fprintln(os.Stderr, "apikey revoke:", err) return 1 } if !ok { fmt.Fprintf(os.Stderr, "apikey revoke: 未找到 owner=%s name=%s 的有效密钥\n", *owner, *name) return 1 } fmt.Println("revoked") return 0 default: fmt.Fprintln(os.Stderr, "apikey: 未知子命令", sub) return 2 } } // deliverCredential 凭据单次交付(KTD14):--out 时写 0600 JSON 文件,stdout 只报 // 属主/用途/文件路径不含明文;缺省沿用 stdout 一次输出。文件含 expiresAt 便于交接。 func deliverCredential(outPath, kind, owner, plaintext string, expiresAt any) error { if outPath == "" { fmt.Println(plaintext) return nil } payload := map[string]any{"kind": kind, "owner": owner, "secret": plaintext} if expiresAt != nil { payload["expiresAt"] = expiresAt } buf, err := json.MarshalIndent(payload, "", " ") if err != nil { return err } if err := writeExclusiveSecret(outPath, string(buf)); err != nil { return err } fmt.Fprintf(os.Stderr, "凭据已写入 %s(0600,仅此一次生成;勿入库勿入日志)\n", outPath) return nil } type credentialOutput struct { path string file *os.File json bool mutated bool } // Reserve the destination before changing the account. O_EXCL also rejects an // existing symlink, so an existing secret can never be overwritten. func reserveCredentialOutput(rawPath, jsonPath string) (*credentialOutput, error) { if rawPath != "" && jsonPath != "" { return nil, fmt.Errorf("--out and --out-file are mutually exclusive") } out := &credentialOutput{} out.path, out.json = rawPath, false if jsonPath != "" { out.path, out.json = jsonPath, true } if out.path == "" { return out, nil } f, err := os.OpenFile(out.path, os.O_WRONLY|os.O_CREATE|os.O_EXCL, 0o600) if err != nil { return nil, err } out.file = f return out, nil } func (out *credentialOutput) abort() { if out == nil || out.file == nil { return } info, err := out.file.Stat() _ = out.file.Close() if out.mutated || err != nil { return } pathInfo, pathErr := os.Stat(out.path) if pathErr == nil && os.SameFile(info, pathInfo) { _ = os.Remove(out.path) } } func (out *credentialOutput) deliver(kind, owner, plaintext string, expiresAt any) error { if out.file == nil { return deliverCredential("", kind, owner, plaintext, expiresAt) } out.mutated = true // Keep the private recovery artifact if delivery fails after DB commit. var body []byte if out.json { payload := map[string]any{"kind": kind, "owner": owner, "secret": plaintext} if expiresAt != nil { payload["expiresAt"] = expiresAt } var err error body, err = json.MarshalIndent(payload, "", " ") if err != nil { return err } } else { body = []byte(plaintext) } body = append(body, '\n') if _, err := out.file.Write(body); err != nil { return err } if err := out.file.Sync(); err != nil { return err } if err := out.file.Close(); err != nil { return err } out.file = nil fmt.Fprintf(os.Stderr, "凭据已写入 %s(0600,仅此一次生成;勿入库勿入日志)\n", out.path) return nil } // runUser:create / reset-password / list / issue-login-key / revoke-login-key / bootstrap-admin。 // 默认 create 无初始密码并签发激活 Key;--generate-password 兼容旧随机密码路径。 func runUser(args []string) int { if len(args) == 0 { fmt.Fprintln(os.Stderr, "用法:agent-system user [flags]") return 2 } sub := args[0] fs := flag.NewFlagSet("user "+sub, flag.ExitOnError) data := fs.String("data", "./data", "数据目录") dsn := fs.String("dsn", "", "PostgreSQL 档 DSN(空 = 嵌入式档;已弃用,优先 AGENT_SYSTEM_DSN 或 --dsn-file)") dsnFile := fs.String("dsn-file", "", "0600 权限的 DSN 文件") owner := fs.String("owner", "", "用户名(owner)") name := fs.String("name", "", "显示名") password := fs.String("password", "", "密码(已弃用;请用 --password-stdin)") passwordStdin := fs.Bool("password-stdin", false, "从 stdin 读取密码") outFile := fs.String("out-file", "", "把明文凭据独占写入该文件(0600)") instanceID := fs.String("instance-id", "", "严格核验库内服务身份") admin := fs.Bool("admin", false, "管理员") role := fs.String("role", "user", "角色 user|admin") openExisting := fs.Bool("open-existing", false, "兼容旧参数;账户命令始终只打开既有存储") generatePassword := fs.Bool("generate-password", false, "create 时生成随机初始密码") purpose := fs.String("purpose", "recover", "issue-login-key:activate|recover") ttlMin := fs.Int("ttl-minutes", 30, "登录 Key 有效分钟(5–120)") revokeAll := fs.Bool("revoke-all-credentials", false, "签发时撤销全部长期 API Key 与会话") keyID := fs.String("id", "", "revoke-login-key 凭据编号") out := fs.String("out", "", "凭据交付文件(0600 JSON,代替 stdout 明文;非交互推荐)") fs.Parse(args[1:]) resolvedDSN, err := resolveAccountDSN(*dsn, *dsnFile) if err != nil { fmt.Fprintln(os.Stderr, "user:", err) return 1 } resolvedPassword, err := resolveAccountPassword(*password, *passwordStdin) if err != nil { fmt.Fprintln(os.Stderr, "user:", err) return 1 } if *instanceID != "" { _ = os.Setenv("AGENT_SYSTEM_INSTANCE_ID", *instanceID) } *dsn = resolvedDSN *password = resolvedPassword credentialAction := sub == "create" || sub == "reset-password" || sub == "issue-login-key" || sub == "bootstrap-admin" if !credentialAction && (*outFile != "" || *out != "") { fmt.Fprintln(os.Stderr, "user: 此子命令不生成凭据") return 2 } var delivery *credentialOutput if credentialAction { delivery, err = reserveCredentialOutput(*outFile, *out) if err != nil { fmt.Fprintln(os.Stderr, "user:", err) return 2 } defer delivery.abort() } switch sub { case "create": if *owner == "" { fmt.Fprintln(os.Stderr, "user create: --owner 必填") return 2 } isAdmin := *admin switch *role { case "user": case "admin": isAdmin = true default: fmt.Fprintf(os.Stderr, "user create: 未知角色 %q\n", *role) return 2 } pw, opID, err := app.CreateUserAccount(*dsn, *data, *owner, *name, *password, isAdmin, *openExisting, *generatePassword) if err != nil { fmt.Fprintln(os.Stderr, "user create:", err) return 1 } if err := delivery.deliver("login-key", *owner, pw, nil); err != nil { fmt.Fprintln(os.Stderr, "user create:", err) return 1 } fmt.Fprintf(os.Stderr, "operation_id=%s\n", opID) if *out == "" && *outFile == "" { if *password == "" && !*generatePassword { fmt.Fprintln(os.Stderr, "一次性激活 Key 仅此一次输出;兑换后设置密码。") } else if *password == "" { fmt.Fprintln(os.Stderr, "初始密码仅此一次输出;该用户首次登录须修改密码。") } } return 0 case "reset-password": if *owner == "" { fmt.Fprintln(os.Stderr, "user reset-password: --owner 必填") return 2 } pw, err := app.ResetUserPassword(*dsn, *data, *owner, *password, *openExisting) if err != nil { fmt.Fprintln(os.Stderr, "user reset-password:", err) return 1 } if err := delivery.deliver("password", *owner, pw, nil); err != nil { fmt.Fprintln(os.Stderr, "user reset-password:", err) return 1 } if *out == "" && *outFile == "" && *password == "" { fmt.Fprintln(os.Stderr, "新密码仅此一次输出;该用户下次登录须修改密码(其余会话已吊销)。") } return 0 case "list": users, err := app.ListUserAccounts(*dsn, *data, *openExisting) if err != nil { fmt.Fprintln(os.Stderr, "user list:", err) return 1 } out, _ := json.MarshalIndent(users, "", " ") fmt.Println(string(out)) return 0 case "issue-login-key": if *owner == "" { fmt.Fprintln(os.Stderr, "user issue-login-key: --owner 必填") return 2 } if *purpose != store.LoginKeyPurposeActivate && *purpose != store.LoginKeyPurposeRecover { fmt.Fprintf(os.Stderr, "user issue-login-key: 无效 purpose %q\n", *purpose) return 2 } plain, err := app.IssueUserLoginKey(*dsn, *data, *owner, *purpose, *openExisting, *revokeAll, time.Duration(*ttlMin)*time.Minute) if err != nil { fmt.Fprintln(os.Stderr, "user issue-login-key:", err) return 1 } if err := delivery.deliver("login-key", *owner, plain, time.Now().Add(time.Duration(*ttlMin)*time.Minute).UTC()); err != nil { fmt.Fprintln(os.Stderr, "user issue-login-key:", err) return 1 } if *out == "" && *outFile == "" { fmt.Fprintln(os.Stderr, "一次性登录 Key 仅此一次输出。") } return 0 case "revoke-login-key": if *keyID == "" { fmt.Fprintln(os.Stderr, "user revoke-login-key: --id 必填") return 2 } if err := app.RevokeUserLoginKey(*dsn, *data, *keyID, *openExisting); err != nil { fmt.Fprintln(os.Stderr, "user revoke-login-key:", err) return 1 } fmt.Println("revoked") return 0 case "bootstrap-admin": if *owner == "" { fmt.Fprintln(os.Stderr, "user bootstrap-admin: --owner 必填") return 2 } plain, err := app.BootstrapAdmin(*dsn, *data, *owner, *openExisting) if err != nil { fmt.Fprintln(os.Stderr, "user bootstrap-admin:", err) return 1 } if err := delivery.deliver("bootstrap-admin", *owner, plain, time.Now().Add(30*time.Minute).UTC()); err != nil { fmt.Fprintln(os.Stderr, "user bootstrap-admin:", err) return 1 } if *out == "" && *outFile == "" { fmt.Fprintln(os.Stderr, "首位管理员激活 Key 仅此一次输出。") } return 0 default: fmt.Fprintln(os.Stderr, "user: 未知子命令", sub) return 2 } } // runIdentity:U1 I3 服务实例库内身份(manifest.instance_id ↔ agent_system_identity)。 // 与 serve --instance(runs 进程标识)无关。实现在 pkg/identity(公共化:下游 // 组合仓共用同一 CLI,未知子命令 usage 退出、不落服务路径——2026-10-12 裁定)。 func runIdentity(args []string) int { return identity.Run(args) }