package controllers

import (
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/shopspring/decimal"
	"gorm.io/gorm"
	"sort"
	"strconv"
	"time"
	"wms/constvar"
	"wms/extend/code"
	"wms/extend/util"
	"wms/models"
	"wms/pkg/logx"
	"wms/pkg/structx"
	"wms/request"
)

type OperationController struct {
}

// Add
// @Tags      入库/出库
// @Summary   添加入库/出库
// @Produce   application/json
// @Param     object  body  request.AddOperation true  "入库/出库信息"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/operation/operation [post]
func (slf OperationController) Add(c *gin.Context) {
	var reqParams request.AddOperation
	var params models.Operation
	if err := c.BindJSON(&reqParams); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	if err := structx.AssignTo(reqParams, &params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "数据转换错误"+err.Error())
		return
	}
	if err := slf.CheckParams(params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}
	if err := slf.FormatLocation(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}

	if CheckDetailsRepeat(params.Details) {
		util.ResponseFormat(c, code.RequestParamError, "明细中不能存在重复的产品")
		return
	}

	operationType, err := models.NewOperationTypeSearch().SetID(uint(params.OperationTypeId)).First()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}

	params.Status = constvar.OperationStatus_Ready
	params.Number = strconv.FormatInt(time.Now().Unix(), 10)
	params.BaseOperationType = operationType.BaseOperationType
	if err := models.NewOperationSearch().Create(&params); err != nil {
		logx.Errorf("Operation create err: %v", err)
		util.ResponseFormat(c, code.SaveFail, "添加失败："+err.Error())
		return
	}
	util.ResponseFormat(c, code.Success, "添加成功")
}

func CheckDetailsRepeat(details []*models.OperationDetails) bool {
	detailIDs := []string{}
	var tempID string
	for _, v := range details {
		detailIDs = append(detailIDs, v.ProductId)
	}
	sort.Strings(detailIDs)
	for _, v := range detailIDs {
		if v != tempID {
			tempID = v
		} else {
			return true
		}
	}
	return false
}

func (slf OperationController) FormatLocation(params *models.Operation) error {
	operationType, err := models.NewOperationTypeSearch().SetID(uint(params.OperationTypeId)).First()
	if err != nil {
		return err
	}
	if operationType.BaseOperationType == constvar.BaseOperationTypeIncoming {
		if location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeVendor)).First(); err != nil {
			return err
		} else {
			params.FromLocationID = location.Id
		}
		if params.ToLocationID == 0 {
			return errors.New("请选择目标位置")
		}
	}
	if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing {
		if location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeCustomer)).First(); err != nil {
			return err
		} else {
			params.ToLocationID = location.Id
		}
		if params.FromLocationID == 0 {
			return errors.New("请选择源位置")
		}
	}
	if operationType.BaseOperationType == constvar.BaseOperationTypeInternal {
		if params.ToLocationID == 0 {
			return errors.New("请选择目标位置")
		}
		if params.FromLocationID == 0 {
			return errors.New("请选择源位置")
		}
	}
	return nil
}

func (slf OperationController) CheckParams(params models.Operation) error {
	if params.SourceNumber == "" {
		return errors.New("请填入源单号")
	}

	if params.OperationTypeId == 0 {
		return errors.New("operationTypeId为0")
	}

	if params.OperationDate == "" {
		return errors.New("请选择安排日期")
	}

	if len(params.Details) <= 0 {
		return errors.New("请添加明细信息")
	}

	//检查明细部分
	for _, v := range params.Details {
		if v.ProductId == "" {
			return errors.New("productID为空")
		}
		//if v.ProductName == "" {
		//	return errors.New("产品名称异常")
		//}
		if v.Amount.IsNegative() {
			return errors.New("产品数量出错")
		}
	}

	return nil
}

// List
// @Tags      入库/出库
// @Summary   入库/出库列表
// @Produce   application/json
// @Param		 object  body  request.OperationList true  "查询参数"
// @Success   200 {object} util.Response	"成功"
// @Router    /api-wms/v1/operation/list [post]
func (slf OperationController) List(c *gin.Context) {
	var params request.OperationList
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误:"+err.Error())
		return
	}
	if err := slf.CheckListParams(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}
	search := models.NewOperationSearch()
	search.SetPage(params.Page, params.PageSize)
	if params.Number != "" {
		search.SetKeyword(params.Number)
	}
	list, total, err := search.SetOperationTypeId(params.OperationTypeId).SetPreload(true).SetOrder("created_at desc").Find()
	if err != nil {
		util.ResponseFormat(c, code.RequestError, "查找失败:"+err.Error())
		return
	}

	util.ResponseFormatListWithPage(c, code.Success, list, int(total), params.Page, params.PageSize)
}

func (slf OperationController) CheckListParams(params *request.OperationList) error {
	if !params.PageInfo.Check() {
		return errors.New("数据分页信息错误")
	}
	if params.OperationTypeId == 0 {
		return errors.New("operationTypeId为0")
	}
	return nil
}

// Update
// @Tags      入库/出库
// @Summary   修改入库/出库信息
// @Produce   application/json
// @Param     object  body request.UpdateOperation true  "入库信息"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/operation/update [post]
func (slf OperationController) Update(c *gin.Context) {
	var reqParams request.UpdateOperation
	var params models.Operation
	if err := c.BindJSON(&reqParams); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误："+err.Error())
		return
	}
	if reqParams.Status != constvar.OperationStatus_Ready {
		util.ResponseFormat(c, code.RequestParamError, "该入库信息已完成，无法进行修改")
		return
	}
	if err := structx.AssignTo(reqParams, &params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "数据转换错误"+err.Error())
		return
	}
	if err := slf.CheckParams(params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}
	if CheckDetailsRepeat(params.Details) {
		util.ResponseFormat(c, code.RequestParamError, "明细中不能存在重复的产品")
		return
	}
	if err := slf.FormatLocation(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}
	if err := models.WithTransaction(func(tx *gorm.DB) error {
		if err := models.NewOperationDetailsSearch().SetOrm(tx).SetOperationId(params.Id).Delete(); err != nil {
			return err
		}
		operationSearch := models.NewOperationSearch().SetOrm(tx)
		if err := operationSearch.Orm.Model(&params).Association("Details").Replace(params.Details); err != nil {
			return err
		}
		if err := models.NewOperationSearch().SetOrm(tx).SetID(params.Id).Save(&params); err != nil {
			return err
		}
		return nil
	}); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "修改失败："+err.Error())
		return
	}

	util.ResponseFormat(c, code.Success, "修改成功")
}

// DeleteDevice
//
//	@Tags		入库/出库
//	@Summary	删除入库/出库信息
//	@Produce	application/json
//	@Param		id	path		int			true	"id"
//	@Success	200	{object}	util.Response	"成功"
//	@Router		/api-wms/v1/operation/operation/{id} [delete]
func (slf OperationController) Delete(c *gin.Context) {
	id, err := strconv.Atoi(c.Param("id"))
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "错误的id值")
		return
	}
	if id == 0 {
		util.ResponseFormat(c, code.RequestParamError, "id为0")
		return
	}
	operation, err := models.NewOperationSearch().SetID(id).First()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "未找到相关出入库信息:"+err.Error())
		return
	}
	if operation.Status != constvar.OperationStatus_Ready {
		util.ResponseFormat(c, code.RequestError, "该入库信息无法进行删除")
		return
	}
	if err := models.WithTransaction(func(tx *gorm.DB) error {
		if err := models.NewOperationDetailsSearch().SetOrm(tx).SetOperationId(id).Delete(); err != nil {
			return err
		}
		if err := models.NewOperationSearch().SetOrm(tx).SetID(id).Delete(); err != nil {
			return err
		}
		return nil
	}); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "修改失败："+err.Error())
		return
	}
	util.ResponseFormat(c, code.Success, "删除成功")
}

// DeleteDevice
//
//	@Tags		入库/出库
//	@Summary	更改记录状态
//	@Produce	application/json
//	@Param		id	path		int			true	"id"
//	@Success	200	{object}	util.Response	"成功"
//	@Router		/api-wms/v1/operation/finish/{id} [put]
func (slf OperationController) Finish(c *gin.Context) {
	id, err := strconv.Atoi(c.Param("id"))
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "错误的id值")
		return
	}
	if id == 0 {
		util.ResponseFormat(c, code.RequestParamError, "id为0")
		return
	}
	operation, err := models.NewOperationSearch().SetID(id).First()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "未找到相关出入库信息:"+err.Error())
		return
	}
	if operation.Status != constvar.OperationStatus_Ready {
		util.ResponseFormat(c, code.RequestError, "该出入库信息无法完成")
		return
	}
	operationType, err := models.NewOperationTypeSearch().SetID(uint(operation.OperationTypeId)).First()
	if err != nil {
		util.ResponseFormat(c, code.RequestError, err.Error())
		return
	}
	if err := models.WithTransaction(func(tx *gorm.DB) error {

		if err := models.NewOperationSearch().SetOrm(tx).SetID(id).Update(&models.Operation{Status: constvar.OperationStatus_Finish}); err != nil {
			return err
		}
		//if operationType.BaseOperationType != constvar.BaseOperationTypeInternal {
		var listProdtId []string
		var listProdt []*models.Material
		mapProdt := make(map[string]decimal.Decimal)
		listDetails, err := models.NewOperationDetailsSearch().SetOperationId(operation.Id).FindAll()
		if err != nil {
			return err
		}
		for _, v := range listDetails {
			listProdtId = append(listProdtId, v.ProductId)
			mapProdt[v.ProductId] = v.Amount
		}
		if err := models.NewMaterialSearch().Orm.Where("id IN ?", listProdtId).Find(&listProdt).Error; err != nil {
			return err
		}
		if operationType.BaseOperationType == constvar.BaseOperationTypeIncoming {
			for k, v := range listProdt {
				if value, ok := mapProdt[v.ID]; !ok {
					return errors.New("产品种类异常")
				} else {
					listProdt[k].Amount = listProdt[k].Amount.Add(value)
					if err := tx.Save(listProdt[k]).Error; err != nil {
						return err
					}
					//TODO:出入库的finish和报废的finish都要增加对location_product_amount表数量的更新,因为此表有ProductCategory字段，所以operation_details表中要增加ProductCategoryId字段
					var locAmount models.LocationProductAmount
					if err := models.NewLocationProductAmountSearch().Orm.
						Table("wms_location_product_amount").
						Joins("inner join wms_location_product on wms_location_product.id=wms_location_product_amount.location_product_id").
						Where("wms_location_product.product_id=? and wms_location_product.location_id=?", v.ID, operation.ToLocationID).
						First(&locAmount).Error; err != nil {
						return err
					}
					locAmount.Amount = locAmount.Amount.Add(value)
					if err := models.NewLocationProductAmountSearch().SetID(locAmount.Id).Update(&locAmount); err != nil {
						return err
					}
				}
			}

		}
		if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing {
			for k, v := range listProdt {
				if value, ok := mapProdt[v.ID]; !ok {
					return errors.New("产品种类异常")
				} else {
					if v.Amount.LessThan(value) {
						return errors.New(fmt.Sprintf("产品：%v,库存：%v,出库：%v,数量不够，无法完成出库操作", v.Name, v.Amount.String(), value.String()))
					}
					listProdt[k].Amount = listProdt[k].Amount.Sub(value)
					if err := tx.Save(listProdt[k]).Error; err != nil {
						return err
					}
					var locAmount models.LocationProductAmount
					if err := models.NewLocationProductAmountSearch().Orm.Table("wms_location_product_amount").Joins("inner join wms_location_product on wms_location_product.id=wms_location_product_amount.location_product_id").Where("wms_location_product.product_id=? and wms_location_product.location_id=?", v.ID, operation.FromLocationID).First(&locAmount).Error; err != nil {
						return err
					}
					if locAmount.Amount.LessThan(value) {
						return errors.New(fmt.Sprintf("产品：%v,库存：%v,出库：%v,数量不够，无法完成出库操作", v.Name, v.Amount.String(), value.String()))
					}
					locAmount.Amount = locAmount.Amount.Sub(value)
					if err := models.NewLocationProductAmountSearch().SetID(locAmount.Id).Update(&locAmount); err != nil {
						return err
					}
				}
			}
		}
		if operationType.BaseOperationType == constvar.BaseOperationTypeInternal {
			for k, v := range listProdt {
				if value, ok := mapProdt[v.ID]; !ok {
					return errors.New("产品种类异常")
				} else {
					listProdt[k].Amount = listProdt[k].Amount.Add(value)
					if err := tx.Save(listProdt[k]).Error; err != nil {
						return err
					}
					//TODO:出入库的finish和报废的finish都要增加对location_product_amount表数量的更新,因为此表有ProductCategory字段，所以operation_details表中要增加ProductCategoryId字段
					var fromAmount, toAmount models.LocationProductAmount
					if err := models.NewLocationProductAmountSearch().Orm.Table("wms_location_product_amount").Joins("inner join wms_location_product on wms_location_product.id=wms_location_product_amount.location_product_id").Where("wms_location_product.product_id=? and wms_location_product.location_id=?", v.ID, operation.FromLocationID).Find(&fromAmount).Error; err != nil {
						return err
					}
					if fromAmount.Amount.LessThan(value) {
						return errors.New(fmt.Sprintf("产品：%v,库存：%v,调拨：%v,数量不够，无法完成调拨操作", v.Name, v.Amount.String(), value.String()))
					}
					fromAmount.Amount = fromAmount.Amount.Sub(value)
					if err := models.NewLocationProductAmountSearch().SetID(fromAmount.Id).Update(&fromAmount); err != nil {
						return err
					}

					if err := models.NewLocationProductAmountSearch().Orm.Table("wms_location_product_amount").Joins("inner join wms_location_product on wms_location_product.id=wms_location_product_amount.location_product_id").Where("wms_location_product.product_id=? and wms_location_product.location_id=?", v.ID, operation.ToLocationID).Find(&toAmount).Error; err != nil {
						return err
					}
					toAmount.Amount = toAmount.Amount.Add(value)
					if err := models.NewLocationProductAmountSearch().SetID(toAmount.Id).Update(&toAmount); err != nil {
						return err
					}
				}
			}

		}

		//}
		return nil
	}); err != nil {
		util.ResponseFormat(c, code.RequestError, err.Error())
		return
	}
	util.ResponseFormat(c, code.Success, "操作成功")
}

// ListAll
// @Tags      入库/出库
// @Summary   调拨
// @Produce   application/json
// @Param     object  body  request.OperationAllList true  "参数"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/operation/listAll [post]
func (slf OperationController) ListAll(c *gin.Context) {
	var params request.OperationAllList
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误:"+err.Error())
		return
	}
	if !params.PageInfo.Check() {
		util.ResponseFormat(c, code.RequestParamError, "数据分页信息错误")
		return
	}
	search := models.NewOperationSearch()
	search.SetPage(params.Page, params.PageSize)
	search.SetPage(params.Page, params.PageSize)
	if params.Number != "" {
		search.SetKeyword(params.Number)
	}
	list, total, err := search.SetPreload(true).SetOrder("created_at desc").Find()
	if err != nil {
		util.ResponseFormat(c, code.RequestError, "查找失败:"+err.Error())
		return
	}
	util.ResponseFormatListWithPage(c, code.Success, list, int(total), params.Page, params.PageSize)
}