package order

import (
	"context"
	"testing"

	"github.com/perfect-panel/server/internal/model/promo"
	"github.com/perfect-panel/server/internal/svc"
	"github.com/perfect-panel/server/internal/types"
	"gorm.io/gorm"
)

type fakePromoModel struct {
	rules []*promo.RuleWithPrice
}

func (m fakePromoModel) QueryEligibleRules(context.Context, int64, int64) ([]*promo.RuleWithPrice, error) {
	return m.rules, nil
}

func (m fakePromoModel) InsertUsage(context.Context, *promo.Usage, ...*gorm.DB) error {
	return nil
}

func (m fakePromoModel) InsertRule(context.Context, *promo.Rule) error {
	return nil
}

func (m fakePromoModel) FindRule(context.Context, int64) (*promo.Rule, error) {
	return nil, gorm.ErrRecordNotFound
}

func (m fakePromoModel) UpdateRule(context.Context, *promo.Rule) error {
	return nil
}

func (m fakePromoModel) DeleteRule(context.Context, int64) error {
	return nil
}

func (m fakePromoModel) QueryRuleList(context.Context, int, int, string, *bool, string) (int64, []*promo.Rule, error) {
	return 0, nil, nil
}

func (m fakePromoModel) UpsertPrices(context.Context, int64, []*promo.SubscribePromo) error {
	return nil
}

func (m fakePromoModel) FindPrice(context.Context, int64) (*promo.SubscribePromo, error) {
	return nil, gorm.ErrRecordNotFound
}

func (m fakePromoModel) DeletePrice(context.Context, int64) error {
	return nil
}

func (m fakePromoModel) QueryPriceList(context.Context, int64, int, int) (int64, []*promo.SubscribePromo, error) {
	return 0, nil, nil
}

func (m fakePromoModel) QueryUsageList(context.Context, promo.UsageFilter) (int64, []*promo.Usage, error) {
	return 0, nil, nil
}

func (m fakePromoModel) Transaction(context.Context, func(*gorm.DB) error) error {
	return nil
}

func TestCalculatePurchasePricePromoSkipsPercentDiscount(t *testing.T) {
	svcCtx := &svc.ServiceContext{
		DB: &gorm.DB{},
		PromoModel: fakePromoModel{rules: []*promo.RuleWithPrice{
			{
				Rule: promo.Rule{
					Id:      9,
					Name:    "campaign",
					Type:    promo.RuleTypeCampaign,
					Enabled: true,
				},
				PromoPrice: 600,
			},
		}},
	}

	result, err := calculatePurchasePrice(
		context.Background(),
		svcCtx,
		1,
		2,
		1000,
		3,
		[]types.SubscribeDiscount{{Quantity: 3, Discount: 50}},
		true,
		true,
	)
	if err != nil {
		t.Fatalf("calculatePurchasePrice returned error: %v", err)
	}

	if result.OriginalPrice != 3000 {
		t.Fatalf("OriginalPrice = %d, want 3000", result.OriginalPrice)
	}
	if result.PayableBase != 1800 {
		t.Fatalf("PayableBase = %d, want 1800", result.PayableBase)
	}
	if result.DiscountAmount != 0 {
		t.Fatalf("DiscountAmount = %d, want 0", result.DiscountAmount)
	}
	if result.PromoRuleId != 9 {
		t.Fatalf("PromoRuleId = %d, want 9", result.PromoRuleId)
	}
	if result.PromoDiscount != 1200 {
		t.Fatalf("PromoDiscount = %d, want 1200", result.PromoDiscount)
	}
}

func TestCalculatePurchasePriceIgnoresInvalidPromoPrice(t *testing.T) {
	svcCtx := &svc.ServiceContext{
		DB: &gorm.DB{},
		PromoModel: fakePromoModel{rules: []*promo.RuleWithPrice{
			{
				Rule: promo.Rule{
					Id:      10,
					Name:    "invalid campaign",
					Type:    promo.RuleTypeCampaign,
					Enabled: true,
				},
				PromoPrice: 1000,
			},
		}},
	}

	result, err := calculatePurchasePrice(
		context.Background(),
		svcCtx,
		1,
		2,
		1000,
		3,
		[]types.SubscribeDiscount{{Quantity: 3, Discount: 50}},
		true,
		true,
	)
	if err != nil {
		t.Fatalf("calculatePurchasePrice returned error: %v", err)
	}

	if result.PayableBase != 1500 {
		t.Fatalf("PayableBase = %d, want 1500", result.PayableBase)
	}
	if result.DiscountAmount != 1500 {
		t.Fatalf("DiscountAmount = %d, want 1500", result.DiscountAmount)
	}
	if result.PromoRuleId != 0 || result.PromoDiscount != 0 {
		t.Fatalf("promo fields = (%d, %d), want (0, 0)", result.PromoRuleId, result.PromoDiscount)
	}
}