Add: get alternative avatar from directory

pkg/api/handle_avatar.go

@@ -1,11 +1,17 @@
 package api
 
 import (
+	"bytes"
 	"errors"
+	"io"
 	"log"
+	"msw-open-music/pkg/database"
 	"net/http"
+	"os"
 	"os/exec"
+	"path"
 	"strconv"
+	"strings"
 )
 
 func (api *API) HandelGetFileAvatar(w http.ResponseWriter, r *http.Request) {
@@ -33,14 +39,70 @@ func (api *API) HandelGetFileAvatar(w http.ResponseWriter, r *http.Request) {
 		return
 	}
 	log.Println("[api] Get avatar of file", path)
+	buff := make([]byte, 0)
+	cache := bytes.NewBuffer(buff)
 	cmd := exec.Command("ffmpeg", "-i", path, "-c:v", "libwebp_anim", "-update", "1", "-f", "image2pipe", "-")
+	cmd.Stdout = cache
+
+	err = cmd.Run()
+	if err != nil {
+		api.HandleGetAlternativeFileAvatar(w, r, file)
+		return
+	}
+
+	w.Header().Set("Content-Type", "image/webp")
+	io.Copy(w, cache)
+}
+
+func (api *API) HandleGetAlternativeFileAvatar(w http.ResponseWriter, r *http.Request, f *database.File) {
+	var err error
+	dir, err := f.Dir()
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+	log.Println("[api] Get alternative avatar in dir", dir)
+	files, err := os.ReadDir(dir)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+	avatar, err := findAvatarFile(files)
+	avatarPath := path.Join(dir, avatar)
+	if err != nil {
+		api.HandleError(w, r, err)
+		return
+	}
+	cmd := exec.Command("ffmpeg", "-i", avatarPath, "-c:v", "libwebp_anim", "-f", "image2pipe", "-")
 	cmd.Stdout = w
 	w.Header().Set("Content-Type", "image/webp")
 
 	err = cmd.Run()
 	if err != nil {
-		w.Header().Set("Content-Type", "application/json")
 		api.HandleError(w, r, err)
 		return
 	}
 }
+
+func findAvatarFile(files []os.DirEntry) (string, error) {
+	for _, file := range files {
+		if isAvatarType(file.Name()) {
+			return file.Name(), nil
+		}
+	}
+	return "", errors.New("Cannot find avatar file")
+}
+
+var avatarFileTypes = []string{
+	".jpg",
+	".png",
+}
+
+func isAvatarType(filename string) bool {
+	for _, t := range avatarFileTypes {
+		if strings.HasSuffix(strings.ToLower(filename), t) {
+			return true
+		}
+	}
+	return false
+}