package purchase_wms

import (
	"context"
	"github.com/shopspring/decimal"
	"gorm.io/gorm"
	"strconv"
	"time"
	"wms/constvar"
	"wms/models"
	"wms/pkg/timex"
)

type Server struct {
	UnimplementedPurchaseServiceServer
}

func (s *Server) PurchaseToWms(c context.Context, req *PurchaseToWmsRequest) (*PurchaseToWmsResponse, error) {
	var operation models.Operation
	var details []*models.OperationDetails
	operation.SourceNumber = req.Number
	operation.OperationDate = timex.TimeToString2(time.Now())
	operation.Number = strconv.FormatInt(time.Now().Unix(), 10)
	operation.Status = constvar.OperationStatus_Ready
	operation.CompanyName = req.SupplierName
	operation.Source = req.Source
	warehouse, err := models.NewWarehouseSearch().First()
	if err != nil {
		return nil, err
	}
	operationType, err := models.NewOperationTypeSearch().SetWarehouseId(warehouse.Id).SetBaseOperationType(constvar.BaseOperationTypeIncoming).First()
	if err != nil {
		return nil, err
	}
	operation.OperationTypeName = operationType.Name
	operation.OperationTypeId = operationType.Id
	location, err := models.NewLocationSearch().SetID(warehouse.LocationId).First()
	if err != nil {
		return nil, err
	}
	first, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeVendor)).First()
	if err != nil {
		return nil, err
	}
	operation.LocationID = location.Id
	operation.BaseOperationType = constvar.BaseOperationTypeIncoming
	for _, product := range req.Product {
		var detail models.OperationDetails
		detail.ProductId = product.Id
		detail.Amount = decimal.NewFromInt(product.Amount)
		detail.FromLocationID = first.Id
		detail.ToLocationID = location.Id
		details = append(details, &detail)
	}
	err = models.WithTransaction(func(db *gorm.DB) error {
		err := models.NewOperationSearch().SetOrm(db).Create(&operation)
		if err != nil {
			return err
		}
		for _, detail := range details {
			detail.OperationID = operation.Id
		}
		err = models.NewOperationDetailsSearch().SetOrm(db).CreateBatch(details)
		return err
	})
	if err != nil {
		return nil, err
	}
	resp := new(PurchaseToWmsResponse)
	resp.Warehouse = warehouse.Name
	return resp, nil
}