Skip to content

Commit bc05ddc

Browse files
AndrewBezold6543zeripathlunny
authored
Redirect on changed user and org name (#11649)
* Add redirect for user * Add redirect for orgs * Add user redirect test * Appease linter * Add comment to DeleteUserRedirect function * Fix locale changes * Fix GetUserByParams * Fix orgAssignment * Remove debug logging * Add redirect prompt * Dont Export DeleteUserRedirect & only use it within a session * Unexport newUserRedirect * cleanup * Fix & Dedub API code * Format Template * Add Migration & rm dublicat * Refactor: unexport newRepoRedirect() & rm dedub del exec * if this fails we'll need to re-rename the user directory Co-authored-by: 6543 <[email protected]> Co-authored-by: zeripath <[email protected]> Co-authored-by: Lunny Xiao <[email protected]>
1 parent 4f608ad commit bc05ddc

File tree

25 files changed

+325
-64
lines changed

25 files changed

+325
-64
lines changed

models/error.go

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,21 @@ func (err ErrUserNotExist) Error() string {
146146
return fmt.Sprintf("user does not exist [uid: %d, name: %s, keyid: %d]", err.UID, err.Name, err.KeyID)
147147
}
148148

149+
// ErrUserRedirectNotExist represents a "UserRedirectNotExist" kind of error.
150+
type ErrUserRedirectNotExist struct {
151+
Name string
152+
}
153+
154+
// IsErrUserRedirectNotExist check if an error is an ErrUserRedirectNotExist.
155+
func IsErrUserRedirectNotExist(err error) bool {
156+
_, ok := err.(ErrUserRedirectNotExist)
157+
return ok
158+
}
159+
160+
func (err ErrUserRedirectNotExist) Error() string {
161+
return fmt.Sprintf("user redirect does not exist [name: %s]", err.Name)
162+
}
163+
149164
// ErrUserProhibitLogin represents a "ErrUserProhibitLogin" kind of error.
150165
type ErrUserProhibitLogin struct {
151166
UID int64

models/fixtures/user_redirect.yml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,4 @@
1+
-
2+
id: 1
3+
lower_name: olduser1
4+
redirect_user_id: 1

models/migrations/migrations.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -279,6 +279,8 @@ var migrations = []Migration{
279279
NewMigration("Convert hook task type from char(16) to varchar(16) and trim the column", convertHookTaskTypeToVarcharAndTrim),
280280
// v166 -> v167
281281
NewMigration("Where Password is Valid with Empty String delete it", recalculateUserEmptyPWD),
282+
// v167 -> v168
283+
NewMigration("Add user redirect", addUserRedirect),
282284
}
283285

284286
// GetCurrentDBVersion returns the current db version

models/migrations/v167.go

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,24 @@
1+
// Copyright 2021 The Gitea Authors. All rights reserved.
2+
// Use of this source code is governed by a MIT-style
3+
// license that can be found in the LICENSE file.
4+
5+
package migrations
6+
7+
import (
8+
"fmt"
9+
10+
"xorm.io/xorm"
11+
)
12+
13+
func addUserRedirect(x *xorm.Engine) (err error) {
14+
type UserRedirect struct {
15+
ID int64 `xorm:"pk autoincr"`
16+
LowerName string `xorm:"UNIQUE(s) INDEX NOT NULL"`
17+
RedirectUserID int64
18+
}
19+
20+
if err := x.Sync2(new(UserRedirect)); err != nil {
21+
return fmt.Errorf("Sync2: %v", err)
22+
}
23+
return nil
24+
}

models/models.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ func init() {
128128
new(Task),
129129
new(LanguageStat),
130130
new(EmailHash),
131+
new(UserRedirect),
131132
new(Project),
132133
new(ProjectBoard),
133134
new(ProjectIssue),

models/org.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -171,6 +171,10 @@ func CreateOrganization(org, owner *User) (err error) {
171171
return err
172172
}
173173

174+
if err = deleteUserRedirect(sess, org.Name); err != nil {
175+
return err
176+
}
177+
174178
if _, err = sess.Insert(org); err != nil {
175179
return fmt.Errorf("insert organization: %v", err)
176180
}

models/repo.go

Lines changed: 3 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1312,8 +1312,8 @@ func TransferOwnership(doer *User, newOwnerName string, repo *Repository) error
13121312
return fmt.Errorf("delete repo redirect: %v", err)
13131313
}
13141314

1315-
if err := NewRepoRedirect(DBContext{sess}, oldOwner.ID, repo.ID, repo.Name, repo.Name); err != nil {
1316-
return fmt.Errorf("NewRepoRedirect: %v", err)
1315+
if err := newRepoRedirect(sess, oldOwner.ID, repo.ID, repo.Name, repo.Name); err != nil {
1316+
return fmt.Errorf("newRepoRedirect: %v", err)
13171317
}
13181318

13191319
return sess.Commit()
@@ -1361,12 +1361,7 @@ func ChangeRepositoryName(doer *User, repo *Repository, newRepoName string) (err
13611361
return fmt.Errorf("sess.Begin: %v", err)
13621362
}
13631363

1364-
// If there was previously a redirect at this location, remove it.
1365-
if err = deleteRepoRedirect(sess, repo.OwnerID, newRepoName); err != nil {
1366-
return fmt.Errorf("delete repo redirect: %v", err)
1367-
}
1368-
1369-
if err := NewRepoRedirect(DBContext{sess}, repo.Owner.ID, repo.ID, oldRepoName, newRepoName); err != nil {
1364+
if err := newRepoRedirect(sess, repo.Owner.ID, repo.ID, oldRepoName, newRepoName); err != nil {
13701365
return err
13711366
}
13721367

models/repo_redirect.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -28,16 +28,16 @@ func LookupRepoRedirect(ownerID int64, repoName string) (int64, error) {
2828
return redirect.RedirectRepoID, nil
2929
}
3030

31-
// NewRepoRedirect create a new repo redirect
32-
func NewRepoRedirect(ctx DBContext, ownerID, repoID int64, oldRepoName, newRepoName string) error {
31+
// newRepoRedirect create a new repo redirect
32+
func newRepoRedirect(e Engine, ownerID, repoID int64, oldRepoName, newRepoName string) error {
3333
oldRepoName = strings.ToLower(oldRepoName)
3434
newRepoName = strings.ToLower(newRepoName)
3535

36-
if err := deleteRepoRedirect(ctx.e, ownerID, newRepoName); err != nil {
36+
if err := deleteRepoRedirect(e, ownerID, newRepoName); err != nil {
3737
return err
3838
}
3939

40-
if _, err := ctx.e.Insert(&RepoRedirect{
40+
if _, err := e.Insert(&RepoRedirect{
4141
OwnerID: ownerID,
4242
LowerName: oldRepoName,
4343
RedirectRepoID: repoID,

models/repo_redirect_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -26,7 +26,7 @@ func TestNewRepoRedirect(t *testing.T) {
2626
assert.NoError(t, PrepareTestDatabase())
2727

2828
repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository)
29-
assert.NoError(t, NewRepoRedirect(DefaultDBContext(), repo.OwnerID, repo.ID, repo.Name, "newreponame"))
29+
assert.NoError(t, newRepoRedirect(x, repo.OwnerID, repo.ID, repo.Name, "newreponame"))
3030

3131
AssertExistsAndLoadBean(t, &RepoRedirect{
3232
OwnerID: repo.OwnerID,
@@ -45,7 +45,7 @@ func TestNewRepoRedirect2(t *testing.T) {
4545
assert.NoError(t, PrepareTestDatabase())
4646

4747
repo := AssertExistsAndLoadBean(t, &Repository{ID: 1}).(*Repository)
48-
assert.NoError(t, NewRepoRedirect(DefaultDBContext(), repo.OwnerID, repo.ID, repo.Name, "oldrepo1"))
48+
assert.NoError(t, newRepoRedirect(x, repo.OwnerID, repo.ID, repo.Name, "oldrepo1"))
4949

5050
AssertExistsAndLoadBean(t, &RepoRedirect{
5151
OwnerID: repo.OwnerID,
@@ -64,7 +64,7 @@ func TestNewRepoRedirect3(t *testing.T) {
6464
assert.NoError(t, PrepareTestDatabase())
6565

6666
repo := AssertExistsAndLoadBean(t, &Repository{ID: 2}).(*Repository)
67-
assert.NoError(t, NewRepoRedirect(DefaultDBContext(), repo.OwnerID, repo.ID, repo.Name, "newreponame"))
67+
assert.NoError(t, newRepoRedirect(x, repo.OwnerID, repo.ID, repo.Name, "newreponame"))
6868

6969
AssertExistsAndLoadBean(t, &RepoRedirect{
7070
OwnerID: repo.OwnerID,

models/user.go

Lines changed: 20 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -863,6 +863,10 @@ func CreateUser(u *User) (err error) {
863863
return ErrUserAlreadyExist{u.Name}
864864
}
865865

866+
if err = deleteUserRedirect(sess, u.Name); err != nil {
867+
return err
868+
}
869+
866870
u.Email = strings.ToLower(u.Email)
867871
isExist, err = sess.
868872
Where("email=?", u.Email).
@@ -973,6 +977,7 @@ func VerifyActiveEmailCode(code, email string) *EmailAddress {
973977

974978
// ChangeUserName changes all corresponding setting from old user name to new one.
975979
func ChangeUserName(u *User, newUserName string) (err error) {
980+
oldUserName := u.Name
976981
if err = IsUsableUsername(newUserName); err != nil {
977982
return err
978983
}
@@ -990,16 +995,28 @@ func ChangeUserName(u *User, newUserName string) (err error) {
990995
return ErrUserAlreadyExist{newUserName}
991996
}
992997

993-
if _, err = sess.Exec("UPDATE `repository` SET owner_name=? WHERE owner_name=?", newUserName, u.Name); err != nil {
998+
if _, err = sess.Exec("UPDATE `repository` SET owner_name=? WHERE owner_name=?", newUserName, oldUserName); err != nil {
994999
return fmt.Errorf("Change repo owner name: %v", err)
9951000
}
9961001

9971002
// Do not fail if directory does not exist
998-
if err = os.Rename(UserPath(u.Name), UserPath(newUserName)); err != nil && !os.IsNotExist(err) {
1003+
if err = os.Rename(UserPath(oldUserName), UserPath(newUserName)); err != nil && !os.IsNotExist(err) {
9991004
return fmt.Errorf("Rename user directory: %v", err)
10001005
}
10011006

1002-
return sess.Commit()
1007+
if err = newUserRedirect(sess, u.ID, oldUserName, newUserName); err != nil {
1008+
return err
1009+
}
1010+
1011+
if err = sess.Commit(); err != nil {
1012+
if err2 := os.Rename(UserPath(newUserName), UserPath(oldUserName)); err2 != nil && !os.IsNotExist(err2) {
1013+
log.Critical("Unable to rollback directory change during failed username change from: %s to: %s. DB Error: %v. Filesystem Error: %v", oldUserName, newUserName, err, err2)
1014+
return fmt.Errorf("failed to rollback directory change during failed username change from: %s to: %s. DB Error: %w. Filesystem Error: %v", oldUserName, newUserName, err, err2)
1015+
}
1016+
return err
1017+
}
1018+
1019+
return nil
10031020
}
10041021

10051022
// checkDupEmail checks whether there are the same email with the user

0 commit comments

Comments
 (0)