Fix bug on avatar (#31008)
Co-authored-by: silverwind <me@silverwind.io>
(cherry picked from commit 58a03e9fadb345de5653345c2a68ecfd0750940a)
(cherry picked from commit 1be797faba
)
This commit is contained in:
parent
3ba58114c7
commit
32d8ada0e7
1 changed files with 21 additions and 11 deletions
|
@ -5,8 +5,10 @@ package user
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
|
"os"
|
||||||
|
|
||||||
"code.gitea.io/gitea/models/db"
|
"code.gitea.io/gitea/models/db"
|
||||||
user_model "code.gitea.io/gitea/models/user"
|
user_model "code.gitea.io/gitea/models/user"
|
||||||
|
@ -48,16 +50,24 @@ func UploadAvatar(ctx context.Context, u *user_model.User, data []byte) error {
|
||||||
func DeleteAvatar(ctx context.Context, u *user_model.User) error {
|
func DeleteAvatar(ctx context.Context, u *user_model.User) error {
|
||||||
aPath := u.CustomAvatarRelativePath()
|
aPath := u.CustomAvatarRelativePath()
|
||||||
log.Trace("DeleteAvatar[%d]: %s", u.ID, aPath)
|
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: %w", aPath, err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
u.UseCustomAvatar = false
|
return db.WithTx(ctx, func(ctx context.Context) error {
|
||||||
u.Avatar = ""
|
hasAvatar := len(u.Avatar) > 0
|
||||||
if _, err := db.GetEngine(ctx).ID(u.ID).Cols("avatar, use_custom_avatar").Update(u); err != nil {
|
u.UseCustomAvatar = false
|
||||||
return fmt.Errorf("DeleteAvatar: %w", err)
|
u.Avatar = ""
|
||||||
}
|
if _, err := db.GetEngine(ctx).ID(u.ID).Cols("avatar, use_custom_avatar").Update(u); err != nil {
|
||||||
return nil
|
return fmt.Errorf("DeleteAvatar: %w", err)
|
||||||
|
}
|
||||||
|
|
||||||
|
if hasAvatar {
|
||||||
|
if err := storage.Avatars.Delete(aPath); err != nil {
|
||||||
|
if !errors.Is(err, os.ErrNotExist) {
|
||||||
|
return fmt.Errorf("failed to remove %s: %w", aPath, err)
|
||||||
|
}
|
||||||
|
log.Warn("Deleting avatar %s but it doesn't exist", aPath)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil
|
||||||
|
})
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue