package handler

import (
	"encoding/json"
	"net/http"
	"net/url"
	"strings"
	"time"

	"kra/internal/server/httpx"
	"kra/internal/service"
	"kra/internal/service/dto"

	"github.com/gin-gonic/gin"
	"github.com/google/uuid"
)

type Export struct {
	settings *service.SettingsService
	service  *service.ExportService
}

func NewExport(settings *service.SettingsService, export *service.ExportService) *Export {
	return &Export{settings: settings, service: export}
}

type exportToken struct {
	TemplateID string            `json:"templateID"`
	Params     map[string]string `json:"params"`
	Blank      bool              `json:"blank"`
}

func exportParams(values url.Values) map[string]string {
	out := map[string]string{}
	nested, _ := url.ParseQuery(values.Get("params"))
	for key, items := range nested {
		if len(items) > 0 {
			out[key] = items[0]
		}
	}
	for key, items := range values {
		if key != "params" && len(items) > 0 {
			out[key] = items[0]
		}
	}
	return out
}

func (h *Export) Create(c *gin.Context) {
	var req dto.ExportTemplateRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		httpx.Fail(c, err.Error())
		return
	}
	if req.Name == "" {
		httpx.Fail(c, "Name值不能为空")
		return
	}
	if err := h.service.CreateRequest(c.Request.Context(), &req); err != nil {
		httpx.Fail(c, "创建失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "创建成功")
}
func (h *Export) Update(c *gin.Context) {
	var req dto.ExportTemplateRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		httpx.Fail(c, err.Error())
		return
	}
	if req.Name == "" {
		httpx.Fail(c, "Name值不能为空")
		return
	}
	if err := h.service.UpdateRequest(c.Request.Context(), &req); err != nil {
		httpx.Fail(c, "更新失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "更新成功")
}
func (h *Export) Delete(c *gin.Context) {
	var req dto.IDRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		httpx.Fail(c, err.Error())
		return
	}
	if err := h.service.Delete(c.Request.Context(), []uint{req.ID}); err != nil {
		httpx.Fail(c, "删除失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "删除成功")
}
func (h *Export) DeleteMany(c *gin.Context) {
	var req dto.IDsRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		httpx.Fail(c, err.Error())
		return
	}
	if err := h.service.Delete(c.Request.Context(), req.IDs); err != nil {
		httpx.Fail(c, "批量删除失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "批量删除成功")
}
func (h *Export) Find(c *gin.Context) {
	var req dto.ExportTemplateRequest
	if err := c.ShouldBindQuery(&req); err != nil {
		httpx.Fail(c, err.Error())
		return
	}
	item, err := h.service.Template(c.Request.Context(), req.ID, "")
	if err != nil {
		httpx.Fail(c, "查询失败")
		return
	}
	httpx.OKWithData(c, gin.H{"resysExportTemplate": item})
}
func (h *Export) List(c *gin.Context) {
	p, size := page(c)
	start, startErr := parseTime(c.Query("startCreatedAt"))
	end, endErr := parseTime(c.Query("endCreatedAt"))
	if startErr != nil || endErr != nil {
		httpx.Fail(c, "参数错误")
		return
	}
	items, total, err := h.service.TemplatesFilter(c.Request.Context(), p, size, c.Query("name"), c.Query("tableName"), c.Query("templateID"), start, end)
	if err != nil {
		httpx.Fail(c, "获取失败")
		return
	}
	httpx.Write(c, httpx.CodeSuccess, httpx.PageResult{List: items, Total: total, Page: p, PageSize: size}, "获取成功")
}
func (h *Export) Preview(c *gin.Context) {
	templateID := c.Query("templateID")
	if templateID == "" {
		httpx.Fail(c, "模板ID不能为空")
		return
	}
	sql, err := h.service.Preview(c.Request.Context(), templateID, exportParams(c.Request.URL.Query()))
	if err != nil {
		httpx.Fail(c, "获取失败")
		return
	}
	httpx.OKWithData(c, gin.H{"sql": sql})
}

func (h *Export) Issue(blank bool) gin.HandlerFunc {
	return func(c *gin.Context) {
		templateID := c.Query("templateID")
		if templateID == "" {
			httpx.Fail(c, "模板ID不能为空")
			return
		}
		token := strings.ReplaceAll(uuid.NewString(), "-", "")
		raw, _ := json.Marshal(exportToken{TemplateID: templateID, Params: exportParams(c.Request.URL.Query()), Blank: blank})
		if err := h.settings.CacheSet(c.Request.Context(), "export:"+token, string(raw), 30*time.Minute); err != nil {
			httpx.Fail(c, "导出令牌创建失败")
			return
		}
		path := "/sysExportTemplate/exportExcelByToken?token=" + token
		if blank {
			path = "/sysExportTemplate/exportTemplateByToken?token=" + token
		}
		httpx.OKWithData(c, path)
	}
}

func (h *Export) Import(c *gin.Context) {
	templateID := c.Query("templateID")
	if templateID == "" {
		httpx.Fail(c, "模板ID不能为空")
		return
	}
	file, err := c.FormFile("file")
	if err != nil {
		httpx.Fail(c, "文件获取失败")
		return
	}
	opened, err := file.Open()
	if err != nil {
		httpx.Fail(c, "读取文件失败")
		return
	}
	defer opened.Close()
	if err = h.service.Import(c.Request.Context(), templateID, opened); err != nil {
		httpx.Fail(c, err.Error())
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "导入成功")
}

func (h *Export) Download(expectBlank bool) gin.HandlerFunc {
	return func(c *gin.Context) {
		token := c.Query("token")
		if token == "" {
			httpx.Fail(c, "导出token不能为空")
			return
		}
		raw, ok, err := h.settings.CacheGet(c.Request.Context(), "export:"+token)
		if err != nil || !ok {
			httpx.Fail(c, "导出token无效或已过期")
			return
		}
		var value exportToken
		if json.Unmarshal([]byte(raw), &value) != nil {
			httpx.Fail(c, "解析导出参数失败")
			return
		}
		if value.Blank != expectBlank {
			httpx.Fail(c, "token类型错误")
			return
		}
		_ = h.settings.CacheDelete(c.Request.Context(), "export:"+token)
		var data []byte
		var name string
		if expectBlank {
			data, name, err = h.service.ExportBlankTemplate(c.Request.Context(), value.TemplateID)
		} else {
			data, name, err = h.service.Export(c.Request.Context(), value.TemplateID, value.Params)
		}
		if err != nil {
			httpx.Fail(c, "获取失败")
			return
		}
		if expectBlank {
			name = strings.TrimSuffix(name, "_template.xlsx") + "模板.xlsx"
		} else {
			name = strings.TrimSuffix(name, ".xlsx") + strings.ReplaceAll(uuid.NewString(), "-", "")[:6] + ".xlsx"
		}
		c.Header("Content-Disposition", "attachment; filename="+name)
		c.Header("success", "true")
		c.Data(http.StatusOK, "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", data)
	}
}