From 70842e18d8694dc9b2aec42b23d67b294edde04f Mon Sep 17 00:00:00 2001
From: LinkinStar <linkinstar@foxmail.com>
Date: Tue, 15 Nov 2022 15:20:41 +0800
Subject: [PATCH] feat: branding support upload icon image

---
 internal/controller/upload_controller.go | 4 ++++
 internal/service/uploader/upload.go      | 3 ++-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/internal/controller/upload_controller.go b/internal/controller/upload_controller.go
index 20e55fe3..8f701441 100644
--- a/internal/controller/upload_controller.go
+++ b/internal/controller/upload_controller.go
@@ -57,5 +57,9 @@ func (uc *UploadController) UploadFile(ctx *gin.Context) {
 		handler.HandleResponse(ctx, errors.BadRequest(reason.UploadFileSourceUnsupported), nil)
 		return
 	}
+	if err != nil {
+		handler.HandleResponse(ctx, err, nil)
+		return
+	}
 	handler.HandleResponse(ctx, err, url)
 }
diff --git a/internal/service/uploader/upload.go b/internal/service/uploader/upload.go
index 3751623f..53cc5573 100644
--- a/internal/service/uploader/upload.go
+++ b/internal/service/uploader/upload.go
@@ -169,7 +169,8 @@ func (us *UploaderService) UploadBrandingFile(ctx *gin.Context) (
 		return
 	}
 	fileExt := strings.ToLower(path.Ext(file.Filename))
-	if _, ok := FormatExts[fileExt]; !ok {
+	_, ok := FormatExts[fileExt]
+	if !ok && fileExt != ".icon" {
 		handler.HandleResponse(ctx, errors.BadRequest(reason.RequestFormatError), nil)
 		return
 	}