package purchase

import (
	"context"
	"errors"
	"github.com/shopspring/decimal"
	"github.com/spf13/cast"
	"gorm.io/gorm"
	"srm/global"
	"srm/model/purchase"
	purchaserequest "srm/model/purchase/request"
	"srm/proto/qualityinspect"
	"srm/service/test"
	"srm/utils"
	"time"
)

type PurchaseService struct{}

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

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

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

	return err
}

func DealPrice(record *purchase.Purchase, productList []*purchase.PurchaseProducts) error {
	var quantity decimal.Decimal
	var totalPrice decimal.Decimal
	var realTotalPrice decimal.Decimal
	for _, product := range productList {
		quantity = quantity.Add(product.Amount)
		totalPrice = totalPrice.Add(product.Price.Mul(product.Amount))
	}
	record.Quantity = quantity
	if !totalPrice.Equal(record.TotalPrice) {
		return errors.New("价税总计计算错误")
	}
	realTotalPrice = record.CalcRealTotalPrice()
	if !realTotalPrice.Equal(record.RealTotalPrice) {
		return errors.New("最终价格计算错误")
	}
	record.UnInvoiceAmount = record.RealTotalPrice
	record.ShouldPayAmount = record.RealTotalPrice
	return nil
}

//@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 = DealPrice(params, productList)
	if err != nil {
		return err
	}
	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
}

func (slf *PurchaseService) GetPurchaseByNumber(number string) (purchase purchase.Purchase, err error) {
	err = global.GVA_DB.Where("number = ?", number).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 purchaserequest.PurchaseSearch) (list interface{}, total int64, err error) {
	limit := info.PageSize
	offset := info.PageSize * (info.Page - 1)
	db := global.GVA_DB.Model(&purchase.Purchase{})
	var ids []uint
	var purchaseList = make([]*purchase.Purchase, 0)
	if info.Keyword != "" {
		kw := "%" + info.Keyword + "%"
		db = db.Select("srm_purchase.id").Joins("left join srm_purchase_products on srm_purchase_products.purchase_id = srm_purchase.id").
			Joins("left join srm_supplier_material on srm_supplier_material.supplier_id = srm_purchase.id").
			Joins("left join srm_supplier on srm_supplier.Id = srm_purchase.supplier_id").
			Where("srm_purchase.name like ? or srm_purchase.source_order like ? or srm_purchase.number like ?", kw, kw, kw).
			Or("srm_supplier_material.name like ?", kw).
			Or("srm_supplier.name like ?", kw)
		err = db.Find(&ids).Error
		if err != nil {
			return purchaseList, total, err
		}
	} else if info.SupplierId != 0 {
		db = db.Where("supplier_id = ?", info.SupplierId)
	}
	if len(ids) != 0 {
		db = global.GVA_DB.Model(&purchase.Purchase{}).Where("id in (?)", ids)
		err = db.Count(&total).Error
		if err != nil {
			return purchaseList, total, err
		}
		err = db.Preload("Supplier").Order("updated_at desc").Limit(limit).Offset(offset).Find(&purchaseList).Error
	} else {
		err = db.Count(&total).Error
		if err != nil {
			return purchaseList, total, err
		}
		err = db.Limit(limit).Offset(offset).Preload("Supplier").Order("updated_at desc").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).Preload("Product").Find(&list).Error
	return list, err
}

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

func (slf *PurchaseService) Submit(id int, status purchase.OrderStatus, warehouse string) (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 {
		m := make(map[string]interface{})
		m["status"] = status
		if warehouse != "" {
			m["warehouse"] = warehouse
		}
		err = tx.Where("id = ?", id).Model(&purchase.Purchase{}).Updates(m).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, product.ID)
	}
	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[productItem.ID]
		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(),
			PurchaseOrderId: record.Number,
		}
		inspectOrders = append(inspectOrders, inspectOrder)
	}
	if len(inspectOrders) == 0 {
		return nil
	}
	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) {
	ids := make([]uint, 0)
	for _, item := range list {
		if item.ID != 0 {
			ids = append(ids, item.ID)
			item.ID = 0
		}
	}
	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		err = tx.Where("id in (?)", ids).Delete(&purchase.PurchaseType{}).Error
		if err != nil {
			return err
		}

		err = tx.Create(list).Error
		if err != nil {
			return err
		}
		return nil
	})
	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
}

func (slf *PurchaseService) MaxAutoIncr() (int, error) {
	var total int64
	err := global.GVA_DB.Model(&purchase.Purchase{}).Count(&total).Error
	return int(total), err
}

func (slf *PurchaseService) SavePurchaseProductConfirm(list []*purchase.PurchaseProductConfirm, first bool) (err error) {
	if len(list) == 0 {
		return errors.New("产品列表不能为空")
	}
	purchaseNumber := list[0].PurchaseNumber
	var pur purchase.Purchase
	err = global.GVA_DB.Where("number = ?", purchaseNumber).First(&pur).Error
	if err != nil {
		return err
	}

	m := make(map[string]interface{})
	if first {
		m["status"] = purchase.OrderStatusWaitReceive
	} else {
		yu := int64(0)
		for _, confirm := range list {
			confirm.OverReceiveAmount = confirm.OverReceiveAmount.Add(confirm.NowReceiveAmount)
			confirm.NotReceiveAmount = confirm.Amount.Sub(confirm.OverReceiveAmount)
			if !confirm.SurplusReceiveAmount.IsZero() {
				yu = confirm.SurplusReceiveAmount.IntPart()
			}
		}
		if yu > 0 {
			m["status"] = purchase.OrderStatusPartReceive
		}
		if yu == 0 {
			m["status"] = purchase.OrderStatusWaitQuality
		}
	}

	err = global.GVA_DB.Transaction(func(tx *gorm.DB) error {
		err = tx.Where("purchase_number = ?", purchaseNumber).Delete(&purchase.PurchaseProductConfirm{}).Error
		if err != nil {
			return err
		}

		err = tx.Create(list).Error
		if err != nil {
			return err
		}
		err = tx.Where("number = ?", purchaseNumber).Model(&purchase.Purchase{}).Updates(m).Error
		if err != nil {
			return err
		}

		return nil
	})
	return err
}

func (slf *PurchaseService) GetPurchaseProductConfirmInfo(number string) ([]*purchase.PurchaseProductConfirm, error) {
	var list []*purchase.PurchaseProductConfirm
	err := global.GVA_DB.Model(&purchase.PurchaseProductConfirm{}).Where("purchase_number = ?", number).Find(&list).Error
	return list, err
}

func (slf *PurchaseService) SavePurchaseQualityInspection(list []*purchase.PurchaseProductConfirm) ([]*purchase.PurchaseQualityInspection, error) {
	if len(list) == 0 {
		return nil, errors.New("产品列表不能为空")
	}
	purchaseNumber := list[0].PurchaseNumber
	qualityList := make([]*purchase.PurchaseQualityInspection, 0)
	for _, confirm := range list {
		if confirm.NowReceiveAmount.IsZero() {
			continue
		}
		var pqi purchase.PurchaseQualityInspection
		pqi.PurchaseNumber = purchaseNumber
		pqi.Principal = confirm.Principal
		pqi.ProductId = confirm.ProductId
		pqi.ProductName = confirm.ProductName
		pqi.Amount = confirm.NowReceiveAmount
		pqi.Status = purchase.WaitInspection
		pqi.CreateTime = utils.TimeToString(time.Now())
		pqi.Unit = confirm.Unit
		pqi.Specs = confirm.Specs
		pqi.Type = confirm.Type
		qualityList = append(qualityList, &pqi)
	}
	err := global.GVA_DB.Model(purchase.PurchaseQualityInspection{}).Create(qualityList).Error
	return qualityList, err
}

func (slf *PurchaseService) GetPurchaseQualityInspection(params purchaserequest.GetQualityInspectionInfo) ([]*purchase.PurchaseQualityInspection, error) {
	var list []*purchase.PurchaseQualityInspection
	db := global.GVA_DB.Model(&purchase.PurchaseQualityInspection{}).Where("purchase_number = ?", params.PurchaseNumber)
	if len(params.Status) > 0 {
		db = db.Where("status in (?)", params.Status)
	}
	if len(params.Times) > 0 {
		db = db.Where("create_time in (?)", params.Times)
	}
	err := db.Find(&list).Error
	return list, err
}

func (slf *PurchaseService) UpdatePurchaseQualityInspection(ids []int, status int) error {
	err := global.GVA_DB.Model(&purchase.PurchaseQualityInspection{}).Where("id in (?)", ids).Updates(map[string]interface{}{"status": status}).Error
	return err
}

func (slf *PurchaseService) GetPurchaseQualityInspectionList(ids []int) ([]*purchase.PurchaseQualityInspection, error) {
	var list []*purchase.PurchaseQualityInspection
	err := global.GVA_DB.Model(&purchase.PurchaseQualityInspection{}).Where("id in (?)", ids).Find(&list).Error
	return list, err
}

func (slf *PurchaseService) UpdatePurchaseStatus(number string, status int) error {
	var total int64
	err := global.GVA_DB.Model(&purchase.Purchase{}).Where("number = ?", number).Where("status = ?", purchase.OrderStatusWaitQuality).Count(&total).Error
	if err != nil {
		return err
	}
	if total > 0 {
		err = global.GVA_DB.Model(&purchase.PurchaseQualityInspection{}).Where("purchase_number = ?", number).Where("status = ?", status).Count(&total).Error
		if err != nil {
			return err
		}

		if total == 0 {
			err = global.GVA_DB.Model(&purchase.Purchase{}).Where("number = ?", number).Updates(map[string]interface{}{"status": purchase.OrderStatusReceived}).Error
			if err != nil {
				return err
			}
		}
	}
	return nil
}