实现封面缓存

internal/util/ffmpeg.go

@@ -1,20 +1,30 @@
 package util
 
 import (
+	"butterfliu/config"
+	"os"
 	"os/exec"
+	"path"
 	"strconv"
 )
 
-func ExtractCover(filePath string, id int) bool {
-	if _, err := exec.LookPath("ffmpeg"); err != nil {
-		return false
+func ExtractCover(filePath string, id int) (string, error) {
+	config := config.LoadConfig()
+	coverPath := path.Join(config.GetCachePath("cover"), strconv.Itoa(id)+".jpg")
+
+	_, err := os.Stat(coverPath)
+	if err == nil {
+		return coverPath, nil
+	}
+
+	if _, err := exec.LookPath("ffmpeg"); err != nil {
+		return "", err
 	}
-	coverPath := "cache/cover/" + strconv.Itoa(id) + ".jpg"
 	cmd := exec.Command("ffmpeg", "-i", filePath, "-an", "-vcodec", "mjpeg", "-q:v", "5", "-f", "image2", "-y", coverPath)
 	if err := cmd.Run(); err != nil {
-		return false
+		return "", err
 	}
 
-	return true
+	return coverPath, nil
 
 }