HomeAgent/internal/plugins/webui/starmap_perf_test.go

package webui

import (
	"os"
	"path/filepath"
	"regexp"
	"strings"
	"testing"
)

// 星图渲染必须避免「每帧全量」——这些判据钉住 2026-10-08 修「非常卡」的三处。
//
// 为何用源码断言而不是运行时测量：dashboard.js 是嵌入的浏览器脚本，
// 用 Node 跑不了（依赖 three.js / DOM）。而这三条都是**结构性**缺陷
// （循环扫全表、循环不停、每帧重建），源码里能可靠判定，且回归时立刻红。
func TestStarmapNoPerFrameFullScan(t *testing.T) {
	src := readDashboardJS(t)

	// ① starmapTickMorph 不得再全量遍历 starmapNodeMeshes。
	//    原先 3190 节点 × 60fps = 每秒 19 万次 userData 读取与分支，
	//    而绝大多数帧里「正在形变」的节点是 0 个。
	morph := extractFunc(t, src, "starmapTickMorph")
	if strings.Contains(morph, "starmapNodeMeshes") {
		t.Error("starmapTickMorph 又全量遍历 starmapNodeMeshes 了：应只遍历 starmapMorphSet 活跃集，" +
			"空集时 O(1) 返回（否则每帧 3190 次无谓遍历）")
	}
	if !strings.Contains(morph, "starmapMorphSet") {
		t.Error("starmapTickMorph 必须走 starmapMorphSet 活跃集合")
	}
	if !regexp.MustCompile(`starmapMorphSet\.size\s*===\s*0`).MatchString(morph) {
		t.Error("starmapTickMorph 必须在活跃集为空时提前返回（这是常见路径，应当零开销）")
	}

	// ② 渲染循环必须能被暂停（否则切走页签/后台仍 60fps 空转）。
	anim := extractFunc(t, src, "starmapAnimate")
	if !strings.Contains(anim, "starmapAnimateRunning") {
		t.Error("starmapAnimate 缺少 starmapAnimateRunning 开关：切走页签后循环不会停")
	}
	if !strings.Contains(anim, "starmapActiveContainer") {
		t.Error("starmapAnimate 必须按「星图容器是否可见」判断暂停，" +
			"不能用 document.hidden（星图可能在未激活页签里被搬走）")
	}

	// ③ 重建图时必须清空形变活跃集，否则集合单调增长、
	//    「空集 O(1)」快速路径永不生效，每帧还遍历已废弃 mesh。
	//    断言必须在**重建函数体内**且真的在调用 clear（不能只看全仓有这串，
	//    否则改成 `if (false) ...clear()` 也能骗过——变异测试实测过）。
	rebuild := extractFunc(t, src, "buildChatStarmapGraph")
	if !regexp.MustCompile(`(?m)^\s*if \(starmapMorphSet\) starmapMorphSet\.clear\(\);`).MatchString(rebuild) {
		t.Error("buildChatStarmapGraph 必须无条件清空 starmapMorphSet（`if (starmapMorphSet) starmapMorphSet.clear();`）：" +
			"旧 mesh 已从场景摘掉，却会留在活跃集里被永远遍历")
	}

	// ④ 标签层必须有节流（相机拖动时 camKey 每帧都变 → 否则每帧重建全量 plan）。
	labels := extractFunc(t, src, "starmapDrawLabels")
	if !strings.Contains(labels, "SM_LABEL_MIN_INTERVAL") {
		t.Error("starmapDrawLabels 缺少节流：OrbitControls 有阻尼，拖动时每帧重建 3190 项 plan")
	}

	// ⑤ ★★ 2026-10-13：力导向布局必须是网格近邻（O(n·k)），不得再出现
	//    全对全（O(n²)）的双层循环。
	//
	//    为何单独钉这一条：生产库 3754 块时全对全 50 轮 = 3.52 亿次配对，
	//    CDP Profiler 实测单次阻塞主线程 10.9 秒。而斥力体里本来就有
	//    `if (d < 25)` —— 只对近邻施力，没必要全对全。
	//
	//    ★ 判据必须同时覆盖**两处**实现（串行回退 + Worker 脚本）：
	//    只盯着 dashboard.js 的话，把回退改回全对全就没人拦 ——
	//    而那正是“有 Worker 与没 Worker 看到的图不一样”的成因。
	relax := extractFunc(t, src, "starmapLayoutRelax")
	if !strings.Contains(relax, "grid") || !strings.Contains(relax, "CUTOFF") {
		t.Error("starmapLayoutRelax 必须走网格近邻（grid + CUTOFF）：" +
			"全对全斥力在生产库（3754 块）实测阻塞主线程 10.9 秒")
	}
	// 全对全的特征：内层循环遍历外层指针的副本（Object.keys(pos) / ids.length）
	if strings.Contains(relax, "Object.keys") {
		t.Error("starmapLayoutRelax 又回到 Object.keys 全量遍历：那是 O(n²) 的写法")
	}
	if !strings.Contains(src, "starmapLayoutIterationsFor") {
		t.Error("缺少迭代数自适应（starmapLayoutIterationsFor）：" +
			"固定 50 轮在大图上绝大多数轮次是在重复收敛后的状态")
	}

	// Worker 脚本必须存在且用同一套网格算法
	wb, err := os.ReadFile(filepath.Join("static", "starmap-layout-worker.js"))
	if err != nil {
		t.Fatalf("缺少 static/starmap-layout-worker.js: %v", err)
	}
	wsrc := string(wb)
	if !strings.Contains(wsrc, "CUTOFF") || !strings.Contains(wsrc, "iterationsFor") {
		t.Error("starmap-layout-worker.js 必须用同一套网格法（CUTOFF）与自适应迭代数，" +
			"否则 Worker 与串行回退会给出不同布局")
	}
	if regexp.MustCompile(`for \(var j = i \+ 1; j < ids\.length`).MatchString(wsrc) {
		t.Error("starmap-layout-worker.js 里仍有全对全双层循环（O(n²)）")
	}

	// ⑥ 建图必须在容器可见时才做（否则聊天页签也被拖死）。
	render := extractFunc(t, src, "renderChatStarmap")
	if !strings.Contains(render, "offsetParent") {
		t.Error("renderChatStarmap 必须在容器不可见时提前返回（offsetParent 判据）：" +
			"starmapActiveContainer 会 fallback 到聊天面板，使未打开星图也被迫建图")
	}
}

// extractFunc 取一个顶层函数的源码体（到下一个行首 `}` 为止）。
func extractFunc(t *testing.T, src, name string) string {
	t.Helper()
	marker := "function " + name + "("
	i := strings.Index(src, marker)
	if i < 0 {
		t.Fatalf("找不到函数 %s", name)
	}
	rest := src[i:]
	end := strings.Index(rest, "\n}\n")
	if end < 0 {
		return rest
	}
	return rest[:end]
}

func readDashboardJS(t *testing.T) string {
	t.Helper()
	b, err := os.ReadFile("dashboard.js")
	if err != nil {
		t.Fatal(err)
	}
	return string(b)
}