junhong_cmp_fiber/internal/application/agentrecharge/approval_decision.go

package agentrecharge

import (
	"context"
	"strings"

	"gorm.io/gorm"
	"gorm.io/gorm/clause"

	approvalapp "github.com/break/junhong_cmp_fiber/internal/application/approval"
	walletapp "github.com/break/junhong_cmp_fiber/internal/application/wallet"
	"github.com/break/junhong_cmp_fiber/internal/model"
	"github.com/break/junhong_cmp_fiber/pkg/auditcontext"
	"github.com/break/junhong_cmp_fiber/pkg/constants"
	"github.com/break/junhong_cmp_fiber/pkg/errors"
)

// ApprovalDecisionHandler 将渠道无关审批终态应用到员工线下代充值业务。
type ApprovalDecisionHandler struct {
	db      *gorm.DB
	posting *walletapp.PostingService
	audit   RechargeAuditWriter
}

// NewApprovalDecisionHandler 创建员工线下代充值审批终态消费者。
func NewApprovalDecisionHandler(db *gorm.DB, posting *walletapp.PostingService, audit RechargeAuditWriter) *ApprovalDecisionHandler {
	return &ApprovalDecisionHandler{db: db, posting: posting, audit: audit}
}

// Handle 幂等处理标准审批终态；只有 approved 首次入账，其他终态不修改钱包。
func (h *ApprovalDecisionHandler) Handle(ctx context.Context, event approvalapp.TerminalDecisionEvent) error {
	if h == nil || h.db == nil || h.posting == nil || h.audit == nil {
		return errors.New(errors.CodeInternalError, "员工线下代充值审批终态能力未配置")
	}
	if event.BusinessType != constants.ApprovalBusinessTypeOfflineRecharge || event.BusinessID == 0 || event.InstanceID == 0 {
		return errors.New(errors.CodeInvalidParam, "员工线下代充值审批终态参数无效")
	}
	ctx = auditcontext.With(ctx, auditcontext.Context{CorrelationID: event.CorrelationID, ParentEventID: event.EventID})
	return h.db.WithContext(ctx).Transaction(func(tx *gorm.DB) error {
		var record model.AgentRechargeRecord
		if err := tx.WithContext(ctx).Clauses(clause.Locking{Strength: "UPDATE"}).
			Where("id = ? AND payment_method = ?", event.BusinessID, constants.RechargeMethodOffline).
			First(&record).Error; err != nil {
			if err == gorm.ErrRecordNotFound {
				return errors.New(errors.CodeForbidden, "无权限操作该资源或资源不存在")
			}
			return errors.Wrap(errors.CodeDatabaseError, err, "锁定员工线下代充值申请失败")
		}
		if record.ApprovalInstanceID == nil || *record.ApprovalInstanceID != event.InstanceID {
			return errors.New(errors.CodeConflict, "线下代充值申请关联的审批实例不一致")
		}
		switch event.Decision {
		case constants.ApprovalDecisionApproved:
			return h.applyApproved(ctx, tx, &record, event)
		case constants.ApprovalDecisionRejected:
			return h.closeOfflineRecharge(ctx, tx, &record, event, constants.RechargeStatusRejected, "企业微信审批已拒绝")
		case constants.ApprovalDecisionCancelled:
			return h.closeOfflineRecharge(ctx, tx, &record, event, constants.RechargeStatusClosed, "企业微信审批已撤销")
		case constants.ApprovalDecisionDeleted:
			return h.closeOfflineRecharge(ctx, tx, &record, event, constants.RechargeStatusClosed, "企业微信审批已删除")
		case constants.ApprovalDecisionRevokedAfterApproved:
			return nil
		default:
			return errors.New(errors.CodeInvalidParam, "不支持的线下代充值审批终态")
		}
	})
}

func (h *ApprovalDecisionHandler) applyApproved(
	ctx context.Context,
	tx *gorm.DB,
	record *model.AgentRechargeRecord,
	event approvalapp.TerminalDecisionEvent,
) error {
	if record.Status != constants.RechargeStatusPending && record.Status != constants.RechargeStatusCompleted {
		return errors.New(errors.CodeInvalidStatus, "线下代充值申请状态不允许审批入账")
	}
	if record.Status == constants.RechargeStatusPending {
		result := tx.WithContext(ctx).Model(&model.AgentRechargeRecord{}).
			Where("id = ? AND status = ?", record.ID, constants.RechargeStatusPending).
			Updates(map[string]any{
				"status": constants.RechargeStatusCompleted, "paid_at": event.OccurredAt,
				"completed_at": event.OccurredAt, "updated_at": event.OccurredAt,
			})
		if result.Error != nil {
			return errors.Wrap(errors.CodeDatabaseError, result.Error, "完成线下代充值审批申请失败")
		}
		if result.RowsAffected != 1 {
			return errors.New(errors.CodeConflict, "线下代充值申请状态已变化")
		}
	}
	posting, err := h.posting.PostInTx(ctx, tx, walletapp.PostingCommand{
		ShopID: record.ShopID, WalletID: record.AgentWalletID, Amount: record.Amount,
		ReferenceType: constants.ReferenceTypeTopup, ReferenceID: record.ID,
		TransactionType: constants.AgentTransactionTypeRecharge,
		UserID:          event.SubmitterAccountID, Creator: event.SubmitterAccountID,
		Remark: "企业微信审批通过线下充值", CorrelationID: event.CorrelationID,
	})
	if err != nil {
		return err
	}
	if record.Status == constants.RechargeStatusCompleted && posting.AlreadyApplied {
		return nil
	}
	return h.appendTerminalAudit(ctx, tx, record, event, constants.AuditActionAgentRechargeCredited, "企业微信审批通过，代理充值已入账", constants.RechargeStatusCompleted, true)
}

func (h *ApprovalDecisionHandler) closeOfflineRecharge(ctx context.Context, tx *gorm.DB, record *model.AgentRechargeRecord, event approvalapp.TerminalDecisionEvent, status int, reason string) error {
	if record.Status == status {
		return nil
	}
	if record.Status != constants.RechargeStatusPending {
		return errors.New(errors.CodeInvalidStatus, "线下代充值申请状态不允许结束审批")
	}
	reason = strings.TrimSpace(reason)
	result := tx.WithContext(ctx).Model(&model.AgentRechargeRecord{}).
		Where("id = ? AND status = ?", record.ID, constants.RechargeStatusPending).
		Updates(map[string]any{"status": status, "rejection_reason": reason})
	if result.Error != nil {
		return errors.Wrap(errors.CodeDatabaseError, result.Error, "结束线下代充值审批申请失败")
	}
	if result.RowsAffected != 1 {
		return errors.New(errors.CodeConflict, "线下代充值申请状态已变化")
	}
	return h.appendTerminalAudit(ctx, tx, record, event, constants.AuditActionAgentRechargeClosed, reason, status, false)
}

func (h *ApprovalDecisionHandler) appendTerminalAudit(ctx context.Context, tx *gorm.DB, record *model.AgentRechargeRecord, event approvalapp.TerminalDecisionEvent, actionCode, summary string, status int, withWallet bool) error {
	after := *record
	after.Status = status
	change := RechargeAudit{
		ActionCode: actionCode, Summary: summary, Record: &after,
		BeforeData: map[string]any{"status": record.Status}, AfterData: map[string]any{"status": status},
	}
	var approval model.ApprovalInstance
	if err := tx.WithContext(ctx).First(&approval, event.InstanceID).Error; err != nil {
		return errors.Wrap(errors.CodeDatabaseError, err, "查询线下代充值审批审计快照失败")
	}
	change.Approval = &approval
	if withWallet {
		var wallet model.AgentWallet
		if err := tx.WithContext(ctx).First(&wallet, record.AgentWalletID).Error; err != nil {
			return errors.Wrap(errors.CodeDatabaseError, err, "查询代理充值钱包审计快照失败")
		}
		var transaction model.AgentWalletTransaction
		if err := tx.WithContext(ctx).Where("reference_type = ? AND reference_id = ? AND transaction_type = ? AND status = ?",
			constants.ReferenceTypeTopup, record.ID, constants.AgentTransactionTypeRecharge, constants.TransactionStatusSuccess).
			First(&transaction).Error; err != nil {
			return errors.Wrap(errors.CodeDatabaseError, err, "查询代理充值入账流水审计快照失败")
		}
		change.Wallet, change.Transaction = &wallet, &transaction
	}
	return h.audit.WriteAgentRecharge(ctx, tx, change)
}

var _ approvalapp.BusinessDecisionHandler = (*ApprovalDecisionHandler)(nil)