fix: handle for different image format

services/avatar/get_gravatar.go

@@ -6,7 +6,6 @@ import (
 	"github.com/imroc/req/v3"
 	"github.com/samber/lo"
 	"image"
-	"image/jpeg"
 	"strconv"
 	"time"
 
@@ -71,9 +70,9 @@ func (s *service) getGravatar(ctx context.Context, hash string, args GetAvatarAr
 
 	defer resp.Body.Close()
 
-	img, err := jpeg.Decode(resp.Body)
+	img, err := parseImage(resp.GetHeader("Content-Type"), resp.Body)
 	if err != nil {
-		otelzap.L().Ctx(ctx).Error("failed to decode gravatar", zap.Error(err))
+		otelzap.L().Ctx(ctx).Error("parse image failed", zap.Error(err))
 		return GetGravatarResult{}, err
 	}
 

services/avatar/get_qq_avatar.go

@@ -3,14 +3,13 @@ package avatar
 import (
 	"context"
 	"fmt"
-	"github.com/imroc/req/v3"
-	"go.uber.org/zap"
 	"image"
-	"image/jpeg"
 	"strconv"
 
+	"github.com/imroc/req/v3"
 	"github.com/nfnt/resize"
 	"github.com/uptrace/opentelemetry-go-extra/otelzap"
+	"go.uber.org/zap"
 )
 
 func initQQClient() *req.Client {
@@ -67,9 +66,9 @@ func (s *service) getQQAvatar(ctx context.Context, hash string, args GetAvatarAr
 
 	defer resp.Body.Close()
 
-	img, err := jpeg.Decode(resp.Body)
+	img, err := parseImage(resp.GetHeader("Content-Type"), resp.Body)
 	if err != nil {
-		otelzap.L().Ctx(ctx).Error("decode qq avatar failed", zap.Error(err))
+		otelzap.L().Ctx(ctx).Error("parse image failed", zap.Error(err))
 		return nil, err
 	}
 

services/avatar/utils.go

@@ -0,0 +1,27 @@
+package avatar
+
+import (
+	"github.com/kolesa-team/go-webp/webp"
+	"image"
+	"image/gif"
+	"image/jpeg"
+	"image/png"
+	"io"
+)
+
+func parseImage(contentType string, reader io.Reader) (img image.Image, err error) {
+	switch contentType {
+	case "image/jpeg":
+		img, err = jpeg.Decode(reader)
+	case "image/png":
+		img, err = png.Decode(reader)
+	case "image/webp":
+		img, err = webp.Decode(reader, nil)
+	case "image/gif":
+		img, err = gif.Decode(reader)
+	default:
+		img, err = jpeg.Decode(reader)
+	}
+
+	return
+}