package service

import (
	"aps_crm/constvar"
	"aps_crm/model"
	"aps_crm/model/request"
	"aps_crm/pkg/ecode"
	"github.com/shopspring/decimal"
	"gorm.io/gorm"
)

type InvoiceService struct{}

func NewInvoiceService() InvoiceService {
	return InvoiceService{}
}

func (InvoiceService) AddInvoice(invoice *model.Invoice) int {

	if invoice.SourceType == constvar.InvoiceSourceTypeServiceContract { //更新服务合同已开票金额
		serviceContract, err := model.NewServiceContractSearch().SetId(invoice.SourceId).SetPreload(true).First()
		if err != nil {
			return ecode.DBErr
		}
		var amountInvoiced, amountNotInvoiced decimal.Decimal
		rightProducts := NewProductsService().PickRightProducts(invoice.Products, serviceContract.Products)
		for _, product := range rightProducts {
			amountInvoiced = serviceContract.AmountInvoiced.Add(product.Amount.Mul(product.Price))
			amountNotInvoiced = serviceContract.AmountUnInvoiced.Sub(product.Amount.Mul(product.Price))
		}
		amountInvoiced = amountInvoiced.Round(2)
		amountNotInvoiced = amountNotInvoiced.Round(2)
		err = model.WithTransaction(func(db *gorm.DB) error {
			err = model.NewInvoiceSearch().Create(invoice)
			if err != nil {
				return err
			}
			return model.NewServiceContractSearch().SetId(invoice.SourceId).UpdateByMap(map[string]interface{}{
				"amount_invoiced":     amountInvoiced,
				"amount_not_invoiced": amountNotInvoiced,
			})
		})
		if err != nil {
			return ecode.DBErr
		}
	} else if invoice.SourceType == constvar.InvoiceSourceTypeSaleDetail { //更新销售明细已开票金额
		salesDetails, err := model.NewSalesDetailsSearch().SetId(invoice.SourceId).SetPreload(true).First()
		if err != nil {
			return ecode.DBErr
		}
		var amountInvoiced, amountNotInvoiced decimal.Decimal
		rightProducts := NewProductsService().PickRightProducts(invoice.Products, salesDetails.Products)
		for _, product := range rightProducts {
			amountInvoiced = salesDetails.AmountInvoiced.Add(product.Amount.Mul(product.Price))
			amountNotInvoiced = salesDetails.AmountUnInvoiced.Sub(product.Amount.Mul(product.Price))
		}
		amountInvoiced = amountInvoiced.Round(2)
		amountNotInvoiced = amountNotInvoiced.Round(2)
		err = model.WithTransaction(func(db *gorm.DB) error {
			err = model.NewInvoiceSearch().Create(invoice)
			if err != nil {
				return err
			}
			return model.NewSalesDetailsSearch().SetId(invoice.SourceId).UpdateByMap(map[string]interface{}{
				"amount_invoiced":     amountInvoiced,
				"amount_not_invoiced": amountNotInvoiced,
			})
		})
		if err != nil {
			return ecode.DBErr
		}
	}

	return ecode.OK
}

func (InvoiceService) DeleteInvoice(id int) int {
	invoice, err := model.NewInvoiceSearch().SetId(id).First()
	if err != nil {
		return ecode.DBErr
	}
	if invoice.SourceType == constvar.InvoiceSourceTypeServiceContract { //更新服务合同已开票金额
		serviceContract, err := model.NewServiceContractSearch().SetId(invoice.SourceId).SetPreload(true).First()
		if err != nil {
			return ecode.DBErr
		}
		var amountInvoiced, amountNotInvoiced decimal.Decimal
		rightProducts := NewProductsService().PickRightProducts(invoice.Products, serviceContract.Products)
		for _, product := range rightProducts {
			amountInvoiced = serviceContract.AmountInvoiced.Sub(product.Amount.Mul(product.Price))
			amountNotInvoiced = serviceContract.AmountUnInvoiced.Add(product.Amount.Mul(product.Price))
		}
		amountInvoiced = amountInvoiced.Round(2)
		amountNotInvoiced = amountNotInvoiced.Round(2)
		err = model.WithTransaction(func(db *gorm.DB) error {
			err := model.NewInvoiceSearch().SetId(id).Delete()
			if err != nil {
				return err
			}
			return model.NewServiceContractSearch().SetId(invoice.SourceId).UpdateByMap(map[string]interface{}{
				"amount_invoiced":     amountInvoiced,
				"amount_not_invoiced": amountNotInvoiced,
			})
		})
		if err != nil {
			return ecode.DBErr
		}
	} else if invoice.SourceType == constvar.InvoiceSourceTypeServiceContract { //更新销售明细已开票金额
		salesDetails, err := model.NewSalesDetailsSearch().SetId(invoice.SourceId).SetPreload(true).First()
		if err != nil {
			return ecode.DBErr
		}
		var amountInvoiced, amountNotInvoiced decimal.Decimal
		rightProducts := NewProductsService().PickRightProducts(invoice.Products, salesDetails.Products)
		for _, product := range rightProducts {
			amountInvoiced = salesDetails.AmountInvoiced.Sub(product.Amount.Mul(product.Price))
			amountNotInvoiced = salesDetails.AmountUnInvoiced.Add(product.Amount.Mul(product.Price))
		}
		amountInvoiced = amountInvoiced.Round(2)
		amountNotInvoiced = amountNotInvoiced.Round(2)
		err = model.WithTransaction(func(db *gorm.DB) error {
			err := model.NewInvoiceSearch().SetId(id).Delete()
			if err != nil {
				return err
			}
			return model.NewSalesDetailsSearch().SetId(invoice.SourceId).UpdateByMap(map[string]interface{}{
				"amount_invoiced":     amountInvoiced,
				"amount_not_invoiced": amountNotInvoiced,
			})
		})
		if err != nil {
			return ecode.DBErr
		}
	}

	if err != nil {
		return ecode.DBErr
	}
	return ecode.OK
}

func (InvoiceService) GetInvoiceList(params request.GetInvoiceList) ([]*model.Invoice, int64, int) {
	list, total, err := model.NewInvoiceSearch().SetSourceId(params.SourceId).SetSourceType(params.SourceType).SetPrincipalIds(params.PrincipalIds).Find()
	if err != nil {
		return nil, 0, ecode.DBErr
	}

	return list, total, ecode.OK
}

func (InvoiceService) UpdateInvoices(Invoices []*request.UpdateInvoice) int {
	for _, v := range Invoices {
		// check Invoice exist
		_, err := model.NewInvoiceSearch().SetId(v.Id).First()
		if err != nil {
			return ecode.DBErr
		}

		err = model.NewInvoiceSearch().SetId(v.Id).Updates(map[string]interface{}{})
		if err != nil {
			return ecode.DBErr
		}
	}

	return ecode.OK
}

func (InvoiceService) UpdateInvoice(invoice *model.Invoice) int {
	if invoice.SourceType == constvar.InvoiceSourceTypeServiceContract {
		serviceContract, err := model.NewServiceContractSearch().SetId(invoice.SourceId).SetPreload(true).First()
		if err != nil {
			return ecode.DBErr
		}
		var amountInvoiced, amountNotInvoiced decimal.Decimal
		newProducts, removedProducts := NewProductsService().PickDiffProducts(invoice.Products, serviceContract.Products)
		for _, product := range newProducts {
			amountInvoiced = serviceContract.AmountInvoiced.Add(product.Amount.Mul(product.Price))
			amountNotInvoiced = serviceContract.AmountUnInvoiced.Sub(product.Amount.Mul(product.Price))
		}
		removedProductIds := make([]uint, 0, len(removedProducts))
		for _, product := range removedProducts {
			amountInvoiced = serviceContract.AmountInvoiced.Sub(product.Amount.Mul(product.Price))
			amountNotInvoiced = serviceContract.AmountUnInvoiced.Add(product.Amount.Mul(product.Price))
			removedProductIds = append(removedProductIds, product.Id)
		}
		amountInvoiced = amountInvoiced.Round(2)
		amountNotInvoiced = amountNotInvoiced.Round(2)
		err = model.WithTransaction(func(db *gorm.DB) error {
			err = model.NewInvoiceSearch().SetId(invoice.Id).Update(invoice)
			if err != nil {
				return err
			}
			if len(removedProductIds) > 0 {
				err = model.NewInvoiceProductSearch().SetInvoiceId(invoice.Id).SetProductIds(removedProductIds).Delete()
				if err != nil {
					return err
				}
			}
			return model.NewServiceContractSearch().SetId(invoice.SourceId).UpdateByMap(map[string]interface{}{
				"amount_invoiced":     amountInvoiced,
				"amount_not_invoiced": amountNotInvoiced,
			})
		})
		if err != nil {
			return ecode.DBErr
		}
	} else if invoice.SourceType == constvar.InvoiceSourceTypeServiceContract { //更新销售明细已开票金额
		salesDetails, err := model.NewSalesDetailsSearch().SetId(invoice.SourceId).SetPreload(true).First()
		if err != nil {
			return ecode.DBErr
		}
		var amountInvoiced, amountNotInvoiced decimal.Decimal
		newProducts, removedProducts := NewProductsService().PickDiffProducts(invoice.Products, salesDetails.Products)
		for _, product := range newProducts {
			amountInvoiced = salesDetails.AmountInvoiced.Add(product.Amount.Mul(product.Price))
			amountNotInvoiced = salesDetails.AmountUnInvoiced.Sub(product.Amount.Mul(product.Price))
		}
		removedProductIds := make([]uint, 0, len(removedProducts))
		for _, product := range removedProducts {
			amountInvoiced = salesDetails.AmountInvoiced.Sub(product.Amount.Mul(product.Price))
			amountNotInvoiced = salesDetails.AmountUnInvoiced.Add(product.Amount.Mul(product.Price))
			removedProductIds = append(removedProductIds, product.Id)
		}
		amountInvoiced = amountInvoiced.Round(2)
		err = model.WithTransaction(func(db *gorm.DB) error {
			err = model.NewInvoiceSearch().SetId(invoice.Id).Update(invoice)
			if err != nil {
				return err
			}
			if len(removedProductIds) > 0 {
				err = model.NewInvoiceProductSearch().SetInvoiceId(invoice.Id).SetProductIds(removedProductIds).Delete()
				if err != nil {
					return err
				}
			}
			return model.NewSalesDetailsSearch().SetId(invoice.SourceId).UpdateByMap(map[string]interface{}{
				"amount_invoiced":     amountInvoiced,
				"amount_not_invoiced": amountNotInvoiced,
			})
		})
		if err != nil {
			return ecode.DBErr
		}
	}
	return ecode.OK
}