Move some repository related code into sub package (#19711)
* Move some repository related code into sub package * Move more repository functions out of models * Fix lint * Some performance optimization for webhooks and others * some refactors * Fix lint * Fix * Update modules/repository/delete.go Co-authored-by: delvh <dev.lh@web.de> * Fix test * Merge * Fix test * Fix test * Fix test * Fix test Co-authored-by: delvh <dev.lh@web.de>tokarchuk/v1.17
parent
ebeb6e7c71
commit
26095115f4
@ -0,0 +1,74 @@ |
||||
// Copyright 2017 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_test |
||||
|
||||
import ( |
||||
"testing" |
||||
|
||||
"code.gitea.io/gitea/models/db" |
||||
repo_model "code.gitea.io/gitea/models/repo" |
||||
"code.gitea.io/gitea/models/unittest" |
||||
|
||||
"github.com/stretchr/testify/assert" |
||||
) |
||||
|
||||
func TestRepoAssignees(t *testing.T) { |
||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
||||
|
||||
repo2 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 2}).(*repo_model.Repository) |
||||
users, err := repo_model.GetRepoAssignees(db.DefaultContext, repo2) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, users, 1) |
||||
assert.Equal(t, users[0].ID, int64(2)) |
||||
|
||||
repo21 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 21}).(*repo_model.Repository) |
||||
users, err = repo_model.GetRepoAssignees(db.DefaultContext, repo21) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, users, 3) |
||||
assert.Equal(t, users[0].ID, int64(15)) |
||||
assert.Equal(t, users[1].ID, int64(18)) |
||||
assert.Equal(t, users[2].ID, int64(16)) |
||||
} |
||||
|
||||
func TestRepoGetReviewers(t *testing.T) { |
||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
||||
|
||||
// test public repo
|
||||
repo1 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 1}).(*repo_model.Repository) |
||||
|
||||
ctx := db.DefaultContext |
||||
reviewers, err := repo_model.GetReviewers(ctx, repo1, 2, 2) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, reviewers, 4) |
||||
|
||||
// should include doer if doer is not PR poster.
|
||||
reviewers, err = repo_model.GetReviewers(ctx, repo1, 11, 2) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, reviewers, 4) |
||||
|
||||
// should not include PR poster, if PR poster would be otherwise eligible
|
||||
reviewers, err = repo_model.GetReviewers(ctx, repo1, 11, 4) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, reviewers, 3) |
||||
|
||||
// test private user repo
|
||||
repo2 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 2}).(*repo_model.Repository) |
||||
|
||||
reviewers, err = repo_model.GetReviewers(ctx, repo2, 2, 4) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, reviewers, 1) |
||||
assert.EqualValues(t, reviewers[0].ID, 2) |
||||
|
||||
// test private org repo
|
||||
repo3 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 3}).(*repo_model.Repository) |
||||
|
||||
reviewers, err = repo_model.GetReviewers(ctx, repo3, 2, 1) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, reviewers, 2) |
||||
|
||||
reviewers, err = repo_model.GetReviewers(ctx, repo3, 2, 2) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, reviewers, 1) |
||||
} |
@ -1,118 +0,0 @@ |
||||
// 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 models |
||||
|
||||
import ( |
||||
"bufio" |
||||
"bytes" |
||||
"context" |
||||
"strings" |
||||
|
||||
"code.gitea.io/gitea/models/db" |
||||
repo_model "code.gitea.io/gitea/models/repo" |
||||
"code.gitea.io/gitea/models/webhook" |
||||
"code.gitea.io/gitea/modules/log" |
||||
|
||||
"github.com/gobwas/glob" |
||||
) |
||||
|
||||
// GenerateRepoOptions contains the template units to generate
|
||||
type GenerateRepoOptions struct { |
||||
Name string |
||||
DefaultBranch string |
||||
Description string |
||||
Private bool |
||||
GitContent bool |
||||
Topics bool |
||||
GitHooks bool |
||||
Webhooks bool |
||||
Avatar bool |
||||
IssueLabels bool |
||||
} |
||||
|
||||
// IsValid checks whether at least one option is chosen for generation
|
||||
func (gro GenerateRepoOptions) IsValid() bool { |
||||
return gro.GitContent || gro.Topics || gro.GitHooks || gro.Webhooks || gro.Avatar || gro.IssueLabels // or other items as they are added
|
||||
} |
||||
|
||||
// GiteaTemplate holds information about a .gitea/template file
|
||||
type GiteaTemplate struct { |
||||
Path string |
||||
Content []byte |
||||
|
||||
globs []glob.Glob |
||||
} |
||||
|
||||
// Globs parses the .gitea/template globs or returns them if they were already parsed
|
||||
func (gt GiteaTemplate) Globs() []glob.Glob { |
||||
if gt.globs != nil { |
||||
return gt.globs |
||||
} |
||||
|
||||
gt.globs = make([]glob.Glob, 0) |
||||
scanner := bufio.NewScanner(bytes.NewReader(gt.Content)) |
||||
for scanner.Scan() { |
||||
line := strings.TrimSpace(scanner.Text()) |
||||
if line == "" || strings.HasPrefix(line, "#") { |
||||
continue |
||||
} |
||||
g, err := glob.Compile(line, '/') |
||||
if err != nil { |
||||
log.Info("Invalid glob expression '%s' (skipped): %v", line, err) |
||||
continue |
||||
} |
||||
gt.globs = append(gt.globs, g) |
||||
} |
||||
return gt.globs |
||||
} |
||||
|
||||
// GenerateWebhooks generates webhooks from a template repository
|
||||
func GenerateWebhooks(ctx context.Context, templateRepo, generateRepo *repo_model.Repository) error { |
||||
templateWebhooks, err := webhook.ListWebhooksByOpts(ctx, &webhook.ListWebhookOptions{RepoID: templateRepo.ID}) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
|
||||
for _, templateWebhook := range templateWebhooks { |
||||
generateWebhook := &webhook.Webhook{ |
||||
RepoID: generateRepo.ID, |
||||
URL: templateWebhook.URL, |
||||
HTTPMethod: templateWebhook.HTTPMethod, |
||||
ContentType: templateWebhook.ContentType, |
||||
Secret: templateWebhook.Secret, |
||||
HookEvent: templateWebhook.HookEvent, |
||||
IsActive: templateWebhook.IsActive, |
||||
Type: templateWebhook.Type, |
||||
OrgID: templateWebhook.OrgID, |
||||
Events: templateWebhook.Events, |
||||
Meta: templateWebhook.Meta, |
||||
} |
||||
if err := webhook.CreateWebhook(ctx, generateWebhook); err != nil { |
||||
return err |
||||
} |
||||
} |
||||
return nil |
||||
} |
||||
|
||||
// GenerateIssueLabels generates issue labels from a template repository
|
||||
func GenerateIssueLabels(ctx context.Context, templateRepo, generateRepo *repo_model.Repository) error { |
||||
templateLabels, err := GetLabelsByRepoID(ctx, templateRepo.ID, "", db.ListOptions{}) |
||||
if err != nil { |
||||
return err |
||||
} |
||||
|
||||
for _, templateLabel := range templateLabels { |
||||
generateLabel := &Label{ |
||||
RepoID: generateRepo.ID, |
||||
Name: templateLabel.Name, |
||||
Description: templateLabel.Description, |
||||
Color: templateLabel.Color, |
||||
} |
||||
if err := db.Insert(ctx, generateLabel); err != nil { |
||||
return err |
||||
} |
||||
} |
||||
return nil |
||||
} |
@ -1,704 +0,0 @@ |
||||
// Copyright 2017 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 models |
||||
|
||||
import ( |
||||
"context" |
||||
"fmt" |
||||
"strings" |
||||
|
||||
"code.gitea.io/gitea/models/db" |
||||
"code.gitea.io/gitea/models/organization" |
||||
"code.gitea.io/gitea/models/perm" |
||||
repo_model "code.gitea.io/gitea/models/repo" |
||||
"code.gitea.io/gitea/models/unit" |
||||
user_model "code.gitea.io/gitea/models/user" |
||||
"code.gitea.io/gitea/modules/container" |
||||
"code.gitea.io/gitea/modules/structs" |
||||
"code.gitea.io/gitea/modules/util" |
||||
|
||||
"xorm.io/builder" |
||||
) |
||||
|
||||
// RepositoryListDefaultPageSize is the default number of repositories
|
||||
// to load in memory when running administrative tasks on all (or almost
|
||||
// all) of them.
|
||||
// The number should be low enough to avoid filling up all RAM with
|
||||
// repository data...
|
||||
const RepositoryListDefaultPageSize = 64 |
||||
|
||||
// RepositoryList contains a list of repositories
|
||||
type RepositoryList []*repo_model.Repository |
||||
|
||||
func (repos RepositoryList) Len() int { |
||||
return len(repos) |
||||
} |
||||
|
||||
func (repos RepositoryList) Less(i, j int) bool { |
||||
return repos[i].FullName() < repos[j].FullName() |
||||
} |
||||
|
||||
func (repos RepositoryList) Swap(i, j int) { |
||||
repos[i], repos[j] = repos[j], repos[i] |
||||
} |
||||
|
||||
// FIXME: Remove in favor of maps.values when MIN_GO_VERSION >= 1.18
|
||||
func valuesRepository(m map[int64]*repo_model.Repository) []*repo_model.Repository { |
||||
values := make([]*repo_model.Repository, 0, len(m)) |
||||
for _, v := range m { |
||||
values = append(values, v) |
||||
} |
||||
return values |
||||
} |
||||
|
||||
// RepositoryListOfMap make list from values of map
|
||||
func RepositoryListOfMap(repoMap map[int64]*repo_model.Repository) RepositoryList { |
||||
return RepositoryList(valuesRepository(repoMap)) |
||||
} |
||||
|
||||
func (repos RepositoryList) loadAttributes(ctx context.Context) error { |
||||
if len(repos) == 0 { |
||||
return nil |
||||
} |
||||
|
||||
set := make(map[int64]struct{}) |
||||
repoIDs := make([]int64, len(repos)) |
||||
for i := range repos { |
||||
set[repos[i].OwnerID] = struct{}{} |
||||
repoIDs[i] = repos[i].ID |
||||
} |
||||
|
||||
// Load owners.
|
||||
users := make(map[int64]*user_model.User, len(set)) |
||||
if err := db.GetEngine(ctx). |
||||
Where("id > 0"). |
||||
In("id", container.KeysInt64(set)). |
||||
Find(&users); err != nil { |
||||
return fmt.Errorf("find users: %v", err) |
||||
} |
||||
for i := range repos { |
||||
repos[i].Owner = users[repos[i].OwnerID] |
||||
} |
||||
|
||||
// Load primary language.
|
||||
stats := make(repo_model.LanguageStatList, 0, len(repos)) |
||||
if err := db.GetEngine(ctx). |
||||
Where("`is_primary` = ? AND `language` != ?", true, "other"). |
||||
In("`repo_id`", repoIDs). |
||||
Find(&stats); err != nil { |
||||
return fmt.Errorf("find primary languages: %v", err) |
||||
} |
||||
stats.LoadAttributes() |
||||
for i := range repos { |
||||
for _, st := range stats { |
||||
if st.RepoID == repos[i].ID { |
||||
repos[i].PrimaryLanguage = st |
||||
break |
||||
} |
||||
} |
||||
} |
||||
|
||||
return nil |
||||
} |
||||
|
||||
// LoadAttributes loads the attributes for the given RepositoryList
|
||||
func (repos RepositoryList) LoadAttributes() error { |
||||
return repos.loadAttributes(db.DefaultContext) |
||||
} |
||||
|
||||
// SearchRepoOptions holds the search options
|
||||
type SearchRepoOptions struct { |
||||
db.ListOptions |
||||
Actor *user_model.User |
||||
Keyword string |
||||
OwnerID int64 |
||||
PriorityOwnerID int64 |
||||
TeamID int64 |
||||
OrderBy db.SearchOrderBy |
||||
Private bool // Include private repositories in results
|
||||
StarredByID int64 |
||||
WatchedByID int64 |
||||
AllPublic bool // Include also all public repositories of users and public organisations
|
||||
AllLimited bool // Include also all public repositories of limited organisations
|
||||
// None -> include public and private
|
||||
// True -> include just private
|
||||
// False -> include just public
|
||||
IsPrivate util.OptionalBool |
||||
// None -> include collaborative AND non-collaborative
|
||||
// True -> include just collaborative
|
||||
// False -> include just non-collaborative
|
||||
Collaborate util.OptionalBool |
||||
// None -> include forks AND non-forks
|
||||
// True -> include just forks
|
||||
// False -> include just non-forks
|
||||
Fork util.OptionalBool |
||||
// None -> include templates AND non-templates
|
||||
// True -> include just templates
|
||||
// False -> include just non-templates
|
||||
Template util.OptionalBool |
||||
// None -> include mirrors AND non-mirrors
|
||||
// True -> include just mirrors
|
||||
// False -> include just non-mirrors
|
||||
Mirror util.OptionalBool |
||||
// None -> include archived AND non-archived
|
||||
// True -> include just archived
|
||||
// False -> include just non-archived
|
||||
Archived util.OptionalBool |
||||
// only search topic name
|
||||
TopicOnly bool |
||||
// only search repositories with specified primary language
|
||||
Language string |
||||
// include description in keyword search
|
||||
IncludeDescription bool |
||||
// None -> include has milestones AND has no milestone
|
||||
// True -> include just has milestones
|
||||
// False -> include just has no milestone
|
||||
HasMilestones util.OptionalBool |
||||
// LowerNames represents valid lower names to restrict to
|
||||
LowerNames []string |
||||
} |
||||
|
||||
// SearchOrderBy is used to sort the result
|
||||
type SearchOrderBy string |
||||
|
||||
func (s SearchOrderBy) String() string { |
||||
return string(s) |
||||
} |
||||
|
||||
// Strings for sorting result
|
||||
const ( |
||||
SearchOrderByAlphabetically SearchOrderBy = "name ASC" |
||||
SearchOrderByAlphabeticallyReverse SearchOrderBy = "name DESC" |
||||
SearchOrderByLeastUpdated SearchOrderBy = "updated_unix ASC" |
||||
SearchOrderByRecentUpdated SearchOrderBy = "updated_unix DESC" |
||||
SearchOrderByOldest SearchOrderBy = "created_unix ASC" |
||||
SearchOrderByNewest SearchOrderBy = "created_unix DESC" |
||||
SearchOrderBySize SearchOrderBy = "size ASC" |
||||
SearchOrderBySizeReverse SearchOrderBy = "size DESC" |
||||
SearchOrderByID SearchOrderBy = "id ASC" |
||||
SearchOrderByIDReverse SearchOrderBy = "id DESC" |
||||
SearchOrderByStars SearchOrderBy = "num_stars ASC" |
||||
SearchOrderByStarsReverse SearchOrderBy = "num_stars DESC" |
||||
SearchOrderByForks SearchOrderBy = "num_forks ASC" |
||||
SearchOrderByForksReverse SearchOrderBy = "num_forks DESC" |
||||
) |
||||
|
||||
// userOwnedRepoCond returns user ownered repositories
|
||||
func userOwnedRepoCond(userID int64) builder.Cond { |
||||
return builder.Eq{ |
||||
"repository.owner_id": userID, |
||||
} |
||||
} |
||||
|
||||
// userAssignedRepoCond return user as assignee repositories list
|
||||
func userAssignedRepoCond(id string, userID int64) builder.Cond { |
||||
return builder.And( |
||||
builder.Eq{ |
||||
"repository.is_private": false, |
||||
}, |
||||
builder.In(id, |
||||
builder.Select("issue.repo_id").From("issue_assignees"). |
||||
InnerJoin("issue", "issue.id = issue_assignees.issue_id"). |
||||
Where(builder.Eq{ |
||||
"issue_assignees.assignee_id": userID, |
||||
}), |
||||
), |
||||
) |
||||
} |
||||
|
||||
// userCreateIssueRepoCond return user created issues repositories list
|
||||
func userCreateIssueRepoCond(id string, userID int64, isPull bool) builder.Cond { |
||||
return builder.And( |
||||
builder.Eq{ |
||||
"repository.is_private": false, |
||||
}, |
||||
builder.In(id, |
||||
builder.Select("issue.repo_id").From("issue"). |
||||
Where(builder.Eq{ |
||||
"issue.poster_id": userID, |
||||
"issue.is_pull": isPull, |
||||
}), |
||||
), |
||||
) |
||||
} |
||||
|
||||
// userMentionedRepoCond return user metinoed repositories list
|
||||
func userMentionedRepoCond(id string, userID int64) builder.Cond { |
||||
return builder.And( |
||||
builder.Eq{ |
||||
"repository.is_private": false, |
||||
}, |
||||
builder.In(id, |
||||
builder.Select("issue.repo_id").From("issue_user"). |
||||
InnerJoin("issue", "issue.id = issue_user.issue_id"). |
||||
Where(builder.Eq{ |
||||
"issue_user.is_mentioned": true, |
||||
"issue_user.uid": userID, |
||||
}), |
||||
), |
||||
) |
||||
} |
||||
|
||||
// teamUnitsRepoCond returns query condition for those repo id in the special org team with special units access
|
||||
func teamUnitsRepoCond(id string, userID, orgID, teamID int64, units ...unit.Type) builder.Cond { |
||||
return builder.In(id, |
||||
builder.Select("repo_id").From("team_repo").Where( |
||||
builder.Eq{ |
||||
"team_id": teamID, |
||||
}.And( |
||||
builder.Or( |
||||
// Check if the user is member of the team.
|
||||
builder.In( |
||||
"team_id", builder.Select("team_id").From("team_user").Where( |
||||
builder.Eq{ |
||||
"uid": userID, |
||||
}, |
||||
), |
||||
), |
||||
// Check if the user is in the owner team of the organisation.
|
||||
builder.Exists(builder.Select("team_id").From("team_user"). |
||||
Where(builder.Eq{ |
||||
"org_id": orgID, |
||||
"team_id": builder.Select("id").From("team").Where( |
||||
builder.Eq{ |
||||
"org_id": orgID, |
||||
"lower_name": strings.ToLower(organization.OwnerTeamName), |
||||
}), |
||||
"uid": userID, |
||||
}), |
||||
), |
||||
)).And( |
||||
builder.In( |
||||
"team_id", builder.Select("team_id").From("team_unit").Where( |
||||
builder.Eq{ |
||||
"`team_unit`.org_id": orgID, |
||||
}.And( |
||||
builder.In("`team_unit`.type", units), |
||||
), |
||||
), |
||||
), |
||||
), |
||||
)) |
||||
} |
||||
|
||||
// userCollaborationRepoCond returns user as collabrators repositories list
|
||||
func userCollaborationRepoCond(idStr string, userID int64) builder.Cond { |
||||
return builder.In(idStr, builder.Select("repo_id"). |
||||
From("`access`"). |
||||
Where(builder.And( |
||||
builder.Eq{"`access`.user_id": userID}, |
||||
builder.Gt{"`access`.mode": int(perm.AccessModeNone)}, |
||||
)), |
||||
) |
||||
} |
||||
|
||||
// userOrgTeamRepoCond selects repos that the given user has access to through team membership
|
||||
func userOrgTeamRepoCond(idStr string, userID int64) builder.Cond { |
||||
return builder.In(idStr, userOrgTeamRepoBuilder(userID)) |
||||
} |
||||
|
||||
// userOrgTeamRepoBuilder returns repo ids where user's teams can access.
|
||||
func userOrgTeamRepoBuilder(userID int64) *builder.Builder { |
||||
return builder.Select("`team_repo`.repo_id"). |
||||
From("team_repo"). |
||||
Join("INNER", "team_user", "`team_user`.team_id = `team_repo`.team_id"). |
||||
Where(builder.Eq{"`team_user`.uid": userID}) |
||||
} |
||||
|
||||
// userOrgTeamUnitRepoBuilder returns repo ids where user's teams can access the special unit.
|
||||
func userOrgTeamUnitRepoBuilder(userID int64, unitType unit.Type) *builder.Builder { |
||||
return userOrgTeamRepoBuilder(userID). |
||||
Join("INNER", "team_unit", "`team_unit`.team_id = `team_repo`.team_id"). |
||||
Where(builder.Eq{"`team_unit`.`type`": unitType}) |
||||
} |
||||
|
||||
// userOrgUnitRepoCond selects repos that the given user has access to through org and the special unit
|
||||
func userOrgUnitRepoCond(idStr string, userID, orgID int64, unitType unit.Type) builder.Cond { |
||||
return builder.In(idStr, |
||||
userOrgTeamUnitRepoBuilder(userID, unitType). |
||||
And(builder.Eq{"`team_unit`.org_id": orgID}), |
||||
) |
||||
} |
||||
|
||||
// userOrgPublicRepoCond returns the condition that one user could access all public repositories in organizations
|
||||
func userOrgPublicRepoCond(userID int64) builder.Cond { |
||||
return builder.And( |
||||
builder.Eq{"`repository`.is_private": false}, |
||||
builder.In("`repository`.owner_id", |
||||
builder.Select("`org_user`.org_id"). |
||||
From("org_user"). |
||||
Where(builder.Eq{"`org_user`.uid": userID}), |
||||
), |
||||
) |
||||
} |
||||
|
||||
// userOrgPublicRepoCondPrivate returns the condition that one user could access all public repositories in private organizations
|
||||
func userOrgPublicRepoCondPrivate(userID int64) builder.Cond { |
||||
return builder.And( |
||||
builder.Eq{"`repository`.is_private": false}, |
||||
builder.In("`repository`.owner_id", |
||||
builder.Select("`org_user`.org_id"). |
||||
From("org_user"). |
||||
Join("INNER", "`user`", "`user`.id = `org_user`.org_id"). |
||||
Where(builder.Eq{ |
||||
"`org_user`.uid": userID, |
||||
"`user`.`type`": user_model.UserTypeOrganization, |
||||
"`user`.visibility": structs.VisibleTypePrivate, |
||||
}), |
||||
), |
||||
) |
||||
} |
||||
|
||||
// userOrgPublicUnitRepoCond returns the condition that one user could access all public repositories in the special organization
|
||||
func userOrgPublicUnitRepoCond(userID, orgID int64) builder.Cond { |
||||
return userOrgPublicRepoCond(userID). |
||||
And(builder.Eq{"`repository`.owner_id": orgID}) |
||||
} |
||||
|
||||
// SearchRepositoryCondition creates a query condition according search repository options
|
||||
func SearchRepositoryCondition(opts *SearchRepoOptions) builder.Cond { |
||||
cond := builder.NewCond() |
||||
|
||||
if opts.Private { |
||||
if opts.Actor != nil && !opts.Actor.IsAdmin && opts.Actor.ID != opts.OwnerID { |
||||
// OK we're in the context of a User
|
||||
cond = cond.And(accessibleRepositoryCondition(opts.Actor)) |
||||
} |
||||
} else { |
||||
// Not looking at private organisations and users
|
||||
// We should be able to see all non-private repositories that
|
||||
// isn't in a private or limited organisation.
|
||||
cond = cond.And( |
||||
builder.Eq{"is_private": false}, |
||||
builder.NotIn("owner_id", builder.Select("id").From("`user`").Where( |
||||
builder.Or(builder.Eq{"visibility": structs.VisibleTypeLimited}, builder.Eq{"visibility": structs.VisibleTypePrivate}), |
||||
))) |
||||
} |
||||
|
||||
if opts.IsPrivate != util.OptionalBoolNone { |
||||
cond = cond.And(builder.Eq{"is_private": opts.IsPrivate.IsTrue()}) |
||||
} |
||||
|
||||
if opts.Template != util.OptionalBoolNone { |
||||
cond = cond.And(builder.Eq{"is_template": opts.Template == util.OptionalBoolTrue}) |
||||
} |
||||
|
||||
// Restrict to starred repositories
|
||||
if opts.StarredByID > 0 { |
||||
cond = cond.And(builder.In("id", builder.Select("repo_id").From("star").Where(builder.Eq{"uid": opts.StarredByID}))) |
||||
} |
||||
|
||||
// Restrict to watched repositories
|
||||
if opts.WatchedByID > 0 { |
||||
cond = cond.And(builder.In("id", builder.Select("repo_id").From("watch").Where(builder.Eq{"user_id": opts.WatchedByID}))) |
||||
} |
||||
|
||||
// Restrict repositories to those the OwnerID owns or contributes to as per opts.Collaborate
|
||||
if opts.OwnerID > 0 { |
||||
accessCond := builder.NewCond() |
||||
if opts.Collaborate != util.OptionalBoolTrue { |
||||
accessCond = builder.Eq{"owner_id": opts.OwnerID} |
||||
} |
||||
|
||||
if opts.Collaborate != util.OptionalBoolFalse { |
||||
// A Collaboration is:
|
||||
collaborateCond := builder.And( |
||||
// 1. Repository we don't own
|
||||
builder.Neq{"owner_id": opts.OwnerID}, |
||||
// 2. But we can see because of:
|
||||
builder.Or( |
||||
// A. We have access
|
||||
userCollaborationRepoCond("`repository`.id", opts.OwnerID), |
||||
// B. We are in a team for
|
||||
userOrgTeamRepoCond("`repository`.id", opts.OwnerID), |
||||
// C. Public repositories in organizations that we are member of
|
||||
userOrgPublicRepoCondPrivate(opts.OwnerID), |
||||
), |
||||
) |
||||
if !opts.Private { |
||||
collaborateCond = collaborateCond.And(builder.Expr("owner_id NOT IN (SELECT org_id FROM org_user WHERE org_user.uid = ? AND org_user.is_public = ?)", opts.OwnerID, false)) |
||||
} |
||||
|
||||
accessCond = accessCond.Or(collaborateCond) |
||||
} |
||||
|
||||
if opts.AllPublic { |
||||
accessCond = accessCond.Or(builder.Eq{"is_private": false}.And(builder.In("owner_id", builder.Select("`user`.id").From("`user`").Where(builder.Eq{"`user`.visibility": structs.VisibleTypePublic})))) |
||||
} |
||||
|
||||
if opts.AllLimited { |
||||
accessCond = accessCond.Or(builder.Eq{"is_private": false}.And(builder.In("owner_id", builder.Select("`user`.id").From("`user`").Where(builder.Eq{"`user`.visibility": structs.VisibleTypeLimited})))) |
||||
} |
||||
|
||||
cond = cond.And(accessCond) |
||||
} |
||||
|
||||
if opts.TeamID > 0 { |
||||
cond = cond.And(builder.In("`repository`.id", builder.Select("`team_repo`.repo_id").From("team_repo").Where(builder.Eq{"`team_repo`.team_id": opts.TeamID}))) |
||||
} |
||||
|
||||
if opts.Keyword != "" { |
||||
// separate keyword
|
||||
subQueryCond := builder.NewCond() |
||||
for _, v := range strings.Split(opts.Keyword, ",") { |
||||
if opts.TopicOnly { |
||||
subQueryCond = subQueryCond.Or(builder.Eq{"topic.name": strings.ToLower(v)}) |
||||
} else { |
||||
subQueryCond = subQueryCond.Or(builder.Like{"topic.name", strings.ToLower(v)}) |
||||
} |
||||
} |
||||
subQuery := builder.Select("repo_topic.repo_id").From("repo_topic"). |
||||
Join("INNER", "topic", "topic.id = repo_topic.topic_id"). |
||||
Where(subQueryCond). |
||||
GroupBy("repo_topic.repo_id") |
||||
|
||||
keywordCond := builder.In("id", subQuery) |
||||
if !opts.TopicOnly { |
||||
likes := builder.NewCond() |
||||
for _, v := range strings.Split(opts.Keyword, ",") { |
||||
likes = likes.Or(builder.Like{"lower_name", strings.ToLower(v)}) |
||||
|
||||
// If the string looks like "org/repo", match against that pattern too
|
||||
if opts.TeamID == 0 && strings.Count(opts.Keyword, "/") == 1 { |
||||
pieces := strings.Split(opts.Keyword, "/") |
||||
ownerName := pieces[0] |
||||
repoName := pieces[1] |
||||
likes = likes.Or(builder.And(builder.Like{"owner_name", strings.ToLower(ownerName)}, builder.Like{"lower_name", strings.ToLower(repoName)})) |
||||
} |
||||
|
||||
if opts.IncludeDescription { |
||||
likes = likes.Or(builder.Like{"LOWER(description)", strings.ToLower(v)}) |
||||
} |
||||
} |
||||
keywordCond = keywordCond.Or(likes) |
||||
} |
||||
cond = cond.And(keywordCond) |
||||
} |
||||
|
||||
if opts.Language != "" { |
||||
cond = cond.And(builder.In("id", builder. |
||||
Select("repo_id"). |
||||
From("language_stat"). |
||||
Where(builder.Eq{"language": opts.Language}).And(builder.Eq{"is_primary": true}))) |
||||
} |
||||
|
||||
if opts.Fork != util.OptionalBoolNone { |
||||
cond = cond.And(builder.Eq{"is_fork": opts.Fork == util.OptionalBoolTrue}) |
||||
} |
||||
|
||||
if opts.Mirror != util.OptionalBoolNone { |
||||
cond = cond.And(builder.Eq{"is_mirror": opts.Mirror == util.OptionalBoolTrue}) |
||||
} |
||||
|
||||
if opts.Actor != nil && opts.Actor.IsRestricted { |
||||
cond = cond.And(accessibleRepositoryCondition(opts.Actor)) |
||||
} |
||||
|
||||
if opts.Archived != util.OptionalBoolNone { |
||||
cond = cond.And(builder.Eq{"is_archived": opts.Archived == util.OptionalBoolTrue}) |
||||
} |
||||
|
||||
switch opts.HasMilestones { |
||||
case util.OptionalBoolTrue: |
||||
cond = cond.And(builder.Gt{"num_milestones": 0}) |
||||
case util.OptionalBoolFalse: |
||||
cond = cond.And(builder.Eq{"num_milestones": 0}.Or(builder.IsNull{"num_milestones"})) |
||||
} |
||||
|
||||
return cond |
||||
} |
||||
|
||||
// SearchRepository returns repositories based on search options,
|
||||
// it returns results in given range and number of total results.
|
||||
func SearchRepository(opts *SearchRepoOptions) (RepositoryList, int64, error) { |
||||
cond := SearchRepositoryCondition(opts) |
||||
return SearchRepositoryByCondition(opts, cond, true) |
||||
} |
||||
|
||||
// SearchRepositoryByCondition search repositories by condition
|
||||
func SearchRepositoryByCondition(opts *SearchRepoOptions, cond builder.Cond, loadAttributes bool) (RepositoryList, int64, error) { |
||||
ctx := db.DefaultContext |
||||
sess, count, err := searchRepositoryByCondition(ctx, opts, cond) |
||||
if err != nil { |
||||
return nil, 0, err |
||||
} |
||||
|
||||
defaultSize := 50 |
||||
if opts.PageSize > 0 { |
||||
defaultSize = opts.PageSize |
||||
} |
||||
repos := make(RepositoryList, 0, defaultSize) |
||||
if err := sess.Find(&repos); err != nil { |
||||
return nil, 0, fmt.Errorf("Repo: %v", err) |
||||
} |
||||
|
||||
if opts.PageSize <= 0 { |
||||
count = int64(len(repos)) |
||||
} |
||||
|
||||
if loadAttributes { |
||||
if err := repos.loadAttributes(ctx); err != nil { |
||||
return nil, 0, fmt.Errorf("LoadAttributes: %v", err) |
||||
} |
||||
} |
||||
|
||||
return repos, count, nil |
||||
} |
||||
|
||||
func searchRepositoryByCondition(ctx context.Context, opts *SearchRepoOptions, cond builder.Cond) (db.Engine, int64, error) { |
||||
if opts.Page <= 0 { |
||||
opts.Page = 1 |
||||
} |
||||
|
||||
if len(opts.OrderBy) == 0 { |
||||
opts.OrderBy = db.SearchOrderByAlphabetically |
||||
} |
||||
|
||||
args := make([]interface{}, 0) |
||||
if opts.PriorityOwnerID > 0 { |
||||
opts.OrderBy = db.SearchOrderBy(fmt.Sprintf("CASE WHEN owner_id = ? THEN 0 ELSE owner_id END, %s", opts.OrderBy)) |
||||
args = append(args, opts.PriorityOwnerID) |
||||
} else if strings.Count(opts.Keyword, "/") == 1 { |
||||
// With "owner/repo" search times, prioritise results which match the owner field
|
||||
orgName := strings.Split(opts.Keyword, "/")[0] |
||||
opts.OrderBy = db.SearchOrderBy(fmt.Sprintf("CASE WHEN owner_name LIKE ? THEN 0 ELSE 1 END, %s", opts.OrderBy)) |
||||
args = append(args, orgName) |
||||
} |
||||
|
||||
sess := db.GetEngine(ctx) |
||||
|
||||
var count int64 |
||||
if opts.PageSize > 0 { |
||||
var err error |
||||
count, err = sess. |
||||
Where(cond). |
||||
Count(new(repo_model.Repository)) |
||||
if err != nil { |
||||
return nil, 0, fmt.Errorf("Count: %v", err) |
||||
} |
||||
} |
||||
|
||||
sess = sess.Where(cond).OrderBy(opts.OrderBy.String(), args...) |
||||
if opts.PageSize > 0 { |
||||
sess = sess.Limit(opts.PageSize, (opts.Page-1)*opts.PageSize) |
||||
} |
||||
return sess, count, nil |
||||
} |
||||
|
||||
// accessibleRepositoryCondition takes a user a returns a condition for checking if a repository is accessible
|
||||
func accessibleRepositoryCondition(user *user_model.User) builder.Cond { |
||||
cond := builder.NewCond() |
||||
|
||||
if user == nil || !user.IsRestricted || user.ID <= 0 { |
||||
orgVisibilityLimit := []structs.VisibleType{structs.VisibleTypePrivate} |
||||
if user == nil || user.ID <= 0 { |
||||
orgVisibilityLimit = append(orgVisibilityLimit, structs.VisibleTypeLimited) |
||||
} |
||||
// 1. Be able to see all non-private repositories that either:
|
||||
cond = cond.Or(builder.And( |
||||
builder.Eq{"`repository`.is_private": false}, |
||||
// 2. Aren't in an private organisation or limited organisation if we're not logged in
|
||||
builder.NotIn("`repository`.owner_id", builder.Select("id").From("`user`").Where( |
||||
builder.And( |
||||
builder.Eq{"type": user_model.UserTypeOrganization}, |
||||
builder.In("visibility", orgVisibilityLimit)), |
||||
)))) |
||||
} |
||||
|
||||
if user != nil { |
||||
cond = cond.Or( |
||||
// 2. Be able to see all repositories that we have access to
|
||||
userCollaborationRepoCond("`repository`.id", user.ID), |
||||
// 3. Repositories that we directly own
|
||||
builder.Eq{"`repository`.owner_id": user.ID}, |
||||
// 4. Be able to see all repositories that we are in a team
|
||||
userOrgTeamRepoCond("`repository`.id", user.ID), |
||||
// 5. Be able to see all public repos in private organizations that we are an org_user of
|
||||
userOrgPublicRepoCond(user.ID), |
||||
) |
||||
} |
||||
|
||||
return cond |
||||
} |
||||
|
||||
// SearchRepositoryByName takes keyword and part of repository name to search,
|
||||
// it returns results in given range and number of total results.
|
||||
func SearchRepositoryByName(opts *SearchRepoOptions) (RepositoryList, int64, error) { |
||||
opts.IncludeDescription = false |
||||
return SearchRepository(opts) |
||||
} |
||||
|
||||
// SearchRepositoryIDs takes keyword and part of repository name to search,
|
||||
// it returns results in given range and number of total results.
|
||||
func SearchRepositoryIDs(opts *SearchRepoOptions) ([]int64, int64, error) { |
||||
opts.IncludeDescription = false |
||||
|
||||
cond := SearchRepositoryCondition(opts) |
||||
|
||||
sess, count, err := searchRepositoryByCondition(db.DefaultContext, opts, cond) |
||||
if err != nil { |
||||
return nil, 0, err |
||||
} |
||||
|
||||
defaultSize := 50 |
||||
if opts.PageSize > 0 { |
||||
defaultSize = opts.PageSize |
||||
} |
||||
|
||||
ids := make([]int64, 0, defaultSize) |
||||
err = sess.Select("id").Table("repository").Find(&ids) |
||||
if opts.PageSize <= 0 { |
||||
count = int64(len(ids)) |
||||
} |
||||
|
||||
return ids, count, err |
||||
} |
||||
|
||||
// AccessibleRepoIDsQuery queries accessible repository ids. Usable as a subquery wherever repo ids need to be filtered.
|
||||
func AccessibleRepoIDsQuery(user *user_model.User) *builder.Builder { |
||||
// NB: Please note this code needs to still work if user is nil
|
||||
return builder.Select("id").From("repository").Where(accessibleRepositoryCondition(user)) |
||||
} |
||||
|
||||
// FindUserAccessibleRepoIDs find all accessible repositories' ID by user's id
|
||||
func FindUserAccessibleRepoIDs(user *user_model.User) ([]int64, error) { |
||||
repoIDs := make([]int64, 0, 10) |
||||
if err := db.GetEngine(db.DefaultContext). |
||||
Table("repository"). |
||||
Cols("id"). |
||||
Where(accessibleRepositoryCondition(user)). |
||||
Find(&repoIDs); err != nil { |
||||
return nil, fmt.Errorf("FindUserAccesibleRepoIDs: %v", err) |
||||
} |
||||
return repoIDs, nil |
||||
} |
||||
|
||||
// GetUserRepositories returns a list of repositories of given user.
|
||||
func GetUserRepositories(opts *SearchRepoOptions) (RepositoryList, int64, error) { |
||||
if len(opts.OrderBy) == 0 { |
||||
opts.OrderBy = "updated_unix DESC" |
||||
} |
||||
|
||||
cond := builder.NewCond() |
||||
cond = cond.And(builder.Eq{"owner_id": opts.Actor.ID}) |
||||
if !opts.Private { |
||||
cond = cond.And(builder.Eq{"is_private": false}) |
||||
} |
||||
|
||||
if opts.LowerNames != nil && len(opts.LowerNames) > 0 { |
||||
cond = cond.And(builder.In("lower_name", opts.LowerNames)) |
||||
} |
||||
|
||||
sess := db.GetEngine(db.DefaultContext) |
||||
|
||||
count, err := sess.Where(cond).Count(new(repo_model.Repository)) |
||||
if err != nil { |
||||
return nil, 0, fmt.Errorf("Count: %v", err) |
||||
} |
||||
|
||||
sess = sess.Where(cond).OrderBy(opts.OrderBy.String()) |
||||
repos := make(RepositoryList, 0, opts.PageSize) |
||||
return repos, count, db.SetSessionPagination(sess, opts).Find(&repos) |
||||
} |
@ -0,0 +1,33 @@ |
||||
// 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 repository |
||||
|
||||
import ( |
||||
"code.gitea.io/gitea/models/db" |
||||
"code.gitea.io/gitea/models/organization" |
||||
repo_model "code.gitea.io/gitea/models/repo" |
||||
user_model "code.gitea.io/gitea/models/user" |
||||
) |
||||
|
||||
// CanUserDelete returns true if user could delete the repository
|
||||
func CanUserDelete(repo *repo_model.Repository, user *user_model.User) (bool, error) { |
||||
if user.IsAdmin || user.ID == repo.OwnerID { |
||||
return true, nil |
||||
} |
||||
|
||||
if err := repo.GetOwner(db.DefaultContext); err != nil { |
||||
return false, err |
||||
} |
||||
|
||||
if repo.Owner.IsOrganization() { |
||||
isOwner, err := organization.OrgFromUser(repo.Owner).IsOwnedBy(user.ID) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
return isOwner, nil |
||||
} |
||||
|
||||
return false, nil |
||||
} |
@ -0,0 +1,31 @@ |
||||
// 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 repository |
||||
|
||||
import ( |
||||
"code.gitea.io/gitea/models/organization" |
||||
repo_model "code.gitea.io/gitea/models/repo" |
||||
user_model "code.gitea.io/gitea/models/user" |
||||
) |
||||
|
||||
// CanUserForkRepo returns true if specified user can fork repository.
|
||||
func CanUserForkRepo(user *user_model.User, repo *repo_model.Repository) (bool, error) { |
||||
if user == nil { |
||||
return false, nil |
||||
} |
||||
if repo.OwnerID != user.ID && !repo_model.HasForkedRepo(user.ID, repo.ID) { |
||||
return true, nil |
||||
} |
||||
ownedOrgs, err := organization.GetOrgsCanCreateRepoByUserID(user.ID) |
||||
if err != nil { |
||||
return false, err |
||||
} |
||||
for _, org := range ownedOrgs { |
||||
if repo.OwnerID != org.ID && !repo_model.HasForkedRepo(org.ID, repo.ID) { |
||||
return true, nil |
||||
} |
||||
} |
||||
return false, nil |
||||
} |
@ -0,0 +1,43 @@ |
||||
// 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 repository |
||||
|
||||
import ( |
||||
"testing" |
||||
|
||||
"code.gitea.io/gitea/models/db" |
||||
repo_model "code.gitea.io/gitea/models/repo" |
||||
"code.gitea.io/gitea/models/unit" |
||||
"code.gitea.io/gitea/models/unittest" |
||||
|
||||
"github.com/stretchr/testify/assert" |
||||
) |
||||
|
||||
func TestLinkedRepository(t *testing.T) { |
||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
||||
testCases := []struct { |
||||
name string |
||||
attachID int64 |
||||
expectedRepo *repo_model.Repository |
||||
expectedUnitType unit.Type |
||||
}{ |
||||
{"LinkedIssue", 1, &repo_model.Repository{ID: 1}, unit.TypeIssues}, |
||||
{"LinkedComment", 3, &repo_model.Repository{ID: 1}, unit.TypePullRequests}, |
||||
{"LinkedRelease", 9, &repo_model.Repository{ID: 1}, unit.TypeReleases}, |
||||
{"Notlinked", 10, nil, -1}, |
||||
} |
||||
for _, tc := range testCases { |
||||
t.Run(tc.name, func(t *testing.T) { |
||||
attach, err := repo_model.GetAttachmentByID(db.DefaultContext, tc.attachID) |
||||
assert.NoError(t, err) |
||||
repo, unitType, err := LinkedRepository(attach) |
||||
assert.NoError(t, err) |
||||
if tc.expectedRepo != nil { |
||||
assert.Equal(t, tc.expectedRepo.ID, repo.ID) |
||||
} |
||||
assert.Equal(t, tc.expectedUnitType, unitType) |
||||
}) |
||||
} |
||||
} |
@ -0,0 +1,24 @@ |
||||
// 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 repository |
||||
|
||||
import ( |
||||
"code.gitea.io/gitea/models/db" |
||||
"code.gitea.io/gitea/models/organization" |
||||
"code.gitea.io/gitea/models/perm" |
||||
repo_model "code.gitea.io/gitea/models/repo" |
||||
) |
||||
|
||||
// GetReviewerTeams get all teams can be requested to review
|
||||
func GetReviewerTeams(repo *repo_model.Repository) ([]*organization.Team, error) { |
||||
if err := repo.GetOwner(db.DefaultContext); err != nil { |
||||
return nil, err |
||||
} |
||||
if !repo.Owner.IsOrganization() { |
||||
return nil, nil |
||||
} |
||||
|
||||
return organization.GetTeamsWithAccessToRepo(db.DefaultContext, repo.OwnerID, repo.ID, perm.AccessModeRead) |
||||
} |
@ -0,0 +1,28 @@ |
||||
// 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 repository |
||||
|
||||
import ( |
||||
"testing" |
||||
|
||||
repo_model "code.gitea.io/gitea/models/repo" |
||||
"code.gitea.io/gitea/models/unittest" |
||||
|
||||
"github.com/stretchr/testify/assert" |
||||
) |
||||
|
||||
func TestRepoGetReviewerTeams(t *testing.T) { |
||||
assert.NoError(t, unittest.PrepareTestDatabase()) |
||||
|
||||
repo2 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 2}).(*repo_model.Repository) |
||||
teams, err := GetReviewerTeams(repo2) |
||||
assert.NoError(t, err) |
||||
assert.Empty(t, teams) |
||||
|
||||
repo3 := unittest.AssertExistsAndLoadBean(t, &repo_model.Repository{ID: 3}).(*repo_model.Repository) |
||||
teams, err = GetReviewerTeams(repo3) |
||||
assert.NoError(t, err) |
||||
assert.Len(t, teams, 2) |
||||
} |
Loading…
Reference in new issue