92 lines
2.6 KiB
Go
92 lines
2.6 KiB
Go
package handler
|
||
|
||
import (
|
||
"net/http"
|
||
"strings"
|
||
|
||
"backend/internal/repo"
|
||
"github.com/gin-gonic/gin"
|
||
)
|
||
|
||
// BannedWordsHandler — admin 违禁词管理: list / add / delete prompt blocklist
|
||
// entries. The generation path (V1Service.checkBannedPrompt) enforces them.
|
||
type BannedWordsHandler struct {
|
||
words *repo.BannedWordRepository
|
||
}
|
||
|
||
func NewBannedWordsHandler(words *repo.BannedWordRepository) *BannedWordsHandler {
|
||
return &BannedWordsHandler{words: words}
|
||
}
|
||
|
||
func (h *BannedWordsHandler) List(c *gin.Context) {
|
||
items, err := h.words.List(c.Request.Context())
|
||
if err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"detail": "failed to load banned words"})
|
||
return
|
||
}
|
||
out := make([]gin.H, 0, len(items))
|
||
for _, w := range items {
|
||
out = append(out, gin.H{
|
||
"id": w.ID,
|
||
"word": w.Word,
|
||
"hits": w.Hits,
|
||
"created_at": w.CreatedAt,
|
||
})
|
||
}
|
||
c.JSON(http.StatusOK, gin.H{"data": out})
|
||
}
|
||
|
||
func (h *BannedWordsHandler) Create(c *gin.Context) {
|
||
var body struct {
|
||
Word string `json:"word"`
|
||
}
|
||
if err := c.ShouldBindJSON(&body); err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"detail": "invalid request body"})
|
||
return
|
||
}
|
||
item, err := h.words.Create(c.Request.Context(), body.Word)
|
||
if err != nil {
|
||
c.JSON(http.StatusBadRequest, gin.H{"detail": err.Error()})
|
||
return
|
||
}
|
||
c.JSON(http.StatusOK, gin.H{"data": gin.H{"id": item.ID, "word": item.Word, "hits": item.Hits, "created_at": item.CreatedAt}})
|
||
}
|
||
|
||
// Import bulk-adds words from a free-form text blob — split on newlines,
|
||
// commas (英文/中文), 顿号 and semicolons — skipping blanks and existing entries.
|
||
func (h *BannedWordsHandler) Import(c *gin.Context) {
|
||
var body struct {
|
||
Text string `json:"text"`
|
||
}
|
||
if err := c.ShouldBindJSON(&body); err != nil || strings.TrimSpace(body.Text) == "" {
|
||
c.JSON(http.StatusBadRequest, gin.H{"detail": "请提供要导入的违禁词"})
|
||
return
|
||
}
|
||
words := strings.FieldsFunc(body.Text, func(r rune) bool {
|
||
switch r {
|
||
case '\n', '\r', ',', '，', '、', ';', '；':
|
||
return true
|
||
}
|
||
return false
|
||
})
|
||
added, skipped, err := h.words.BulkCreate(c.Request.Context(), words)
|
||
if err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"detail": "导入失败"})
|
||
return
|
||
}
|
||
c.JSON(http.StatusOK, gin.H{"added": added, "skipped": skipped})
|
||
}
|
||
|
||
func (h *BannedWordsHandler) Delete(c *gin.Context) {
|
||
n, err := h.words.Delete(c.Request.Context(), c.Param("id"))
|
||
if err != nil {
|
||
c.JSON(http.StatusInternalServerError, gin.H{"detail": "delete failed"})
|
||
return
|
||
}
|
||
if n == 0 {
|
||
c.JSON(http.StatusNotFound, gin.H{"detail": "not found"})
|
||
return
|
||
}
|
||
c.JSON(http.StatusOK, gin.H{"ok": true})
|
||
}
|