123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171 |
- // Copyright 2020 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 models
-
- import (
- "crypto/md5"
- "fmt"
- "image/png"
- "io"
- "strconv"
- "strings"
-
- "code.gitea.io/gitea/modules/avatar"
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/setting"
- "code.gitea.io/gitea/modules/storage"
- )
-
- // CustomAvatarRelativePath returns user custom avatar relative path.
- func (u *User) CustomAvatarRelativePath() string {
- return u.Avatar
- }
-
- // GenerateRandomAvatar generates a random avatar for user.
- func (u *User) GenerateRandomAvatar() error {
- return u.generateRandomAvatar(x)
- }
-
- func (u *User) generateRandomAvatar(e Engine) error {
- seed := u.Email
- if len(seed) == 0 {
- seed = u.Name
- }
-
- img, err := avatar.RandomImage([]byte(seed))
- if err != nil {
- return fmt.Errorf("RandomImage: %v", err)
- }
-
- u.Avatar = HashEmail(seed)
-
- // Don't share the images so that we can delete them easily
- if err := storage.SaveFrom(storage.Avatars, u.CustomAvatarRelativePath(), func(w io.Writer) error {
- if err := png.Encode(w, img); err != nil {
- log.Error("Encode: %v", err)
- }
- return err
- }); err != nil {
- return fmt.Errorf("Failed to create dir %s: %v", u.CustomAvatarRelativePath(), err)
- }
-
- if _, err := e.ID(u.ID).Cols("avatar").Update(u); err != nil {
- return err
- }
-
- log.Info("New random avatar created: %d", u.ID)
- return nil
- }
-
- // 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 DefaultAvatarLink()
- }
-
- switch {
- case u.UseCustomAvatar:
- if u.Avatar == "" {
- return DefaultAvatarLink()
- }
- if size > 0 {
- return setting.AppSubURL + "/avatars/" + u.Avatar + "?size=" + strconv.Itoa(size)
- }
- return setting.AppSubURL + "/avatars/" + u.Avatar
- case setting.DisableGravatar, setting.OfflineMode:
- if u.Avatar == "" {
- if err := u.GenerateRandomAvatar(); err != nil {
- log.Error("GenerateRandomAvatar: %v", err)
- }
- }
- if size > 0 {
- return setting.AppSubURL + "/avatars/" + u.Avatar + "?size=" + strconv.Itoa(size)
- }
- return setting.AppSubURL + "/avatars/" + u.Avatar
- }
- return SizedAvatarLink(u.AvatarEmail, size)
- }
-
- // RelAvatarLink returns a relative link to the user's avatar. The link
- // may either be a sub-URL to this site, or a full URL to an external avatar
- // service.
- func (u *User) RelAvatarLink() string {
- return u.SizedRelAvatarLink(DefaultAvatarSize)
- }
-
- // AvatarLink returns user avatar absolute link.
- func (u *User) AvatarLink() string {
- link := u.RelAvatarLink()
- if link[0] == '/' && link[1] != '/' {
- return setting.AppURL + strings.TrimPrefix(link, setting.AppSubURL)[1:]
- }
- return link
- }
-
- // UploadAvatar saves custom avatar for user.
- // FIXME: split uploads to different subdirs in case we have massive users.
- func (u *User) UploadAvatar(data []byte) error {
- m, err := avatar.Prepare(data)
- if err != nil {
- return err
- }
-
- sess := x.NewSession()
- defer sess.Close()
- if err = sess.Begin(); err != nil {
- return err
- }
-
- u.UseCustomAvatar = true
- // Different users can upload same image as avatar
- // If we prefix it with u.ID, it will be separated
- // Otherwise, if any of the users delete his avatar
- // Other users will lose their avatars too.
- u.Avatar = fmt.Sprintf("%x", md5.Sum([]byte(fmt.Sprintf("%d-%x", u.ID, md5.Sum(data)))))
- if err = updateUserCols(sess, u, "use_custom_avatar", "avatar"); err != nil {
- return fmt.Errorf("updateUser: %v", err)
- }
-
- if err := storage.SaveFrom(storage.Avatars, u.CustomAvatarRelativePath(), func(w io.Writer) error {
- if err := png.Encode(w, *m); err != nil {
- log.Error("Encode: %v", err)
- }
- return err
- }); err != nil {
- return fmt.Errorf("Failed to create dir %s: %v", u.CustomAvatarRelativePath(), err)
- }
-
- return sess.Commit()
- }
-
- // DeleteAvatar deletes the user's custom avatar.
- func (u *User) DeleteAvatar() error {
- aPath := u.CustomAvatarRelativePath()
- log.Trace("DeleteAvatar[%d]: %s", u.ID, aPath)
- if len(u.Avatar) > 0 {
- if err := storage.Avatars.Delete(aPath); err != nil {
- return fmt.Errorf("Failed to remove %s: %v", aPath, err)
- }
- }
-
- u.UseCustomAvatar = false
- u.Avatar = ""
- if _, err := x.ID(u.ID).Cols("avatar, use_custom_avatar").Update(u); err != nil {
- return fmt.Errorf("UpdateUser: %v", err)
- }
- return nil
- }
|