package controllers

import (
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/shopspring/decimal"
	"gorm.io/gorm"
	"strconv"
	"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
	}

	params.Status = constvar.OperationStatus_Ready
	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 (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
// @Accept	  json
// @Param     object  query  request.OperationList true  "参数"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/operation/operation [get]
func (slf OperationController) List(c *gin.Context) {
	var params request.OperationList
	if err := c.ShouldBindQuery(&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)
	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.ResponseFormatList(c, code.Success, list, int(total))

}

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 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.BaseOperationTypeIncoming {
			if err := models.NewMaterialSearch().Orm.Exec("update material INNER JOIN wms_operation_details on wms_operation_details.product_id=material.id INNER JOIN wms_operation on wms_operation.id=wms_operation_details.operation_id set material.amount=material.amount + wms_operation_details.quantity where wms_operation.id=?", id).Error; err != nil {
				return err
			}
		} else if operationType.BaseOperationType == constvar.BaseOperationTypeOutgoing {
			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
			}
			for _, 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()))
					}
				}
			}
			if err := models.NewMaterialSearch().Orm.Exec("update material INNER JOIN wms_operation_details on wms_operation_details.product_id=material.id INNER JOIN wms_operation on wms_operation.id=wms_operation_details.operation_id set material.amount=material.amount - wms_operation_details.quantity where wms_operation.id=?", id).Error; err != nil {
				return err
			}
		}
		return nil
	}); err != nil {
		util.ResponseFormat(c, code.RequestError, err.Error())
		return
	}
	util.ResponseFormat(c, code.Success, "操作成功")
}