使用了AI重构项目,并完善了一部分后台问题

This commit is contained in:
2026-08-05 04:47:11 +08:00
parent 146b6b1e6c
commit ac740c24fd
35 changed files with 2876 additions and 980 deletions
+112 -9
View File
@@ -3,8 +3,10 @@ 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"
@@ -34,6 +36,95 @@ func GetSites(db *database.Database) gin.HandlerFunc {
}
}
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 {
@@ -47,6 +138,11 @@ func CreateSite(db *database.Database) gin.HandlerFunc {
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().
@@ -90,22 +186,25 @@ func UpdateSite(db *database.Database) gin.HandlerFunc {
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)
if req.Name != "" {
update.SetName(req.Name)
}
if req.URL != "" {
update.SetURL(req.URL)
}
if req.Icon != "" {
update.SetIcon(req.Icon)
}
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
}
@@ -130,6 +229,10 @@ func DeleteSite(db *database.Database) gin.HandlerFunc {
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
}