Home-Vue-Go/internal/api/sites.go

package api

import (
	"net/http"
	"strconv"
	"strings"

	"home-vue-go/internal/database"
	"home-vue-go/internal/ent"
	"home-vue-go/internal/ent/site"

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

func GetSites(db *database.Database) gin.HandlerFunc {
	return func(c *gin.Context) {
		ctx := c.Request.Context()
		sites, err := db.Client.Site.Query().Order(site.BySortOrder(), site.ByID()).All(ctx)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		result := make([]gin.H, len(sites))
		for i, site := range sites {
			result[i] = gin.H{
				"id":        site.ID,
				"name":      site.Name,
				"url":       site.URL,
				"icon":      site.Icon,
				"sortOrder": site.SortOrder,
			}
		}

		c.JSON(http.StatusOK, result)
	}
}

func ReorderSites(db *database.Database) gin.HandlerFunc {
	return func(c *gin.Context) {
		var req struct {
			IDs []int `json:"ids" binding:"required"`
		}

		if err := c.ShouldBindJSON(&req); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
			return
		}
		if len(req.IDs) == 0 {
			c.JSON(http.StatusBadRequest, gin.H{"error": "排序列表不能为空"})
			return
		}

		seen := make(map[int]bool, len(req.IDs))
		for _, id := range req.IDs {
			if id <= 0 {
				c.JSON(http.StatusBadRequest, gin.H{"error": "排序列表包含无效ID"})
				return
			}
			if seen[id] {
				c.JSON(http.StatusBadRequest, gin.H{"error": "排序列表包含重复ID"})
				return
			}
			seen[id] = true
		}

		ctx := c.Request.Context()
		existingIDs, err := db.Client.Site.Query().IDs(ctx)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}
		if len(existingIDs) != len(req.IDs) {
			c.JSON(http.StatusBadRequest, gin.H{"error": "排序列表与当前站点数量不一致"})
			return
		}
		existing := make(map[int]bool, len(existingIDs))
		for _, id := range existingIDs {
			existing[id] = true
		}
		for _, id := range req.IDs {
			if !existing[id] {
				c.JSON(http.StatusBadRequest, gin.H{"error": "排序列表包含不存在的站点"})
				return
			}
		}

		tx, err := db.Client.Tx(ctx)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		for index, id := range req.IDs {
			if _, err := tx.Site.UpdateOneID(id).SetSortOrder((index + 1) * 10).Save(ctx); err != nil {
				_ = tx.Rollback()
				c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
				return
			}
		}

		if err := tx.Commit(); err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		sites, err := db.Client.Site.Query().Order(site.BySortOrder(), site.ByID()).All(ctx)
		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		result := make([]gin.H, len(sites))
		for i, site := range sites {
			result[i] = gin.H{
				"id":        site.ID,
				"name":      site.Name,
				"url":       site.URL,
				"icon":      site.Icon,
				"sortOrder": site.SortOrder,
			}
		}

		c.JSON(http.StatusOK, result)
	}
}

func CreateSite(db *database.Database) gin.HandlerFunc {
	return func(c *gin.Context) {
		var req struct {
			Name      string `json:"name" binding:"required"`
			URL       string `json:"url" binding:"required"`
			Icon      string `json:"icon" binding:"required"`
			SortOrder int    `json:"sortOrder"`
		}

		if err := c.ShouldBindJSON(&req); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
			return
		}
		req.Name, req.URL, req.Icon = strings.TrimSpace(req.Name), strings.TrimSpace(req.URL), strings.TrimSpace(req.Icon)
		if req.Name == "" || req.URL == "" || req.Icon == "" || !validHTTPURL(req.URL) {
			c.JSON(http.StatusBadRequest, gin.H{"error": "站点名称、图标不能为空，URL必须是有效的http/https地址"})
			return
		}

		ctx := c.Request.Context()
		site, err := db.Client.Site.Create().
			SetName(req.Name).
			SetURL(req.URL).
			SetIcon(req.Icon).
			SetSortOrder(req.SortOrder).
			Save(ctx)

		if err != nil {
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		c.JSON(http.StatusOK, gin.H{
			"id":        site.ID,
			"name":      site.Name,
			"url":       site.URL,
			"icon":      site.Icon,
			"sortOrder": site.SortOrder,
		})
	}
}

func UpdateSite(db *database.Database) gin.HandlerFunc {
	return func(c *gin.Context) {
		id, err := strconv.Atoi(c.Param("id"))
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "无效的ID"})
			return
		}

		var req struct {
			Name      string `json:"name"`
			URL       string `json:"url"`
			Icon      string `json:"icon"`
			SortOrder int    `json:"sortOrder"`
		}

		if err := c.ShouldBindJSON(&req); err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
			return
		}
		req.Name, req.URL, req.Icon = strings.TrimSpace(req.Name), strings.TrimSpace(req.URL), strings.TrimSpace(req.Icon)
		if req.Name == "" || req.URL == "" || req.Icon == "" || !validHTTPURL(req.URL) {
			c.JSON(http.StatusBadRequest, gin.H{"error": "站点名称、图标不能为空，URL必须是有效的http/https地址"})
			return
		}

		ctx := c.Request.Context()
		update := db.Client.Site.UpdateOneID(id)
		update.SetName(req.Name)
		update.SetURL(req.URL)
		update.SetIcon(req.Icon)
		update.SetSortOrder(req.SortOrder)

		site, err := update.Save(ctx)
		if err != nil {
			if ent.IsNotFound(err) {
				c.JSON(http.StatusNotFound, gin.H{"error": "站点不存在"})
				return
			}
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		c.JSON(http.StatusOK, gin.H{
			"id":        site.ID,
			"name":      site.Name,
			"url":       site.URL,
			"icon":      site.Icon,
			"sortOrder": site.SortOrder,
		})
	}
}

func DeleteSite(db *database.Database) gin.HandlerFunc {
	return func(c *gin.Context) {
		id, err := strconv.Atoi(c.Param("id"))
		if err != nil {
			c.JSON(http.StatusBadRequest, gin.H{"error": "无效的ID"})
			return
		}

		ctx := c.Request.Context()
		if err := db.Client.Site.DeleteOneID(id).Exec(ctx); err != nil {
			if ent.IsNotFound(err) {
				c.JSON(http.StatusNotFound, gin.H{"error": "站点不存在"})
				return
			}
			c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
			return
		}

		c.JSON(http.StatusOK, gin.H{"message": "删除成功"})
	}
}