package controllers

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

type LocationProductAmountController struct {
}

// List
// @Tags      库存盘点
// @Summary   库存盘点列表
// @Produce   application/json
// @Param		 object  body  request.GetList true  "查询参数"
// @Success   200 {object} util.Response	"成功"
// @Router    /api-wms/v1/locationProductAmount/list [post]
func (slf LocationProductAmountController) List(c *gin.Context) {
	var params request.GetList
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误:"+err.Error())
		return
	}
	if !params.Check() {
		util.ResponseFormat(c, code.RequestParamError, "参数异常")
		return
	}
	productIds := make([]string, 0)
	locationIds := make([]int, 0)
	if params.Keyword != "" {
		materials, _ := models.NewMaterialSearch().SetKeyword(params.Keyword).FindNotTotal()
		for _, material := range materials {
			productIds = append(productIds, material.ID)
		}
		locations, _ := models.NewLocationSearch().SetKeyword(params.Keyword).FindNotTotal()
		for _, location := range locations {
			locationIds = append(locationIds, location.Id)
		}
	}

	search := models.NewLocationProductAmountSearch()
	//search.SetPage(params.Page, params.PageSize)

	//wms_operation_details.amount as adjust_amount,wms_operation.id as operation_id,wms_operation.status
	search.Orm = search.Orm.
		Table("wms_location_product_amount").
		Select(`wms_location_product_amount.id as location_product_amount_id,wms_location.id as location_id,wms_location.joint_name as location_name,
		material.id as product_id,material.name as product_name,wms_location_product_amount.amount,material.unit,
		wms_location_product_amount.create_date,material.weight`).
		InnerJoins("inner join material on material.id=wms_location_product_amount.product_id").
		InnerJoins("inner join wms_location on wms_location.id=wms_location_product_amount.location_id")

	var (
		records = make([]*models.LocationProductAmountWithOperation, 0)
		total   int64
	)
	if len(productIds) > 0 {
		search.Orm = search.Orm.Where("wms_location_product_amount.product_id in (?)", productIds)

	}
	if len(locationIds) > 0 {
		search.Orm = search.Orm.Where("wms_location_product_amount.location_id in (?)", locationIds)
	}

	if err := search.Orm.Count(&total).Error; err != nil {
		util.ResponseFormat(c, code.RequestError, fmt.Errorf("find count err: %v", err))
		return
	}
	//search.Orm = search.Orm.Preload("LocationProduct").Preload("LocationProduct.Location").Preload("LocationProduct.Product")
	if params.Page*params.PageSize > 0 {
		search.Orm = search.Orm.Offset((params.Page - 1) * params.PageSize).Limit(params.PageSize)
	}
	if err := search.Orm.Order("wms_location_product_amount.updated_at desc").Find(&records).Error; err != nil {
		util.ResponseFormat(c, code.RequestError, fmt.Errorf("find count err: %v", err))
		return
	}

	for k, v := range records {
		var operation models.Operation
		db := mysqlx.GetDB()
		if res := db.Table("wms_operation").Preload("Details").InnerJoins("inner join wms_operation_details "+
			"on wms_operation_details.operation_id=wms_operation.id").Where("wms_operation.base_operation_type=? and "+
			"wms_operation.status=? and wms_operation_details.to_location_id=? and wms_operation_details.product_id=?", constvar.BaseOperationTypeAdjust,
			constvar.OperationStatus_Ready, v.LocationId, v.ProductId).First(&operation); res.Error != nil {
			if errors.Is(res.Error, gorm.ErrRecordNotFound) {
				records[k].Status = constvar.OperationStatus_Finish
				records[k].BaseOperationType = constvar.BaseOperationTypeAdjust
				continue
			} else {
				util.ResponseFormat(c, code.RequestError, fmt.Errorf("find operation err: %v", res.Error.Error()))
				return
			}
		} else {
			if len(operation.Details) > 0 {
				records[k].DifferenceAmount = operation.Details[0].Amount
				records[k].AdjustAmount = v.DifferenceAmount.Sub(v.Amount)
			}
			records[k].OperationId = operation.Id
			records[k].Status = operation.Status
		}
		records[k].BaseOperationType = constvar.BaseOperationTypeAdjust
	}
	util.ResponseFormatListWithPage(c, code.Success, records, int(total), params.Page, params.PageSize)
}

// Add
// @Tags      库存盘点
// @Summary   添加库存盘点信息
// @Produce   application/json
// @Param     object  body  request.UpdateLocationProductAmount true  "入库/出库信息"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/locationProductAmount/add [post]
func (slf LocationProductAmountController) Add(c *gin.Context) {
	var reqParams request.UpdateLocationProductAmount
	if err := c.BindJSON(&reqParams); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}

	if err := AddLocationProductAmount(reqParams); err != nil {
		util.ResponseFormat(c, code.RequestError, err)
		return
	}

	util.ResponseFormat(c, code.Success, "添加成功")
}

func AddLocationProductAmount(reqParams request.UpdateLocationProductAmount) error {
	if reqParams.LocationId == 0 {
		return errors.New("参数异常，位置id信息不存在")
	}
	if reqParams.ProductId == "" {
		return errors.New("参数异常，产品id信息不存在")
	}

	var existCount int64
	if err := models.NewOperationSearch().Orm.
		Table("wms_operation").
		InnerJoins("inner join wms_operation_details on wms_operation_details.operation_id=wms_operation.id").
		Where("wms_operation_details.to_location_id=? and wms_operation_details.product_id=? and wms_operation.base_operation_type=? and wms_operation.status=?", reqParams.LocationId, reqParams.ProductId, constvar.BaseOperationTypeAdjust, constvar.OperationStatus_Ready).
		Count(&existCount).Error; err != nil {
		return err
	}
	if existCount > 0 {
		return errors.New("该商品在已选中的位置存在未验证的库存盘点信息，无法继续添加相关信息")
	}

	existAmount, res := models.NewLocationProductAmountSearch().SetLocationId(reqParams.LocationId).SetProductId(reqParams.ProductId).FirstRes()
	if res.Error != nil && !errors.Is(res.Error, gorm.ErrRecordNotFound) {
		return res.Error
	}
	location, err := models.NewLocationSearch().SetType(int(constvar.LocationTypeAdjust)).First()
	if err != nil {
		return err
	}
	var locAmount models.LocationProductAmount
	if res.Error != nil && errors.Is(res.Error, gorm.ErrRecordNotFound) {
		locAmount = models.LocationProductAmount{
			LocationId: reqParams.LocationId,
			//TODO:此处新增了ProductCategoryId入参，目前不清楚这个值的来源
			ProductCategoryID: reqParams.ProductCategoryId,
			ProductId:         reqParams.ProductId,
			Amount:            decimal.NewFromInt(0),
			CreateDate:        time.Now().Format("2006-01-02 15:04:05"),
		}
	}

	detail := &models.OperationDetails{
		ProductId:      reqParams.ProductId,
		Amount:         reqParams.DifferenceAmount,
		FromLocationID: location.Id,
		ToLocationID:   reqParams.LocationId,
	}
	operation := models.Operation{
		Number:            strconv.FormatInt(time.Now().Unix(), 10),
		Status:            constvar.OperationStatus_Ready,
		OperationDate:     time.Now().Format("2006-01-02 15:04:05"),
		Comment:           "库存盘点",
		BaseOperationType: constvar.BaseOperationTypeAdjust,
		Details:           []*models.OperationDetails{detail},
		LocationID:        location.Id,
		OperationTypeName: "库存调整",
	}

	if err := models.WithTransaction(func(tx *gorm.DB) error {
		if err := models.NewOperationSearch().SetOrm(tx).Create(&operation); err != nil {
			return err
		}
		if existAmount.Id == 0 {
			if err := models.NewLocationProductAmountSearch().SetOrm(tx).Create(&locAmount); err != nil {
				return err
			}
		}
		return nil
	}); err != nil {
		return err
	}

	service.AddNewLocationReportRecord(locAmount.Id)
	return nil
}

// GetRuleList
// @Tags      库存盘点
// @Summary   获取上架规则
// @Produce   application/json
// @Param     object  body  request.GetRuleList true  "查询信息"
// @Success   200 {object} util.ResponseList{data=[]response.RuleList}	"成功"
// @Router    /api-wms/v1/locationProductAmount/getRuleList [post]
func (slf LocationProductAmountController) GetRuleList(c *gin.Context) {
	var params request.GetRuleList
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	list, err := models.NewLocationProductSearch().SetProductId(params.ProductId).SetLocationId(params.LocationId).Find()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}
	var ruleList response.RuleList
	for _, l := range list {
		ruleList.LocationList = append(ruleList.LocationList, l.Location)
		ruleList.ProductList = append(ruleList.ProductList, l.Product)
	}
	util.ResponseFormat(c, code.Success, ruleList)
}

// Update
// @Tags      库存盘点
// @Summary   修改库存盘点信息
// @Produce   application/json
// @Param     object  body  request.UpdateLocationProductAmount true  "入库/出库信息"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/locationProductAmount/update [post]
func (slf LocationProductAmountController) Update(c *gin.Context) {
	var reqParams request.UpdateLocationProductAmount
	if err := c.BindJSON(&reqParams); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	if reqParams.OperationId == 0 {
		if err := AddLocationProductAmount(reqParams); err != nil {
			util.ResponseFormat(c, code.RequestError, err.Error())
			return
		}
		util.ResponseFormat(c, code.Success, "添加成功")
		return
	}
	operation, err := models.NewOperationSearch().SetID(reqParams.OperationId).First()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "未找到相关数据")
		return
	}
	if operation.Status != constvar.OperationStatus_Ready {
		util.ResponseFormat(c, code.RequestParamError, "该记录已被验证,请刷新页面查看")
		return
	}

	if err := models.NewOperationDetailsSearch().SetOperationId(reqParams.OperationId).SetProductId(reqParams.ProductId).Update(&models.OperationDetails{Amount: reqParams.DifferenceAmount}); err != nil {
		util.ResponseFormat(c, code.RequestError, "修改失败："+err.Error())
		return
	}

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

// Finish
//
//	@Tags		库存盘点
//	@Summary	应用、验证
//	@Produce	application/json
//	@Param		object  body  request.FinishLocationProductAmount true  "入参"
//	@Success	200	{object}	util.Response	"成功"
//	@Router		/api-wms/v1/locationProductAmount/finish [post]
func (slf LocationProductAmountController) Finish(c *gin.Context) {
	var reqParams request.FinishLocationProductAmount
	if err := c.BindJSON(&reqParams); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	if reqParams.LocationProductAmountId == 0 {
		util.ResponseFormat(c, code.RequestParamError, "locationProductAmountId为0")
		return
	}
	if reqParams.OperationId == 0 {
		util.ResponseFormat(c, code.RequestParamError, "operationId为0")
		return
	}
	operationSearch := models.NewOperationSearch().SetPreload(true)
	operationSearch.SetOrm(operationSearch.Orm.InnerJoins("inner join wms_operation_details on wms_operation_details.operation_id="+
		"wms_operation.id").Where("wms_operation_details.product_id=? and wms_operation.id=?", reqParams.ProductId, reqParams.OperationId))
	operation, err := operationSearch.First()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "未找到相关信息:"+err.Error())
		return
	}
	if operation.Status != constvar.OperationStatus_Ready {
		util.ResponseFormat(c, code.RequestError, "该验证无法完成")
		return
	}
	material, err := models.NewMaterialSearch().SetID(reqParams.ProductId).First()
	if err != nil {
		util.ResponseFormat(c, code.RequestError, err.Error())
		return
	}
	locProductAmount, err := models.NewLocationProductAmountSearch().SetID(reqParams.LocationProductAmountId).First()
	if err != nil {
		util.ResponseFormat(c, code.RequestError, err.Error())
		return
	}
	adjustAmount := operation.Details[0].Amount.Sub(locProductAmount.Amount)
	if err := models.WithTransaction(func(tx *gorm.DB) error {
		if err := models.NewOperationSearch().SetOrm(tx).SetID(reqParams.OperationId).Update(&models.Operation{Status: constvar.OperationStatus_Finish}); err != nil {
			return err
		}
		if err := models.NewMaterialSearch().SetOrm(tx).SetID(reqParams.ProductId).Update(&models.Material{Amount: material.Amount.Add(adjustAmount)}); err != nil {
			return err
		}
		if err := models.NewLocationProductAmountSearch().SetOrm(tx).SetID(reqParams.LocationProductAmountId).Update(&models.LocationProductAmount{Amount: operation.Details[0].Amount}); err != nil {
			return err
		}
		return nil
	}); err != nil {
		util.ResponseFormat(c, code.RequestError, err.Error())
		return
	}
	util.ResponseFormat(c, code.Success, "操作成功")
}

// GetLocationProductAmount
//
//	@Tags		库存盘点
//	@Summary	获取位置产品信息
//	@Produce	application/json
//	@Param		object  body 	request.GetRuleList true  "入参"
//	@Success	200	{object}	util.Response{data=models.LocationProductAmount}	"成功"
//	@Router		/api-wms/v1/locationProductAmount/getLocationProductAmount [post]
func (slf LocationProductAmountController) GetLocationProductAmount(c *gin.Context) {
	var params request.GetRuleList
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	if params.LocationId == 0 || params.ProductId == "" {
		util.ResponseFormat(c, code.Success, nil)
		return
	}
	first, err := models.NewLocationProductAmountSearch().SetLocationId(params.LocationId).SetProductId(params.ProductId).First()
	if err != nil {
		if err == gorm.ErrRecordNotFound {
			util.ResponseFormat(c, code.Success, nil)
			return
		}
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}
	util.ResponseFormat(c, code.Success, first)
}