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(),