add option for avatar/profilebg change

packages/httpRequest/user.go

@@ -2,11 +2,12 @@ package httpRequest
 
 import (
 	json2 "encoding/json"
+	"net/http"
+	"time"
+
 	"go-socket/packages/convertions"
 	"go-socket/packages/globals"
 	"go-socket/packages/minio"
-	"net/http"
-	"time"
 
 	"go-socket/packages/cache"
 	"go-socket/packages/passwords"
@@ -18,7 +19,7 @@ import (
 )
 
 func HandleUserNewToken(response http.ResponseWriter, request *http.Request) {
-	if !postValidCheckWithResponseOnFail(&response, request, false) {
+	if !postValidCheckWithResponseOnFail(&response, request, postNormal) {
 		return
 	}
 
@@ -77,7 +78,7 @@ func HandleUserNewToken(response http.ResponseWriter, request *http.Request) {
 }
 
 func HandleUserNew(response http.ResponseWriter, request *http.Request) {
-	if !postValidCheckWithResponseOnFail(&response, request, false) {
+	if !postValidCheckWithResponseOnFail(&response, request, postNormal) {
 		return
 	}
 
@@ -118,7 +119,7 @@ func HandleUserNew(response http.ResponseWriter, request *http.Request) {
 }
 
 func HandleUserDelete(response http.ResponseWriter, request *http.Request) {
-	if !postValidCheckWithResponseOnFail(&response, request, false) {
+	if !postValidCheckWithResponseOnFail(&response, request, postNormal) {
 		return
 	}
 	ctx := request.Context()
@@ -140,7 +141,7 @@ func HandleUserDelete(response http.ResponseWriter, request *http.Request) {
 }
 
 func HandleUserModProfile(response http.ResponseWriter, request *http.Request) {
-	if !postValidCheckWithResponseOnFail(&response, request, false) {
+	if !postValidCheckWithResponseOnFail(&response, request, postNormal) {
 		return
 	}
 
@@ -190,8 +191,7 @@ func HandleUserModProfile(response http.ResponseWriter, request *http.Request) {
 }
 
 func HandleUserModAvatar(response http.ResponseWriter, request *http.Request) {
-
-	if !postValidCheckWithResponseOnFail(&response, request, true) {
+	if !postValidCheckWithResponseOnFail(&response, request, postAvatar) {
 		return
 	}
 	ctx := request.Context()
@@ -202,7 +202,7 @@ func HandleUserModAvatar(response http.ResponseWriter, request *http.Request) {
 		return
 	}
 
-	request.Body = http.MaxBytesReader(response, request.Body, int64(globals.MaxPostWithFileBytes))
+	request.Body = http.MaxBytesReader(response, request.Body, int64(globals.MaxPostWithAvatar))
 
 	if err = request.ParseMultipartForm(int64(globals.MaxPostBytes)); err != nil {
 		http.Error(response, "invalid multipart form", http.StatusBadRequest)
@@ -221,9 +221,21 @@ func HandleUserModAvatar(response http.ResponseWriter, request *http.Request) {
 	}
 	defer file.Close()
 
-	contentType := header.Header.Get("Content-Type")
-	key := minio.GetKey(conn.Id, contentType, minio.File)
+	isImg, contentType, err := isImage(file)
+	if err != nil || !isImg {
+		http.Error(response, "invalid file", http.StatusBadRequest)
+		return
+	}
 
+	if user.Avatar != "" {
+		err = minio.Delete(ctx, string(minio.UserAvatarPrefix)+user.Avatar)
+		if err != nil {
+			http.Error(response, "internal server error", http.StatusInternalServerError)
+			return
+		}
+	}
+
+	key := minio.GetKey(conn.Id, contentType, minio.File)
 	if err = minio.Upload(ctx, key, file, header.Size, contentType, map[string]string{
 		"originalName": header.Filename,
 		"uploaderId":   user.Id.String(),
@@ -231,4 +243,77 @@ func HandleUserModAvatar(response http.ResponseWriter, request *http.Request) {
 		http.Error(response, "upload failed", http.StatusInternalServerError)
 		return
 	}
+
+	user.Avatar = key[len(minio.UserAvatarPrefix):]
+	err = postgresql.UserUpdateProfile(ctx, user, types.UserProfileUpdateList{Avatar: true})
+	if err != nil {
+		http.Error(response, "internal server error", http.StatusInternalServerError)
+		return
+	}
+
+	response.WriteHeader(http.StatusAccepted)
+}
+
+func HandleUserModProfileBg(response http.ResponseWriter, request *http.Request) {
+	if !postValidCheckWithResponseOnFail(&response, request, postProfileBg) {
+		return
+	}
+	ctx := request.Context()
+
+	user, err := getUserByToken(ctx, request.Header.Get("token"))
+	if err != nil {
+		http.Error(response, "invalid token", http.StatusUnauthorized)
+		return
+	}
+
+	request.Body = http.MaxBytesReader(response, request.Body, int64(globals.MaxPostWithProfileBg))
+
+	if err = request.ParseMultipartForm(int64(globals.MaxPostBytes)); err != nil {
+		http.Error(response, "invalid multipart form", http.StatusBadRequest)
+		return
+	}
+
+	conn, ok := getConnectionWithResponseOnFail(&response, request, user)
+	if !ok {
+		return
+	}
+
+	file, header, err := request.FormFile("file")
+	if err != nil {
+		http.Error(response, "missing file", http.StatusBadRequest)
+		return
+	}
+	defer file.Close()
+
+	isImg, contentType, err := isImage(file)
+	if err != nil || !isImg {
+		http.Error(response, "invalid file", http.StatusBadRequest)
+		return
+	}
+
+	if user.ProfileBg != "" {
+		err = minio.Delete(ctx, string(minio.UserProfileBgPrefix)+user.ProfileBg)
+		if err != nil {
+			http.Error(response, "internal server error", http.StatusInternalServerError)
+			return
+		}
+	}
+
+	key := minio.GetKey(conn.Id, contentType, minio.UserProfileBg)
+	if err = minio.Upload(ctx, key, file, header.Size, contentType, map[string]string{
+		"originalName": header.Filename,
+		"uploaderId":   user.Id.String(),
+	}); err != nil {
+		http.Error(response, "upload failed", http.StatusInternalServerError)
+		return
+	}
+
+	user.ProfileBg = key[len(minio.UserProfileBgPrefix):]
+	err = postgresql.UserUpdateProfile(ctx, user, types.UserProfileUpdateList{ProfileBg: true})
+	if err != nil {
+		http.Error(response, "internal server error", http.StatusInternalServerError)
+		return
+	}
+	
+	response.WriteHeader(http.StatusAccepted)
 }