package business

import (
	"context"
	"fmt"
	"math/big"
	"sort"
	"testing"
	"time"
)

type workbenchRepoFake struct {
	totals *WorkbenchTotals
	data   *WorkbenchRankData
	window WorkbenchWindow
	filter AnalyticsFilter
	scope  AnalyticsScope
	named  []*WorkbenchRankRow
}

func TestWorkbenchTopTenMatchesFullStableSort(t *testing.T) {
	for _, kind := range []WorkbenchRankKind{WorkbenchPayHigh, WorkbenchPayLow} {
		uc, repo := workbenchFixture()
		for i := 0; i < 3000; i++ {
			repo.data.Rows = append(repo.data.Rows, &WorkbenchRankRow{AnalyticsKey: AnalyticsKey{ID: int64(i%11 + 1), SourceDB: []string{"lx", "hw", "yq"}[i%3]}, Name: fmt.Sprintf("candidate-%04d", i), Value: big.NewRat(int64(i%7-2), 100).FloatString(2)})
		}
		expected := make([]*WorkbenchRankRow, 0, len(repo.data.Rows))
		for _, row := range repo.data.Rows {
			n, _ := new(big.Rat).SetString(row.Value)
			if kind == WorkbenchPayLow || n.Sign() > 0 {
				expected = append(expected, row)
			}
		}
		sort.SliceStable(expected, func(i, j int) bool {
			a, b := expected[i], expected[j]
			av, _ := new(big.Rat).SetString(a.Value)
			bv, _ := new(big.Rat).SetString(b.Value)
			if order := av.Cmp(bv); order != 0 {
				if kind == WorkbenchPayLow {
					return order < 0
				}
				return order > 0
			}
			if a.ID != b.ID {
				return a.ID < b.ID
			}
			return a.SourceDB < b.SourceDB
		})
		actual, err := uc.Ranks(accountContext(7, 901), []string{"lx", "hw", "yq"}, "last30", kind)
		if err != nil || len(actual) != 10 {
			t.Fatalf("rank failed: %v", err)
		}
		for i, row := range actual {
			if row.AnalyticsKey != expected[i].AnalyticsKey || row.Value != expected[i].Value || row.Name != expected[i].Name || row.Rank != i+1 {
				t.Fatalf("%s rank %d differs from full stable sort: %+v vs %+v", kind, i, row, expected[i])
			}
		}
	}
}

func (r *workbenchRepoFake) FillRankNames(_ context.Context, rows []*WorkbenchRankRow) error {
	r.named = rows
	return nil
}

func (r *workbenchRepoFake) Summary(_ context.Context, s AnalyticsScope, _ []string, w WorkbenchWindow) (*WorkbenchTotals, error) {
	r.window, r.scope = w, s
	c := *r.totals
	return &c, nil
}
func (r *workbenchRepoFake) Ranks(_ context.Context, s AnalyticsScope, f AnalyticsFilter, _ WorkbenchRankKind) (*WorkbenchRankData, error) {
	r.filter, r.scope = f, s
	out := &WorkbenchRankData{Available: r.data.Available}
	for _, row := range r.data.Rows {
		c := *row
		out.Rows = append(out.Rows, &c)
	}
	return out, nil
}
func workbenchFixture() (*WorkbenchUsecase, *workbenchRepoFake) {
	r := &workbenchRepoFake{totals: &WorkbenchTotals{Available: true, Today: WorkbenchDay{Counts: PortraitCounts{Register: 3, Active: 4}, Pay: "20.005", ActiveChannels: 2, NewChannels: 1}, Yesterday: WorkbenchDay{Counts: PortraitCounts{Register: 2}, Pay: "10", ActiveChannels: 1}, MonthPay: "300", PreviousMonthPay: "200"}, data: &WorkbenchRankData{Available: true}}
	a := &accountRepoFake{accounts: map[uint]*Account{7: {ID: 7, Level: 1, IDPath: "0"}}}
	uc := NewWorkbenchUsecase(r, NewAccountUsecase(a), NewUserPreferenceUsecase(&channelPreferenceFake{UserPreference{IncludeSubPromote: true, IDCardDedup: true}}))
	uc.now = func() time.Time { return time.Date(2026, 3, 31, 9, 0, 0, 0, portraitLocation) }
	return uc, r
}
func TestWorkbenchSummaryCalendarAndRatios(t *testing.T) {
	uc, r := workbenchFixture()
	out, err := uc.Summary(accountContext(7, 901), []string{"lx"})
	if err != nil || *out.TodayRegister.Value != "3.00" || *out.TodayRegister.ChangeRate != "50.0" || out.TodayActive.ChangeRate != nil || *out.TodayPay.Value != "20.01" || *out.TodayPay.ChangeRate != "100.1" || *out.MonthPay.ChangeRate != "50.0" || *out.ActivePromote.ChangeRate != "100.0" {
		t.Fatalf("卡片精度或环比错误: %+v %v", out, err)
	}
	if r.window.PreviousMonthEnd.Format("2006-01-02") != "2026-02-28" || r.window.PreviousMonthStart.Format("2006-01-02") != "2026-02-01" {
		t.Fatal("上月同期需要截断至月末")
	}
	r.totals.Available = false
	out, err = uc.Summary(accountContext(7, 901), []string{"lx"})
	if err != nil || out.TodayPay.Value != nil || out.TodayPay.CompareValue != nil || out.Available {
		t.Fatal("缺表不能返回零卡片")
	}
	if _, err := uc.Summary(accountContext(7, 901), nil); err != ErrWorkbenchSource {
		t.Fatal("必须选择平台")
	}
}
func TestWorkbenchRankLimitsExactSortDefaultsAndUnlogin(t *testing.T) {
	uc, r := workbenchFixture()
	ctx := accountContext(7, 901)
	for i := 0; i < 14; i++ {
		r.data.Rows = append(r.data.Rows, &WorkbenchRankRow{AnalyticsKey: AnalyticsKey{SourceDB: "lx", ID: int64(i)}, Value: fmt.Sprint(i)})
	}
	rows, err := uc.Ranks(ctx, []string{"lx"}, "", WorkbenchPayHigh)
	if err != nil || len(rows) != 10 || rows[0].ID != 13 || rows[9].Rank != 10 {
		t.Fatal("榜单排序或 TOP10 截断错误")
	}
	if len(r.named) != 10 {
		t.Fatal("仅应补充已入选 TOP10 的渠道名称")
	}
	rows, err = uc.Ranks(ctx, []string{"lx"}, "", WorkbenchPayLow)
	if err != nil || rows[0].ID != 0 || rows[0].Value != "0.00" || rows[0].Name != "官方渠道" || r.filter.Begin != "2026-03-02" {
		t.Fatal("最低榜应包含零值，默认近30天")
	}
	r.data.Rows = []*WorkbenchRankRow{{AnalyticsKey: AnalyticsKey{SourceDB: "lx", ID: 1}, Value: "9007199254740993.00"}, {AnalyticsKey: AnalyticsKey{SourceDB: "hw", ID: 2}, Value: "9007199254740992.99"}}
	rows, err = uc.Ranks(ctx, []string{"lx", "hw"}, "last7", WorkbenchPayHigh)
	if err != nil || rows[0].ID != 1 || r.filter.Begin != "2026-03-25" {
		t.Fatal("金额排序丢失精度或近7天范围错误")
	}
	if _, err := uc.Ranks(ctx, []string{"lx"}, "invalid", WorkbenchRegister); err != ErrWorkbenchPeriod {
		t.Fatal("无效时间维度必须拒绝")
	}
	today := workbenchWindow(uc.now()).Today
	created := today.AddDate(0, 0, -5).Unix()
	future := today.AddDate(0, 0, 1).Unix()
	r.data.Rows = []*WorkbenchRankRow{{AnalyticsKey: AnalyticsKey{ID: 1}}, {AnalyticsKey: AnalyticsKey{ID: 2}, CreatedUnix: &created}, {AnalyticsKey: AnalyticsKey{ID: 3}, CreatedUnix: &created, LastLoginUnix: &future}}
	rows, err = uc.Ranks(ctx, []string{"lx"}, "ignored", WorkbenchUnlogin)
	if err != nil || rows[0].Value != "99999" || rows[1].Value != "5" || rows[2].Value != "0" {
		t.Fatal("未登录回退创建时间、无时间或未来时间错误")
	}
	r.data.Available = false
	if _, err := uc.Ranks(ctx, []string{"lx"}, "", WorkbenchUnlogin); err != ErrWorkbenchUnavailable {
		t.Fatal("缺数据不能返回空榜单")
	}
}