From 7e49accf2bfa8ad865decab4eb98d5b97c855805 Mon Sep 17 00:00:00 2001
From: Odboy <tianjun@odboy.cn>
Date: Tue, 15 Apr 2025 21:05:26 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20BMP=E5=A4=B4=E8=A7=A3=E6=9E=90=E3=80=81?=
 =?UTF-8?q?=E7=A7=BB=E5=8A=A8=E8=AF=AD=E4=B9=89=E4=B8=8E=E7=A9=BA=E6=96=87?=
 =?UTF-8?q?=E4=BB=B6=E5=A4=84=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

- probe/bmp: 兼容 BITMAPCOREHEADER(12B) 等 info 头变体，按 Size 字段分支解析
- mover: moveOne 返回三态结果，复制成功但删源失败时明确告警，不再静默视为完全成功
- scanner: 跳过 0 字节文件，避免空文件进入 MIME/ffprobe 探测
---
 internal/mover/mover.go     | 24 +++++++++++++------
 internal/probe/bmp.go       | 46 +++++++++++++++++++++++++------------
 internal/scanner/scanner.go |  4 ++++
 3 files changed, 52 insertions(+), 22 deletions(-)

diff --git a/internal/mover/mover.go b/internal/mover/mover.go
index 832e91e..0059569 100644
--- a/internal/mover/mover.go
+++ b/internal/mover/mover.go
@@ -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 逐块复制文件；失败时清理已写入的目标文件，避免残留半截文件
diff --git a/internal/probe/bmp.go b/internal/probe/bmp.go
index f02c0e3..ca9dfb3 100644
--- a/internal/probe/bmp.go
+++ b/internal/probe/bmp.go
@@ -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)
 	}
diff --git a/internal/scanner/scanner.go b/internal/scanner/scanner.go
index 0c8610b..8a69bf5 100644
--- a/internal/scanner/scanner.go
+++ b/internal/scanner/scanner.go
@@ -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(),