Skip to content

Commit

Permalink
Always return local url for users avatar
Browse files Browse the repository at this point in the history
Avoids having to wait for DNS lookups when libravatar is activated
fixing #6046
  • Loading branch information
strk committed Sep 23, 2019
1 parent 63ff616 commit b489b99
Show file tree
Hide file tree
Showing 4 changed files with 53 additions and 3 deletions.
2 changes: 1 addition & 1 deletion models/action_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -173,7 +173,7 @@ func TestPushCommits_AvatarLink(t *testing.T) {
pushCommits.Len = len(pushCommits.Commits)

assert.Equal(t,
"https://secure.gravatar.com/avatar/ab53a2911ddf9b4817ac01ddcd3d975f?d=identicon",
"/user/avatar/user2/-1",
pushCommits.AvatarLink("user2@example.com"))

assert.Equal(t,
Expand Down
16 changes: 14 additions & 2 deletions models/user.go
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@ import (
"image/png"
"os"
"path/filepath"
"strconv"
"strings"
"time"
"unicode/utf8"
Expand Down Expand Up @@ -373,9 +374,20 @@ func (u *User) generateRandomAvatar(e Engine) error {
return nil
}

// SizedRelAvatarLink returns a relative link to the user's avatar. When
// applicable, the link is for an avatar of the indicated size (in pixels).
// SizedRelAvatarLink returns a link to the user's avatar via
// the local explore page. Function returns immediately.
// When applicable, the link is for an avatar of the indicated size (in pixels).
func (u *User) SizedRelAvatarLink(size int) string {
return setting.AppSubURL + "/user/avatar/" + u.Name + "/" + strconv.Itoa(size)
}

// RealSizedAvatarLink returns a link to the user's avatar. When
// applicable, the link is for an avatar of the indicated size (in pixels).
//
// This function make take time to return when federated avatars
// are in use, due to a DNS lookup need
//
func (u *User) RealSizedAvatarLink(size int) string {
if u.ID == -1 {
return base.DefaultAvatarLink()
}
Expand Down
1 change: 1 addition & 0 deletions routers/routes/routes.go
Original file line number Diff line number Diff line change
Expand Up @@ -278,6 +278,7 @@ func RegisterRoutes(m *macaron.Macaron) {

// ***** START: User *****
m.Group("/user", func() {
m.Get("/avatar/:username/:size", user.Avatar)
m.Get("/login", user.SignIn)
m.Post("/login", bindIgnErr(auth.SignInForm{}), user.SignInPost)
m.Group("", func() {
Expand Down
37 changes: 37 additions & 0 deletions routers/user/avatar.go
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
// 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 (
"strconv"

"code.gitea.io/gitea/models"
"code.gitea.io/gitea/modules/context"
"code.gitea.io/gitea/modules/log"
)

// Avatar redirect browser to user avatar of requested size
func Avatar(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)

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)
}
return
}

ctx.Redirect(user.RealSizedAvatarLink(size))
}

0 comments on commit b489b99

Please sign in to comment.