feat: 色情识别结果可视化与日志;超长路径修复;旋转开关、明暗5档等

This commit is contained in:
2025-06-08 21:07:42 +08:00
parent c766a948da
commit d4f9dc362e
15 changed files with 477 additions and 34 deletions
+75
View File
@@ -0,0 +1,75 @@
// 色情识别相关的两个接口:查记录、看图片预览。
// 前端「色情识别」页面用它展示每次扫描判出的色情图片(检测分 + 路径 + 预览),
// 方便用户核对识别结果准不准。只读数据库 + 读本地图片文件,不干别的活。
package api
import (
"net/http"
"os"
"strings"
"github.com/gin-gonic/gin"
"kenaito-media-picker/internal/store"
)
// NsfwController 色情识别相关接口的"窗口"。
type NsfwController struct {
store *store.Store // 数据库(只读)
}
// NewNsfwController 创建色情识别控制器,注入数据库。
func NewNsfwController(st *store.Store) *NsfwController {
return &NsfwController{store: st}
}
// nsfwView 列表里展示的一条记录:路径用"整理后位置",没有就用源路径。
type nsfwView struct {
Id int64 `json:"id"` // 主键
FileName string `json:"fileName"` // 文件名
Score float64 `json:"score"` // 检测分 0~1
Threshold float64 `json:"threshold"` // 判定阈值
Path string `json:"path"` // 预览/打开用的完整路径
CreatedAt string `json:"createdAt"` // 检测时间
}
// List 处理 GET /api/nsfw/list:查全部色情识别记录(按检测分从高到低)。
func (c *NsfwController) List(ctx *gin.Context) {
list, err := c.store.ListNsfw()
if err != nil {
Error(ctx, http.StatusInternalServerError, "查询色情识别记录失败:"+err.Error())
return
}
views := make([]nsfwView, 0, len(list))
for _, r := range list {
path := r.MovedPath
if path == "" {
path = r.FilePath
}
views = append(views, nsfwView{
Id: r.Id,
FileName: r.FileName,
Score: r.Score,
Threshold: r.Threshold,
Path: path,
CreatedAt: r.CreatedAt.Format("2006-01-02 15:04:05"),
})
}
Success(ctx, map[string]interface{}{"list": views, "total": len(views)})
}
// Preview 处理 GET /api/nsfw/preview?path=xxx:把指定图片文件返回给浏览器做预览。
// 这是本地单机工具,直接按路径读文件;文件不存在/已被移动时返回 404。
func (c *NsfwController) Preview(ctx *gin.Context) {
path := strings.TrimSpace(ctx.Query("path"))
if path == "" {
Error(ctx, http.StatusBadRequest, "缺少 path 参数")
return
}
if _, err := os.Stat(path); err != nil {
Error(ctx, http.StatusNotFound, "图片不存在或已被移动: "+path)
return
}
// 交给 gin 读文件并带上正确的 Content-Type / 断点续传
ctx.File(path)
}
+5
View File
@@ -33,6 +33,11 @@ func NewRouter(st *store.Store) *gin.Engine {
apiGroup.POST("/media/list", mc.List)
apiGroup.POST("/media/stats", mc.Stats)
// 色情识别(查看检测分 + 图片预览)
nc := NewNsfwController(st)
apiGroup.GET("/nsfw/list", nc.List)
apiGroup.GET("/nsfw/preview", nc.Preview)
// 扫描整理(开始 / 中止 / 查进度 / 实时进度)
sc := NewScanController(service.NewScanService(st))
apiGroup.POST("/scan/start", sc.Start)