148 lines
4.1 KiB
Go
148 lines
4.1 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, 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
|
||
}
|