From 91c0282040bbc0e95170183a2ffedb849a0f0fa0 Mon Sep 17 00:00:00 2001
From: Gucheng Wang <nomeguy@qq.com>
Date: Sat, 20 Nov 2021 15:35:33 +0800
Subject: [PATCH] Support username arg in UploadResource().

---
 controllers/resource.go | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/controllers/resource.go b/controllers/resource.go
index 2c506968..3d4be49e 100644
--- a/controllers/resource.go
+++ b/controllers/resource.go
@@ -111,6 +111,11 @@ func (c *ApiController) UploadResource() {
 	}
 	defer file.Close()
 
+	if username == "" || fullFilePath == "" {
+		c.ResponseError(fmt.Sprintf("username or fullFilePath is empty: username = %s, fullFilePath = %s", username, fullFilePath))
+		return
+	}
+
 	filename := filepath.Base(fullFilePath)
 	fileBuffer := bytes.NewBuffer(nil)
 	if _, err = io.Copy(fileBuffer, file); err != nil {
@@ -161,8 +166,11 @@ func (c *ApiController) UploadResource() {
 	switch tag {
 	case "avatar":
 		if user == nil {
-			c.ResponseError("user is nil for tag: \"avatar\"")
-			return
+			user = object.GetUser(username)
+			if user == nil {
+				c.ResponseError("user is nil for tag: \"avatar\"")
+				return
+			}
 		}
 
 		user.Avatar = fileUrl