hfb_sys/backend/internal/modules/paymentconfig/query.go

package paymentconfig

import (
	"context"
	"errors"

	"hfb_sys/backend/internal/model"

	"gorm.io/gorm"
)

// List 获取配置列表
func (r *Repository) List(ctx context.Context, query ListQuery) ([]ConfigDTO, int64, error) {
	var items []model.PaymentMerchantConfig
	var total int64

	db := r.db.WithContext(ctx).Model(&model.PaymentMerchantConfig{})

	// 过滤条件
	if query.Provider != "" {
		db = db.Where("provider = ?", query.Provider)
	}
	if query.PayWay != "" {
		db = db.Where("pay_way = ?", query.PayWay)
	}
	if query.Status != "" {
		db = db.Where("status = ?", query.Status)
	}
	if query.Environment != "" {
		db = db.Where("environment = ?", query.Environment)
	}

	// 获取总数
	if err := db.Count(&total).Error; err != nil {
		return nil, 0, err
	}

	// 分页
	page := query.Page
	if page < 1 {
		page = 1
	}
	pageSize := query.PageSize
	if pageSize < 1 {
		pageSize = 20
	}
	if pageSize > 100 {
		pageSize = 100
	}

	offset := (page - 1) * pageSize
	if err := db.Order("pay_way ASC, is_default DESC, id DESC").Offset(offset).Limit(pageSize).Find(&items).Error; err != nil {
		return nil, 0, err
	}

	dtos := make([]ConfigDTO, 0, len(items))
	for _, item := range items {
		dto, err := r.toDTO(item, false)
		if err != nil {
			return nil, 0, err
		}
		dtos = append(dtos, dto)
	}

	return dtos, total, nil
}

// FindByID 根据 ID 查询配置
func (r *Repository) FindByID(ctx context.Context, id uint64, includeSecret bool, actorID uint64, meta AuditMeta) (*ConfigDTO, error) {
	var item model.PaymentMerchantConfig
	if err := r.db.WithContext(ctx).Where("id = ?", id).First(&item).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrConfigNotFound
		}
		return nil, err
	}
	dto, err := r.toDTO(item, includeSecret)
	if err != nil {
		return nil, err
	}
	if includeSecret {
		if err := appendAuditLog(r.db.WithContext(ctx), actorID, "payment_config.view_secret", item.ID, meta, map[string]any{
			"name":        item.Name,
			"provider":    item.Provider,
			"merchant_id": item.MerchantID,
		}); err != nil {
			return nil, err
		}
	}
	return &dto, nil
}

// FindRuntimeByID 根据配置 ID 查询运行时配置，不写审计日志，供支付单回溯原配置使用。
func (r *Repository) FindRuntimeByID(ctx context.Context, id uint64, includeSecret bool) (*ConfigDTO, error) {
	var item model.PaymentMerchantConfig
	if err := r.db.WithContext(ctx).Where("id = ?", id).First(&item).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrConfigNotFound
		}
		return nil, err
	}
	dto, err := r.toDTO(item, includeSecret)
	if err != nil {
		return nil, err
	}
	return &dto, nil
}

// FindDefault 查询指定服务商的默认启用配置，兼容旧调用路径。
func (r *Repository) FindDefault(ctx context.Context, provider string) (*model.PaymentMerchantConfig, error) {
	var item model.PaymentMerchantConfig
	if err := r.db.WithContext(ctx).Where("provider = ? AND is_default = ? AND status = ?", provider, true, "active").First(&item).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrNoActiveConfigFound
		}
		return nil, err
	}
	return &item, nil
}

// FindDefaultByPayWay 查询指定支付方式的默认启用配置。
func (r *Repository) FindDefaultByPayWay(ctx context.Context, payWay string, includeSecret bool) (*ConfigDTO, error) {
	var item model.PaymentMerchantConfig
	if err := r.db.WithContext(ctx).
		Where("pay_way = ? AND status = ?", payWay, "active").
		Order("is_default DESC, id DESC").
		First(&item).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrNoActiveConfigFound
		}
		return nil, err
	}
	dto, err := r.toDTO(item, includeSecret)
	if err != nil {
		return nil, err
	}
	return &dto, nil
}

// FindDefaultByProvider 查询指定服务商的默认启用配置。
func (r *Repository) FindDefaultByProvider(ctx context.Context, provider string, includeSecret bool) (*ConfigDTO, error) {
	item, err := r.FindDefault(ctx, provider)
	if err != nil {
		return nil, err
	}
	dto, err := r.toDTO(*item, includeSecret)
	if err != nil {
		return nil, err
	}
	return &dto, nil
}

// FindDefaultByProviderPayWay 查询指定服务商和支付方式的默认启用配置。
func (r *Repository) FindDefaultByProviderPayWay(ctx context.Context, provider string, payWay string, includeSecret bool) (*ConfigDTO, error) {
	var item model.PaymentMerchantConfig
	if err := r.db.WithContext(ctx).
		Where("provider = ? AND pay_way = ? AND status = ?", provider, payWay, "active").
		Order("is_default DESC, id DESC").
		First(&item).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrNoActiveConfigFound
		}
		return nil, err
	}
	dto, err := r.toDTO(item, includeSecret)
	if err != nil {
		return nil, err
	}
	return &dto, nil
}

// FindByProviderMerchantPayWay 根据服务商、商户号和支付方式查配置，用于旧支付单回溯原商户密钥。
func (r *Repository) FindByProviderMerchantPayWay(ctx context.Context, provider string, merchantID string, payWay string, includeSecret bool) (*ConfigDTO, error) {
	var item model.PaymentMerchantConfig
	if err := r.db.WithContext(ctx).Where("provider = ? AND merchant_id = ? AND pay_way = ?", provider, merchantID, payWay).
		Order("status = 'active' DESC, is_default DESC, id DESC").
		First(&item).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrConfigNotFound
		}
		return nil, err
	}
	dto, err := r.toDTO(item, includeSecret)
	if err != nil {
		return nil, err
	}
	return &dto, nil
}

// FindByProviderMerchant 根据服务商和商户号查配置，保留给历史数据没有支付方式时兜底使用。
func (r *Repository) FindByProviderMerchant(ctx context.Context, provider string, merchantID string, includeSecret bool) (*ConfigDTO, error) {
	var item model.PaymentMerchantConfig
	if err := r.db.WithContext(ctx).Where("provider = ? AND merchant_id = ?", provider, merchantID).
		Order("status = 'active' DESC, is_default DESC, id DESC").
		First(&item).Error; err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, ErrConfigNotFound
		}
		return nil, err
	}
	dto, err := r.toDTO(item, includeSecret)
	if err != nil {
		return nil, err
	}
	return &dto, nil
}

// FindActiveByProvider 查询提供商的所有激活配置
func (r *Repository) FindActiveByProvider(ctx context.Context, provider string) ([]model.PaymentMerchantConfig, error) {
	var items []model.PaymentMerchantConfig
	if err := r.db.WithContext(ctx).Where("provider = ? AND status = ?", provider, "active").Order("is_default DESC, id DESC").Find(&items).Error; err != nil {
		return nil, err
	}
	return items, nil
}

// Create 创建配置