MailUI4Agents/gateway/internal/repo/participants_test.go

package repo

import (
	"context"
	"testing"

	"github.com/agentmail/gateway/internal/models"
	"github.com/google/uuid"
)

// 参与方列表要回答的是「这条线索上现在有谁、用什么地址找到他、谁还没开口」。
// 三个问题里每一个都曾经答错过：
//   - 有谁：sessions 表只有 from_agent 一个名字，抄送方与转发引入的人都不在里面
//   - 什么地址：拿 from_workspace 当 path 会拼出 dsh@dsh.alias 这种投不出去的东西
//   - 谁还没回：只留最后一个身份的话，既发过信又被抄送过的人会被算成纯配合方

func TestSessionParticipantsIncludesCC(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()

	sid, _ := CreateSession(ctx, nil, "admin", "抄收联调", "/home/program/llmsproxy")
	// 复刻线上那封：admin 主发 dsh，抄送 opencode
	mustMail(t, sid, "admin", "", "dsh", "/home/program/llmsproxy",
		[]models.Address{{Name: "opencode", Path: "/home", Session: "new", Raw: "opencode@/home.new"}})

	parts, err := SessionParticipants(ctx, sid)
	if err != nil {
		t.Fatalf("列参与方: %v", err)
	}

	byName := map[string]Participant{}
	for _, p := range parts {
		byName[p.Name] = p
	}
	for _, want := range []string{"admin", "dsh", "opencode"} {
		if _, ok := byName[want]; !ok {
			t.Errorf("参与方缺 %s，实得 %+v", want, parts)
		}
	}
	// 抄送方的 path 必须是它自己那个地址的 path 位，不是主收件人的。
	// 用错的后果：对方在别人的工作目录里开会话。
	if got := byName["opencode"].Path; got != "/home" {
		t.Errorf("opencode 的 path = %q，应为 /home（它自己地址的 path 位）", got)
	}
	if got := byName["dsh"].Path; got != "/home/program/llmsproxy" {
		t.Errorf("dsh 的 path = %q，应为 /home/program/llmsproxy", got)
	}
}

func TestSessionParticipantsSenderPathStaysEmpty(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()

	sid, _ := CreateSession(ctx, nil, "admin", "回信", "/tmp/ws")
	// Agent 回信时 from_workspace 存的是 Agent 名而非路径（历史遗留）。
	// 若把它当 path，地址会拼成 dsh@dsh.alias —— 投不出去。
	mustMail(t, sid, "dsh", "dsh", "admin", "", nil)

	parts, _ := SessionParticipants(ctx, sid)
	for _, p := range parts {
		if p.Name == "dsh" && p.Path == "dsh" {
			t.Fatal("发件人的 path 取了 from_workspace（那列存的是 Agent 名），会拼出无效地址")
		}
	}
}

func TestSessionParticipantsMergesRoles(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()

	sid, _ := CreateSession(ctx, nil, "admin", "往返", "/tmp/ws")
	mustMail(t, sid, "admin", "", "dsh", "/tmp/ws", nil) // admin=from, dsh=to
	mustMail(t, sid, "dsh", "dsh", "admin", "", nil)     // dsh=from, admin=to

	parts, _ := SessionParticipants(ctx, sid)
	for _, p := range parts {
		if len(p.Roles) != 2 {
			t.Errorf("%s 的 roles = %v，双方都该同时有 from 与 to", p.Name, p.Roles)
		}
	}
}

func TestSessionParticipantsCountsOnlySends(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()

	sid, _ := CreateSession(ctx, nil, "admin", "谁还没回", "/tmp/ws")
	mustMail(t, sid, "admin", "", "dsh", "/tmp/ws",
		[]models.Address{{Name: "opencode", Path: "/tmp/ws", Raw: "opencode@/tmp/ws"}})
	mustMail(t, sid, "dsh", "dsh", "admin", "", nil)

	parts, _ := SessionParticipants(ctx, sid)
	got := map[string]int{}
	for _, p := range parts {
		got[p.Name] = p.MailCount
	}
	// MailCount 只数「作为发件人」的邮件：抄送方 opencode 一封都没发，
	// 计数为 0 正是「还没开口的人」这个判断的依据。
	if got["opencode"] != 0 {
		t.Errorf("opencode 只被抄送未发信，MailCount 应为 0，实为 %d", got["opencode"])
	}
	if got["admin"] != 1 || got["dsh"] != 1 {
		t.Errorf("admin/dsh 各发过一封，实为 %d/%d", got["admin"], got["dsh"])
	}
}

func TestSessionParticipantsKeepsFirstSeenOrder(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()

	sid, _ := CreateSession(ctx, nil, "admin", "顺序", "/tmp/ws")
	mustMail(t, sid, "admin", "", "dsh", "/tmp/ws",
		[]models.Address{{Name: "opencode", Path: "/home", Raw: "opencode@/home"}})

	parts, _ := SessionParticipants(ctx, sid)
	// 按首次出现排序，让主收件人稳定排在抄送方之前 ——
	// 模型据此判断谁是负责人、谁是配合方；按名字排序会丢掉这个信息。
	want := []string{"admin", "dsh", "opencode"}
	if len(parts) != len(want) {
		t.Fatalf("参与方数量 %d，期望 %d: %+v", len(parts), len(want), parts)
	}
	for i, w := range want {
		if parts[i].Name != w {
			t.Errorf("第 %d 位是 %s，期望 %s", i, parts[i].Name, w)
		}
	}
}

func TestSessionParticipantsPrefersFirstNonEmptyPath(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()

	sid, _ := CreateSession(ctx, nil, "admin", "改指", "/tmp/a")
	// 同一个人先被抄送到 /home，后被主发到 /tmp/b。
	// 保留首个非空值，与排序口径一致，也避免一封转发把地址改指到别处。
	mustMail(t, sid, "admin", "", "dsh", "/tmp/a",
		[]models.Address{{Name: "opencode", Path: "/home", Raw: "opencode@/home"}})
	mustMail(t, sid, "admin", "", "opencode", "/tmp/b", nil)

	parts, _ := SessionParticipants(ctx, sid)
	for _, p := range parts {
		if p.Name == "opencode" && p.Path != "/home" {
			t.Fatalf("opencode 的 path = %q，应保持首次出现的 /home", p.Path)
		}
	}
}

func TestSessionParticipantsEmptySession(t *testing.T) {
	setupTestDB(t)
	ctx := context.Background()

	// 会话刚建、还没有邮件。返回空列表而不是报错：
	// 调用方拿到空表能正常渲染「暂无参与方」，拿到 error 只能整个失败。
	sid, _ := CreateSession(ctx, nil, "admin", "空会话", "/tmp/ws")
	parts, err := SessionParticipants(ctx, sid)
	if err != nil {
		t.Fatalf("空会话应正常返回: %v", err)
	}
	if len(parts) != 0 {
		t.Fatalf("空会话应无参与方，实得 %+v", parts)
	}
}

func mustMail(t *testing.T, sid uuid.UUID, from, fromWS, to, toWS string, cc []models.Address) {
	t.Helper()
	if _, err := CreateMail(context.Background(), sid, nil,
		from, fromWS, to, toWS, "主题", "正文", cc); err != nil {
		t.Fatalf("建邮件: %v", err)
	}
}