82 lines
2.1 KiB
Go
82 lines
2.1 KiB
Go
package handler
|
||
|
||
import (
|
||
"net/http"
|
||
"strings"
|
||
"unicode/utf8"
|
||
|
||
"soul-api/internal/database"
|
||
"soul-api/internal/model"
|
||
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// escapeLike 转义 LIKE 中的 % _ \，防止注入与通配符滥用
|
||
func escapeLike(s string) string {
|
||
s = strings.ReplaceAll(s, "\\", "\\\\")
|
||
s = strings.ReplaceAll(s, "%", "\\%")
|
||
s = strings.ReplaceAll(s, "_", "\\_")
|
||
return s
|
||
}
|
||
|
||
// SearchGet GET /api/search?q= 从 chapters 表搜索（GORM，参数化）
|
||
func SearchGet(c *gin.Context) {
|
||
q := strings.TrimSpace(c.Query("q"))
|
||
if q == "" {
|
||
c.JSON(http.StatusBadRequest, gin.H{"success": false, "error": "请输入搜索关键词"})
|
||
return
|
||
}
|
||
pattern := "%" + escapeLike(q) + "%"
|
||
var list []model.Chapter
|
||
err := database.DB().Model(&model.Chapter{}).
|
||
Where("section_title LIKE ? OR content LIKE ?", pattern, pattern).
|
||
Order("sort_order ASC, id ASC").
|
||
Limit(50).
|
||
Find(&list).Error
|
||
if err != nil {
|
||
c.JSON(http.StatusOK, gin.H{"success": true, "data": gin.H{"keyword": q, "total": 0, "results": []interface{}{}}})
|
||
return
|
||
}
|
||
lowerQ := strings.ToLower(q)
|
||
results := make([]gin.H, 0, len(list))
|
||
for _, ch := range list {
|
||
matchType := "content"
|
||
score := 5
|
||
if strings.Contains(strings.ToLower(ch.SectionTitle), lowerQ) {
|
||
matchType = "title"
|
||
score = 10
|
||
}
|
||
snippet := ""
|
||
pos := strings.Index(strings.ToLower(ch.Content), lowerQ)
|
||
if pos >= 0 && len(ch.Content) > 0 {
|
||
start := pos - 50
|
||
if start < 0 {
|
||
start = 0
|
||
}
|
||
end := pos + utf8.RuneCountInString(q) + 50
|
||
if end > len(ch.Content) {
|
||
end = len(ch.Content)
|
||
}
|
||
snippet = ch.Content[start:end]
|
||
if start > 0 {
|
||
snippet = "..." + snippet
|
||
}
|
||
if end < len(ch.Content) {
|
||
snippet = snippet + "..."
|
||
}
|
||
}
|
||
price := 1.0
|
||
if ch.Price != nil {
|
||
price = *ch.Price
|
||
}
|
||
results = append(results, gin.H{
|
||
"id": ch.ID, "title": ch.SectionTitle, "partTitle": ch.PartTitle, "chapterTitle": ch.ChapterTitle,
|
||
"price": price, "isFree": ch.IsFree, "matchType": matchType, "score": score, "snippet": snippet,
|
||
})
|
||
}
|
||
c.JSON(http.StatusOK, gin.H{
|
||
"success": true,
|
||
"data": gin.H{"keyword": q, "total": len(results), "results": results},
|
||
})
|
||
}
|