Add migration renameExistingUserAvatarName

Author: masudur-rahman

models/migrations/migrations.go

@@ -284,6 +284,8 @@ var migrations = []Migration{
 	NewMigration("new feature: change target branch of pull requests", featureChangeTargetBranch),
 	// v114 -> v115
 	NewMigration("Remove authentication credentials from stored URL", sanitizeOriginalURL),
+	// v115 -> v116
+	NewMigration("add user_id prefix to existing user avatar name", renameExistingUserAvatarName),
 }
 
 // Migrate database to current version

models/migrations/v115.go

@@ -0,0 +1,73 @@
+// 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 migrations
+
+import (
+	"fmt"
+	"io"
+	"os"
+	"path/filepath"
+	"strconv"
+	"strings"
+
+	"code.gitea.io/gitea/models"
+	"code.gitea.io/gitea/modules/setting"
+
+	"xorm.io/xorm"
+)
+
+func renameExistingUserAvatarName(x *xorm.Engine) error {
+	sess := x.NewSession()
+	defer sess.Close()
+	if err := sess.Begin(); err != nil {
+		return err
+	}
+
+	users := make([]models.User, 0)
+	if err := sess.Find(&users); err != nil {
+		return err
+	}
+
+	deleteList := make([]string, 0, len(users))
+	for _, user := range users {
+		if !strings.HasPrefix(user.Avatar, strconv.FormatInt(user.ID, 10)+"-") {
+			oldAvatar := user.Avatar
+			newAvatar := fmt.Sprintf("%d-%s", user.ID, user.Avatar)
+
+			if _, err := os.Stat(filepath.Join(setting.AvatarUploadPath, oldAvatar)); err != nil {
+				continue
+			}
+
+			fr, err := os.Open(filepath.Join(setting.AvatarUploadPath, oldAvatar))
+			if err != nil {
+				return err
+			}
+			defer fr.Close()
+
+			fw, err := os.Create(filepath.Join(setting.AvatarUploadPath, newAvatar))
+			if err != nil {
+				return err
+			}
+			defer fw.Close()
+
+			if _, err := io.Copy(fw, fr); err != nil {
+				return err
+			}
+
+			user.Avatar = newAvatar
+			if _, err := sess.ID(user.ID).Update(&user); err != nil {
+				return err
+			}
+
+			deleteList = append(deleteList, filepath.Join(setting.AvatarUploadPath, oldAvatar))
+		}
+	}
+	for _, file := range deleteList {
+		if err := os.Remove(file); err != nil {
+			return err
+		}
+	}
+	return sess.Commit()
+}