fix: BMP头解析、移动语义与空文件处理
- probe/bmp: 兼容 BITMAPCOREHEADER(12B) 等 info 头变体,按 Size 字段分支解析 - mover: moveOne 返回三态结果,复制成功但删源失败时明确告警,不再静默视为完全成功 - scanner: 跳过 0 字节文件,避免空文件进入 MIME/ffprobe 探测
This commit is contained in:
+17
-7
@@ -15,6 +15,15 @@ import (
|
||||
// Mover 文件移动器
|
||||
type Mover struct{}
|
||||
|
||||
// moveResult 移动结果状态
|
||||
type moveResult int
|
||||
|
||||
const (
|
||||
moveOK moveResult = iota // 移动成功(rename 或 copy+delete 完整完成)
|
||||
movePartial // 复制成功但删除源失败,源与目标同时存在
|
||||
moveFailed // 移动/复制失败
|
||||
)
|
||||
|
||||
// MoveToDir 将一批文件移动到目标目录(目标文件名使用 uuid 前缀防止同名覆盖)。
|
||||
// 返回移动成功的 源路径 → 目标目录 映射,供记录去向使用。
|
||||
func (m *Mover) MoveToDir(filePaths []string, targetDir string) map[string]string {
|
||||
@@ -32,7 +41,8 @@ func (m *Mover) MoveToDir(filePaths []string, targetDir string) map[string]strin
|
||||
bar.SetGraph(">")
|
||||
total := len(filePaths)
|
||||
for i, src := range filePaths {
|
||||
if moveOne(src, targetDir) {
|
||||
switch moveOne(src, targetDir) {
|
||||
case moveOK, movePartial:
|
||||
moved[src] = targetDir
|
||||
}
|
||||
bar.PrintBar(percent(i+1, total))
|
||||
@@ -42,21 +52,21 @@ func (m *Mover) MoveToDir(filePaths []string, targetDir string) map[string]strin
|
||||
}
|
||||
|
||||
// moveOne 移动单个文件;跨文件系统时 fallback 为 copy+delete
|
||||
func moveOne(src, targetDir string) bool {
|
||||
func moveOne(src, targetDir string) moveResult {
|
||||
fileName := strings.ReplaceAll(uuid.NewV4().String(), "-", "") + "_" + filepath.Base(src)
|
||||
target := filepath.Join(targetDir, fileName)
|
||||
if err := os.Rename(src, target); err == nil {
|
||||
return true
|
||||
return moveOK
|
||||
}
|
||||
if err := copyFile(src, target); err != nil {
|
||||
fmt.Printf("=== 复制文件失败, 源: %s, 目标: %s, 错误: %v\n", src, target, err)
|
||||
return false
|
||||
return moveFailed
|
||||
}
|
||||
if err := os.Remove(src); err != nil {
|
||||
fmt.Printf("=== 删除源文件失败: %s, 错误: %v\n", src, err)
|
||||
// 复制成功但删除失败,文件已存在于两处,视为移动成功
|
||||
fmt.Printf("=== 警告: 复制成功但删除源文件失败, 源与目标同时存在, 源: %s, 错误: %v\n", src, err)
|
||||
return movePartial
|
||||
}
|
||||
return true
|
||||
return moveOK
|
||||
}
|
||||
|
||||
// copyFile 逐块复制文件;失败时清理已写入的目标文件,避免残留半截文件
|
||||
|
||||
+31
-15
@@ -6,7 +6,8 @@ import (
|
||||
"os"
|
||||
)
|
||||
|
||||
// probeBmp 读取 BMP 文件头中的宽高;自顶向下 BMP 的高度为负值,取绝对值
|
||||
// probeBmp 读取 BMP 文件头中的宽高;自顶向下 BMP 的高度为负值,取绝对值。
|
||||
// 兼容常见 info 头:BITMAPCOREHEADER(12B, OS/2)、BITMAPINFOHEADER(40B) 及更长的 V4/V5 变体。
|
||||
func probeBmp(filePath string) (*ProbeResult, error) {
|
||||
f, err := os.Open(filePath)
|
||||
if err != nil {
|
||||
@@ -27,24 +28,39 @@ func probeBmp(filePath string) (*ProbeResult, error) {
|
||||
if string(fileHeader.FileType[:]) != "BM" {
|
||||
return nil, fmt.Errorf("not a valid BMP file")
|
||||
}
|
||||
var infoHeader struct {
|
||||
Size uint32
|
||||
Width int32
|
||||
Height int32
|
||||
Planes uint16
|
||||
BitsPerPixel uint16
|
||||
Compression uint32
|
||||
SizeImage uint32
|
||||
XPelsPerMeter int32
|
||||
YPelsPerMeter int32
|
||||
}
|
||||
if err := binary.Read(f, binary.LittleEndian, &infoHeader); err != nil {
|
||||
|
||||
var infoSize uint32
|
||||
if err := binary.Read(f, binary.LittleEndian, &infoSize); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &ProbeResult{Width: abs(infoHeader.Width), Height: abs(infoHeader.Height)}, nil
|
||||
|
||||
switch {
|
||||
case infoSize >= 40:
|
||||
// BITMAPINFOHEADER(40B)/BITMAPV4HEADER(108B)/BITMAPV5HEADER(124B)/OS2(64B):宽高为 int32
|
||||
var width, height int32
|
||||
if err := binary.Read(f, binary.LittleEndian, &width); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := binary.Read(f, binary.LittleEndian, &height); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &ProbeResult{Width: abs32(width), Height: abs32(height)}, nil
|
||||
case infoSize >= 12:
|
||||
// BITMAPCOREHEADER(12B, OS/2 1.x):宽高为 uint16
|
||||
var width, height uint16
|
||||
if err := binary.Read(f, binary.LittleEndian, &width); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := binary.Read(f, binary.LittleEndian, &height); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &ProbeResult{Width: int(width), Height: int(height)}, nil
|
||||
default:
|
||||
return nil, fmt.Errorf("unsupported BMP info header size: %d", infoSize)
|
||||
}
|
||||
}
|
||||
|
||||
func abs(v int32) int {
|
||||
func abs32(v int32) int {
|
||||
if v < 0 {
|
||||
return int(-v)
|
||||
}
|
||||
|
||||
@@ -55,6 +55,10 @@ func Scan(opts Options) (*Result, error) {
|
||||
return nil
|
||||
}
|
||||
}
|
||||
if info.Size() == 0 {
|
||||
// 空文件不可能是媒体,直接跳过,避免无谓的 MIME/ffprobe 探测
|
||||
return nil
|
||||
}
|
||||
res.Files = append(res.Files, File{
|
||||
Path: currentPath,
|
||||
Name: info.Name(),
|
||||
|
||||
Reference in New Issue
Block a user