summaryrefslogtreecommitdiffstats
path: root/routers
diff options
context:
space:
mode:
authorwxiaoguang <wxiaoguang@gmail.com>2021-10-06 07:25:46 +0800
committerGitHub <noreply@github.com>2021-10-06 01:25:46 +0200
commitf0ba87fda88c7bb601eee17f3e3a72bea8a71521 (patch)
treecb5caf3a22fceaec8d2d421bec0cb094ecdb9c53 /routers
parent48c2578bd8ce4ddd90bf926bd40388f57c90fe14 (diff)
downloadgitea-f0ba87fda88c7bb601eee17f3e3a72bea8a71521.tar.gz
gitea-f0ba87fda88c7bb601eee17f3e3a72bea8a71521.zip
Avatar refactor, move avatar code from `models` to `models.avatars`, remove duplicated code (#17123)
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.
Diffstat (limited to 'routers')
-rw-r--r--routers/web/repo/issue.go2
-rw-r--r--routers/web/user/avatar.go84
-rw-r--r--routers/web/web.go2
3 files changed, 19 insertions, 69 deletions
diff --git a/routers/web/repo/issue.go b/routers/web/repo/issue.go
index 03bdbec80c..64c4ae41f4 100644
--- a/routers/web/repo/issue.go
+++ b/routers/web/repo/issue.go
@@ -2614,5 +2614,5 @@ func handleTeamMentions(ctx *context.Context) {
ctx.Data["MentionableTeams"] = ctx.Repo.Owner.Teams
ctx.Data["MentionableTeamsOrg"] = ctx.Repo.Owner.Name
- ctx.Data["MentionableTeamsOrgAvatar"] = ctx.Repo.Owner.RelAvatarLink()
+ ctx.Data["MentionableTeamsOrgAvatar"] = ctx.Repo.Owner.AvatarLink()
}
diff --git a/routers/web/user/avatar.go b/routers/web/user/avatar.go
index 2df5c148f7..f39bcc36d3 100644
--- a/routers/web/user/avatar.go
+++ b/routers/web/user/avatar.go
@@ -5,100 +5,50 @@
package user
import (
- "errors"
- "net/url"
- "path"
- "strconv"
"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"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/setting"
)
func cacheableRedirect(ctx *context.Context, location string) {
- ctx.Resp.Header().Set("Cache-Control", httpcache.GetCacheControl())
+ // 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)
}
-// Avatar redirect browser to user avatar of requested size
-func Avatar(ctx *context.Context) {
+// AvatarByUserName redirect browser to user avatar of requested size
+func AvatarByUserName(ctx *context.Context) {
userName := ctx.Params(":username")
- size, err := strconv.Atoi(ctx.Params(":size"))
- if err != nil {
- ctx.ServerError("Invalid avatar size", err)
- return
- }
-
- log.Debug("Asked avatar for user %v and size %v", userName, size)
+ size := int(ctx.ParamsInt64(":size"))
var user *models.User
if strings.ToLower(userName) != "ghost" {
- user, err = models.GetUserByName(userName)
- if err != nil {
- if models.IsErrUserNotExist(err) {
- ctx.ServerError("Requested avatar for invalid user", err)
- } else {
- ctx.ServerError("Retrieving user by name", err)
- }
+ 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.RealSizedAvatarLink(size))
+ cacheableRedirect(ctx, user.AvatarLinkWithSize(size))
}
-// AvatarByEmailHash redirects the browser to the appropriate Avatar link
+// AvatarByEmailHash redirects the browser to the email avatar link
func AvatarByEmailHash(ctx *context.Context) {
- var err error
-
hash := ctx.Params(":hash")
- if len(hash) == 0 {
- ctx.ServerError("invalid avatar hash", errors.New("hash cannot be empty"))
- return
- }
-
- var email string
- email, err = models.GetEmailForHash(hash)
+ email, err := avatars.GetEmailForHash(hash)
if err != nil {
- ctx.ServerError("invalid avatar hash", err)
- return
- }
- if len(email) == 0 {
- cacheableRedirect(ctx, models.DefaultAvatarLink())
+ ctx.ServerError("invalid avatar hash: "+hash, err)
return
}
size := ctx.FormInt("size")
- if size == 0 {
- size = models.DefaultAvatarSize
- }
-
- var avatarURL *url.URL
-
- if setting.EnableFederatedAvatar && setting.LibravatarService != nil {
- avatarURL, err = models.LibravatarURL(email)
- if err != nil {
- avatarURL, err = url.Parse(models.DefaultAvatarLink())
- if err != nil {
- ctx.ServerError("invalid default avatar url", err)
- return
- }
- }
- } else if !setting.DisableGravatar {
- copyOfGravatarSourceURL := *setting.GravatarSourceURL
- avatarURL = &copyOfGravatarSourceURL
- avatarURL.Path = path.Join(avatarURL.Path, hash)
- } else {
- avatarURL, err = url.Parse(models.DefaultAvatarLink())
- if err != nil {
- ctx.ServerError("invalid default avatar url", err)
- return
- }
- }
-
- cacheableRedirect(ctx, models.MakeFinalAvatarURL(avatarURL, size))
+ cacheableRedirect(ctx, avatars.GenerateEmailAvatarFinalLink(email, size))
}
diff --git a/routers/web/web.go b/routers/web/web.go
index a56a2250bd..e12552f4d4 100644
--- a/routers/web/web.go
+++ b/routers/web/web.go
@@ -366,7 +366,7 @@ func RegisterRoutes(m *web.Route) {
m.Get("/activate", user.Activate, reqSignIn)
m.Post("/activate", user.ActivatePost, reqSignIn)
m.Any("/activate_email", user.ActivateEmail)
- m.Get("/avatar/{username}/{size}", user.Avatar)
+ m.Get("/avatar/{username}/{size}", user.AvatarByUserName)
m.Get("/email2user", user.Email2User)
m.Get("/recover_account", user.ResetPasswd)
m.Post("/recover_account", user.ResetPasswdPost)