package logic

import (
	"context"

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

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

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

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

func (l *ConfirmPurchaseOrderLogic) ConfirmPurchaseOrder(in *pb.ConfirmPurchaseOrderReq) (*pb.Empty, error) {
	order, err := l.svcCtx.OrderModel.FindOneByOrderId(l.ctx, in.OrderId)
	if err != nil {
		return nil, ErrOrderNotFound
	}
	if order.Status != 0 {
		return nil, ErrOrderNotDraft
	}

	order.Status = 1 // confirmed
	if err := l.svcCtx.OrderModel.Update(l.ctx, order); err != nil {
		return nil, err
	}

	// Update supplier stats: purchase_count++, payable_amount += contract_amount
	if err := l.svcCtx.SupplierModel.UpdateStats(l.ctx, order.SupplierId, 1, order.ContractAmount, 0, 0); err != nil {
		l.Errorf("update supplier stats failed: %v", err)
	}

	return &pb.Empty{}, nil
}