forked from Shiloh/githaven
f0ba87fda8
Why this refactor The goal is to move most files from `models` package to `models.xxx` package. Many models depend on avatar model, so just move this first. And the existing logic is not clear, there are too many function like `AvatarLink`, `RelAvatarLink`, `SizedRelAvatarLink`, `SizedAvatarLink`, `MakeFinalAvatarURL`, `HashedAvatarLink`, etc. This refactor make everything clear: * user.AvatarLink() * user.AvatarLinkWithSize(size) * avatars.GenerateEmailAvatarFastLink(email, size) * avatars.GenerateEmailAvatarFinalLink(email, size) And many duplicated code are deleted in route handler, the handler and the model share the same avatar logic now.
55 lines
1.7 KiB
Go
55 lines
1.7 KiB
Go
// Copyright 2019 The Gitea Authors. All rights reserved.
|
|
// Use of this source code is governed by a MIT-style
|
|
// license that can be found in the LICENSE file.
|
|
|
|
package user
|
|
|
|
import (
|
|
"strings"
|
|
"time"
|
|
|
|
"code.gitea.io/gitea/models"
|
|
"code.gitea.io/gitea/models/avatars"
|
|
"code.gitea.io/gitea/modules/context"
|
|
"code.gitea.io/gitea/modules/httpcache"
|
|
)
|
|
|
|
func cacheableRedirect(ctx *context.Context, location string) {
|
|
// here we should not use `setting.StaticCacheTime`, it is pretty long (default: 6 hours)
|
|
// we must make sure the redirection cache time is short enough, otherwise a user won't see the updated avatar in 6 hours
|
|
// it's OK to make the cache time short, it is only a redirection, and doesn't cost much to make a new request
|
|
httpcache.AddCacheControlToHeader(ctx.Resp.Header(), 5*time.Minute)
|
|
ctx.Redirect(location)
|
|
}
|
|
|
|
// AvatarByUserName redirect browser to user avatar of requested size
|
|
func AvatarByUserName(ctx *context.Context) {
|
|
userName := ctx.Params(":username")
|
|
size := int(ctx.ParamsInt64(":size"))
|
|
|
|
var user *models.User
|
|
if strings.ToLower(userName) != "ghost" {
|
|
var err error
|
|
if user, err = models.GetUserByName(userName); err != nil {
|
|
ctx.ServerError("Invalid user: "+userName, err)
|
|
return
|
|
}
|
|
} else {
|
|
user = models.NewGhostUser()
|
|
}
|
|
|
|
cacheableRedirect(ctx, user.AvatarLinkWithSize(size))
|
|
}
|
|
|
|
// AvatarByEmailHash redirects the browser to the email avatar link
|
|
func AvatarByEmailHash(ctx *context.Context) {
|
|
hash := ctx.Params(":hash")
|
|
email, err := avatars.GetEmailForHash(hash)
|
|
if err != nil {
|
|
ctx.ServerError("invalid avatar hash: "+hash, err)
|
|
return
|
|
}
|
|
size := ctx.FormInt("size")
|
|
cacheableRedirect(ctx, avatars.GenerateEmailAvatarFinalLink(email, size))
|
|
}
|