package handler

import (
	"fmt"
	"github.com/gin-gonic/gin"
	"hpds-iot-web/internal/proto"
	"hpds-iot-web/internal/service"
	"hpds-iot-web/model"
	e "hpds-iot-web/pkg/err"
)

func (s HandlerService) ReportList(c *gin.Context) (data interface{}, err error) {
	repo := service.NewReportService(s.Engine, s.Logger)
	us, _ := c.Get("operatorUser")
	userInfo := us.(*model.SystemUser)
	var req proto.ReportRequest
	err = c.ShouldBindJSON(&req)
	if err != nil {
		go s.SaveLog("ReportList", "Manage", "", "", req.ToString(), fmt.Sprintf("%d", userInfo.UserId), c.Request.RemoteAddr, "")
		return nil, e.NewValidErr(err)
	}
	data, err = repo.ReportList(c, req)
	go s.SaveLog("报表列表", "Manage", "", "", "", fmt.Sprintf("%d", userInfo.UserId), c.Request.RemoteAddr, "")
	return
}

func (s HandlerService) GenerateReport(c *gin.Context) (data interface{}, err error) {
	repo := service.NewReportService(s.Engine, s.Logger)
	us, _ := c.Get("operatorUser")
	userInfo := us.(*model.SystemUser)
	var req proto.ReportRequest
	err = c.ShouldBindJSON(&req)
	if err != nil {
		go s.SaveLog("GenerateReport", "Manage", "", "", req.ToString(), fmt.Sprintf("%d", userInfo.UserId), c.Request.RemoteAddr, "")
		return nil, e.NewValidErr(err)
	}
	xlsx, err := repo.GenerateReport(c, req)
	if err != nil {

	}
	c.Header("Content-Type", "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")
	c.Header("Content-Disposition", "attachment; filename=模型报表.xlsx")
	//c.Header("")
	c.Header("Content-Transfer-Encoding", "binary")
	err = xlsx.Write(c.Writer)
	return nil, err
}