kra-oa/internal/modules/system/transport/handler/version.go

package handler

import (
	"errors"
	"fmt"
	"net/http"
	"strconv"
	"time"

	"kra/internal/modules/system/biz"
	"kra/internal/modules/system/dto"
	"kra/internal/modules/system/service"
	"kra/internal/modules/system/transport/httpx"

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

type Version struct{ service *service.VersionService }

func NewVersion(service *service.VersionService) *Version { return &Version{service: service} }

func versionStage(err error) (biz.VersionStage, bool) {
	var stageErr *biz.VersionStageError
	if !errors.As(err, &stageErr) {
		return "", false
	}
	return stageErr.Stage, true
}

func (h *Version) Delete(c *gin.Context) {
	id, _ := strconv.ParseUint(c.Query("ID"), 10, 64)
	if err := h.service.DeleteVersions(c.Request.Context(), []uint{uint(id)}); err != nil {
		httpx.Fail(c, "删除失败:"+err.Error())
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "删除成功")
}
func (h *Version) DeleteMany(c *gin.Context) {
	raw := c.QueryArray("IDs[]")
	ids := make([]uint, 0, len(raw))
	for _, value := range raw {
		id, _ := strconv.ParseUint(value, 10, 64)
		ids = append(ids, uint(id))
	}
	if err := h.service.DeleteVersions(c.Request.Context(), ids); err != nil {
		httpx.Fail(c, "批量删除失败:"+err.Error())
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "批量删除成功")
}
func (h *Version) Find(c *gin.Context) {
	id, _ := strconv.ParseUint(c.Query("ID"), 10, 64)
	item, err := h.service.Version(c.Request.Context(), uint(id))
	if err != nil {
		httpx.Fail(c, "查询失败:"+err.Error())
		return
	}
	httpx.OKWithData(c, item)
}
func (h *Version) List(c *gin.Context) {
	var req dto.VersionSearchRequest
	if err := c.ShouldBindQuery(&req); err != nil {
		httpx.Fail(c, err.Error())
		return
	}
	name, code := "", ""
	if req.VersionName != nil {
		name = *req.VersionName
	}
	if req.VersionCode != nil {
		code = *req.VersionCode
	}
	createdAtRange := make([]*time.Time, 0, len(req.CreatedAtRange))
	for i := range req.CreatedAtRange {
		createdAtRange = append(createdAtRange, &req.CreatedAtRange[i])
	}
	items, total, err := h.service.Versions(c.Request.Context(), req.Page, req.PageSize, name, code, createdAtRange)
	if err != nil {
		httpx.Fail(c, "获取失败:"+err.Error())
		return
	}
	httpx.Write(c, httpx.CodeSuccess, httpx.PageResult{List: items, Total: total, Page: req.Page, PageSize: req.PageSize}, "获取成功")
}

func (h *Version) Export(c *gin.Context) {
	var req dto.ExportVersionRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		httpx.Fail(c, err.Error())
		return
	}
	if err := h.service.Export(c.Request.Context(), req.VersionName, req.VersionCode, req.Description, req.MenuIDs, req.APIIDs, req.DictIDs); err != nil {
		switch stage, _ := versionStage(err); stage {
		case biz.VersionStageMenus:
			httpx.Fail(c, "获取菜单数据失败:"+err.Error())
		case biz.VersionStageAPIs:
			httpx.Fail(c, "获取API数据失败:"+err.Error())
		case biz.VersionStageDictionaries:
			httpx.Fail(c, "获取字典数据失败:"+err.Error())
		case biz.VersionStageJSON:
			httpx.Fail(c, "JSON序列化失败:"+err.Error())
		case biz.VersionStageSave:
			httpx.Fail(c, "保存版本记录失败:"+err.Error())
		default:
			httpx.Fail(c, "创建发版失败:"+err.Error())
		}
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "创建发版成功")
}
func (h *Version) Download(c *gin.Context) {
	rawID := c.Query("ID")
	if rawID == "" {
		httpx.Fail(c, "版本ID不能为空")
		return
	}
	id, _ := strconv.ParseUint(rawID, 10, 64)
	raw, code, err := h.service.VersionData(c.Request.Context(), uint(id))
	if err != nil {
		httpx.Fail(c, "获取版本记录失败:"+err.Error())
		return
	}
	name := fmt.Sprintf("version_%s_%s.json", code, time.Now().Format("20060102150405"))
	c.Header("Content-Type", "application/json")
	c.Header("Content-Disposition", fmt.Sprintf("attachment; filename=%s", name))
	c.Header("Content-Length", strconv.Itoa(len(raw)))
	c.Data(http.StatusOK, "application/json", []byte(raw))
}
func (h *Version) Import(c *gin.Context) {
	var req dto.ImportVersionRequest
	if err := c.ShouldBindJSON(&req); err != nil {
		httpx.Fail(c, "解析JSON数据失败:"+err.Error())
		return
	}
	if err := h.service.ImportRequest(c.Request.Context(), &req); err != nil {
		if errors.Is(err, biz.ErrInvalidVersion) {
			httpx.Fail(c, "版本信息格式错误")
			return
		}
		switch stage, _ := versionStage(err); stage {
		case biz.VersionStageMenus:
			httpx.Fail(c, "导入菜单失败: "+err.Error())
		case biz.VersionStageAPIs:
			httpx.Fail(c, "导入API失败: "+err.Error())
		case biz.VersionStageDictionaries:
			httpx.Fail(c, "导入字典失败: "+err.Error())
		default:
			httpx.Fail(c, "导入失败:"+err.Error())
		}
		return
	}
	httpx.Write(c, httpx.CodeSuccess, gin.H{}, "导入成功")
}