odboy
4da4b738e6
- 删除 vars 全局状态，按职责拆分 internal/ 包：scanner/probe/classify/organize/mover/store/digest - 数据驱动归类规则替代图片/视频重复分支，store 改为结构体实例 - build.sh 按系统类型编译：Windows 用 upx 压缩，macOS/Linux 不压缩 - 忽略目录拆分：图片_已忽略 / 视频_已忽略 - 输出构建产物加入 .gitignore
77 lines
1.8 KiB
Go
77 lines
1.8 KiB
Go
package probe
|
|
|
|
import (
|
|
"fmt"
|
|
"image"
|
|
"os"
|
|
"path/filepath"
|
|
"strings"
|
|
)
|
|
|
|
// supportImageTypes 支持的标准图片扩展名（大小写不敏感）
|
|
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/psb）走专用解码器
|
|
// 2. 其余支持格式走 Go 标准库 image.DecodeConfig
|
|
// 3. 标准库失败时用 ffprobe 兜底（ffmpeg 支持的图片格式更广，如 tiff/svg 等）
|
|
//
|
|
// ffprobePath 为空表示不启用 ffprobe 兜底；扩展名不属于支持格式时返回错误。
|
|
func ProbeImage(filePath, ffprobePath string) (*ProbeResult, error) {
|
|
ext := strings.ToLower(filepath.Ext(filePath))
|
|
switch ext {
|
|
case ".webp":
|
|
return probeWebp(filePath)
|
|
case ".bmp":
|
|
return probeBmp(filePath)
|
|
case ".psd", ".psb":
|
|
return probePsd(filePath)
|
|
}
|
|
if !IsSupportImage(ext) {
|
|
return nil, fmt.Errorf("不支持的图片格式: %s", ext)
|
|
}
|
|
if w, h, err := decodeConfig(filePath); err == nil {
|
|
return &ProbeResult{Width: w, Height: h}, nil
|
|
}
|
|
if ffprobePath != "" {
|
|
if res, err := ProbeMediaFile(ffprobePath, filePath); err == nil && res.HasVideo {
|
|
return res, nil
|
|
}
|
|
}
|
|
return nil, fmt.Errorf("无法读取图片尺寸: %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
|
|
}
|