odboy
8979c744a8
- 重构后端与前端，全程大白话注释（不懂程序也能看懂） - 移动阶段改为全局并行（MoveAll，工人数=CPU核心数） - 去掉记录每个文件搬去哪了（删除去向回填，解决写库卡顿） - 去掉重复文件标记与统计：重复判定改为 MD5+路径都相同才算，重复则原地忽略不移动；MD5 相同但路径不同仍照常搬 - 媒体库去掉归类目录列，看板去掉重复/归类分布统计
104 lines
2.9 KiB
Go
104 lines
2.9 KiB
Go
package probe
|
|
|
|
import (
|
|
"fmt"
|
|
"image"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// supportImageTypes 认识的图片扩展名（大小写不敏感）。
|
|
// 这些格式可以自己读出尺寸；列表之外的（如 tiff）在 ffprobe 可用时兜底探测。
|
|
var supportImageTypes = []string{
|
|
".bmp",
|
|
".gif",
|
|
".jpg",
|
|
".jpeg",
|
|
".jpe",
|
|
".png",
|
|
".webp",
|
|
".psb",
|
|
".jfif",
|
|
}
|
|
|
|
// IsSupportImage 判断扩展名是不是认识的图片格式。
|
|
func IsSupportImage(ext string) bool {
|
|
for _, t := range supportImageTypes {
|
|
if strings.EqualFold(t, ext) {
|
|
return true
|
|
}
|
|
}
|
|
return false
|
|
}
|
|
|
|
// ProbeImage 探测一张图片的尺寸（宽高）。
|
|
//
|
|
// 流程（从快到慢、从准到兜底）：
|
|
// 1. 特殊格式（WebP/BMP/PSD）用专门写的读取函数，它们标准库不一定认识；
|
|
// 2. 其余认识的格式用 Go 标准库读（最轻量）；
|
|
// 3. 上面都失败、或者格式不常见时，用 ffprobe 兜底（它认识的图片格式更多）。
|
|
//
|
|
// ffprobePath 为空表示不启用兜底；遇到不认识的格式就报"不支持"。
|
|
func ProbeImage(filePath, ffprobePath string) (*ProbeResult, error) {
|
|
ext := strings.ToLower(filepath.Ext(filePath))
|
|
// 先试试专门的读取函数
|
|
switch ext {
|
|
case ".webp":
|
|
if res, err := probeWebp(filePath); err == nil {
|
|
return res, nil
|
|
}
|
|
case ".bmp":
|
|
if res, err := probeBmp(filePath); err == nil {
|
|
return res, nil
|
|
}
|
|
case ".psd", ".psb":
|
|
if res, err := probePsd(filePath); err == nil {
|
|
return res, nil
|
|
}
|
|
}
|
|
// 不在认识列表里的格式：只能用 ffprobe 碰运气
|
|
if !IsSupportImage(ext) {
|
|
if res, err := probeByFFprobe(ffprobePath, filePath); err == nil {
|
|
return res, nil
|
|
}
|
|
return nil, fmt.Errorf("不支持的图片格式: %s", ext)
|
|
}
|
|
// 用标准库读宽高
|
|
if w, h, err := decodeConfig(filePath); err == nil {
|
|
return &ProbeResult{Width: w, Height: h}, nil
|
|
}
|
|
// 标准库读不出 → 用 ffprobe 兜底
|
|
if res, err := probeByFFprobe(ffprobePath, filePath); err == nil {
|
|
return res, nil
|
|
}
|
|
return nil, fmt.Errorf("无法读取图片尺寸: %s", filePath)
|
|
}
|
|
|
|
// probeByFFprobe 用 ffprobe 兜底读图片尺寸。
|
|
// ffprobePath 为空或探测失败时返回错误。
|
|
func probeByFFprobe(ffprobePath, filePath string) (*ProbeResult, error) {
|
|
if ffprobePath == "" {
|
|
return nil, fmt.Errorf("未启用 ffprobe 兜底")
|
|
}
|
|
// 图片在 ffprobe 眼里会被当成"只有视频流"的视频
|
|
if res, err := ProbeMediaFile(ffprobePath, filePath); err == nil && res.HasVideo {
|
|
return res, nil
|
|
}
|
|
return nil, fmt.Errorf("ffprobe 兜底失败: %s", filePath)
|
|
}
|
|
|
|
// decodeConfig 用 Go 标准库读图片宽高（只解码头部配置，不加载整张图，很快）。
|
|
func decodeConfig(filePath string) (int, int, error) {
|
|
f, err := os.Open(filePath)
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
defer f.Close()
|
|
cfg, _, err := image.DecodeConfig(f)
|
|
if err != nil {
|
|
return 0, 0, err
|
|
}
|
|
return cfg.Width, cfg.Height, nil
|
|
}
|