package export

import (
	"bytes"
	"database/sql"
	"path/filepath"
	"testing"

	"github.com/xuri/excelize/v2"
)

// seedEntries 造一份覆盖筛选维度的报名名单：两个单位、两个小项、两个组别，
// 含批次（有无各一）、中文姓名、带前导零的号码布，以及一条团体报名（无运动员）。
func seedEntries(t *testing.T, db *sql.DB) {
	t.Helper()
	stmts := []string{
		`INSERT INTO teams(code, name, short_name, sort_order, active) VALUES ('T1', '第一分公司', '一公司', 1, 1)`,
		`INSERT INTO teams(code, name, short_name, sort_order, active) VALUES ('T2', '第二分公司', '二公司', 2, 1)`,
		`INSERT INTO athletes(name, gender, team_id, id_card, age) VALUES ('张伟', '男', 1, '110101199001011234', 35)`,
		`INSERT INTO athletes(name, gender, team_id, id_card, age) VALUES ('李强', '男', 2, '110101199202022345', 33)`,
		`INSERT INTO sports(code, name, sort_order, active) VALUES ('SP1', '田径', 1, 1)`,
		`INSERT INTO sports(code, name, sort_order, active) VALUES ('SP2', '游泳', 2, 1)`,
		`INSERT INTO events(sport_id, code, name, status, sort_order) VALUES (1, 'E1', '100米跑', 'ongoing', 1)`,
		`INSERT INTO events(sport_id, code, name, status, sort_order) VALUES (1, 'E2', '4×100米接力', 'ongoing', 2)`,
		`INSERT INTO events(sport_id, code, name, status, sort_order) VALUES (2, 'E3', '100米自由泳', 'pending', 3)`,
		`INSERT INTO event_categories(event_id, name, sort_order) VALUES (1, '男子', 1)`,
		`INSERT INTO event_categories(event_id, name, sort_order) VALUES (1, '女子', 2)`,
		`INSERT INTO event_categories(event_id, name, sort_order) VALUES (2, '男子', 1)`,
		`INSERT INTO event_categories(event_id, name, sort_order) VALUES (3, '男子', 1)`,
		// 批次 2、号码布带前导零。
		`INSERT INTO entries(event_id, category_id, team_id, athlete_id, bib, heat, status, remark)
		 VALUES (1, 1, 1, 1, '007', 2, 'registered', '卫冕冠军')`,
		// 未分批，号码布 0012。
		`INSERT INTO entries(event_id, category_id, team_id, athlete_id, bib, status)
		 VALUES (1, 1, 2, 2, '0012', 'scratched')`,
		// 团体报名：没有运动员，姓名回落成员名单。
		`INSERT INTO entries(event_id, category_id, team_id, member_names, heat, status)
		 VALUES (2, 3, 1, '["王磊","赵雷"]', 1, 'registered')`,
		`UPDATE settings SET value = '测试运动会' WHERE key = 'event.name'`,
	}
	for _, s := range stmts {
		if _, err := db.Exec(s); err != nil {
			t.Fatalf("造数据 %q: %v", s, err)
		}
	}
}

func TestExportEntriesXLSXColumnsAndRows(t *testing.T) {
	db := openEventDB(t)
	seedEntries(t, db)

	dst := filepath.Join(t.TempDir(), "报名名单.xlsx")
	if err := ExportEntriesXLSX(db, dst, EntriesFilter{}, Options{}); err != nil {
		t.Fatalf("导出报名名单: %v", err)
	}
	f, err := excelize.OpenFile(dst)
	if err != nil {
		t.Fatalf("重新打开 xlsx: %v", err)
	}
	defer func() { _ = f.Close() }()

	wantSheet := []string{"报名名单"}
	if got := f.GetSheetList(); !equalStrings(got, wantSheet) {
		t.Fatalf("sheet 列表 = %v，期望 %v", got, wantSheet)
	}
	rows, err := f.GetRows("报名名单")
	if err != nil {
		t.Fatalf("读取报名名单: %v", err)
	}
	// 4 行版式 + 3 条报名。
	if len(rows) != 7 {
		t.Fatalf("行数 = %d，期望 7（4 行版式 + 3 条报名）", len(rows))
	}

	want := []string{"小项编码", "小项名", "组别", "单位编码", "单位名", "姓名", "证件号",
		"号码布", "批次", "性别", "年龄", "状态", "备注"}
	for i, header := range want {
		if got := cell(rows, 3, i); got != header {
			t.Errorf("表头第 %d 列 = %q，期望 %q", i+1, got, header)
		}
	}

	// 排序：未分批的排最前（heat IS NOT NULL 升序 = false < true）。
	if got := cell(rows, 4, 5); got != "李强" {
		t.Fatalf("第一行姓名 = %q，期望 李强（未分批排最前）", got)
	}
	if got := cell(rows, 4, 7); got != "0012" {
		t.Errorf("未分批行号码布 = %q，期望 0012（前导零不能丢）", got)
	}
	if got := cell(rows, 4, 8); got != "" {
		t.Errorf("未分批行批次 = %q，期望留空", got)
	}
	if got := cell(rows, 4, 11); got != "已退赛" {
		t.Errorf("scratched 状态 = %q，期望 已退赛", got)
	}
	if got := cell(rows, 4, 2); got != "男子" {
		t.Errorf("组别 = %q，期望 男子", got)
	}
	if got := cell(rows, 4, 10); got != "33" {
		t.Errorf("年龄 = %q，期望 33", got)
	}

	if got := cell(rows, 5, 5); got != "张伟" {
		t.Fatalf("第二行姓名 = %q，期望 张伟", got)
	}
	if got := cell(rows, 5, 7); got != "007" {
		t.Errorf("号码布 = %q，期望 007", got)
	}
	if got := cell(rows, 5, 8); got != "2" {
		t.Errorf("批次 = %q，期望 2", got)
	}
	if got := cell(rows, 5, 11); got != "正常参赛" {
		t.Errorf("registered 状态 = %q，期望 正常参赛", got)
	}
	if got := cell(rows, 5, 12); got != "卫冕冠军" {
		t.Errorf("备注 = %q，期望 卫冕冠军", got)
	}

	// 团体行：姓名取成员名单，批次 1。
	if got := cell(rows, 6, 5); got != "王磊、赵雷" {
		t.Errorf("团体行姓名 = %q，期望 王磊、赵雷", got)
	}
	if got := cell(rows, 6, 1); got != "4×100米接力" {
		t.Errorf("团体行小项名 = %q，期望 4×100米接力", got)
	}

	// 号码布必须是文本单元格：写成数字会把 007 变成 7。
	if typ, err := f.GetCellType("报名名单", "H5"); err != nil || typ != excelize.CellTypeSharedString {
		t.Errorf("号码布单元格类型 = %v (err=%v)，期望共享字符串（保住前导零）", typ, err)
	}
}

func TestExportEntriesXLSXFilters(t *testing.T) {
	db := openEventDB(t)
	seedEntries(t, db)

	cases := []struct {
		name   string
		filter EntriesFilter
		want   int
		first  string
	}{
		{"全部", EntriesFilter{}, 3, "李强"},
		{"按批次", EntriesFilter{Heat: 2}, 1, "张伟"},
		{"按单位", EntriesFilter{TeamID: 1}, 2, "张伟"},
		{"按小项", EntriesFilter{EventID: 1}, 2, "李强"},
		{"按组别", EntriesFilter{CategoryID: 3}, 1, "王磊、赵雷"},
		{"组合筛选", EntriesFilter{EventID: 1, TeamID: 1, Heat: 2}, 1, "张伟"},
	}
	for _, tc := range cases {
		t.Run(tc.name, func(t *testing.T) {
			wb, err := LoadEntriesWorkbook(db, tc.filter, Options{})
			if err != nil {
				t.Fatalf("读取报名名单: %v", err)
			}
			if len(wb.Rows) != tc.want {
				t.Fatalf("行数 = %d，期望 %d", len(wb.Rows), tc.want)
			}
			if got := wb.Rows[0].Name; got != tc.first {
				t.Errorf("首行姓名 = %q，期望 %q", got, tc.first)
			}
		})
	}

	// 不存在的批次号不是错误，只是没有数据。
	wb, err := LoadEntriesWorkbook(db, EntriesFilter{Heat: 99}, Options{})
	if err != nil {
		t.Fatalf("读取空结果: %v", err)
	}
	if len(wb.Rows) != 0 {
		t.Errorf("批次 99 应为空，得到 %d 行", len(wb.Rows))
	}
}

func TestWriteEntriesXLSXStream(t *testing.T) {
	db := openEventDB(t)
	seedEntries(t, db)

	wb, err := LoadEntriesWorkbook(db, EntriesFilter{}, Options{})
	if err != nil {
		t.Fatalf("读取报名名单: %v", err)
	}
	var buf bytes.Buffer
	if err := WriteEntriesXLSX(&buf, wb); err != nil {
		t.Fatalf("流式导出: %v", err)
	}
	f, err := excelize.OpenReader(bytes.NewReader(buf.Bytes()))
	if err != nil {
		t.Fatalf("打开流式导出的 xlsx: %v", err)
	}
	defer func() { _ = f.Close() }()
	rows, err := f.GetRows("报名名单")
	if err != nil {
		t.Fatalf("读取报名名单: %v", err)
	}
	if len(rows) != 7 {
		t.Fatalf("行数 = %d，期望 7", len(rows))
	}
}