mirror of
				https://github.com/go-gitea/gitea.git
				synced 2025-10-31 19:06:18 +01:00 
			
		
		
		
	Move organization related structs into sub package (#18518)
* Move organization related structs into sub package * Fix test * Fix lint * Move more functions into sub packages * Fix bug * Fix test * Update models/organization/team_repo.go Co-authored-by: KN4CK3R <admin@oldschoolhack.me> * Apply suggestions from code review Co-authored-by: KN4CK3R <admin@oldschoolhack.me> * Fix fmt * Follow suggestion from @Gusted * Fix test * Fix test * Fix bug * Use ctx but db.DefaultContext on routers * Fix bug * Fix bug * fix bug * Update models/organization/team_user.go * Fix bug Co-authored-by: KN4CK3R <admin@oldschoolhack.me> Co-authored-by: wxiaoguang <wxiaoguang@gmail.com>
This commit is contained in:
		
							
								
								
									
										87
									
								
								models/repo/user_repo.go
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										87
									
								
								models/repo/user_repo.go
									
									
									
									
									
										Normal file
									
								
							| @@ -0,0 +1,87 @@ | ||||
| // Copyright 2022 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 repo | ||||
|  | ||||
| import ( | ||||
| 	"context" | ||||
|  | ||||
| 	"code.gitea.io/gitea/models/db" | ||||
| 	user_model "code.gitea.io/gitea/models/user" | ||||
| ) | ||||
|  | ||||
| // GetStarredRepos returns the repos starred by a particular user | ||||
| func GetStarredRepos(userID int64, private bool, listOptions db.ListOptions) ([]*Repository, error) { | ||||
| 	sess := db.GetEngine(db.DefaultContext).Where("star.uid=?", userID). | ||||
| 		Join("LEFT", "star", "`repository`.id=`star`.repo_id") | ||||
| 	if !private { | ||||
| 		sess = sess.And("is_private=?", false) | ||||
| 	} | ||||
|  | ||||
| 	if listOptions.Page != 0 { | ||||
| 		sess = db.SetSessionPagination(sess, &listOptions) | ||||
|  | ||||
| 		repos := make([]*Repository, 0, listOptions.PageSize) | ||||
| 		return repos, sess.Find(&repos) | ||||
| 	} | ||||
|  | ||||
| 	repos := make([]*Repository, 0, 10) | ||||
| 	return repos, sess.Find(&repos) | ||||
| } | ||||
|  | ||||
| // GetWatchedRepos returns the repos watched by a particular user | ||||
| func GetWatchedRepos(userID int64, private bool, listOptions db.ListOptions) ([]*Repository, int64, error) { | ||||
| 	sess := db.GetEngine(db.DefaultContext).Where("watch.user_id=?", userID). | ||||
| 		And("`watch`.mode<>?", WatchModeDont). | ||||
| 		Join("LEFT", "watch", "`repository`.id=`watch`.repo_id") | ||||
| 	if !private { | ||||
| 		sess = sess.And("is_private=?", false) | ||||
| 	} | ||||
|  | ||||
| 	if listOptions.Page != 0 { | ||||
| 		sess = db.SetSessionPagination(sess, &listOptions) | ||||
|  | ||||
| 		repos := make([]*Repository, 0, listOptions.PageSize) | ||||
| 		total, err := sess.FindAndCount(&repos) | ||||
| 		return repos, total, err | ||||
| 	} | ||||
|  | ||||
| 	repos := make([]*Repository, 0, 10) | ||||
| 	total, err := sess.FindAndCount(&repos) | ||||
| 	return repos, total, err | ||||
| } | ||||
|  | ||||
| // CountUserRepositories returns number of repositories user owns. | ||||
| // Argument private only takes effect when it is false, | ||||
| // set it true to count all repositories. | ||||
| func CountUserRepositories(userID int64, private bool) int64 { | ||||
| 	return countRepositories(userID, private) | ||||
| } | ||||
|  | ||||
| func getRepositoryCount(e db.Engine, ownerID int64) (int64, error) { | ||||
| 	return e.Count(&Repository{OwnerID: ownerID}) | ||||
| } | ||||
|  | ||||
| func getPublicRepositoryCount(e db.Engine, u *user_model.User) (int64, error) { | ||||
| 	return e.Where("is_private = ?", false).Count(&Repository{OwnerID: u.ID}) | ||||
| } | ||||
|  | ||||
| func getPrivateRepositoryCount(e db.Engine, u *user_model.User) (int64, error) { | ||||
| 	return e.Where("is_private = ?", true).Count(&Repository{OwnerID: u.ID}) | ||||
| } | ||||
|  | ||||
| // GetRepositoryCount returns the total number of repositories of user. | ||||
| func GetRepositoryCount(ctx context.Context, ownerID int64) (int64, error) { | ||||
| 	return getRepositoryCount(db.GetEngine(ctx), ownerID) | ||||
| } | ||||
|  | ||||
| // GetPublicRepositoryCount returns the total number of public repositories of user. | ||||
| func GetPublicRepositoryCount(u *user_model.User) (int64, error) { | ||||
| 	return getPublicRepositoryCount(db.GetEngine(db.DefaultContext), u) | ||||
| } | ||||
|  | ||||
| // GetPrivateRepositoryCount returns the total number of private repositories of user. | ||||
| func GetPrivateRepositoryCount(u *user_model.User) (int64, error) { | ||||
| 	return getPrivateRepositoryCount(db.GetEngine(db.DefaultContext), u) | ||||
| } | ||||
		Reference in New Issue
	
	Block a user