package controllers

import (
	"errors"
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/spf13/cast"
	"gorm.io/gorm"
	"wms/extend/code"
	"wms/extend/util"
	"wms/models"
	"wms/pkg/logx"
	"wms/pkg/structx"
	"wms/request"
)

type WarehouseController struct{}

// Add
// @Tags      仓库
// @Summary   添加仓库
// @Produce   application/json
// @Param     object  body  request.AddWarehouse true  "仓库信息"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/warehouse/warehouse [post]
func (slf WarehouseController) Add(c *gin.Context) {
	var reqParams request.AddWarehouse
	var params models.Warehouse
	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, "数据转换错误")
		return
	}

	if err := slf.ParamsCheck(params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}
	if err := models.NewWarehouseSearch().Create(&params); err != nil {
		logx.Errorf("warehouse create err: %v", err)
		util.ResponseFormat(c, code.SaveFail, "插入失败")
		return
	}

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

// Update
// @Tags      仓库
// @Summary   编辑仓库
// @Produce   application/json
// @Param     object  body request.UpdateWarehouse true  "仓库信息"
// @Param     id  path string true  "仓库id"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/warehouse/warehouse/{id} [put]
func (slf WarehouseController) Update(c *gin.Context) {
	id := cast.ToUint(c.Param("id"))
	if id == 0 {
		util.ResponseFormat(c, code.RequestParamError, "空的记录id")
		return
	}
	var (
		reqParams request.UpdateWarehouse
		params    models.Warehouse
	)
	if err := c.BindJSON(&reqParams); err != nil {
		util.ResponseFormat(c, code.RequestParamError, fmt.Sprintf("参数解析失败: %v"+err.Error()))
		return
	}
	if err := structx.AssignTo(reqParams, &params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, fmt.Sprintf("数据转换错误: %v", err.Error()))
		return
	}
	params.ID = id
	if err := slf.ParamsCheck(params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}

	err := models.NewWarehouseSearch().SetID(params.ID).Update(&params)

	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "修改失败")
		return
	}

	util.ResponseFormat(c, code.UpdateSuccess, "更新成功")
}

func (slf WarehouseController) ParamsCheck(params models.Warehouse) (err error) {
	var oldRecord *models.Warehouse
	if params.ID != 0 {
		oldRecord, err = models.NewWarehouseSearch().SetID(params.ID).First()
		if err == gorm.ErrRecordNotFound {
			return errors.New("记录不存在")
		}
	}
	if oldRecord == nil || params.Code != oldRecord.Code {
		_, err = models.NewWarehouseSearch().SetCode(params.Code).First()
		if err != gorm.ErrRecordNotFound {
			return errors.New("仓库编号重复")
		}
	}

	return nil
}

// List
// @Tags      仓库
// @Summary   查询仓库列表
// @Produce   application/json
// @Param     object  query    request.GetWarehouseList true  "查询参数"
// @Success   200   {object}  util.ResponseList{data=[]models.Warehouse}  "成功"
// @Router    /api-wms/v1/warehouse/warehouse [get]
func (slf WarehouseController) List(c *gin.Context) {
	var params request.GetWarehouseList
	if err := c.ShouldBindQuery(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, err.Error())
		return
	}
	list, total, err := models.NewWarehouseSearch().SetPage(params.Page, params.PageSize).SetKeyword(params.Keyword).SetOrder("id desc").Find()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "查找失败")
		return
	}

	util.ResponseFormatList(c, code.Success, list, cast.ToInt(total))
}

// Delete
// @Tags      仓库
// @Summary   删除仓库
// @Produce   application/json
// @Param     id  path string true  "仓库id"
// @Success   200 {object} util.Response "成功"
// @Router    /api-wms/v1/warehouse/warehouse/{id} [delete]
func (slf WarehouseController) Delete(c *gin.Context) {
	id := cast.ToUint(c.Param("id"))
	if id == 0 {
		util.ResponseFormat(c, code.RequestParamError, "空的记录id")
		return
	}

	err := models.NewWarehouseSearch().SetID(id).Delete()
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "删除失败")
		return
	}
	util.ResponseFormat(c, code.UpdateSuccess, "删除成功")
}