feat(webui): 首启人格向导(默认 / 自定义 / 稍后)+ 一次性标记

接续人格配置项化(597f07c):现在人格是 core.agent.personal_prompt,
本次加上「首启问一次」的界面,之后不再打扰。

后端(GET/POST /api/v1/persona):
- GET  → {initialized, current_prompt, file_override}
        未设置时 current_prompt 回落到内置默认模板;存在 personal/personal.md
        时报告 file_override(它会覆盖配置项,向导据此提示用户)
- POST → {"mode":"default"|"custom"|"later","content":"…"}
        写配置 + 打一次性标记 core.internal.persona_initialized;
        custom 返回 restart_required=true(人格在启动时载入);
        「稍后」= 保留当前默认 + 打标记,**绝不阻塞任何流程**
- 空内容的 custom 与未知 mode 一律 400,且**不打标记**(否则向导会被跳过)

前端(dashboard.html):
- 首启拉一次 /api/v1/persona,未初始化则弹向导(复用一直没人用的 .confirm-* 样式)
- 「自定义…」第一次点击展开文本域并预填当前人格,再次点击才提交(避免误提交)
- 中英双语走既有 __() 机制;保存失败/空内容用 toast 提示

测试:TestPersonaWizardFlow(首启状态、later 打标记不改人格、custom 写入 + 需重启、
空内容与未知 mode 被拒且不打标记)、TestPersonaWizardReportsFileOverride。

E2E(真实实例):首启 initialized=false → POST later → initialized=true,
config 中标记=1、人格键为默认模板;前端页面含向导函数。
This commit is contained in:
JianFeeeee
2026-09-12 13:02:18 +08:00
parent ce8bc27db8
commit 10367b384e
3 changed files with 361 additions and 2 deletions

View File

@ -0,0 +1,145 @@
package webui
import (
"encoding/json"
"net/http"
"net/http/httptest"
"os"
"path/filepath"
"strings"
"testing"
internalConfig "gitcode.com/JianFeeeee/HomeAgent/internal/config"
"gitcode.com/JianFeeeee/HomeAgent/internal/sdk"
)
func newPersonaHandler(t *testing.T) (*Handler, *internalConfig.ConfigRegistry) {
t.Helper()
dir := t.TempDir()
cfgReg := internalConfig.NewConfigRegistry(filepath.Join(dir, "config.db"))
cfgReg.SeedDefaults(dir)
t.Cleanup(func() { cfgReg.Close() })
h := NewHandler(testSDK(sdk.SDKConfig{Settings: sdk.NewSettings("webui", cfgReg)}))
return h, cfgReg
}
func doPersona(t *testing.T, h *Handler, method, body string) *httptest.ResponseRecorder {
t.Helper()
var rd *strings.Reader
if body == "" {
rd = strings.NewReader("")
} else {
rd = strings.NewReader(body)
}
req := httptest.NewRequest(method, "/api/v1/persona", rd)
w := httptest.NewRecorder()
h.handlePersona(w, req)
return w
}
// 首启向导的后端契约GET 报告状态、POST 三选一、并且**只问一次**。
func TestPersonaWizardFlow(t *testing.T) {
h, cfgReg := newPersonaHandler(t)
// 1. 全新安装未初始化current_prompt 回落到内置默认模板
w := doPersona(t, h, http.MethodGet, "")
if w.Code != http.StatusOK {
t.Fatalf("GET 状态码 %d", w.Code)
}
var got struct {
Initialized bool `json:"initialized"`
CurrentPrompt string `json:"current_prompt"`
FileOverride bool `json:"file_override"`
}
if err := json.Unmarshal(w.Body.Bytes(), &got); err != nil {
t.Fatal(err)
}
if got.Initialized {
t.Fatal("全新安装不应已初始化")
}
if got.CurrentPrompt != internalConfig.DefaultPersonaPrompt {
t.Fatal("未设置时应回落到内置默认模板")
}
if got.FileOverride {
t.Fatal("没有人格文件时不应报告 file_override")
}
// 2. 「稍后再说」= 保留默认、打标记、不再问
w = doPersona(t, h, http.MethodPost, `{"mode":"later"}`)
if w.Code != http.StatusOK {
t.Fatalf("later 状态码 %d: %s", w.Code, w.Body.String())
}
if v := cfgReg.GetString(personaInitMarker, ""); v == "" {
t.Fatal("later 也必须打一次性标记(否则每次启动都问)")
}
if v := cfgReg.GetString(personaPromptKey, ""); v != internalConfig.DefaultPersonaPrompt {
t.Fatalf("later 不应改动人格,实际 %q", v)
}
// 3. 已初始化后 GET 应报 true
w = doPersona(t, h, http.MethodGet, "")
got.Initialized = false
_ = json.Unmarshal(w.Body.Bytes(), &got)
if !got.Initialized {
t.Fatal("打过标记后应报告已初始化")
}
// 4. 自定义:写入内容 + 需要重启(人格在启动时载入)
h2, cfgReg2 := newPersonaHandler(t)
w = doPersona(t, h2, http.MethodPost, `{"mode":"custom","content":"你是测试人格"}`)
if w.Code != http.StatusOK {
t.Fatalf("custom 状态码 %d: %s", w.Code, w.Body.String())
}
var pr struct {
RestartRequired bool `json:"restart_required"`
}
_ = json.Unmarshal(w.Body.Bytes(), &pr)
if !pr.RestartRequired {
t.Fatal("自定义人格应提示需要重启才生效")
}
if v := cfgReg2.GetString(personaPromptKey, ""); v != "你是测试人格" {
t.Fatalf("自定义内容未写库: %q", v)
}
// 5. 空内容的 custom 必须被拒(否则等于静默清空人格)
h3, _ := newPersonaHandler(t)
if w = doPersona(t, h3, http.MethodPost, `{"mode":"custom","content":" "}`); w.Code != http.StatusBadRequest {
t.Fatalf("空内容应 400实际 %d", w.Code)
}
// 6. 未知 mode 必须被拒
if w = doPersona(t, h3, http.MethodPost, `{"mode":"nope"}`); w.Code != http.StatusBadRequest {
t.Fatalf("未知 mode 应 400实际 %d", w.Code)
}
// 7. 被拒的请求不得打标记(否则向导会被跳过)
if v := cfgReg2.GetString(personaInitMarker, ""); v == "" {
t.Fatal("前置条件:第 4 步已打标记")
}
h4, cfgReg4 := newPersonaHandler(t)
_ = doPersona(t, h4, http.MethodPost, `{"mode":"nope"}`)
if v := cfgReg4.GetString(personaInitMarker, ""); v != "" {
t.Fatal("被拒的请求不应打标记")
}
}
// 存在人格文件时 GET 要报告 file_override它会覆盖配置项向导应提示用户
func TestPersonaWizardReportsFileOverride(t *testing.T) {
h, cfgReg := newPersonaHandler(t)
dir := cfgReg.GetString("core.daemon.data_dir", "")
if dir == "" {
t.Fatal("播种应写入 core.daemon.data_dir")
}
if err := os.MkdirAll(filepath.Join(dir, "personal"), 0o755); err != nil {
t.Fatal(err)
}
if err := os.WriteFile(filepath.Join(dir, "personal", "personal.md"), []byte("旧人格"), 0o644); err != nil {
t.Fatal(err)
}
w := doPersona(t, h, http.MethodGet, "")
var got struct {
FileOverride bool `json:"file_override"`
}
_ = json.Unmarshal(w.Body.Bytes(), &got)
if !got.FileOverride {
t.Fatal("存在 personal.md 时必须报告 file_override")
}
}