package controllers

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"github.com/shopspring/decimal"
	"net/url"
	"os"
	"wms/constvar"
	"wms/extend/code"
	"wms/extend/util"
	"wms/models"
	"wms/pkg/logx"
	"wms/request"
	"wms/response"
	"wms/service"
	"wms/task"
)

type ReportFormsController struct {
}

type Detail struct {
	ProductId string                   `json:"productId"`
	Amount    decimal.Decimal          `json:"amount"`
	Status    constvar.OperationStatus `json:"status"`
	//ProductName string                   `json:"productName"`
}

// GetInventoryForms
// @Tags      报表
// @Summary   获取库存报表
// @Produce   application/json
// @Param     Authorization	header string true "token"
// @Param     object  body  request.GetInventoryForms true  "查询参数"
// @Success   200 {object} util.ResponseList{data=[]response.InventoryForms}	"成功"
// @Router    /api-wms/v1/forms/getInventoryForms [post]
func (slf ReportFormsController) GetInventoryForms(c *gin.Context) {
	var params request.GetInventoryForms
	err := c.BindJSON(&params)
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	inventoryReportFormsService := service.NewInventoryReportFormsService()
	total, err := inventoryReportFormsService.Count(params)
	if err != nil {
		logx.Errorf("GetInventoryForms count err:%v", err)
		util.ResponseFormat(c, code.InternalError, "查询总数失败")
		return
	}
	result, err := inventoryReportFormsService.Query(params)
	if err != nil {
		logx.Errorf("GetInventoryForms query err:%v", err)
		util.ResponseFormat(c, code.InternalError, "查询失败")
		return
	}

	util.ResponseFormatList(c, code.Success, result, int(total))
}

// DownloadInventoryForms
// @Tags      报表
// @Summary   下载库存报表
// @Produce   application/json
// @Param     Authorization	header string true "token"
// @Param     object  body  request.GetInventoryForms true  "查询参数"
// @Success   200 {object} util.ResponseList{data=[]response.InventoryForms}	"成功"
// @Router    /api-wms/v1/forms/downloadInventoryForms [post]
func (slf ReportFormsController) DownloadInventoryForms(c *gin.Context) {
	var params request.GetInventoryForms
	err := c.BindJSON(&params)
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	inventoryReportFormsService := service.NewInventoryReportFormsService()
	list, err := inventoryReportFormsService.FetchAll(params)
	if err != nil {
		logx.Errorf("DownloadInventoryForms FetchAll err:%v", err)
		util.ResponseFormat(c, code.InternalError, "查询失败")
		return
	}
	filename, err := inventoryReportFormsService.Export(list)
	if err != nil {
		logx.Errorf("DownloadInventoryForms Export err:%v", err)
		util.ResponseFormat(c, code.InternalError, "导出数据到文件失败")
		return
	}

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

	fileContentDisposition := "attachment;filename=\"" + url.QueryEscape(filename) + "\""
	c.Header("Content-Type", "application/xlsx")
	c.Header("Content-Disposition", fileContentDisposition)
	c.File(filename)
	defer os.Remove(filename)
}

// GetHistory
// @Tags      报表
// @Summary   获取历史信息
// @Produce   application/json
// @Param     object  body  request.GetInventoryHistory true  "查询参数"
// @Success   200 {object} util.ResponseList{data=[]response.InventoryHistory}	"成功"
// @Router    /api-wms/v1/forms/getHistory [post]
func (slf ReportFormsController) GetHistory(c *gin.Context) {
	slf.GetHistoryNew(c)
	return
	//var params request.GetInventoryHistory
	//if err := c.BindJSON(&params); err != nil {
	//	util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
	//	return
	//}
	////获取操作详情
	//detailsSearch := models.NewOperationDetailsSearch()
	//var (
	//	result []*response.InventoryHistory
	//	total  int64
	//	ids    []int
	//	err    error
	//)
	//if params.KeyWord != "" {
	//	ids, err = service.SearchHistoryReport(params.KeyWord, params.BaseOperationType)
	//
	//	if err != nil {
	//		util.ResponseFormat(c, code.InternalError, err.Error())
	//		return
	//	}
	//	if len(ids) == 0 {
	//		util.ResponseFormatList(c, code.Success, result, 0)
	//	}
	//}
	//
	//detailsSearch.Orm = detailsSearch.Orm.Model(&models.OperationDetails{}).
	//	Select("wms_operation.number, wms_operation.updated_at as date, material.name as product_name, wms_operation.from_location_id,"+
	//		"wms_operation.to_location_id, wms_operation_details.amount, material.unit, wms_operation.contacter_name as contacted_name,"+
	//		"wms_operation.base_operation_type, material.weight, wms_operation_details.product_id").
	//	Joins("left join wms_operation on wms_operation_details.operation_id = wms_operation.id").
	//	Joins("left join material on wms_operation_details.product_id = material.id").
	//	Where("wms_operation.status = ?", constvar.OperationStatus_Finish).Order("wms_operation.created_at desc")
	//if len(ids) > 0 {
	//	detailsSearch.Orm = detailsSearch.Orm.Where("wms_operation_details.id in ?", ids)
	//}
	//if params.ProduceId != "" {
	//	detailsSearch.Orm = detailsSearch.Orm.Where("wms_operation_details.product_id = ?", params.ProduceId)
	//}
	//if params.BaseOperationType != 0 {
	//	detailsSearch.Orm = detailsSearch.Orm.Where("wms_operation.base_operation_type = ?", params.BaseOperationType)
	//}
	//
	//err = detailsSearch.Orm.Count(&total).Error
	//if err != nil {
	//	util.ResponseFormat(c, code.RequestParamError, "查询total失败")
	//	return
	//}
	//if params.Page*params.PageSize > 0 {
	//	detailsSearch.Orm = detailsSearch.Orm.Offset((params.Page - 1) * params.PageSize).Limit(params.PageSize)
	//}
	//
	//err = detailsSearch.Orm.Find(&result).Error
	//if err != nil {
	//	util.ResponseFormat(c, code.RequestParamError, fmt.Errorf("查询操作明细失败: %v", err))
	//	return
	//}
	//var locationIds []int
	//for _, history := range result {
	//	locationIds = append(locationIds, history.FromLocationId, history.ToLocationId)
	//}
	//locations, err := models.NewLocationSearch().SetIds(locationIds).FindNotTotal()
	//if err != nil {
	//	util.ResponseFormat(c, code.RequestParamError, fmt.Errorf("查询位置失败: %v", err))
	//	return
	//}
	//for _, history := range result {
	//	history.Status = "完成"
	//	for _, location := range locations {
	//		if history.ToLocationId == location.Id {
	//			history.ToLocation = location.Name
	//		}
	//		if history.FromLocationId == location.Id {
	//			history.FromLocation = location.Name
	//		}
	//	}
	//}
	//util.ResponseFormatList(c, code.Success, result, int(total))
}

func (slf ReportFormsController) GetHistoryNew(c *gin.Context) {
	var params request.GetInventoryHistory
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	//获取操作详情
	detailsSearch := models.NewMoveHistorySearch()
	var (
		result []*response.InventoryHistory
		total  uint64
		ids    []int
		err    error
	)
	if params.KeyWord != "" {
		ids, total, err = service.SearchHistoryReport(params.KeyWord, params.BaseOperationType, params.Page, params.PageSize)
		if err != nil {
			util.ResponseFormat(c, code.InternalError, err.Error())
			return
		}
		if len(ids) == 0 {
			util.ResponseFormatList(c, code.Success, result, 0)
			return
		}
	}

	detailsSearch.Orm = detailsSearch.Orm.Model(&models.MoveHistory{}).
		Select("number, updated_at as date, product_name as product_name, from_location_id, operation_id,to_location_id, amount, " +
			"unit, operator as contacted_name, base_operation_type, weight, product_id, from_location, to_location, operation_type_name, weight").Order("id desc")
	if len(ids) > 0 {
		detailsSearch.Orm = detailsSearch.Orm.Where("id in ?", ids)
	}
	if params.BaseOperationType != 0 {
		detailsSearch.Orm = detailsSearch.Orm.Where("base_operation_type = ?", params.BaseOperationType)
	}
	var t int64
	err = detailsSearch.Orm.Count(&t).Error
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, fmt.Errorf("查询总条数失败: %v", err))
		return
	}
	total = uint64(t)
	if params.Page*params.PageSize > 0 {
		detailsSearch.Orm = detailsSearch.Orm.Offset((params.Page - 1) * params.PageSize).Limit(params.PageSize)
	}
	err = detailsSearch.Orm.Find(&result).Error
	if err != nil {
		util.ResponseFormat(c, code.RequestParamError, fmt.Errorf("查询操作明细失败: %v", err))
		return
	}
	util.ResponseFormatList(c, code.Success, result, int(total))
}

// GetLocationForms
// @Tags      报表
// @Summary   获取位置报表
// @Produce   application/json
// @Param     object  body  request.GetLocationForms true  "查询参数"
// @Success   200 {object} util.ResponseList{data=[]response.LocationForms}	"成功"
// @Router    /api-wms/v1/forms/getLocationForms [post]
func (slf ReportFormsController) GetLocationForms(c *gin.Context) {
	var params request.GetLocationForms
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}
	ids := make([]int, 0)
	if params.LocationId != 0 {
		ids = append(ids, params.LocationId)
	} else {
		//查询位置
		locations, err := models.NewLocationSearch().SetJointName(params.WareHouseCode).SetType(3).FindAll()
		if err != nil {
			util.ResponseFormat(c, code.RequestParamError, "查询位置失败")
			return
		}
		for _, location := range locations {
			ids = append(ids, location.Id)
		}
	}
	var (
		amounts    []*models.LocationProductAmount
		total      uint64
		totalInt64 int64
		err        error
	)
	if params.KeyWord != "" {
		amounts, total, err = service.SearchLocationReport(params.KeyWord, params.Page, params.PageSize)
	} else {
		amounts, totalInt64, err = models.NewLocationProductAmountSearch().SetPage(params.Page, params.PageSize).SetPreload(true).SetKeyword(params.KeyWord).SetProductId(params.ProductId).SetLocationIds(ids).FindByPage()
		if err != nil {
			util.ResponseFormat(c, code.RequestParamError, "查询数量失败")
			return
		}
		total = uint64(totalInt64)
	}

	var result []response.LocationForms
	for _, amount := range amounts {
		var resp response.LocationForms
		resp.Amount = amount.Amount
		resp.LocationId = amount.LocationId
		resp.LocationName = amount.Location.Name
		resp.ProduceId = amount.Product.ID
		resp.ProductName = amount.Product.Name
		resp.ProductTypeName = amount.ProductCategory.Name
		resp.Unit = amount.Product.Unit
		resp.Value = resp.Amount.Mul(amount.Product.Cost)
		result = append(result, resp)
	}

	util.ResponseFormatList(c, code.Success, result, int(total))
}

// MonthStats
// @Tags      报表
// @Summary   月度统计库存报表
// @Produce   application/json
// @Param     object  body  request.GetMonthStats true  "查询参数"
// @Param     Authorization	header string true "token"
// @Success   200 {object} util.ResponseList{data=[]models.MonthStats}	"成功"
// @Router    /api-wms/v1/forms/monthStats [post]
func (slf ReportFormsController) MonthStats(c *gin.Context) {
	var params request.GetMonthStats
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}

	list, total, err := models.NewMonthStatsSearch().SetPage(params.Page, params.PageSize).SetKeyword(params.Keyword).SetDate(params.Date).Find()
	if err != nil {
		util.ResponseFormat(c, code.InternalError, "查询失败")
		return
	}

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

// DoMonthStats
// @Tags      报表
// @Summary   手动跑月度统计库存报表
// @Produce   application/json
// @Param     object  body  request.DoMonthStats true  "查询参数"
// @Param     Authorization	header string true "token"
// @Success   200 {object} util.ResponseList{data=[]models.MonthStats}	"成功"
// @Router    /api-wms/v1/forms/doMonthStats [post]
func (slf ReportFormsController) DoMonthStats(c *gin.Context) {
	var params request.DoMonthStats
	if err := c.BindJSON(&params); err != nil {
		util.ResponseFormat(c, code.RequestParamError, "参数解析失败，数据类型错误")
		return
	}

	if params.Token != constvar.DoMonthStatsToken {
		return
	}

	task.MonthStats()
	util.ResponseFormat(c, code.Success, nil)
}