112 lines
3.5 KiB
Go
112 lines
3.5 KiB
Go
package util
|
|
|
|
import (
|
|
"github.com/gabriel-vasile/mimetype"
|
|
"io/ioutil"
|
|
"log"
|
|
"os"
|
|
"path/filepath"
|
|
)
|
|
|
|
// CreateDir 创建目录
|
|
func CreateDir(dirPath string) bool {
|
|
err := os.Mkdir(dirPath, 0755)
|
|
return err == nil
|
|
}
|
|
|
|
// GetFileDirectory 获取文件所在文件夹
|
|
func GetFileDirectory(filePath string) string {
|
|
directoryPath, _ := filepath.Split(filePath)
|
|
return directoryPath
|
|
}
|
|
|
|
// ReadFileMimeInfo 获取文件mime信息
|
|
func ReadFileMimeInfo(filepath string) *mimetype.MIME {
|
|
mt, err := mimetype.DetectFile(filepath)
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
return mt
|
|
}
|
|
|
|
// CheckFileIsExist 判断文件是否存在，存在返回 true，不存在返回false
|
|
func CheckFileIsExist(filename string) bool {
|
|
var exist = true
|
|
if _, err := os.Stat(filename); os.IsNotExist(err) {
|
|
exist = false
|
|
}
|
|
return exist
|
|
}
|
|
|
|
// WriteByteArraysToFile 写字节数组到文件
|
|
func WriteByteArraysToFile(content []byte, filename string) error {
|
|
return ioutil.WriteFile(filename, content, 0777)
|
|
}
|
|
|
|
//// 移动文件到根目录
|
|
//func renameFile(rootDir string, modelType string, videoList []string, pathSeparator string) {
|
|
// total := len(videoList)
|
|
// var count = 0
|
|
// bar := goPrint.NewBar(100)
|
|
// bar.SetNotice("=== 重命名文件：")
|
|
// bar.SetGraph(">")
|
|
// for _, videoFilePath := range videoList {
|
|
// wh := videoPath2WidthHeightTagMap[videoFilePath]
|
|
// fileName := vars.GlobalFilePath2FileNameMap[videoFilePath]
|
|
// if strings.Contains(fileName, videoTag) { // 处理过了
|
|
// fileNames := strings.Split(fileName, videoTag)
|
|
// if len(fileNames) == 2 {
|
|
// fileName = fileNames[1]
|
|
// targetFilePath := rootDir + pathSeparator + "[" + videoPath2DurationMap[videoFilePath] + "]" + modelType + wh + videoTag + fileName
|
|
// err := os.Rename(videoFilePath, targetFilePath)
|
|
// if err != nil {
|
|
// fmt.Printf("=== 重命名异常: %s \n", videoFilePath)
|
|
// }
|
|
// }
|
|
// } else {
|
|
// targetFilePath := rootDir + pathSeparator + "[" + videoPath2DurationMap[videoFilePath] + "]" + modelType + wh + videoTag + " - " + fileName
|
|
// err := os.Rename(videoFilePath, targetFilePath)
|
|
// if err != nil {
|
|
// fmt.Printf("=== 重命名异常: %s \n", videoFilePath)
|
|
// }
|
|
// }
|
|
// count = count + 1
|
|
// bar.PrintBar(util.CalcPercentage(count, total))
|
|
// }
|
|
// bar.PrintEnd("=== Finish")
|
|
//}
|
|
//
|
|
//// 移动文件到原目录
|
|
//func renameFileV2(modelType string, videoList []string) {
|
|
// total := len(videoList)
|
|
// var count = 0
|
|
// bar := goPrint.NewBar(100)
|
|
// bar.SetNotice("=== 重命名文件：")
|
|
// bar.SetGraph(">")
|
|
// for _, videoFilePath := range videoList {
|
|
// wh := videoPath2WidthHeightTagMap[videoFilePath]
|
|
// fileName := vars.GlobalFilePath2FileNameMap[videoFilePath]
|
|
// filePath := util.GetFileDirectory(videoFilePath)
|
|
// if strings.Contains(fileName, videoTag) { // 处理过了
|
|
// fileNames := strings.Split(fileName, videoTag)
|
|
// if len(fileNames) == 2 {
|
|
// fileName = fileNames[1]
|
|
// targetFilePath := filePath + "[" + videoPath2DurationMap[videoFilePath] + "]" + modelType + wh + videoTag + fileName
|
|
// err := os.Rename(videoFilePath, targetFilePath)
|
|
// if err != nil {
|
|
// fmt.Printf("=== 重命名异常: %s \n", videoFilePath)
|
|
// }
|
|
// }
|
|
// } else {
|
|
// targetFilePath := filePath + "[" + videoPath2DurationMap[videoFilePath] + "]" + modelType + wh + videoTag + " - " + fileName
|
|
// err := os.Rename(videoFilePath, targetFilePath)
|
|
// if err != nil {
|
|
// fmt.Printf("=== 重命名异常: %s \n", videoFilePath)
|
|
// }
|
|
// }
|
|
// count = count + 1
|
|
// bar.PrintBar(util.CalcPercentage(count, total))
|
|
// }
|
|
// bar.PrintEnd("=== Finish")
|
|
//}
|