package logic

import (
	"context"

	"muyu-apiserver/rpc/purchase/internal/svc"
	"muyu-apiserver/rpc/purchase/pb"

	"github.com/zeromicro/go-zero/core/logx"
)

type CancelPurchaseOrderLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewCancelPurchaseOrderLogic(ctx context.Context, svcCtx *svc.ServiceContext) *CancelPurchaseOrderLogic {
	return &CancelPurchaseOrderLogic{ctx: ctx, svcCtx: svcCtx, Logger: logx.WithContext(ctx)}
}

func (l *CancelPurchaseOrderLogic) CancelPurchaseOrder(in *pb.CancelPurchaseOrderReq) (*pb.Empty, error) {
	order, err := l.svcCtx.OrderModel.FindOneByOrderId(l.ctx, in.OrderId)
	if err != nil {
		return nil, ErrOrderNotFound
	}
	// Only confirmed (status=1) with no receipt can be cancelled
	if order.Status != 1 {
		return nil, ErrOrderNotConfirmed
	}
	if order.ReceiptStatus != 0 {
		return nil, ErrOrderAlreadyReceived
	}

	order.Status = 2 // cancelled
	if err := l.svcCtx.OrderModel.Update(l.ctx, order); err != nil {
		return nil, err
	}
	// Reverse supplier payable
	if err := l.svcCtx.SupplierModel.UpdateStats(l.ctx, order.SupplierId, -1, -order.ContractAmount, 0, 0); err != nil {
		l.Errorf("reverse supplier stats failed: %v", err)
	}
	return &pb.Empty{}, nil
}