package purchase

import (
	"context"
	"github.com/spf13/cast"
	"gorm.io/gorm"
	"srm/global"
	"srm/model/common/request"
	"srm/model/purchase"
	"srm/proto/qualityinspect"
	"srm/service/test"
)

type PurchaseService struct{}

func NewPurchaseService() *PurchaseService {
	return &PurchaseService{}
}

//@function: CreatePurchase
//@description: 创建采购单
//@param: params *purchaserequest.AddPurchase
//@return: err error

func (slf *PurchaseService) CreatePurchase(params *purchase.Purchase, productList []*purchase.PurchaseProducts) (err error) {
	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		err = tx.Create(&params).Error
		if err != nil {
			return err
		}
		for _, product := range productList {
			product.PurchaseId = cast.ToInt(params.ID)
		}
		return tx.Create(productList).Error
	})

	return err
}

//@function: DeletePurchase
//@description: 删除采购单
//@param: id uint
//@return: err error

func (slf *PurchaseService) DeletePurchase(id uint) (err error) {
	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		err = tx.Where("id = ?", id).Delete(&purchase.Purchase{}).Error
		if err != nil {
			return err
		}
		return tx.Where("purchase_id = ?", id).Delete(&purchase.PurchaseProducts{}).Error
	})
	return err
}

//@function: UpdatePurchase
//@description: 更新采购单
//@param: params *purchaserequest.AddPurchase
//@return: err error

func (slf *PurchaseService) UpdatePurchase(params *purchase.Purchase, productList []*purchase.PurchaseProducts) (err error) {
	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		err = tx.Where("id = ?", params.ID).Updates(params).Error
		if err != nil {
			return err
		}
		err = tx.Where("purchase_id = ?", params.ID).Delete(&purchase.PurchaseProducts{}).Error
		if err != nil {
			return err
		}
		for _, product := range productList {
			product.ID = 0
			product.PurchaseId = cast.ToInt(params.ID)
		}
		return tx.Create(productList).Error
	})
	return err
}

//@function: GetPurchase
//@description: 获取采购单信息
//@param: id uint
//@return: purchase model.Purchase, err error

func (slf *PurchaseService) GetPurchase(id uint) (purchase purchase.Purchase, err error) {
	err = global.GVA_DB.Where("id = ?", id).Preload("Supplier").First(&purchase).Error
	return
}

//@function: GetPurchaseList
//@description: 分页获取采购单列表
//@param: info request.PageInfo
//@return: list interface{}, total int64, err error

func (slf *PurchaseService) GetPurchaseList(info request.PageInfo) (list interface{}, total int64, err error) {
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
	db := global.GVA_DB.Model(&purchase.Purchase{})
	if info.Keyword != "" {
		db.Distinct("purchases.id").Joins("left join purchase_products on purchase_products.purchase_id = purchases.id").
			Joins("left join Product on Product.Id = purchase_products.product_id").
			Joins("left join supplier on supplier.Id = purchases.supplier_id").
			Where("purchases.name like ?", "%"+info.Keyword+"%").
			Or("Product.name like ?", "%"+info.Keyword+"%").
			Or("supplier.name like ?", "%"+info.Keyword+"%")
	}
	var purchaseList []purchase.Purchase
	err = db.Count(&total).Error
	if err != nil {
		return purchaseList, total, err
	}
	err = db.Limit(limit).Offset(offset).Find(&purchaseList).Error
	return purchaseList, total, err
}

//@function: GetPurchaseProductList
//@description: 分页获取采购单产品列表
//@param: purchaseId int
//@return: list interface{},  err error

func (slf *PurchaseService) GetPurchaseProductList(purchaseId uint) (list []*purchase.PurchaseProducts, err error) {
	db := global.GVA_DB.Model(&purchase.PurchaseProducts{})
	list = make([]*purchase.PurchaseProducts, 0)
	err = db.Where("purchase_id = ?", purchaseId).Find(&list).Error
	return list, err
}

//@function: Submit
//@description: 提交采购单
//@param: id uint
//@return: err error

func (slf *PurchaseService) Submit(id uint) (err error) {

	purchaseData, err := slf.GetPurchase(id)
	if err != nil {
		return err
	}
	var targetStatus purchase.OrderStatus
	switch purchaseData.Status {
	case purchase.OrderStatusConfirmed:
		targetStatus = purchase.OrderStatusReceived
	case purchase.OrderStatusReceived:
		targetStatus = purchase.OrderStatusStored
	case purchase.OrderStatusStored:
		targetStatus = purchase.OrderStatusCompleted
	}
	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		err = tx.Where("id = ?", id).Model(&purchase.Purchase{}).Updates(map[string]interface{}{"status": targetStatus}).Error
		if err != nil {
			return err
		}

		switch targetStatus {
		case purchase.OrderStatusReceived:
			return SendInspect(purchaseData)
		case purchase.OrderStatusStored:
		case purchase.OrderStatusCompleted:
		}
		return nil
	})
	return err
}

func SendInspect(record purchase.Purchase) error {
	productList, err := NewPurchaseService().GetPurchaseProductList(record.ID)
	if err != nil {
		return err
	}
	productIds := make([]uint, 0, len(productList))
	for _, product := range productList {
		productIds = append(productIds, uint(product.ProductId))
	}
	productService := &test.ProductService{}
	_, productMap, err := productService.GetProducts(productIds)
	if err != nil {
		return err
	}
	inspectOrders := make([]*qualityinspect.QualityInspect, 0, len(productList))
	for _, productItem := range productList {
		product := productMap[uint(productItem.ProductId)]
		if product == nil {
			continue
		}
		inspectOrder := &qualityinspect.QualityInspect{
			InspectType:   qualityinspect.InspectType_InspectTypePurchase,
			MaterialType:  qualityinspect.MaterialType_MaterialTypeRaw,
			MaterialName:  product.Name,
			MaterialId:    product.Number,
			MaterialTp:    product.ModelNumber,
			MaterialUnit:  product.Unit,
			Supplier:      record.Supplier.Name,
			WarehouseName: "采购总仓",
			ReportAmount:  productItem.Amount.InexactFloat64(),
			InspectMethod: qualityinspect.InspectMethod_InspectMethodAll,
			InspectAmount: productItem.Amount.InexactFloat64(),
		}
		inspectOrders = append(inspectOrders, inspectOrder)
	}
	inspectRequest := qualityinspect.SendPurchaseInspectRequest{List: inspectOrders}
	_, err = qualityinspect.NewQualityInspectServiceClient(qualityinspect.Conn).SendPurchaseInspect(context.Background(), &inspectRequest)
	return err
}

func (slf *PurchaseService) SavePurchaseType(list []*purchase.PurchaseType) (err error) {
	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		err = tx.Where("1 = ?", 1).Delete(&purchase.PurchaseType{}).Error
		if err != nil {
			return err
		}
		for _, item := range list {
			item.ID = 0
		}
		return tx.Create(list).Error
	})
	return err
}

func (slf *PurchaseService) GetPurchaseTypeList() (list []*purchase.PurchaseType, err error) {
	db := global.GVA_DB.Model(&purchase.PurchaseType{})
	list = make([]*purchase.PurchaseType, 0)
	err = db.Order("pin desc, sort desc, id asc").Find(&list).Error
	return list, err
}