chore: import upstream snapshot with attribution

internal/base/middleware/avatar.go

@@ -0,0 +1,94 @@
+/*
+ * Licensed to the Apache Software Foundation (ASF) under one
+ * or more contributor license agreements.  See the NOTICE file
+ * distributed with this work for additional information
+ * regarding copyright ownership.  The ASF licenses this file
+ * to you under the Apache License, Version 2.0 (the
+ * "License"); you may not use this file except in compliance
+ * with the License.  You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing,
+ * software distributed under the License is distributed on an
+ * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+ * KIND, either express or implied.  See the License for the
+ * specific language governing permissions and limitations
+ * under the License.
+ */
+
+package middleware
+
+import (
+	"fmt"
+	"net/http"
+	"net/url"
+	"os"
+	"path"
+	"path/filepath"
+	"strings"
+
+	"github.com/apache/answer/internal/service/service_config"
+	"github.com/apache/answer/internal/service/uploader"
+	"github.com/apache/answer/pkg/converter"
+	"github.com/gin-gonic/gin"
+	"github.com/segmentfault/pacman/log"
+)
+
+type AvatarMiddleware struct {
+	serviceConfig   *service_config.ServiceConfig
+	uploaderService uploader.UploaderService
+}
+
+// NewAvatarMiddleware new auth user middleware
+func NewAvatarMiddleware(serviceConfig *service_config.ServiceConfig,
+	uploaderService uploader.UploaderService,
+) *AvatarMiddleware {
+	return &AvatarMiddleware{
+		serviceConfig:   serviceConfig,
+		uploaderService: uploaderService,
+	}
+}
+
+func (am *AvatarMiddleware) AvatarThumb() gin.HandlerFunc {
+	return func(ctx *gin.Context) {
+		uri := ctx.Request.RequestURI
+		if strings.Contains(uri, "/uploads/avatar/") {
+			size := converter.StringToInt(ctx.Query("s"))
+			uriWithoutQuery, _ := url.Parse(uri)
+			filename := filepath.Base(uriWithoutQuery.Path)
+			filePath := fmt.Sprintf("%s/avatar/%s", am.serviceConfig.UploadPath, filename)
+			var err error
+			if size != 0 {
+				filePath, err = am.uploaderService.AvatarThumbFile(ctx, filename, size)
+				if err != nil {
+					log.Error(err)
+					ctx.AbortWithStatus(http.StatusNotFound)
+					return
+				}
+			}
+			avatarFile, err := os.ReadFile(filePath)
+			if err != nil {
+				log.Error(err)
+				ctx.Abort()
+				return
+			}
+			ctx.Header("content-type", fmt.Sprintf("image/%s", strings.TrimLeft(path.Ext(filePath), ".")))
+			_, err = ctx.Writer.Write(avatarFile)
+			if err != nil {
+				log.Error(err)
+			}
+			ctx.Abort()
+			return
+		} else {
+			urlInfo, err := url.Parse(uri)
+			if err != nil {
+				ctx.Next()
+				return
+			}
+			ext := strings.TrimPrefix(filepath.Ext(urlInfo.Path), ".")
+			ctx.Header("content-type", fmt.Sprintf("image/%s", ext))
+		}
+		ctx.Next()
+	}
+}