94 lines
2.6 KiB
Go
94 lines
2.6 KiB
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, 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
|
|||
|
|
}
|