mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-10-31 07:21:36 +01:00 
			
		
		
		
	Backport #31788 by @wolfogre Fix #31784. Before: <img width="1648" alt="image" src="https://github.com/user-attachments/assets/03f32545-4a85-42ed-bafc-2b193a5d8023"> After: <img width="1653" alt="image" src="https://github.com/user-attachments/assets/e5bcaf93-49cb-421f-aac1-5122bc488b02"> Co-authored-by: Jason Song <i@wolfogre.com>
This commit is contained in:
		
							parent
							
								
									e563297c34
								
							
						
					
					
						commit
						a39fe53252
					
				| @ -6,6 +6,7 @@ package git | |||||||
| import ( | import ( | ||||||
| 	"context" | 	"context" | ||||||
| 	"errors" | 	"errors" | ||||||
|  | 	"fmt" | ||||||
| 	"strings" | 	"strings" | ||||||
| 	"time" | 	"time" | ||||||
| 
 | 
 | ||||||
| @ -24,6 +25,7 @@ type LFSLock struct { | |||||||
| 	ID      int64            `xorm:"pk autoincr"` | 	ID      int64            `xorm:"pk autoincr"` | ||||||
| 	RepoID  int64            `xorm:"INDEX NOT NULL"` | 	RepoID  int64            `xorm:"INDEX NOT NULL"` | ||||||
| 	OwnerID int64            `xorm:"INDEX NOT NULL"` | 	OwnerID int64            `xorm:"INDEX NOT NULL"` | ||||||
|  | 	Owner   *user_model.User `xorm:"-"` | ||||||
| 	Path    string           `xorm:"TEXT"` | 	Path    string           `xorm:"TEXT"` | ||||||
| 	Created time.Time        `xorm:"created"` | 	Created time.Time        `xorm:"created"` | ||||||
| } | } | ||||||
| @ -37,6 +39,35 @@ func (l *LFSLock) BeforeInsert() { | |||||||
| 	l.Path = util.PathJoinRel(l.Path) | 	l.Path = util.PathJoinRel(l.Path) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | // LoadAttributes loads attributes of the lock. | ||||||
|  | func (l *LFSLock) LoadAttributes(ctx context.Context) error { | ||||||
|  | 	// Load owner | ||||||
|  | 	if err := l.LoadOwner(ctx); err != nil { | ||||||
|  | 		return fmt.Errorf("load owner: %w", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // LoadOwner loads owner of the lock. | ||||||
|  | func (l *LFSLock) LoadOwner(ctx context.Context) error { | ||||||
|  | 	if l.Owner != nil { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	owner, err := user_model.GetUserByID(ctx, l.OwnerID) | ||||||
|  | 	if err != nil { | ||||||
|  | 		if user_model.IsErrUserNotExist(err) { | ||||||
|  | 			l.Owner = user_model.NewGhostUser() | ||||||
|  | 			return nil | ||||||
|  | 		} | ||||||
|  | 		return err | ||||||
|  | 	} | ||||||
|  | 	l.Owner = owner | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
| // CreateLFSLock creates a new lock. | // CreateLFSLock creates a new lock. | ||||||
| func CreateLFSLock(ctx context.Context, repo *repo_model.Repository, lock *LFSLock) (*LFSLock, error) { | func CreateLFSLock(ctx context.Context, repo *repo_model.Repository, lock *LFSLock) (*LFSLock, error) { | ||||||
| 	dbCtx, committer, err := db.TxContext(ctx) | 	dbCtx, committer, err := db.TxContext(ctx) | ||||||
| @ -94,7 +125,7 @@ func GetLFSLockByID(ctx context.Context, id int64) (*LFSLock, error) { | |||||||
| } | } | ||||||
| 
 | 
 | ||||||
| // GetLFSLockByRepoID returns a list of locks of repository. | // GetLFSLockByRepoID returns a list of locks of repository. | ||||||
| func GetLFSLockByRepoID(ctx context.Context, repoID int64, page, pageSize int) ([]*LFSLock, error) { | func GetLFSLockByRepoID(ctx context.Context, repoID int64, page, pageSize int) (LFSLockList, error) { | ||||||
| 	e := db.GetEngine(ctx) | 	e := db.GetEngine(ctx) | ||||||
| 	if page >= 0 && pageSize > 0 { | 	if page >= 0 && pageSize > 0 { | ||||||
| 		start := 0 | 		start := 0 | ||||||
| @ -103,7 +134,7 @@ func GetLFSLockByRepoID(ctx context.Context, repoID int64, page, pageSize int) ( | |||||||
| 		} | 		} | ||||||
| 		e.Limit(pageSize, start) | 		e.Limit(pageSize, start) | ||||||
| 	} | 	} | ||||||
| 	lfsLocks := make([]*LFSLock, 0, pageSize) | 	lfsLocks := make(LFSLockList, 0, pageSize) | ||||||
| 	return lfsLocks, e.Find(&lfsLocks, &LFSLock{RepoID: repoID}) | 	return lfsLocks, e.Find(&lfsLocks, &LFSLock{RepoID: repoID}) | ||||||
| } | } | ||||||
| 
 | 
 | ||||||
|  | |||||||
							
								
								
									
										54
									
								
								models/git/lfs_lock_list.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										54
									
								
								models/git/lfs_lock_list.go
									
									
									
									
									
										Normal file
									
								
							| @ -0,0 +1,54 @@ | |||||||
|  | // Copyright 2024 The Gitea Authors. All rights reserved. | ||||||
|  | // SPDX-License-Identifier: MIT | ||||||
|  | 
 | ||||||
|  | package git | ||||||
|  | 
 | ||||||
|  | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
|  | 
 | ||||||
|  | 	"code.gitea.io/gitea/models/db" | ||||||
|  | 	user_model "code.gitea.io/gitea/models/user" | ||||||
|  | 	"code.gitea.io/gitea/modules/container" | ||||||
|  | ) | ||||||
|  | 
 | ||||||
|  | // LFSLockList is a list of LFSLock | ||||||
|  | type LFSLockList []*LFSLock | ||||||
|  | 
 | ||||||
|  | // LoadAttributes loads the attributes for the given locks | ||||||
|  | func (locks LFSLockList) LoadAttributes(ctx context.Context) error { | ||||||
|  | 	if len(locks) == 0 { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	if err := locks.LoadOwner(ctx); err != nil { | ||||||
|  | 		return fmt.Errorf("load owner: %w", err) | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
|  | 
 | ||||||
|  | // LoadOwner loads the owner of the locks | ||||||
|  | func (locks LFSLockList) LoadOwner(ctx context.Context) error { | ||||||
|  | 	if len(locks) == 0 { | ||||||
|  | 		return nil | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	usersIDs := container.FilterSlice(locks, func(lock *LFSLock) (int64, bool) { | ||||||
|  | 		return lock.OwnerID, true | ||||||
|  | 	}) | ||||||
|  | 	users := make(map[int64]*user_model.User, len(usersIDs)) | ||||||
|  | 	if err := db.GetEngine(ctx). | ||||||
|  | 		In("id", usersIDs). | ||||||
|  | 		Find(&users); err != nil { | ||||||
|  | 		return fmt.Errorf("find users: %w", err) | ||||||
|  | 	} | ||||||
|  | 	for _, v := range locks { | ||||||
|  | 		v.Owner = users[v.OwnerID] | ||||||
|  | 		if v.Owner == nil { // not exist | ||||||
|  | 			v.Owner = user_model.NewGhostUser() | ||||||
|  | 		} | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
|  | 	return nil | ||||||
|  | } | ||||||
| @ -95,6 +95,11 @@ func LFSLocks(ctx *context.Context) { | |||||||
| 		ctx.ServerError("LFSLocks", err) | 		ctx.ServerError("LFSLocks", err) | ||||||
| 		return | 		return | ||||||
| 	} | 	} | ||||||
|  | 	if err := lfsLocks.LoadAttributes(ctx); err != nil { | ||||||
|  | 		ctx.ServerError("LFSLocks", err) | ||||||
|  | 		return | ||||||
|  | 	} | ||||||
|  | 
 | ||||||
| 	ctx.Data["LFSLocks"] = lfsLocks | 	ctx.Data["LFSLocks"] = lfsLocks | ||||||
| 
 | 
 | ||||||
| 	if len(lfsLocks) == 0 { | 	if len(lfsLocks) == 0 { | ||||||
|  | |||||||
| @ -30,9 +30,9 @@ | |||||||
| 								{{end}} | 								{{end}} | ||||||
| 							</td> | 							</td> | ||||||
| 							<td> | 							<td> | ||||||
| 								<a href="{{$.Owner.HomeLink}}"> | 								<a href="{{$lock.Owner.HomeLink}}"> | ||||||
| 									{{ctx.AvatarUtils.Avatar $.Owner}} | 									{{ctx.AvatarUtils.Avatar $lock.Owner}} | ||||||
| 									{{$.Owner.DisplayName}} | 									{{$lock.Owner.DisplayName}} | ||||||
| 								</a> | 								</a> | ||||||
| 							</td> | 							</td> | ||||||
| 							<td>{{TimeSince .Created ctx.Locale}}</td> | 							<td>{{TimeSince .Created ctx.Locale}}</td> | ||||||
|  | |||||||
| @ -4,12 +4,21 @@ | |||||||
| package integration | package integration | ||||||
| 
 | 
 | ||||||
| import ( | import ( | ||||||
|  | 	"context" | ||||||
|  | 	"fmt" | ||||||
| 	"net/http" | 	"net/http" | ||||||
|  | 	"strings" | ||||||
| 	"testing" | 	"testing" | ||||||
| 
 | 
 | ||||||
|  | 	repo_model "code.gitea.io/gitea/models/repo" | ||||||
|  | 	"code.gitea.io/gitea/models/unittest" | ||||||
|  | 	user_model "code.gitea.io/gitea/models/user" | ||||||
|  | 	"code.gitea.io/gitea/modules/lfs" | ||||||
|  | 	api "code.gitea.io/gitea/modules/structs" | ||||||
| 	"code.gitea.io/gitea/tests" | 	"code.gitea.io/gitea/tests" | ||||||
| 
 | 
 | ||||||
| 	"github.com/stretchr/testify/assert" | 	"github.com/stretchr/testify/assert" | ||||||
|  | 	"github.com/stretchr/testify/require" | ||||||
| ) | ) | ||||||
| 
 | 
 | ||||||
| // check that files stored in LFS render properly in the web UI | // check that files stored in LFS render properly in the web UI | ||||||
| @ -81,3 +90,56 @@ func TestLFSRender(t *testing.T) { | |||||||
| 		assert.Contains(t, content, "Testing READMEs in LFS") | 		assert.Contains(t, content, "Testing READMEs in LFS") | ||||||
| 	}) | 	}) | ||||||
| } | } | ||||||
|  | 
 | ||||||
|  | // TestLFSLockView tests the LFS lock view on settings page of repositories | ||||||
|  | func TestLFSLockView(t *testing.T) { | ||||||
|  | 	defer tests.PrepareTestEnv(t)() | ||||||
|  | 
 | ||||||
|  | 	user2 := unittest.AssertExistsAndLoadBean(t, &user_model.User{ID: 2})       // in org 3 | ||||||
|  | 	repo3 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 3}) // own by org 3 | ||||||
|  | 	session := loginUser(t, user2.Name) | ||||||
|  | 
 | ||||||
|  | 	// create a lock | ||||||
|  | 	lockPath := "test_lfs_lock_view.zip" | ||||||
|  | 	lockID := "" | ||||||
|  | 	{ | ||||||
|  | 		req := NewRequestWithJSON(t, "POST", fmt.Sprintf("/%s.git/info/lfs/locks", repo3.FullName()), map[string]string{"path": lockPath}) | ||||||
|  | 		req.Header.Set("Accept", lfs.AcceptHeader) | ||||||
|  | 		req.Header.Set("Content-Type", lfs.MediaType) | ||||||
|  | 		resp := session.MakeRequest(t, req, http.StatusCreated) | ||||||
|  | 		lockResp := &api.LFSLockResponse{} | ||||||
|  | 		DecodeJSON(t, resp, lockResp) | ||||||
|  | 		lockID = lockResp.Lock.ID | ||||||
|  | 	} | ||||||
|  | 	defer func() { | ||||||
|  | 		// release the lock | ||||||
|  | 		req := NewRequestWithJSON(t, "POST", fmt.Sprintf("/%s.git/info/lfs/locks/%s/unlock", repo3.FullName(), lockID), map[string]string{}) | ||||||
|  | 		req.Header.Set("Accept", lfs.AcceptHeader) | ||||||
|  | 		req.Header.Set("Content-Type", lfs.MediaType) | ||||||
|  | 		session.MakeRequest(t, req, http.StatusOK) | ||||||
|  | 	}() | ||||||
|  | 
 | ||||||
|  | 	t.Run("owner name", func(t *testing.T) { | ||||||
|  | 		defer tests.PrintCurrentTest(t)() | ||||||
|  | 
 | ||||||
|  | 		// make sure the display names are different, or the test is meaningless | ||||||
|  | 		require.NoError(t, repo3.LoadOwner(context.Background())) | ||||||
|  | 		require.NotEqual(t, user2.DisplayName(), repo3.Owner.DisplayName()) | ||||||
|  | 
 | ||||||
|  | 		req := NewRequest(t, "GET", fmt.Sprintf("/%s/settings/lfs/locks", repo3.FullName())) | ||||||
|  | 		resp := session.MakeRequest(t, req, http.StatusOK) | ||||||
|  | 
 | ||||||
|  | 		doc := NewHTMLParser(t, resp.Body).doc | ||||||
|  | 
 | ||||||
|  | 		tr := doc.Find("table#lfs-files-locks-table tbody tr") | ||||||
|  | 		require.Equal(t, 1, tr.Length()) | ||||||
|  | 
 | ||||||
|  | 		td := tr.First().Find("td") | ||||||
|  | 		require.Equal(t, 4, td.Length()) | ||||||
|  | 
 | ||||||
|  | 		// path | ||||||
|  | 		assert.Equal(t, lockPath, strings.TrimSpace(td.Eq(0).Text())) | ||||||
|  | 		// owner name | ||||||
|  | 		assert.Equal(t, user2.DisplayName(), strings.TrimSpace(td.Eq(1).Text())) | ||||||
|  | 	}) | ||||||
|  | } | ||||||
|  | |||||||
		Loading…
	
	
			
			x
			
			
		
	
		Reference in New Issue
	
	Block a user