media-picker/core/file_scanner.go

package core

import (
	"OdMediaPicker/util"
	"OdMediaPicker/vars"
	"fmt"
	"github.com/redmask-hb/GoSimplePrint/goPrint"
	"log"
	"os"
	"path"
	"path/filepath"
	"strings"
)

type FileScanner struct {
}

func (FileScanner) DoScan(rootDir string) {
	fmt.Printf("=== 开始扫描文件 \n")
	if err := filepath.Walk(rootDir, visitFile); err != nil {
		log.Fatal(err)
	}
	doReadFileMimeInfo()
}

func doReadFileMimeInfo() {
	total := len(vars.GlobalFilePathList)
	fmt.Printf("=== 文件总数: %d \n", total)
	// 扫描文件mime信息
	var count = 0
	bar := goPrint.NewBar(100)
	bar.SetNotice("=== 扫描文件Mime：")
	bar.SetGraph(">")
	bar.SetNoticeColor(goPrint.FontColor.Red)
	for _, currentPath := range vars.GlobalFilePathList {
		// mime
		vars.GlobalFilePath2MimeInfoMap[currentPath] = util.ReadFileMimeInfo(currentPath).String()
		count = count + 1
		bar.PrintBar(util.CalcPercentage(count, total))
	}
	bar.PrintEnd("=== Finish")
}

func (FileScanner) DoFilter() {
	total := len(vars.GlobalFilePathList)
	var count = 0
	bar := goPrint.NewBar(100)
	bar.SetNotice("=== 过滤已支持的媒体：")
	bar.SetGraph(">")
	for _, globalFilePath := range vars.GlobalFilePathList {
		fileMime := vars.GlobalFilePath2MimeInfoMap[globalFilePath]
		count = count + 1
		if strings.Contains(fileMime, "video/") { // 视频
			vars.GlobalVideoPathList = append(vars.GlobalVideoPathList, globalFilePath)
			bar.PrintBar(util.CalcPercentage(count, total))
			continue
		}
		// mime格式为application/octet-stream的视频
		ext := path.Ext(globalFilePath)
		if isSupportVideo(ext) {
			vars.GlobalVideoPathList = append(vars.GlobalVideoPathList, globalFilePath)
			bar.PrintBar(util.CalcPercentage(count, total))
			continue
		}
		if strings.Contains(fileMime, "image/") { // 图片
			vars.GlobalImagePathList = append(vars.GlobalImagePathList, globalFilePath)
			bar.PrintBar(util.CalcPercentage(count, total))
			continue
		}
		if isSupportImage(ext) {
			vars.GlobalImagePathList = append(vars.GlobalImagePathList, globalFilePath)
			bar.PrintBar(util.CalcPercentage(count, total))
			continue
		}
		// 其他的文件不处理
	}
	bar.PrintEnd("=== Finish")
}
func countFiles(dir string) (int, error) {
	var count int
	err := filepath.Walk(dir, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		if !info.IsDir() {
			count++
		}
		return nil
	})
	return count, err
}

func (s FileScanner) DoPickerDir(dir string, fileCnt int) {
	if err := filepath.Walk(dir, visitDir); err != nil {
		log.Fatal(err)
	}
	for _, s2 := range vars.GlobalSubDirPathList {
		// 是整理过的文件夹
		if strings.Contains(s2, "_图片_") {
			cnt, err := countFiles(s2)
			// 满足文件数量
			if err == nil && cnt >= fileCnt {
				vars.GlobalNeedMoveSubDirPathList = append(vars.GlobalNeedMoveSubDirPathList, s2)
			} else {
				vars.GlobalNeedReScanSubDirPathList = append(vars.GlobalNeedReScanSubDirPathList, s2)
			}
		}
	}
	pathSeparator := string(os.PathSeparator)
	pickerOkDir := dir + pathSeparator + "PickerOk"
	pickerRescanDir := dir + pathSeparator + "PickerRescanDir"
	util.CreateDir(pickerOkDir)
	util.CreateDir(pickerRescanDir)
	for _, originPath := range vars.GlobalNeedMoveSubDirPathList {
		targetPath := pickerOkDir + pathSeparator + filepath.Base(originPath)
		os.Rename(originPath, targetPath)
	}
	for _, originPath := range vars.GlobalNeedReScanSubDirPathList {
		targetPath := pickerRescanDir + pathSeparator + filepath.Base(originPath)
		os.Rename(originPath, targetPath)
	}
}

func visitDir(currentPath string, info os.FileInfo, err error) error {
	if err != nil {
		return err // 如果有错误，直接返回
	}
	if info.IsDir() {
		vars.GlobalSubDirPathList = append(vars.GlobalSubDirPathList, currentPath)
	}
	return nil
}

// 定义walkFn回调函数visit
func visitFile(currentPath string, info os.FileInfo, err error) error {
	if err != nil {
		return err // 如果有错误，直接返回
	}
	if !info.IsDir() {
		vars.GlobalFilePathList = append(vars.GlobalFilePathList, currentPath)
		// filename, include ext
		vars.GlobalFilePath2FileNameMap[currentPath] = filepath.Base(currentPath)
		// file ext
		vars.GlobalFilePath2FileExtMap[currentPath] = path.Ext(currentPath)
	}
	return nil
}