package logic

import (
	"context"
	"database/sql"
	"fmt"
	"strconv"
	"time"

	"muyu-apiserver/model"
	"muyu-apiserver/pkg/tenantctx"
	"muyu-apiserver/pkg/uid"
	"muyu-apiserver/rpc/purchase/internal/svc"
	"muyu-apiserver/rpc/purchase/pb"

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

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

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

func (l *CreatePurchasePaymentLogic) CreatePurchasePayment(in *pb.CreatePurchasePaymentReq) (*pb.IdResp, error) {
	tenantId := tenantctx.ExtractTenantId(l.ctx)

	order, err := l.svcCtx.OrderModel.FindOneByOrderId(l.ctx, in.OrderId)
	if err != nil {
		return nil, ErrOrderNotFound
	}

	amount, _ := strconv.ParseFloat(in.Amount, 64)

	paymentDate, err := time.Parse("2006-01-02", in.PaymentDate)
	if err != nil {
		paymentDate = time.Now()
	}

	// Generate payment number
	today := time.Now().Format("20060102")
	count, err := l.svcCtx.PaymentModel.CountTodayPayments(l.ctx, tenantId, time.Now().Format("2006-01-02"))
	if err != nil {
		count = 0
	}
	_ = fmt.Sprintf("PAY-%s-%04d", today, count+1)

	paymentId := uid.Generate()
	_, err = l.svcCtx.PaymentModel.Insert(l.ctx, &model.PurPurchasePayment{
		PaymentId:     paymentId,
		TenantId:      tenantId,
		OrderId:       in.OrderId,
		SupplierId:    order.SupplierId,
		PaymentDate:   paymentDate,
		Amount:        amount,
		PaymentMethod: in.PaymentMethod,
		Operator:      tenantId,
		Remark:        sql.NullString{String: in.Remark, Valid: in.Remark != ""},
	})
	if err != nil {
		return nil, err
	}

	// Update order paid_amount and payment_status
	if err := l.svcCtx.OrderModel.UpdatePaymentProgress(l.ctx, in.OrderId, amount); err != nil {
		l.Errorf("update order payment progress failed: %v", err)
	}
	// Update supplier paid_amount
	if err := l.svcCtx.SupplierModel.UpdateStats(l.ctx, order.SupplierId, 0, 0, amount, 0); err != nil {
		l.Errorf("update supplier paid_amount failed: %v", err)
	}

	return &pb.IdResp{Id: paymentId}, nil
}