package gateway_test

import (
	"context"
	"encoding/json"
	"testing"

	"github.com/wangjia/pay/config"
	"github.com/wangjia/pay/internal/accounts"
	"github.com/wangjia/pay/internal/gateway"
	"github.com/wangjia/pay/internal/model"
	"github.com/wangjia/pay/internal/provider"
	"github.com/wangjia/pay/internal/provider/fake"
	"github.com/wangjia/pay/internal/store"
)

func attemptRef(t *testing.T, orders interface {
	ListAttemptsByStatus(model.AttemptStatus, int) ([]model.Attempt, error)
}) string {
	t.Helper()
	atts, _ := orders.ListAttemptsByStatus(model.AttemptPending, 10)
	if len(atts) == 0 {
		t.Fatalf("无 pending 尝试")
	}
	return atts[0].ProviderRef
}

func TestSettleHappyIdempotentAndWebhook(t *testing.T) {
	g, _, spy, orders := newGateway(t)
	ctx := context.Background()
	res, _ := g.CreateOrder(ctx, gateway.CreateOrderInput{SKU: "pro_year", Method: "fake", BizSystem: "pangolin", BizRef: "u-1"})
	ref := attemptRef(t, orders)

	ev := &provider.PaidEvent{ProviderRef: ref, Status: provider.PaidSucceeded, PaidAmountMinor: 29990000, PaidCurrency: "USDT"}
	got, err := g.Settle(ctx, ev)
	if err != nil || got != gateway.SettleProcessed {
		t.Fatalf("settle#1 = %v, %v", got, err)
	}
	// 订单已 paid
	o, _ := orders.GetOrder(res.OrderNo)
	if o.Status != model.OrderPaidV2 {
		t.Fatalf("order 应 paid, got %v", o.Status)
	}
	// webhook 入队一次,payload 带 event_type
	if len(spy.calls) != 1 || spy.calls[0]["event_type"] != "payment.succeeded" || spy.calls[0]["out_trade_no"] != res.OrderNo {
		t.Fatalf("webhook calls = %+v", spy.calls)
	}

	// 幂等:再 settle → duplicate,不重复入队
	got2, _ := g.Settle(ctx, ev)
	if got2 != gateway.SettleDuplicate || len(spy.calls) != 1 {
		t.Fatalf("settle#2 = %v, calls=%d", got2, len(spy.calls))
	}
}

func TestSettleWebhookCarriesProductBizCode(t *testing.T) {
	g, fp, spy, orders := newGateway(t)
	ctx := context.Background()
	res, _ := g.CreateOrder(ctx, gateway.CreateOrderInput{
		SKU: "pro_year", Method: "fake", BizSystem: "pangolin", BizRef: "u-9",
	})
	ref := attemptRef(t, orders)
	fp.SetQueryResult(ref, provider.PaidEvent{
		ProviderRef: ref, Status: provider.PaidSucceeded,
		PaidAmountMinor: 29990000, PaidCurrency: "USDT",
	})
	if _, err := g.SyncPendingAttempts(ctx, 10); err != nil {
		t.Fatalf("sync: %v", err)
	}
	if len(spy.calls) != 1 {
		t.Fatalf("want 1 webhook, got %d", len(spy.calls))
	}
	if spy.calls[0]["product_biz_code"] != "pro_year" {
		t.Fatalf("payload product_biz_code = %v, want pro_year", spy.calls[0]["product_biz_code"])
	}
	_ = res
}

func TestSettleGuards(t *testing.T) {
	g, _, spy, orders := newGateway(t)
	ctx := context.Background()
	g.CreateOrder(ctx, gateway.CreateOrderInput{SKU: "pro_year", Method: "fake", BizSystem: "pangolin", BizRef: "u-1"})
	ref := attemptRef(t, orders)

	// 未 succeeded → ignored
	if got, _ := g.Settle(ctx, &provider.PaidEvent{ProviderRef: ref, Status: provider.PaidPending}); got != gateway.SettleIgnored {
		t.Fatalf("pending 应 ignored, got %v", got)
	}
	// 未知 ref → not_found
	if got, _ := g.Settle(ctx, &provider.PaidEvent{ProviderRef: "GHOST", Status: provider.PaidSucceeded, PaidCurrency: "USDT", PaidAmountMinor: 1}); got != gateway.SettleNotFound {
		t.Fatalf("未知 ref 应 not_found, got %v", got)
	}
	// 少付 → amount_mismatch
	if got, err := g.Settle(ctx, &provider.PaidEvent{ProviderRef: ref, Status: provider.PaidSucceeded, PaidCurrency: "USDT", PaidAmountMinor: 1}); got != gateway.SettleAmountMismatch || err == nil {
		t.Fatalf("少付应 amount_mismatch, got %v %v", got, err)
	}
	// 错币种 → amount_mismatch
	if got, _ := g.Settle(ctx, &provider.PaidEvent{ProviderRef: ref, Status: provider.PaidSucceeded, PaidCurrency: "CNY", PaidAmountMinor: 29990000}); got != gateway.SettleAmountMismatch {
		t.Fatalf("错币种应 amount_mismatch, got %v", got)
	}
	if len(spy.calls) != 0 {
		t.Fatalf("守卫失败路径不应入队 webhook, got %d", len(spy.calls))
	}
}

// 资金命脉不变量:outbox 入队失败 → 绝不翻转订单(否则"已付但永不通知")。
// 渠道拿不到 200 会重投,重投时入队+翻转都幂等,自然恢复。
func TestSettleEnqueueFailureKeepsOrderPending(t *testing.T) {
	g, _, spy, orders := newGateway(t)
	ctx := context.Background()
	res, _ := g.CreateOrder(ctx, gateway.CreateOrderInput{SKU: "pro_year", Method: "fake", BizSystem: "pangolin", BizRef: "u-1"})
	ref := attemptRef(t, orders)
	ev := &provider.PaidEvent{ProviderRef: ref, Status: provider.PaidSucceeded, PaidAmountMinor: 29990000, PaidCurrency: "USDT"}

	spy.failNext = true
	if got, err := g.Settle(ctx, ev); got != gateway.SettleFailed || err == nil {
		t.Fatalf("入队失败应 SettleFailed+err, got %v, %v", got, err)
	}
	o, _ := orders.GetOrder(res.OrderNo)
	if o.Status != model.OrderPendingV2 {
		t.Fatalf("入队失败后订单必须仍 pending, got %v", o.Status)
	}

	// 渠道重投 → 入队成功 → 翻转
	if got, err := g.Settle(ctx, ev); err != nil || got != gateway.SettleProcessed {
		t.Fatalf("重投应 processed, got %v, %v", got, err)
	}
	if len(spy.calls) != 1 {
		t.Fatalf("恢复后应恰入队 1 次, got %d", len(spy.calls))
	}
}

// 瞬时读库失败(非 store.ErrAttemptNotFound 哨兵)必须归 SettleFailed(可重试),
// 不能与"查无此单"的终态 SettleNotFound 混淆——否则调用方按结果值决定 ack,
// 会把已付订单永久丢弃。构造方式:先建好订单/尝试,再直接关掉底层连接,
// 让 AttemptByProviderRef 打到一个已关闭的 DB 上,产出非哨兵错误。
func TestSettleTransientReadErrorIsFailed(t *testing.T) {
	db := model.OpenTestDB(t)
	orders := store.NewOrderStore(db)
	preg := provider.NewRegistry()
	fp := fake.New()
	preg.Register(fp)
	areg := accounts.New([]config.AccountConfig{
		{AccountID: "fake-a1", Channel: "fake", Region: "global", Enabled: true, Weight: 1},
	})
	picker := accounts.NewRouter(areg, nil, nil)
	spy := &spyEnqueuer{}
	g := gateway.New(orders, preg, picker, stubResolver{}, spy, "global")

	ctx := context.Background()
	g.CreateOrder(ctx, gateway.CreateOrderInput{SKU: "pro_year", Method: "fake", BizSystem: "pangolin", BizRef: "u-1"})
	ref := attemptRef(t, orders)

	sqlDB, err := db.DB()
	if err != nil {
		t.Fatalf("db.DB(): %v", err)
	}
	if err := sqlDB.Close(); err != nil {
		t.Fatalf("close db: %v", err)
	}

	ev := &provider.PaidEvent{ProviderRef: ref, Status: provider.PaidSucceeded, PaidAmountMinor: 29990000, PaidCurrency: "USDT"}
	got, err := g.Settle(ctx, ev)
	if got != gateway.SettleFailed || err == nil {
		t.Fatalf("瞬时读库失败应 SettleFailed+err, got %v, %v", got, err)
	}
}

func TestHandleCallbackAndSync(t *testing.T) {
	g, fp, _, orders := newGateway(t)
	ctx := context.Background()
	g.CreateOrder(ctx, gateway.CreateOrderInput{SKU: "pro_year", Method: "fake", BizSystem: "pangolin", BizRef: "u-1"})
	ref := attemptRef(t, orders)

	// 回调路径:fake.VerifyCallback 解析 JSON → Settle
	body, _ := json.Marshal(map[string]any{"provider_ref": ref, "status": "succeeded", "amount_minor": 29990000, "currency": "USDT"})
	got, err := g.HandleCallback(ctx, "fake", provider.CallbackInput{Raw: body})
	if err != nil || got != gateway.SettleProcessed {
		t.Fatalf("HandleCallback = %v, %v", got, err)
	}

	// 查单兜底:另起一单,预置 query 命中 → SyncPendingAttempts 收敛
	res2, _ := g.CreateOrder(ctx, gateway.CreateOrderInput{SKU: "pro_year", Method: "fake", BizSystem: "pangolin", BizRef: "u-2"})
	atts, _ := orders.ListAttemptsByStatus(model.AttemptPending, 10)
	ref2 := atts[0].ProviderRef
	fp.SetQueryResult(ref2, provider.PaidEvent{ProviderRef: ref2, Status: provider.PaidSucceeded, PaidAmountMinor: 29990000, PaidCurrency: "USDT"})
	n, err := g.SyncPendingAttempts(ctx, 10)
	if err != nil || n < 1 {
		t.Fatalf("SyncPendingAttempts = %d, %v", n, err)
	}
	o2, _ := orders.GetOrder(res2.OrderNo)
	if o2.Status != model.OrderPaidV2 {
		t.Fatalf("查单兜底后 order 应 paid, got %v", o2.Status)
	}
}