584e3af237
Fix getting avatar based on email Remove avatarUrl from user struct Fix staticcheck Add default avatar size Add config option for caching avatars go mod vendor Add swagger docs Add proxying gravatar requests for user avatars Co-authored-by: kolaente <k@knt.li> Reviewed-on: https://kolaente.dev/vikunja/api/pulls/148
75 lines
2.5 KiB
Go
75 lines
2.5 KiB
Go
// Vikunja is a to-do-list application to facilitate your life.
|
|
// Copyright 2018-2020 Vikunja and contributors. All rights reserved.
|
|
//
|
|
// This program is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
package v1
|
|
|
|
import (
|
|
"code.vikunja.io/api/pkg/log"
|
|
"code.vikunja.io/api/pkg/modules/avatar/gravatar"
|
|
user2 "code.vikunja.io/api/pkg/user"
|
|
"code.vikunja.io/web/handler"
|
|
"github.com/labstack/echo/v4"
|
|
"net/http"
|
|
"strconv"
|
|
)
|
|
|
|
// GetAvatar returns a user's avatar
|
|
// @Summary User Avatar
|
|
// @Description Returns the user avatar as image.
|
|
// @tags user
|
|
// @Produce octet-stream
|
|
// @Param username path string true "The username of the user who's avatar you want to get"
|
|
// @Param size query int false "The size of the avatar you want to get"
|
|
// @Success 200 {} blob "The avatar"
|
|
// @Failure 404 {object} models.Message "The user does not exist."
|
|
// @Failure 500 {object} models.Message "Internal error"
|
|
// @Router /{username}/avatar [get]
|
|
func GetAvatar(c echo.Context) error {
|
|
// Get the username
|
|
username := c.Param("username")
|
|
|
|
// Get the user
|
|
user, err := user2.GetUserWithEmail(&user2.User{Username: username})
|
|
if err != nil {
|
|
log.Errorf("Error getting user for avatar: %v", err)
|
|
return handler.HandleHTTPError(err, c)
|
|
}
|
|
|
|
// Initialize the avatar provider
|
|
// For now, we only have one avatar provider, in the future there could be multiple which
|
|
// could be changed based on user settings etc.
|
|
avatarProvider := gravatar.Provider{}
|
|
|
|
size := c.QueryParam("size")
|
|
var sizeInt int64 = 250 // Default size of 250
|
|
if size != "" {
|
|
sizeInt, err = strconv.ParseInt(size, 10, 64)
|
|
if err != nil {
|
|
log.Errorf("Error parsing size: %v", err)
|
|
return handler.HandleHTTPError(err, c)
|
|
}
|
|
}
|
|
|
|
// Get the avatar
|
|
avatar, mimeType, err := avatarProvider.GetAvatar(user, sizeInt)
|
|
if err != nil {
|
|
log.Errorf("Error getting avatar for user %d: %v", user.ID, err)
|
|
return handler.HandleHTTPError(err, c)
|
|
}
|
|
|
|
return c.Blob(http.StatusOK, mimeType, avatar)
|
|
}
|