Skip to content

Commit 91dce87

Browse files
JakobDev6543
JakobDev
authored and
6543
committed
Delete repos of org when purge delete user (go-gitea#27273)
Fixes https://codeberg.org/forgejo/forgejo/issues/1514 I had to remove `RenameOrganization` to avoid circular import. We should really add some foreign keys to the database.
1 parent 1848858 commit 91dce87

File tree

8 files changed

+67
-42
lines changed

8 files changed

+67
-42
lines changed

‎routers/api/v1/org/org.go

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -385,7 +385,7 @@ func Delete(ctx *context.APIContext) {
385385
// "404":
386386
// "$ref": "#/responses/notFound"
387387

388-
if err := org.DeleteOrganization(ctx.Org.Organization); err != nil {
388+
if err := org.DeleteOrganization(ctx.Org.Organization, false); err != nil {
389389
ctx.Error(http.StatusInternalServerError, "DeleteOrganization", err)
390390
return
391391
}

‎routers/web/org/setting.go

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ func SettingsPost(ctx *context.Context) {
7575

7676
// Check if organization name has been changed.
7777
if nameChanged {
78-
err := org_service.RenameOrganization(ctx, org, form.Name)
78+
err := user_service.RenameUser(ctx, org.AsUser(), form.Name)
7979
switch {
8080
case user_model.IsErrUserAlreadyExist(err):
8181
ctx.Data["OrgName"] = true
@@ -180,7 +180,7 @@ func SettingsDelete(ctx *context.Context) {
180180
return
181181
}
182182

183-
if err := org_service.DeleteOrganization(ctx.Org.Organization); err != nil {
183+
if err := org_service.DeleteOrganization(ctx.Org.Organization, false); err != nil {
184184
if models.IsErrUserOwnRepos(err) {
185185
ctx.Flash.Error(ctx.Tr("form.org_still_own_repo"))
186186
ctx.Redirect(ctx.Org.OrgLink + "/settings/delete")

‎services/org/org.go

Lines changed: 9 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,6 @@
44
package org
55

66
import (
7-
"context"
87
"fmt"
98

109
"code.gitea.io/gitea/models"
@@ -15,17 +14,24 @@ import (
1514
user_model "code.gitea.io/gitea/models/user"
1615
"code.gitea.io/gitea/modules/storage"
1716
"code.gitea.io/gitea/modules/util"
18-
user_service "code.gitea.io/gitea/services/user"
17+
repo_service "code.gitea.io/gitea/services/repository"
1918
)
2019

2120
// DeleteOrganization completely and permanently deletes everything of organization.
22-
func DeleteOrganization(org *org_model.Organization) error {
21+
func DeleteOrganization(org *org_model.Organization, purge bool) error {
2322
ctx, commiter, err := db.TxContext(db.DefaultContext)
2423
if err != nil {
2524
return err
2625
}
2726
defer commiter.Close()
2827

28+
if purge {
29+
err := repo_service.DeleteOwnerRepositoriesDirectly(ctx, org.AsUser())
30+
if err != nil {
31+
return err
32+
}
33+
}
34+
2935
// Check ownership of repository.
3036
count, err := repo_model.CountRepositories(ctx, repo_model.CountRepositoryOptions{OwnerID: org.ID})
3137
if err != nil {
@@ -67,8 +73,3 @@ func DeleteOrganization(org *org_model.Organization) error {
6773

6874
return nil
6975
}
70-
71-
// RenameOrganization renames an organization.
72-
func RenameOrganization(ctx context.Context, org *org_model.Organization, newName string) error {
73-
return user_service.RenameUser(ctx, org.AsUser(), newName)
74-
}

‎services/org/org_test.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -24,17 +24,17 @@ func TestMain(m *testing.M) {
2424
func TestDeleteOrganization(t *testing.T) {
2525
assert.NoError(t, unittest.PrepareTestDatabase())
2626
org := unittest.AssertExistsAndLoadBean(t, &organization.Organization{ID: 6})
27-
assert.NoError(t, DeleteOrganization(org))
27+
assert.NoError(t, DeleteOrganization(org), false)
2828
unittest.AssertNotExistsBean(t, &organization.Organization{ID: 6})
2929
unittest.AssertNotExistsBean(t, &organization.OrgUser{OrgID: 6})
3030
unittest.AssertNotExistsBean(t, &organization.Team{OrgID: 6})
3131

3232
org = unittest.AssertExistsAndLoadBean(t, &organization.Organization{ID: 3})
33-
err := DeleteOrganization(org)
33+
err := DeleteOrganization(org, false)
3434
assert.Error(t, err)
3535
assert.True(t, models.IsErrUserOwnRepos(err))
3636

3737
user := unittest.AssertExistsAndLoadBean(t, &organization.Organization{ID: 5})
38-
assert.Error(t, DeleteOrganization(user))
38+
assert.Error(t, DeleteOrganization(user), false)
3939
unittest.CheckConsistencyFor(t, &user_model.User{}, &organization.Team{})
4040
}

‎services/repository/delete.go

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -422,3 +422,30 @@ func RemoveRepositoryFromTeam(ctx context.Context, t *organization.Team, repoID
422422

423423
return committer.Commit()
424424
}
425+
426+
// DeleteOwnerRepositoriesDirectly calls DeleteRepositoryDirectly for all repos of the given owner
427+
func DeleteOwnerRepositoriesDirectly(ctx context.Context, owner *user_model.User) error {
428+
for {
429+
repos, _, err := repo_model.GetUserRepositories(&repo_model.SearchRepoOptions{
430+
ListOptions: db.ListOptions{
431+
PageSize: repo_model.RepositoryListDefaultPageSize,
432+
Page: 1,
433+
},
434+
Private: true,
435+
OwnerID: owner.ID,
436+
Actor: owner,
437+
})
438+
if err != nil {
439+
return fmt.Errorf("GetUserRepositories: %w", err)
440+
}
441+
if len(repos) == 0 {
442+
break
443+
}
444+
for _, repo := range repos {
445+
if err := DeleteRepositoryDirectly(ctx, owner, owner.ID, repo.ID); err != nil {
446+
return fmt.Errorf("unable to delete repository %s for %s[%d]. Error: %w", repo.Name, owner.Name, owner.ID, err)
447+
}
448+
}
449+
}
450+
return nil
451+
}

‎services/repository/delete_test.go

Lines changed: 13 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// Copyright 2017 The Gitea Authors. All rights reserved.
22
// SPDX-License-Identifier: MIT
33

4-
package repository
4+
package repository_test
55

66
import (
77
"testing"
@@ -10,6 +10,8 @@ import (
1010
"code.gitea.io/gitea/models/organization"
1111
repo_model "code.gitea.io/gitea/models/repo"
1212
"code.gitea.io/gitea/models/unittest"
13+
user_model "code.gitea.io/gitea/models/user"
14+
repo_service "code.gitea.io/gitea/services/repository"
1315

1416
"github.com/stretchr/testify/assert"
1517
)
@@ -19,7 +21,7 @@ func TestTeam_HasRepository(t *testing.T) {
1921

2022
test := func(teamID, repoID int64, expected bool) {
2123
team := unittest.AssertExistsAndLoadBean(t, &organization.Team{ID: teamID})
22-
assert.Equal(t, expected, HasRepository(db.DefaultContext, team, repoID))
24+
assert.Equal(t, expected, repo_service.HasRepository(db.DefaultContext, team, repoID))
2325
}
2426
test(1, 1, false)
2527
test(1, 3, true)
@@ -35,11 +37,19 @@ func TestTeam_RemoveRepository(t *testing.T) {
3537

3638
testSuccess := func(teamID, repoID int64) {
3739
team := unittest.AssertExistsAndLoadBean(t, &organization.Team{ID: teamID})
38-
assert.NoError(t, RemoveRepositoryFromTeam(db.DefaultContext, team, repoID))
40+
assert.NoError(t, repo_service.RemoveRepositoryFromTeam(db.DefaultContext, team, repoID))
3941
unittest.AssertNotExistsBean(t, &organization.TeamRepo{TeamID: teamID, RepoID: repoID})
4042
unittest.CheckConsistencyFor(t, &organization.Team{ID: teamID}, &repo_model.Repository{ID: repoID})
4143
}
4244
testSuccess(2, 3)
4345
testSuccess(2, 5)
4446
testSuccess(1, unittest.NonexistentID)
4547
}
48+
49+
func TestDeleteOwnerRepositoriesDirectly(t *testing.T) {
50+
unittest.PrepareTestEnv(t)
51+
52+
user := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})
53+
54+
assert.NoError(t, repo_service.DeleteOwnerRepositoriesDirectly(db.DefaultContext, user))
55+
}

‎services/repository/lfs_test.go

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// Copyright 2023 The Gitea Authors. All rights reserved.
22
// SPDX-License-Identifier: MIT
33

4-
package repository
4+
package repository_test
55

66
import (
77
"bytes"
@@ -16,12 +16,13 @@ import (
1616
"code.gitea.io/gitea/modules/lfs"
1717
"code.gitea.io/gitea/modules/setting"
1818
"code.gitea.io/gitea/modules/storage"
19+
repo_service "code.gitea.io/gitea/services/repository"
1920

2021
"github.com/stretchr/testify/assert"
2122
)
2223

2324
func TestGarbageCollectLFSMetaObjects(t *testing.T) {
24-
assert.NoError(t, unittest.PrepareTestDatabase())
25+
unittest.PrepareTestEnv(t)
2526

2627
setting.LFS.StartServer = true
2728
err := storage.Init()
@@ -35,7 +36,7 @@ func TestGarbageCollectLFSMetaObjects(t *testing.T) {
3536
lfsOid := storeObjectInRepo(t, repo.ID, &lfsContent)
3637

3738
// gc
38-
err = GarbageCollectLFSMetaObjects(context.Background(), GarbageCollectLFSMetaObjectsOptions{
39+
err = repo_service.GarbageCollectLFSMetaObjects(context.Background(), repo_service.GarbageCollectLFSMetaObjectsOptions{
3940
AutoFix: true,
4041
OlderThan: time.Now().Add(7 * 24 * time.Hour).Add(5 * 24 * time.Hour),
4142
UpdatedLessRecentlyThan: time.Now().Add(7 * 24 * time.Hour).Add(3 * 24 * time.Hour),

‎services/user/user.go

Lines changed: 8 additions & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ import (
2424
"code.gitea.io/gitea/modules/storage"
2525
"code.gitea.io/gitea/modules/util"
2626
"code.gitea.io/gitea/services/agit"
27+
org_service "code.gitea.io/gitea/services/org"
2728
"code.gitea.io/gitea/services/packages"
2829
container_service "code.gitea.io/gitea/services/packages/container"
2930
repo_service "code.gitea.io/gitea/services/repository"
@@ -158,27 +159,9 @@ func DeleteUser(ctx context.Context, u *user_model.User, purge bool) error {
158159
//
159160
// An alternative option here would be write a DeleteAllRepositoriesForUserID function which would delete all of the repos
160161
// but such a function would likely get out of date
161-
for {
162-
repos, _, err := repo_model.GetUserRepositories(&repo_model.SearchRepoOptions{
163-
ListOptions: db.ListOptions{
164-
PageSize: repo_model.RepositoryListDefaultPageSize,
165-
Page: 1,
166-
},
167-
Private: true,
168-
OwnerID: u.ID,
169-
Actor: u,
170-
})
171-
if err != nil {
172-
return fmt.Errorf("GetUserRepositories: %w", err)
173-
}
174-
if len(repos) == 0 {
175-
break
176-
}
177-
for _, repo := range repos {
178-
if err := repo_service.DeleteRepositoryDirectly(ctx, u, u.ID, repo.ID); err != nil {
179-
return fmt.Errorf("unable to delete repository %s for %s[%d]. Error: %w", repo.Name, u.Name, u.ID, err)
180-
}
181-
}
162+
err := repo_service.DeleteOwnerRepositoriesDirectly(ctx, u)
163+
if err != nil {
164+
return err
182165
}
183166

184167
// Remove from Organizations and delete last owner organizations
@@ -206,7 +189,10 @@ func DeleteUser(ctx context.Context, u *user_model.User, purge bool) error {
206189
for _, org := range orgs {
207190
if err := models.RemoveOrgUser(org.ID, u.ID); err != nil {
208191
if organization.IsErrLastOrgOwner(err) {
209-
err = organization.DeleteOrganization(ctx, org)
192+
err = org_service.DeleteOrganization(ctx, org, true)
193+
if err != nil {
194+
return fmt.Errorf("unable to delete organization %d: %w", org.ID, err)
195+
}
210196
}
211197
if err != nil {
212198
return fmt.Errorf("unable to remove user %s[%d] from org %s[%d]. Error: %w", u.Name, u.ID, org.Name, org.ID, err)

0 commit comments

Comments
 (0)