llvy.ltd/casdoor
1
0
Fork 0
mirror of https://github.com/casdoor/casdoor.git synced 2025-07-03 12:30:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add UploadFile() API.

Browse Source
This commit is contained in:
Yang Luo
2021-08-08 21:58:29 +08:00
parent 40fbca7db4
commit 5c2f96bda0
4 changed files with 33 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

42
controllers/account.go
View File

@ -212,19 +212,23 @@ func (c *ApiController) GetAccount() {
c.ResponseOk(user, organization)
}
// UploadAvatar
// @Title UploadAvatar
// @Description upload avatar
// @Param avatarfile formData string true "The base64 encode of avatarfile"
// @Param password formData string true "The password"
// UploadFile
// @Title UploadFile
// @Description upload file
// @Param folder query string true "The folder"
// @Param subFolder query string true "The sub folder"
// @Param file formData string true "The file"
// @Success 200 {object} controllers.Response The Response object
// @router /upload-avatar [post]
func (c *ApiController) UploadAvatar() {
// @router /upload-file [post]
func (c *ApiController) UploadFile() {
userId, ok := c.RequireSignedIn()
if !ok {
return
}
folder := c.Input().Get("folder")
subFolder := c.Input().Get("subFolder")
user := object.GetUser(userId)
application := object.GetApplicationByUser(user)
provider := application.GetStorageProvider()
@ -233,24 +237,28 @@ func (c *ApiController) UploadAvatar() {
return
}
avatarBase64 := c.Ctx.Request.Form.Get("avatarfile")
index := strings.Index(avatarBase64, ",")
if index < 0 || avatarBase64[0:index] != "data:image/png;base64" {
c.ResponseError("File encoding error")
return
fileString := c.Ctx.Request.Form.Get("file")
var fileBytes []byte
pngHeader := "data:image/png;base64,"
if strings.HasPrefix(fileString, pngHeader) {
fileBytes, _ = base64.StdEncoding.DecodeString(fileString[len(pngHeader):])
} else {
fileBytes = []byte(fileString)
}
dist, _ := base64.StdEncoding.DecodeString(avatarBase64[index+1:])
fileUrl, err := object.UploadFile(provider, "avatar", user.GetId(), dist)
fileUrl, err := object.UploadFile(provider, folder, subFolder, fileBytes)
if err != nil {
c.ResponseError(err.Error())
return
}
user.Avatar = fileUrl
object.UpdateUser(user.GetId(), user)
if folder == "avatar" {
user.Avatar = fileUrl
object.UpdateUser(user.GetId(), user)
}
c.ResponseOk()
c.ResponseOk(fileUrl)
}
// GetHumanCheck ...