package repository

import (
	"context"
	"errors"

	"gorm.io/gorm"
	"pay-bridge/internal/model"
)

// RefundOrderRepository 退款记录数据访问
type RefundOrderRepository struct {
	db *gorm.DB
}

func NewRefundOrderRepository(db *gorm.DB) *RefundOrderRepository {
	return &RefundOrderRepository{db: db}
}

// Create 创建退款单
func (r *RefundOrderRepository) Create(ctx context.Context, refund *model.RefundOrder) error {
	return r.db.WithContext(ctx).Create(refund).Error
}

// GetByRefundNo 按 refund_no 查询
func (r *RefundOrderRepository) GetByRefundNo(ctx context.Context, refundNo string) (*model.RefundOrder, error) {
	var refund model.RefundOrder
	err := r.db.WithContext(ctx).Where("refund_no = ?", refundNo).First(&refund).Error
	if errors.Is(err, gorm.ErrRecordNotFound) {
		return nil, nil
	}
	return &refund, err
}

// SumRefundedAmount 统计某笔交易已退款总额（成功+处理中）
func (r *RefundOrderRepository) SumRefundedAmount(ctx context.Context, tradeNo string) (int64, error) {
	var total int64
	err := r.db.WithContext(ctx).Model(&model.RefundOrder{}).
		Where("trade_no = ? AND status IN ?", tradeNo, []model.RefundStatus{
			model.RefundStatusPending,
			model.RefundStatusProcessing,
			model.RefundStatusSuccess,
		}).
		Select("COALESCE(SUM(refund_amount), 0)").
		Scan(&total).Error
	return total, err
}

// UpdateStatus 更新退款状态
func (r *RefundOrderRepository) UpdateStatus(ctx context.Context, refundNo string, fromStatus, toStatus model.RefundStatus, updates map[string]any) (bool, error) {
	if updates == nil {
		updates = make(map[string]any)
	}
	updates["status"] = toStatus

	result := r.db.WithContext(ctx).Model(&model.RefundOrder{}).
		Where("refund_no = ? AND status = ?", refundNo, fromStatus).
		Updates(updates)
	if result.Error != nil {
		return false, result.Error
	}
	return result.RowsAffected > 0, nil
}