odboy/kenaito-media-picker
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: 目录层级变更

Browse Source
This commit is contained in:
odboy
2025-09-24 21:19:55 +08:00
parent 7792f9281c
commit 72c9d67d9b
12 changed files with 156 additions and 317 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/store/store.go
+26 -28
View File
@@ -31,31 +31,29 @@ func init() {
// MediaInfo 一条媒体文件的记录：文件本身的信息——路径、名字、大小、类型、宽高、编码、格式等。
type MediaInfo struct {
Id int64 `xorm:"pk autoincr 'id'" json:"id"`
FileMd5 string `xorm:"char(32) index 'file_md5'" json:"fileMd5"` // 文件 MD5，内容相同则相同，用来判断是不是同一个文件
FilePath string `xorm:"text 'file_path'" json:"filePath"` // 扫描时的源路径（在哪发现的）
FileName string `xorm:"varchar(255) 'file_name'" json:"fileName"` // 文件名（含扩展名）
FileExt string `xorm:"varchar(32) 'file_ext'" json:"fileExt"` // 扩展名（如 .jpg）
FileSize int64 `xorm:"bigint 'file_size'" json:"fileSize"` // 文件大小（字节）
MimeType string `xorm:"varchar(128) 'mime_type'" json:"mimeType"` // MIME 类型（如 image/jpeg）
MediaType string `xorm:"varchar(16) index 'media_type'" json:"mediaType"` // 大类型：image / video / audio
Width int `xorm:"int index 'width'" json:"width"` // 宽（像素）
Height int `xorm:"int index 'height'" json:"height"` // 高（像素）
Duration float64 `xorm:"real 'duration'" json:"duration"` // 时长（秒）
Codec string `xorm:"varchar(64) 'codec'" json:"codec"` // 编码，如 h264 / hevc
Format string `xorm:"varchar(64) 'format'" json:"format"` // 容器格式，如 mp4 / mkv
RawInfo string `xorm:"text 'raw_info'" json:"rawInfo"` // ffprobe 读出的原始信息（JSON），留底备用
HeadMd5 string `xorm:"char(32) index('idx_fp') 'head_md5'" json:"headMd5"` // 文件头 256KB 的 MD5（指纹用）
TailMd5 string `xorm:"char(32) index('idx_fp') 'tail_md5'" json:"tailMd5"` // 文件尾 256KB 的 MD5（指纹用）
NsfwScore float64 `xorm:"real 'nsfw_score'" json:"nsfwScore"` // NSFW 检测分 0~1（0 表示未检测或非图片）
NsfwThreshold float64 `xorm:"real 'nsfw_threshold'" json:"nsfwThreshold"` // NSFW 判定阈值
NsfwDetections string `xorm:"text 'nsfw_detections'" json:"nsfwDetections"` // NudeNet 检测详情 JSON（类别+分数+位置）
AnimeNsfwScore float64 `xorm:"real 'anime_nsfw_score'" json:"animeNsfwScore"` // 动漫 NSFW 检测分 0~1（0 表示未检测或非动漫）
AnimeNsfwThreshold float64 `xorm:"real 'anime_nsfw_threshold'" json:"animeNsfwThreshold"` // 动漫 NSFW 判定阈值
AnimeNsfwRatings string `xorm:"text 'anime_nsfw_ratings'" json:"animeNsfwRatings"` // WD Tagger 四个 rating 的 JSON（G/S/Q/E）
ManualNsfw string `xorm:"varchar(1) 'manual_nsfw'" json:"manualNsfw"` // 手动标记是否色情：Y/N，默认 N
CreatedAt time.Time `xorm:"created 'created_at'" json:"createdAt"` // 首次入库时间
UpdatedAt time.Time `xorm:"updated 'updated_at'" json:"updatedAt"` // 最近更新时间
Id int64 `xorm:"pk autoincr 'id'" json:"id"`
FileMd5 string `xorm:"char(32) index 'file_md5'" json:"fileMd5"` // 文件 MD5，内容相同则相同，用来判断是不是同一个文件
FilePath string `xorm:"text 'file_path'" json:"filePath"` // 扫描时的源路径（在哪发现的）
FileName string `xorm:"varchar(255) 'file_name'" json:"fileName"` // 文件名（含扩展名）
FileExt string `xorm:"varchar(32) 'file_ext'" json:"fileExt"` // 扩展名（如 .jpg）
FileSize int64 `xorm:"bigint 'file_size'" json:"fileSize"` // 文件大小（字节）
MimeType string `xorm:"varchar(128) 'mime_type'" json:"mimeType"` // MIME 类型（如 image/jpeg）
MediaType string `xorm:"varchar(16) index 'media_type'" json:"mediaType"` // 大类型：image / video / audio
Width int `xorm:"int index 'width'" json:"width"` // 宽（像素）
Height int `xorm:"int index 'height'" json:"height"` // 高（像素）
Duration float64 `xorm:"real 'duration'" json:"duration"` // 时长（秒）
Codec string `xorm:"varchar(64) 'codec'" json:"codec"` // 编码，如 h264 / hevc
Format string `xorm:"varchar(64) 'format'" json:"format"` // 容器格式，如 mp4 / mkv
RawInfo string `xorm:"text 'raw_info'" json:"rawInfo"` // ffprobe 读出的原始信息（JSON），留底备用
HeadMd5 string `xorm:"char(32) index('idx_fp') 'head_md5'" json:"headMd5"` // 文件头 256KB 的 MD5（指纹用）
TailMd5 string `xorm:"char(32) index('idx_fp') 'tail_md5'" json:"tailMd5"` // 文件尾 256KB 的 MD5（指纹用）
NsfwScore float64 `xorm:"real 'nsfw_score'" json:"nsfwScore"` // NSFW 检测分 0~1（0 表示未检测或非图片）
NsfwDetections string `xorm:"text 'nsfw_detections'" json:"nsfwDetections"` // NudeNet 检测详情 JSON（类别+分数+位置）
AnimeNsfwScore float64 `xorm:"real 'anime_nsfw_score'" json:"animeNsfwScore"` // 动漫 NSFW 检测分 0~1（0 表示未检测或非动漫）
AnimeNsfwRatings string `xorm:"text 'anime_nsfw_ratings'" json:"animeNsfwRatings"` // WD Tagger 四个 rating 的 JSON（G/S/Q/E）
ManualNsfw string `xorm:"varchar(1) 'manual_nsfw'" json:"manualNsfw"` // 手动标记是否色情：Y/N，默认 N
CreatedAt time.Time `xorm:"created 'created_at'" json:"createdAt"` // 首次入库时间
UpdatedAt time.Time `xorm:"updated 'updated_at'" json:"updatedAt"` // 最近更新时间
}
// Store 数据库访问的封装。整个程序共用一个实例，通过它读写数据库。
@@ -358,7 +356,7 @@ type MediaQuery struct {
Codec string // 编码
Format string // 容器格式
Resolution string // 分辨率档位（如 1080P / 5K / 普通 / 原图）
NsfwOnly bool // 只看色情图片（nsfw_score >= nsfw_threshold 且 threshold > 0）
NsfwOnly bool // 只看有检测分的图片（nsfw_score > 0 或 anime_nsfw_score > 0）
Page int // 第几页（从 1 开始）
PageSize int // 每页几条
SortField string // 按哪个字段排序：id / fileSize / width / duration / createdAt
@@ -457,9 +455,9 @@ func buildMediaCond(q MediaQuery) (string, []interface{}) {
}
}
}
// 色情筛选：只看检测分 >= 阈值的图片
// 色情筛选：只看有检测分的图片（nsfw_score > 0）
if q.NsfwOnly {
appendCond("nsfw_score >= nsfw_threshold AND nsfw_threshold > 0")
appendCond("nsfw_score > 0 OR anime_nsfw_score > 0")
}
if cond == "" {
return "", nil