使用了AI重构项目,并完善了一部分后台问题
This commit is contained in:
+177
-38
@@ -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/contact"
|
||||
|
||||
"github.com/gin-gonic/gin"
|
||||
@@ -22,13 +24,104 @@ func GetContacts(db *database.Database) gin.HandlerFunc {
|
||||
result := make([]gin.H, len(contacts))
|
||||
for i, contact := range contacts {
|
||||
result[i] = gin.H{
|
||||
"id": contact.ID,
|
||||
"type": contact.Type,
|
||||
"icon": contact.Icon,
|
||||
"url": contact.URL,
|
||||
"qrCode": contact.QrCode,
|
||||
"id": contact.ID,
|
||||
"type": contact.Type,
|
||||
"icon": contact.Icon,
|
||||
"url": contact.URL,
|
||||
"qrCode": contact.QrCode,
|
||||
"hoverColor": contact.HoverColor,
|
||||
"sortOrder": contact.SortOrder,
|
||||
"sortOrder": contact.SortOrder,
|
||||
}
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, result)
|
||||
}
|
||||
}
|
||||
|
||||
func ReorderContacts(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.Contact.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.Contact.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
|
||||
}
|
||||
|
||||
contacts, err := db.Client.Contact.Query().Order(contact.BySortOrder(), contact.ByID()).All(ctx)
|
||||
if err != nil {
|
||||
c.JSON(http.StatusInternalServerError, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
|
||||
result := make([]gin.H, len(contacts))
|
||||
for i, contact := range contacts {
|
||||
result[i] = gin.H{
|
||||
"id": contact.ID,
|
||||
"type": contact.Type,
|
||||
"icon": contact.Icon,
|
||||
"url": contact.URL,
|
||||
"qrCode": contact.QrCode,
|
||||
"hoverColor": contact.HoverColor,
|
||||
"sortOrder": contact.SortOrder,
|
||||
}
|
||||
}
|
||||
|
||||
@@ -51,13 +144,31 @@ func CreateContact(db *database.Database) gin.HandlerFunc {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
req.Type = strings.TrimSpace(req.Type)
|
||||
req.Icon = strings.TrimSpace(req.Icon)
|
||||
req.URL = strings.TrimSpace(req.URL)
|
||||
req.QrCode = strings.TrimSpace(req.QrCode)
|
||||
req.HoverColor = strings.TrimSpace(req.HoverColor)
|
||||
if req.Type == "" || req.Icon == "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "类型和图标不能为空"})
|
||||
return
|
||||
}
|
||||
|
||||
// 验证Email类型必须是mailto格式
|
||||
if req.Type == "Email" && req.URL != "" {
|
||||
if len(req.URL) < 7 || req.URL[:7] != "mailto:" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "Email URL必须是mailto:格式"})
|
||||
return
|
||||
}
|
||||
if req.URL != "" && !validContactURL(req.URL) {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "联系方式URL必须是有效的http/https/mailto/tel地址"})
|
||||
return
|
||||
}
|
||||
if req.QrCode != "" && !validQRCode(req.QrCode) {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "二维码必须使用本地上传路径或安全的http/https图片地址"})
|
||||
return
|
||||
}
|
||||
if req.URL != "" && req.QrCode != "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "联系方式不能同时设置链接和二维码"})
|
||||
return
|
||||
}
|
||||
if req.URL == "" && req.QrCode == "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "链接或二维码不能为空"})
|
||||
return
|
||||
}
|
||||
|
||||
ctx := c.Request.Context()
|
||||
@@ -83,13 +194,13 @@ func CreateContact(db *database.Database) gin.HandlerFunc {
|
||||
}
|
||||
|
||||
c.JSON(http.StatusOK, gin.H{
|
||||
"id": contact.ID,
|
||||
"type": contact.Type,
|
||||
"icon": contact.Icon,
|
||||
"url": contact.URL,
|
||||
"qrCode": contact.QrCode,
|
||||
"id": contact.ID,
|
||||
"type": contact.Type,
|
||||
"icon": contact.Icon,
|
||||
"url": contact.URL,
|
||||
"qrCode": contact.QrCode,
|
||||
"hoverColor": contact.HoverColor,
|
||||
"sortOrder": contact.SortOrder,
|
||||
"sortOrder": contact.SortOrder,
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -103,8 +214,8 @@ func UpdateContact(db *database.Database) gin.HandlerFunc {
|
||||
}
|
||||
|
||||
var req struct {
|
||||
Type string `json:"type"`
|
||||
Icon string `json:"icon"`
|
||||
Type string `json:"type" binding:"required"`
|
||||
Icon string `json:"icon" binding:"required"`
|
||||
URL string `json:"url"`
|
||||
QrCode string `json:"qrCode"`
|
||||
HoverColor string `json:"hoverColor"`
|
||||
@@ -115,48 +226,72 @@ func UpdateContact(db *database.Database) gin.HandlerFunc {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": err.Error()})
|
||||
return
|
||||
}
|
||||
req.Type = strings.TrimSpace(req.Type)
|
||||
req.Icon = strings.TrimSpace(req.Icon)
|
||||
req.URL = strings.TrimSpace(req.URL)
|
||||
req.QrCode = strings.TrimSpace(req.QrCode)
|
||||
req.HoverColor = strings.TrimSpace(req.HoverColor)
|
||||
if req.Type == "" || req.Icon == "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "类型和图标不能为空"})
|
||||
return
|
||||
}
|
||||
|
||||
// 验证Email类型必须是mailto格式
|
||||
if req.Type == "Email" && req.URL != "" {
|
||||
if len(req.URL) < 7 || req.URL[:7] != "mailto:" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "Email URL必须是mailto:格式"})
|
||||
return
|
||||
}
|
||||
if req.URL != "" && !validContactURL(req.URL) {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "联系方式URL必须是有效的http/https/mailto/tel地址"})
|
||||
return
|
||||
}
|
||||
if req.QrCode != "" && !validQRCode(req.QrCode) {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "二维码必须使用本地上传路径或安全的http/https图片地址"})
|
||||
return
|
||||
}
|
||||
if req.URL != "" && req.QrCode != "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "联系方式不能同时设置链接和二维码"})
|
||||
return
|
||||
}
|
||||
if req.URL == "" && req.QrCode == "" {
|
||||
c.JSON(http.StatusBadRequest, gin.H{"error": "链接或二维码不能为空"})
|
||||
return
|
||||
}
|
||||
|
||||
ctx := c.Request.Context()
|
||||
update := db.Client.Contact.UpdateOneID(id)
|
||||
if req.Type != "" {
|
||||
update.SetType(req.Type)
|
||||
}
|
||||
if req.Icon != "" {
|
||||
update.SetIcon(req.Icon)
|
||||
}
|
||||
update.SetType(req.Type)
|
||||
update.SetIcon(req.Icon)
|
||||
if req.URL != "" {
|
||||
update.SetURL(req.URL)
|
||||
} else {
|
||||
update.ClearURL()
|
||||
}
|
||||
if req.QrCode != "" {
|
||||
update.SetQrCode(req.QrCode)
|
||||
} else {
|
||||
update.ClearQrCode()
|
||||
}
|
||||
if req.HoverColor != "" {
|
||||
update.SetHoverColor(req.HoverColor)
|
||||
} else {
|
||||
update.ClearHoverColor()
|
||||
}
|
||||
update.SetSortOrder(req.SortOrder)
|
||||
|
||||
contact, 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": contact.ID,
|
||||
"type": contact.Type,
|
||||
"icon": contact.Icon,
|
||||
"url": contact.URL,
|
||||
"qrCode": contact.QrCode,
|
||||
"id": contact.ID,
|
||||
"type": contact.Type,
|
||||
"icon": contact.Icon,
|
||||
"url": contact.URL,
|
||||
"qrCode": contact.QrCode,
|
||||
"hoverColor": contact.HoverColor,
|
||||
"sortOrder": contact.SortOrder,
|
||||
"sortOrder": contact.SortOrder,
|
||||
})
|
||||
}
|
||||
}
|
||||
@@ -171,6 +306,10 @@ func DeleteContact(db *database.Database) gin.HandlerFunc {
|
||||
|
||||
ctx := c.Request.Context()
|
||||
if err := db.Client.Contact.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
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user