package main

import (
	"net/url"
	"strings"
	"testing"
)

// ★ 收件箱按工作区收窄（2026-09-26）
//
// 我今天先部署了服务端（缺 workspace 直接 400），只修了 pi/opencode/dsh 三个桥，
// 漏了这里 —— 线上随即出现：
//
//	07:42:28 tool read_inbox result: 工具 read_inbox 执行失败:
//	  proc: homeagent-mail-bridge.tool.invoke: HTTP 400:
//	  {"error":"缺少 workspace：收件箱按工作区收窄..."}
//
// 这组判据钉住"读类端点必须带 workspace"，以及补投必须逐工作区。

func TestInboxURLCarriesWorkspace(t *testing.T) {
	p := &Plugin{gwURL: "http://gw", currentSessionID: "s1", currentWorkspace: "/home/program/agentmail"}
	u := p.inboxURL("unread", 5)

	if !strings.Contains(u, "workspace=%2Fhome%2Fprogram%2Fagentmail") {
		t.Fatalf("inboxURL 必须带 workspace，实际: %s", u)
	}
	// 两维都要在：session 防"A 会话标掉 B 会话"，workspace 防跨工作区越界
	if !strings.Contains(u, "session_id=s1") {
		t.Fatalf("inboxURL 必须带 session_id，实际: %s", u)
	}
	// 构造出的 URL 必须可解析（不是靠字符串拼凑蒙对的）
	if _, err := url.Parse(u); err != nil {
		t.Fatalf("URL 不可解析: %v", err)
	}
}

// 拿不到工作区时**不带** —— 服务端会 400，那是刻意的：
// 错误可见，好过静默跨工作区拿到别处的信。
func TestInboxURLOmitsWorkspaceWhenUnknown(t *testing.T) {
	p := &Plugin{gwURL: "http://gw", currentSessionID: "s1"}
	u := p.inboxURL("unread", 5)
	if strings.Contains(u, "workspace=") {
		t.Fatalf("没有工作区时不该带 workspace 参数: %s", u)
	}
}

func TestScopeQueryCarriesWorkspace(t *testing.T) {
	p := &Plugin{currentSessionID: "s1", currentWorkspace: "/w"}
	q := p.scopeQuery("&")
	if !strings.Contains(q, "workspace=%2Fw") {
		t.Fatalf("scopeQuery 必须带 workspace，实际: %q", q)
	}
}

// ★ 补投不再读一次全局收件箱
func TestCatchUpIsPerWorkspace(t *testing.T) {
	src := readSource(t, "plugin.go")

	// catchUp 必须有 workspaces 参数并逐个分发
	if !strings.Contains(src, "func (p *Plugin) catchUp(pending int, workspaces []string)") {
		t.Fatal("catchUp 必须接 workspaces 清单（心跳给的 pending_workspaces）")
	}
	if !strings.Contains(src, "p.catchUpWorkspace(ws, limit, pending)") {
		t.Fatal("catchUp 必须逐工作区分发")
	}

	// 真正的拉取必须带 workspace 参数
	if !strings.Contains(src, "status=unread&limit=%d&workspace=%s") {
		t.Fatal("补投的收件箱请求必须带 workspace")
	}

	// 反向对照：旧的全局读法不能残留
	if strings.Contains(src, `"/api/v1/mail/inbox?status=unread&limit=%d"`) {
		t.Fatal("仍残留不带 workspace 的全局收件箱读法")
	}

	// 拿不到清单就不补投（带空 workspace 去拉必然 400）
	if !strings.Contains(src, "len(workspaces) == 0") {
		t.Fatal("没有 pending_workspaces 时必须跳过补投")
	}
}

// 两处投递路径都要在回合期间设上工作区 —— 模型那轮会调 read_inbox。
func TestDeliverySetsWorkspaceDuringTurn(t *testing.T) {
	src := readSource(t, "plugin.go")
	if strings.Count(src, "p.currentWorkspace = evt.Workspace") < 1 {
		t.Fatal("SSE 投递路径必须设 currentWorkspace（信封的 to_workspace）")
	}
	if strings.Count(src, "p.currentWorkspace = workspace") < 1 {
		t.Fatal("补投路径必须设 currentWorkspace")
	}
	// 两处都要清空 —— 残留会让下一轮读到上一封的工作区
	if strings.Count(src, `p.currentWorkspace = ""`) < 2 {
		t.Fatal("两处都必须清空 currentWorkspace（残留会串到下一轮）")
	}
}