package export

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

// seedSecondSport 再造一个大项下的小项（1 个组别 1 条成绩），
// 用于验证「大项 / 小项 / 组别」三列在多个小项之间不会串行。
func seedSecondSport(t *testing.T, db *sql.DB) {
	t.Helper()
	stmts := []string{
		`INSERT INTO sports(code, name) VALUES ('SP2', '游泳')`,
		`INSERT INTO events(sport_id, code, name, event_type, result_mode, result_type, compare, status)
		 VALUES (2, 'E2', '50米自由泳', 'individual', 'auto', 'time', 'asc', 'published')`,
		`INSERT INTO event_categories(event_id, name, sort_order) VALUES (2, '女子', 1)`,
		`INSERT INTO event_score_channels(event_id, scope, basis, profile_id)
		 SELECT 2, 'athlete', 'result', id FROM scoring_profiles WHERE code = 'individual_top6'`,
		`INSERT INTO entries(event_id, category_id, team_id, athlete_id, bib) VALUES (2, 2, 1, 1, '201')`,
		`INSERT INTO results(event_id, entry_id, rank, result_text, result_value, unit, state)
		 VALUES (2, 3, 1, '28.30', 28.30, 's', 'published')`,
		`INSERT INTO score_ledger(team_id, athlete_id, event_id, entry_id, result_id, source_type, source_id, points)
		 VALUES (1, 1, 2, 3, 3, 'result', 3, 9)`,
	}
	for _, s := range stmts {
		if _, err := db.Exec(s); err != nil {
			t.Fatalf("造数据 %q: %v", s, err)
		}
	}
}

// 单页导出：只有一个 sheet，行数是全部小项 sheet 行数之和，前三列标注所属小项。
func TestExportFlatXLSX(t *testing.T) {
	db := openEventDB(t)
	seedEvent(t, db)
	seedSecondSport(t, db)

	dst := filepath.Join(t.TempDir(), "单页.xlsx")
	if err := ExportFlatXLSX(db, dst, Options{}); err != nil {
		t.Fatalf("导出单页总表: %v", err)
	}
	f := openXLSX(t, dst)
	defer func() { _ = f.Close() }()

	if got := f.GetSheetList(); !equalStrings(got, []string{flatSheet}) {
		t.Fatalf("sheet 列表 = %v，期望只有 %s", got, flatSheet)
	}
	rows, err := f.GetRows(flatSheet)
	if err != nil {
		t.Fatalf("读取单页总表: %v", err)
	}
	// 4 行版式 + 小项 1 的 2 条成绩（两个单位各 1 条）+ 小项 2 的 1 条成绩
	if len(rows) != 7 {
		t.Fatalf("单页总表行数 = %d，期望 7", len(rows))
	}
	wantHeader := []string{"大项", "小项", "组别", "名次", "号码", "单位", "运动员/队", "成绩", "积分", "状态", "备注"}
	for i, want := range wantHeader {
		if got := cell(rows, 3, i); got != want {
			t.Errorf("表头第 %d 列 = %q，期望 %q", i, got, want)
		}
	}
	// 小项 1：田径 / 100米跑 / 男子。
	first := []string{"田径", "100米跑", "男子", "1", "101", "第一分公司", "张三", "11.58 秒", "10", "有效"}
	for i, want := range first {
		if got := cell(rows, 4, i); got != want {
			t.Errorf("第 1 行第 %d 列 = %q，期望 %q", i, got, want)
		}
	}
	// 同一个小项的第二条成绩：三列保持该小项，不随行变。
	sameEvent := []string{"田径", "100米跑", "男子", "2", "102", "第二分公司", "李四", "11.90 秒", "8", "有效"}
	for i, want := range sameEvent {
		if got := cell(rows, 5, i); got != want {
			t.Errorf("第 2 行第 %d 列 = %q，期望 %q", i, got, want)
		}
	}
	// 小项 2：游泳 / 50米自由泳 / 女子——三列必须跟着换，不能沿用上一个小项。
	second := []string{"游泳", "50米自由泳", "女子", "1", "201", "第一分公司", "张三", "28.30 秒", "9", "有效"}
	for i, want := range second {
		if got := cell(rows, 6, i); got != want {
			t.Errorf("第 3 行第 %d 列 = %q，期望 %q", i, got, want)
		}
	}
}

// 与「全部」（all.xlsx）逐格一致：同一个小项的行，除新增三列外每个单元格都相同。
func TestFlatMatchesAllXLSX(t *testing.T) {
	db := openEventDB(t)
	seedEvent(t, db)
	seedSecondSport(t, db)

	allDst := filepath.Join(t.TempDir(), "全部.xlsx")
	if err := ExportAllXLSX(db, allDst, Options{}); err != nil {
		t.Fatalf("导出全部: %v", err)
	}
	flatDst := filepath.Join(t.TempDir(), "单页.xlsx")
	if err := ExportFlatXLSX(db, flatDst, Options{}); err != nil {
		t.Fatalf("导出单页总表: %v", err)
	}

	allF := openXLSX(t, allDst)
	defer func() { _ = allF.Close() }()
	flatF := openXLSX(t, flatDst)
	defer func() { _ = flatF.Close() }()

	flatRows, err := flatF.GetRows(flatSheet)
	if err != nil {
		t.Fatalf("读取单页总表: %v", err)
	}

	// 判断哪个分项 sheet 对应哪组结构化数据：直接重新加载一次，顺序与写 sheet 时一致，
	// 不用去解析 sheet 名或标题字符串。
	wb, err := LoadAllWorkbook(db, Options{})
	if err != nil {
		t.Fatalf("读取全部: %v", err)
	}
	// 「全部」前 3 张是总表（总积分榜 / 单位奖牌榜 / 个人积分奖牌榜），其后才是分项 sheet。
	sheets := allF.GetSheetList()
	if len(sheets) != 3+len(wb.Events.Sheets) {
		t.Fatalf("全部导出 sheet 列表 = %v，与 %d 个分项对不上", sheets, len(wb.Events.Sheets))
	}

	flatRow, compared := 4, 0
	for si, name := range sheets[3:] {
		sh := wb.Events.Sheets[si]
		rows, err := allF.GetRows(name)
		if err != nil {
			t.Fatalf("读取分项 %s: %v", name, err)
		}
		// 版式为 1 标题 / 2 副标题 / 3 空行 / 4 表头，数据从第 5 行（索引 4）起。
		dataRows := len(rows) - 4
		for i := 0; i < dataRows; i++ {
			for c := 0; c < len(eventColumns); c++ {
				if got, want := cell(flatRows, flatRow+i, c+3), cell(rows, 4+i, c); got != want {
					t.Errorf("分项 %s 第 %d 行第 %d 列：单页 %q，全部 %q", name, i, c, got, want)
				}
			}
			// 新增的三列必须是该 sheet 所属的大项 / 小项 / 组别。
			for c, want := range []string{sh.SportName, sh.EventName, sh.Category} {
				if got := cell(flatRows, flatRow+i, c); got != want {
					t.Errorf("分项 %s 第 %d 行第 %d 列 = %q，期望 %q", name, i, c, got, want)
				}
			}
			compared++
		}
		flatRow += dataRows
	}

	// 行数 = 全部小项 sheet 的行数之和（+ 4 行版式）。
	if flatRow != len(flatRows) {
		t.Errorf("单页总表行数 = %d，期望 %d", len(flatRows), flatRow)
	}
	if compared != 3 {
		t.Fatalf("比对的数据行数 = %d，期望 3", compared)
	}
	t.Logf("单页数据行三列: %v / %v / %v", flatRows[4][:3], flatRows[5][:3], flatRows[6][:3])
}

// 空赛事：返回一张只有标题与表头的工作簿，不报错。
func TestExportFlatXLSXEmptyEvent(t *testing.T) {
	db := openEventDB(t)

	dst := filepath.Join(t.TempDir(), "单页.xlsx")
	if err := ExportFlatXLSX(db, dst, Options{}); err != nil {
		t.Fatalf("空赛事导出单页总表: %v", err)
	}
	f := openXLSX(t, dst)
	defer func() { _ = f.Close() }()

	if got := f.GetSheetList(); !equalStrings(got, []string{flatSheet}) {
		t.Fatalf("sheet 列表 = %v，期望只有 %s", got, flatSheet)
	}
	rows, err := f.GetRows(flatSheet)
	if err != nil {
		t.Fatalf("读取单页总表: %v", err)
	}
	if len(rows) != 4 { // 标题 / 副标题 / 空行 / 表头，没有数据行
		t.Fatalf("空赛事单页总表行数 = %d，期望 4", len(rows))
	}
	if got := cell(rows, 3, 0); got != "大项" {
		t.Errorf("空赛事表头首列 = %q，期望 大项", got)
	}
}