upd

core/file_scanner.go

@@ -0,0 +1,93 @@
+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, visit); 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")
+}
+
+// 定义walkFn回调函数visit
+func visit(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
+}