2022-03-29 14:29:02 +08:00
|
|
|
// Copyright 2014 The Gogs Authors. All rights reserved.
|
|
|
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
2022-11-28 02:20:29 +08:00
|
|
|
// SPDX-License-Identifier: MIT
|
2022-03-29 14:29:02 +08:00
|
|
|
|
|
|
|
package organization
|
|
|
|
|
|
|
|
import (
|
|
|
|
"context"
|
|
|
|
"fmt"
|
|
|
|
"strings"
|
|
|
|
|
2024-04-11 16:01:44 +08:00
|
|
|
actions_model "code.gitea.io/gitea/models/actions"
|
2022-03-29 14:29:02 +08:00
|
|
|
"code.gitea.io/gitea/models/db"
|
|
|
|
"code.gitea.io/gitea/models/perm"
|
|
|
|
repo_model "code.gitea.io/gitea/models/repo"
|
2022-12-20 17:07:13 +08:00
|
|
|
secret_model "code.gitea.io/gitea/models/secret"
|
2022-03-29 14:29:02 +08:00
|
|
|
"code.gitea.io/gitea/models/unit"
|
|
|
|
user_model "code.gitea.io/gitea/models/user"
|
|
|
|
"code.gitea.io/gitea/modules/log"
|
|
|
|
"code.gitea.io/gitea/modules/setting"
|
|
|
|
"code.gitea.io/gitea/modules/structs"
|
2022-10-18 13:50:37 +08:00
|
|
|
"code.gitea.io/gitea/modules/util"
|
2022-03-29 14:29:02 +08:00
|
|
|
|
|
|
|
"xorm.io/builder"
|
2024-11-12 11:44:24 +08:00
|
|
|
"xorm.io/xorm"
|
2022-03-29 14:29:02 +08:00
|
|
|
)
|
|
|
|
|
|
|
|
// ________ .__ __ .__
|
|
|
|
// \_____ \_______ _________ ____ |__|____________ _/ |_|__| ____ ____
|
|
|
|
// / | \_ __ \/ ___\__ \ / \| \___ /\__ \\ __\ |/ _ \ / \
|
|
|
|
// / | \ | \/ /_/ > __ \| | \ |/ / / __ \| | | ( <_> ) | \
|
|
|
|
// \_______ /__| \___ (____ /___| /__/_____ \(____ /__| |__|\____/|___| /
|
|
|
|
// \/ /_____/ \/ \/ \/ \/ \/
|
|
|
|
|
|
|
|
// ErrOrgNotExist represents a "OrgNotExist" kind of error.
|
|
|
|
type ErrOrgNotExist struct {
|
|
|
|
ID int64
|
|
|
|
Name string
|
|
|
|
}
|
|
|
|
|
|
|
|
// IsErrOrgNotExist checks if an error is a ErrOrgNotExist.
|
|
|
|
func IsErrOrgNotExist(err error) bool {
|
|
|
|
_, ok := err.(ErrOrgNotExist)
|
|
|
|
return ok
|
|
|
|
}
|
|
|
|
|
|
|
|
func (err ErrOrgNotExist) Error() string {
|
|
|
|
return fmt.Sprintf("org does not exist [id: %d, name: %s]", err.ID, err.Name)
|
|
|
|
}
|
|
|
|
|
2022-10-18 13:50:37 +08:00
|
|
|
func (err ErrOrgNotExist) Unwrap() error {
|
|
|
|
return util.ErrNotExist
|
|
|
|
}
|
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
// ErrLastOrgOwner represents a "LastOrgOwner" kind of error.
|
|
|
|
type ErrLastOrgOwner struct {
|
|
|
|
UID int64
|
|
|
|
}
|
|
|
|
|
|
|
|
// IsErrLastOrgOwner checks if an error is a ErrLastOrgOwner.
|
|
|
|
func IsErrLastOrgOwner(err error) bool {
|
|
|
|
_, ok := err.(ErrLastOrgOwner)
|
|
|
|
return ok
|
|
|
|
}
|
|
|
|
|
|
|
|
func (err ErrLastOrgOwner) Error() string {
|
|
|
|
return fmt.Sprintf("user is the last member of owner team [uid: %d]", err.UID)
|
|
|
|
}
|
|
|
|
|
|
|
|
// ErrUserNotAllowedCreateOrg represents a "UserNotAllowedCreateOrg" kind of error.
|
|
|
|
type ErrUserNotAllowedCreateOrg struct{}
|
|
|
|
|
|
|
|
// IsErrUserNotAllowedCreateOrg checks if an error is an ErrUserNotAllowedCreateOrg.
|
|
|
|
func IsErrUserNotAllowedCreateOrg(err error) bool {
|
|
|
|
_, ok := err.(ErrUserNotAllowedCreateOrg)
|
|
|
|
return ok
|
|
|
|
}
|
|
|
|
|
|
|
|
func (err ErrUserNotAllowedCreateOrg) Error() string {
|
|
|
|
return "user is not allowed to create organizations"
|
|
|
|
}
|
|
|
|
|
2022-10-18 13:50:37 +08:00
|
|
|
func (err ErrUserNotAllowedCreateOrg) Unwrap() error {
|
|
|
|
return util.ErrPermissionDenied
|
|
|
|
}
|
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
// Organization represents an organization
|
|
|
|
type Organization user_model.User
|
|
|
|
|
|
|
|
// OrgFromUser converts user to organization
|
|
|
|
func OrgFromUser(user *user_model.User) *Organization {
|
|
|
|
return (*Organization)(user)
|
|
|
|
}
|
|
|
|
|
|
|
|
// TableName represents the real table name of Organization
|
|
|
|
func (Organization) TableName() string {
|
|
|
|
return "user"
|
|
|
|
}
|
|
|
|
|
|
|
|
// IsOwnedBy returns true if given user is in the owner team.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) IsOwnedBy(ctx context.Context, uid int64) (bool, error) {
|
|
|
|
return IsOrganizationOwner(ctx, org.ID, uid)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
2023-04-07 23:21:02 +08:00
|
|
|
// IsOrgAdmin returns true if given user is in the owner team or an admin team.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) IsOrgAdmin(ctx context.Context, uid int64) (bool, error) {
|
|
|
|
return IsOrganizationAdmin(ctx, org.ID, uid)
|
2023-04-07 23:21:02 +08:00
|
|
|
}
|
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
// IsOrgMember returns true if given user is member of organization.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) IsOrgMember(ctx context.Context, uid int64) (bool, error) {
|
|
|
|
return IsOrganizationMember(ctx, org.ID, uid)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// CanCreateOrgRepo returns true if given user can create repo in organization
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) CanCreateOrgRepo(ctx context.Context, uid int64) (bool, error) {
|
|
|
|
return CanCreateOrgRepo(ctx, org.ID, uid)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetTeam returns named team of organization.
|
2023-02-08 14:44:42 +08:00
|
|
|
func (org *Organization) GetTeam(ctx context.Context, name string) (*Team, error) {
|
|
|
|
return GetTeam(ctx, org.ID, name)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetOwnerTeam returns owner team of organization.
|
2023-02-08 14:44:42 +08:00
|
|
|
func (org *Organization) GetOwnerTeam(ctx context.Context) (*Team, error) {
|
|
|
|
return org.GetTeam(ctx, OwnerTeamName)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// FindOrgTeams returns all teams of a given organization
|
|
|
|
func FindOrgTeams(ctx context.Context, orgID int64) ([]*Team, error) {
|
|
|
|
var teams []*Team
|
|
|
|
return teams, db.GetEngine(ctx).
|
|
|
|
Where("org_id=?", orgID).
|
|
|
|
OrderBy("CASE WHEN name LIKE '" + OwnerTeamName + "' THEN '' ELSE name END").
|
|
|
|
Find(&teams)
|
|
|
|
}
|
|
|
|
|
|
|
|
// LoadTeams load teams if not loaded.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) LoadTeams(ctx context.Context) ([]*Team, error) {
|
|
|
|
return FindOrgTeams(ctx, org.ID)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetMembers returns all members of organization.
|
2024-11-11 08:38:30 +08:00
|
|
|
func (org *Organization) GetMembers(ctx context.Context, doer *user_model.User) (user_model.UserList, map[int64]bool, error) {
|
2023-09-15 01:09:32 +08:00
|
|
|
return FindOrgMembers(ctx, &FindOrgMembersOpts{
|
2024-11-11 08:38:30 +08:00
|
|
|
Doer: doer,
|
2022-03-29 14:29:02 +08:00
|
|
|
OrgID: org.ID,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
// HasMemberWithUserID returns true if user with userID is part of the u organisation.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) HasMemberWithUserID(ctx context.Context, userID int64) bool {
|
|
|
|
return org.hasMemberWithUserID(ctx, userID)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
func (org *Organization) hasMemberWithUserID(ctx context.Context, userID int64) bool {
|
|
|
|
isMember, err := IsOrganizationMember(ctx, org.ID, userID)
|
|
|
|
if err != nil {
|
|
|
|
log.Error("IsOrganizationMember: %v", err)
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
return isMember
|
|
|
|
}
|
|
|
|
|
|
|
|
// AvatarLink returns the full avatar link with http host
|
Add context cache as a request level cache (#22294)
To avoid duplicated load of the same data in an HTTP request, we can set
a context cache to do that. i.e. Some pages may load a user from a
database with the same id in different areas on the same page. But the
code is hidden in two different deep logic. How should we share the
user? As a result of this PR, now if both entry functions accept
`context.Context` as the first parameter and we just need to refactor
`GetUserByID` to reuse the user from the context cache. Then it will not
be loaded twice on an HTTP request.
But of course, sometimes we would like to reload an object from the
database, that's why `RemoveContextData` is also exposed.
The core context cache is here. It defines a new context
```go
type cacheContext struct {
ctx context.Context
data map[any]map[any]any
lock sync.RWMutex
}
var cacheContextKey = struct{}{}
func WithCacheContext(ctx context.Context) context.Context {
return context.WithValue(ctx, cacheContextKey, &cacheContext{
ctx: ctx,
data: make(map[any]map[any]any),
})
}
```
Then you can use the below 4 methods to read/write/del the data within
the same context.
```go
func GetContextData(ctx context.Context, tp, key any) any
func SetContextData(ctx context.Context, tp, key, value any)
func RemoveContextData(ctx context.Context, tp, key any)
func GetWithContextCache[T any](ctx context.Context, cacheGroupKey string, cacheTargetID any, f func() (T, error)) (T, error)
```
Then let's take a look at how `system.GetString` implement it.
```go
func GetSetting(ctx context.Context, key string) (string, error) {
return cache.GetWithContextCache(ctx, contextCacheKey, key, func() (string, error) {
return cache.GetString(genSettingCacheKey(key), func() (string, error) {
res, err := GetSettingNoCache(ctx, key)
if err != nil {
return "", err
}
return res.SettingValue, nil
})
})
}
```
First, it will check if context data include the setting object with the
key. If not, it will query from the global cache which may be memory or
a Redis cache. If not, it will get the object from the database. In the
end, if the object gets from the global cache or database, it will be
set into the context cache.
An object stored in the context cache will only be destroyed after the
context disappeared.
2023-02-15 21:37:34 +08:00
|
|
|
func (org *Organization) AvatarLink(ctx context.Context) string {
|
|
|
|
return org.AsUser().AvatarLink(ctx)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// HTMLURL returns the organization's full link.
|
|
|
|
func (org *Organization) HTMLURL() string {
|
|
|
|
return org.AsUser().HTMLURL()
|
|
|
|
}
|
|
|
|
|
|
|
|
// OrganisationLink returns the organization sub page link.
|
|
|
|
func (org *Organization) OrganisationLink() string {
|
|
|
|
return org.AsUser().OrganisationLink()
|
|
|
|
}
|
|
|
|
|
|
|
|
// ShortName ellipses username to length
|
|
|
|
func (org *Organization) ShortName(length int) string {
|
|
|
|
return org.AsUser().ShortName(length)
|
|
|
|
}
|
|
|
|
|
|
|
|
// HomeLink returns the user or organization home page link.
|
|
|
|
func (org *Organization) HomeLink() string {
|
|
|
|
return org.AsUser().HomeLink()
|
|
|
|
}
|
|
|
|
|
|
|
|
// CanCreateRepo returns if user login can create a repository
|
|
|
|
// NOTE: functions calling this assume a failure due to repository count limit; if new checks are added, those functions should be revised
|
|
|
|
func (org *Organization) CanCreateRepo() bool {
|
|
|
|
return org.AsUser().CanCreateRepo()
|
|
|
|
}
|
|
|
|
|
|
|
|
// FindOrgMembersOpts represensts find org members conditions
|
|
|
|
type FindOrgMembersOpts struct {
|
|
|
|
db.ListOptions
|
2024-11-11 08:38:30 +08:00
|
|
|
Doer *user_model.User
|
|
|
|
IsDoerMember bool
|
|
|
|
OrgID int64
|
|
|
|
}
|
|
|
|
|
|
|
|
func (opts FindOrgMembersOpts) PublicOnly() bool {
|
|
|
|
return opts.Doer == nil || !(opts.IsDoerMember || opts.Doer.IsAdmin)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
2024-11-12 11:44:24 +08:00
|
|
|
// applyTeamMatesOnlyFilter make sure restricted users only see public team members and there own team mates
|
|
|
|
func (opts FindOrgMembersOpts) applyTeamMatesOnlyFilter(sess *xorm.Session) {
|
|
|
|
if opts.Doer != nil && opts.IsDoerMember && opts.Doer.IsRestricted {
|
|
|
|
teamMates := builder.Select("DISTINCT team_user.uid").
|
|
|
|
From("team_user").
|
|
|
|
Where(builder.In("team_user.team_id", getUserTeamIDsQueryBuilder(opts.OrgID, opts.Doer.ID))).
|
|
|
|
And(builder.Eq{"team_user.org_id": opts.OrgID})
|
|
|
|
|
|
|
|
sess.And(
|
|
|
|
builder.In("org_user.uid", teamMates).
|
|
|
|
Or(builder.Eq{"org_user.is_public": true}),
|
|
|
|
)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
// CountOrgMembers counts the organization's members
|
2023-10-03 18:30:41 +08:00
|
|
|
func CountOrgMembers(ctx context.Context, opts *FindOrgMembersOpts) (int64, error) {
|
|
|
|
sess := db.GetEngine(ctx).Where("org_id=?", opts.OrgID)
|
2024-11-11 08:38:30 +08:00
|
|
|
if opts.PublicOnly() {
|
2024-11-12 11:44:24 +08:00
|
|
|
sess = sess.And("is_public = ?", true)
|
|
|
|
} else {
|
|
|
|
opts.applyTeamMatesOnlyFilter(sess)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
2024-11-11 08:38:30 +08:00
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
return sess.Count(new(OrgUser))
|
|
|
|
}
|
|
|
|
|
|
|
|
// FindOrgMembers loads organization members according conditions
|
2023-09-15 01:09:32 +08:00
|
|
|
func FindOrgMembers(ctx context.Context, opts *FindOrgMembersOpts) (user_model.UserList, map[int64]bool, error) {
|
|
|
|
ous, err := GetOrgUsersByOrgID(ctx, opts)
|
2022-03-29 14:29:02 +08:00
|
|
|
if err != nil {
|
|
|
|
return nil, nil, err
|
|
|
|
}
|
|
|
|
|
|
|
|
ids := make([]int64, len(ous))
|
|
|
|
idsIsPublic := make(map[int64]bool, len(ous))
|
|
|
|
for i, ou := range ous {
|
|
|
|
ids[i] = ou.UID
|
|
|
|
idsIsPublic[ou.UID] = ou.IsPublic
|
|
|
|
}
|
|
|
|
|
2023-09-15 01:09:32 +08:00
|
|
|
users, err := user_model.GetUsersByIDs(ctx, ids)
|
2022-03-29 14:29:02 +08:00
|
|
|
if err != nil {
|
|
|
|
return nil, nil, err
|
|
|
|
}
|
|
|
|
return users, idsIsPublic, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// AsUser returns the org as user object
|
|
|
|
func (org *Organization) AsUser() *user_model.User {
|
|
|
|
return (*user_model.User)(org)
|
|
|
|
}
|
|
|
|
|
|
|
|
// DisplayName returns full name if it's not empty,
|
|
|
|
// returns username otherwise.
|
|
|
|
func (org *Organization) DisplayName() string {
|
|
|
|
return org.AsUser().DisplayName()
|
|
|
|
}
|
|
|
|
|
|
|
|
// CustomAvatarRelativePath returns user custom avatar relative path.
|
|
|
|
func (org *Organization) CustomAvatarRelativePath() string {
|
|
|
|
return org.Avatar
|
|
|
|
}
|
|
|
|
|
2023-03-10 23:18:20 +08:00
|
|
|
// UnitPermission returns unit permission
|
|
|
|
func (org *Organization) UnitPermission(ctx context.Context, doer *user_model.User, unitType unit.Type) perm.AccessMode {
|
|
|
|
if doer != nil {
|
|
|
|
teams, err := GetUserOrgTeams(ctx, org.ID, doer.ID)
|
|
|
|
if err != nil {
|
|
|
|
log.Error("GetUserOrgTeams: %v", err)
|
|
|
|
return perm.AccessModeNone
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := teams.LoadUnits(ctx); err != nil {
|
|
|
|
log.Error("LoadUnits: %v", err)
|
|
|
|
return perm.AccessModeNone
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(teams) > 0 {
|
|
|
|
return teams.UnitMaxAccess(unitType)
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if org.Visibility.IsPublic() {
|
|
|
|
return perm.AccessModeRead
|
|
|
|
}
|
|
|
|
|
|
|
|
return perm.AccessModeNone
|
|
|
|
}
|
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
// CreateOrganization creates record of a new organization.
|
2023-10-03 18:30:41 +08:00
|
|
|
func CreateOrganization(ctx context.Context, org *Organization, owner *user_model.User) (err error) {
|
2022-03-29 14:29:02 +08:00
|
|
|
if !owner.CanCreateOrganization() {
|
|
|
|
return ErrUserNotAllowedCreateOrg{}
|
|
|
|
}
|
|
|
|
|
|
|
|
if err = user_model.IsUsableUsername(org.Name); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-03 18:30:41 +08:00
|
|
|
isExist, err := user_model.IsUserExist(ctx, 0, org.Name)
|
2022-03-29 14:29:02 +08:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
} else if isExist {
|
|
|
|
return user_model.ErrUserAlreadyExist{Name: org.Name}
|
|
|
|
}
|
|
|
|
|
|
|
|
org.LowerName = strings.ToLower(org.Name)
|
|
|
|
if org.Rands, err = user_model.GetUserSalt(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
if org.Salt, err = user_model.GetUserSalt(); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
org.UseCustomAvatar = true
|
|
|
|
org.MaxRepoCreation = -1
|
|
|
|
org.NumTeams = 1
|
|
|
|
org.NumMembers = 1
|
|
|
|
org.Type = user_model.UserTypeOrganization
|
|
|
|
|
2023-10-03 18:30:41 +08:00
|
|
|
ctx, committer, err := db.TxContext(ctx)
|
2022-03-29 14:29:02 +08:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
defer committer.Close()
|
|
|
|
|
|
|
|
if err = user_model.DeleteUserRedirect(ctx, org.Name); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if err = db.Insert(ctx, org); err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return fmt.Errorf("insert organization: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
2022-05-20 22:08:52 +08:00
|
|
|
if err = user_model.GenerateRandomAvatar(ctx, org.AsUser()); err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return fmt.Errorf("generate random avatar: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Add initial creator to organization and owner team.
|
|
|
|
if err = db.Insert(ctx, &OrgUser{
|
2024-03-24 05:59:58 +08:00
|
|
|
UID: owner.ID,
|
|
|
|
OrgID: org.ID,
|
|
|
|
IsPublic: setting.Service.DefaultOrgMemberVisible,
|
2022-03-29 14:29:02 +08:00
|
|
|
}); err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return fmt.Errorf("insert org-user relation: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// Create default owner team.
|
|
|
|
t := &Team{
|
|
|
|
OrgID: org.ID,
|
|
|
|
LowerName: strings.ToLower(OwnerTeamName),
|
|
|
|
Name: OwnerTeamName,
|
|
|
|
AccessMode: perm.AccessModeOwner,
|
|
|
|
NumMembers: 1,
|
|
|
|
IncludesAllRepositories: true,
|
|
|
|
CanCreateOrgRepo: true,
|
|
|
|
}
|
|
|
|
if err = db.Insert(ctx, t); err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return fmt.Errorf("insert owner team: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// insert units for team
|
|
|
|
units := make([]TeamUnit, 0, len(unit.AllRepoUnitTypes))
|
|
|
|
for _, tp := range unit.AllRepoUnitTypes {
|
2023-04-20 07:19:13 +08:00
|
|
|
up := perm.AccessModeOwner
|
|
|
|
if tp == unit.TypeExternalTracker || tp == unit.TypeExternalWiki {
|
|
|
|
up = perm.AccessModeRead
|
|
|
|
}
|
2022-03-29 14:29:02 +08:00
|
|
|
units = append(units, TeamUnit{
|
2023-04-03 22:36:35 +08:00
|
|
|
OrgID: org.ID,
|
|
|
|
TeamID: t.ID,
|
|
|
|
Type: tp,
|
2023-04-20 07:19:13 +08:00
|
|
|
AccessMode: up,
|
2022-03-29 14:29:02 +08:00
|
|
|
})
|
|
|
|
}
|
|
|
|
|
|
|
|
if err = db.Insert(ctx, &units); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if err = db.Insert(ctx, &TeamUser{
|
|
|
|
UID: owner.ID,
|
|
|
|
OrgID: org.ID,
|
|
|
|
TeamID: t.ID,
|
|
|
|
}); err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return fmt.Errorf("insert team-user relation: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
return committer.Commit()
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetOrgByName returns organization by given name.
|
2023-02-08 14:44:42 +08:00
|
|
|
func GetOrgByName(ctx context.Context, name string) (*Organization, error) {
|
2022-03-29 14:29:02 +08:00
|
|
|
if len(name) == 0 {
|
|
|
|
return nil, ErrOrgNotExist{0, name}
|
|
|
|
}
|
|
|
|
u := &Organization{
|
|
|
|
LowerName: strings.ToLower(name),
|
|
|
|
Type: user_model.UserTypeOrganization,
|
|
|
|
}
|
2023-02-08 14:44:42 +08:00
|
|
|
has, err := db.GetEngine(ctx).Get(u)
|
2022-03-29 14:29:02 +08:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
} else if !has {
|
|
|
|
return nil, ErrOrgNotExist{0, name}
|
|
|
|
}
|
|
|
|
return u, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// DeleteOrganization deletes models associated to an organization.
|
|
|
|
func DeleteOrganization(ctx context.Context, org *Organization) error {
|
|
|
|
if org.Type != user_model.UserTypeOrganization {
|
|
|
|
return fmt.Errorf("%s is a user not an organization", org.Name)
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := db.DeleteBeans(ctx,
|
|
|
|
&Team{OrgID: org.ID},
|
|
|
|
&OrgUser{OrgID: org.ID},
|
|
|
|
&TeamUser{OrgID: org.ID},
|
|
|
|
&TeamUnit{OrgID: org.ID},
|
2022-10-19 20:40:28 +08:00
|
|
|
&TeamInvite{OrgID: org.ID},
|
2022-12-20 17:07:13 +08:00
|
|
|
&secret_model.Secret{OwnerID: org.ID},
|
2024-03-04 16:16:03 +08:00
|
|
|
&user_model.Blocking{BlockerID: org.ID},
|
2024-04-11 16:01:44 +08:00
|
|
|
&actions_model.ActionRunner{OwnerID: org.ID},
|
|
|
|
&actions_model.ActionRunnerToken{OwnerID: org.ID},
|
2022-03-29 14:29:02 +08:00
|
|
|
); err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return fmt.Errorf("DeleteBeans: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
if _, err := db.GetEngine(ctx).ID(org.ID).Delete(new(user_model.User)); err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return fmt.Errorf("Delete: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetOrgUserMaxAuthorizeLevel returns highest authorize level of user in an organization
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) GetOrgUserMaxAuthorizeLevel(ctx context.Context, uid int64) (perm.AccessMode, error) {
|
2022-03-29 14:29:02 +08:00
|
|
|
var authorize perm.AccessMode
|
2023-10-03 18:30:41 +08:00
|
|
|
_, err := db.GetEngine(ctx).
|
2022-03-29 14:29:02 +08:00
|
|
|
Select("max(team.authorize)").
|
|
|
|
Table("team").
|
|
|
|
Join("INNER", "team_user", "team_user.team_id = team.id").
|
|
|
|
Where("team_user.uid = ?", uid).
|
|
|
|
And("team_user.org_id = ?", org.ID).
|
|
|
|
Get(&authorize)
|
|
|
|
return authorize, err
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetUsersWhoCanCreateOrgRepo returns users which are able to create repo in organization
|
2023-01-30 18:12:45 +08:00
|
|
|
func GetUsersWhoCanCreateOrgRepo(ctx context.Context, orgID int64) (map[int64]*user_model.User, error) {
|
|
|
|
// Use a map, in order to de-duplicate users.
|
|
|
|
users := make(map[int64]*user_model.User)
|
2022-03-29 14:29:02 +08:00
|
|
|
return users, db.GetEngine(ctx).
|
|
|
|
Join("INNER", "`team_user`", "`team_user`.uid=`user`.id").
|
|
|
|
Join("INNER", "`team`", "`team`.id=`team_user`.team_id").
|
|
|
|
Where(builder.Eq{"team.can_create_org_repo": true}.Or(builder.Eq{"team.authorize": perm.AccessModeOwner})).
|
2023-01-30 18:12:45 +08:00
|
|
|
And("team_user.org_id = ?", orgID).Find(&users)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// SearchOrganizationsOptions options to filter organizations
|
|
|
|
type SearchOrganizationsOptions struct {
|
|
|
|
db.ListOptions
|
|
|
|
All bool
|
|
|
|
}
|
|
|
|
|
|
|
|
// FindOrgOptions finds orgs options
|
|
|
|
type FindOrgOptions struct {
|
|
|
|
db.ListOptions
|
|
|
|
UserID int64
|
|
|
|
IncludePrivate bool
|
|
|
|
}
|
|
|
|
|
|
|
|
func queryUserOrgIDs(userID int64, includePrivate bool) *builder.Builder {
|
|
|
|
cond := builder.Eq{"uid": userID}
|
|
|
|
if !includePrivate {
|
|
|
|
cond["is_public"] = true
|
|
|
|
}
|
|
|
|
return builder.Select("org_id").From("org_user").Where(cond)
|
|
|
|
}
|
|
|
|
|
2023-11-24 11:49:41 +08:00
|
|
|
func (opts FindOrgOptions) ToConds() builder.Cond {
|
2022-05-20 22:08:52 +08:00
|
|
|
var cond builder.Cond = builder.Eq{"`user`.`type`": user_model.UserTypeOrganization}
|
2022-03-29 14:29:02 +08:00
|
|
|
if opts.UserID > 0 {
|
|
|
|
cond = cond.And(builder.In("`user`.`id`", queryUserOrgIDs(opts.UserID, opts.IncludePrivate)))
|
|
|
|
}
|
|
|
|
if !opts.IncludePrivate {
|
|
|
|
cond = cond.And(builder.Eq{"`user`.visibility": structs.VisibleTypePublic})
|
|
|
|
}
|
|
|
|
return cond
|
|
|
|
}
|
|
|
|
|
2023-11-24 11:49:41 +08:00
|
|
|
func (opts FindOrgOptions) ToOrders() string {
|
|
|
|
return "`user`.name ASC"
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// HasOrgOrUserVisible tells if the given user can see the given org or user
|
|
|
|
func HasOrgOrUserVisible(ctx context.Context, orgOrUser, user *user_model.User) bool {
|
2022-11-18 03:29:33 +08:00
|
|
|
// If user is nil, it's an anonymous user/request.
|
|
|
|
// The Ghost user is handled like an anonymous user.
|
|
|
|
if user == nil || user.IsGhost() {
|
2022-03-29 14:29:02 +08:00
|
|
|
return orgOrUser.Visibility == structs.VisibleTypePublic
|
|
|
|
}
|
|
|
|
|
|
|
|
if user.IsAdmin || orgOrUser.ID == user.ID {
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
if (orgOrUser.Visibility == structs.VisibleTypePrivate || user.IsRestricted) && !OrgFromUser(orgOrUser).hasMemberWithUserID(ctx, user.ID) {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
return true
|
|
|
|
}
|
|
|
|
|
|
|
|
// HasOrgsVisible tells if the given user can see at least one of the orgs provided
|
2023-10-03 18:30:41 +08:00
|
|
|
func HasOrgsVisible(ctx context.Context, orgs []*Organization, user *user_model.User) bool {
|
2022-03-29 14:29:02 +08:00
|
|
|
if len(orgs) == 0 {
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
for _, org := range orgs {
|
2023-10-03 18:30:41 +08:00
|
|
|
if HasOrgOrUserVisible(ctx, org.AsUser(), user) {
|
2022-03-29 14:29:02 +08:00
|
|
|
return true
|
|
|
|
}
|
|
|
|
}
|
|
|
|
return false
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetOrgsCanCreateRepoByUserID returns a list of organizations where given user ID
|
|
|
|
// are allowed to create repos.
|
2023-09-15 01:09:32 +08:00
|
|
|
func GetOrgsCanCreateRepoByUserID(ctx context.Context, userID int64) ([]*Organization, error) {
|
2022-03-29 14:29:02 +08:00
|
|
|
orgs := make([]*Organization, 0, 10)
|
|
|
|
|
2023-09-15 01:09:32 +08:00
|
|
|
return orgs, db.GetEngine(ctx).Where(builder.In("id", builder.Select("`user`.id").From("`user`").
|
2022-03-29 14:29:02 +08:00
|
|
|
Join("INNER", "`team_user`", "`team_user`.org_id = `user`.id").
|
|
|
|
Join("INNER", "`team`", "`team`.id = `team_user`.team_id").
|
|
|
|
Where(builder.Eq{"`team_user`.uid": userID}).
|
|
|
|
And(builder.Eq{"`team`.authorize": perm.AccessModeOwner}.Or(builder.Eq{"`team`.can_create_org_repo": true})))).
|
|
|
|
Asc("`user`.name").
|
|
|
|
Find(&orgs)
|
|
|
|
}
|
|
|
|
|
|
|
|
// GetOrgUsersByOrgID returns all organization-user relations by organization ID.
|
2022-05-20 22:08:52 +08:00
|
|
|
func GetOrgUsersByOrgID(ctx context.Context, opts *FindOrgMembersOpts) ([]*OrgUser, error) {
|
|
|
|
sess := db.GetEngine(ctx).Where("org_id=?", opts.OrgID)
|
2024-11-11 08:38:30 +08:00
|
|
|
if opts.PublicOnly() {
|
2024-11-12 11:44:24 +08:00
|
|
|
sess = sess.And("is_public = ?", true)
|
|
|
|
} else {
|
|
|
|
opts.applyTeamMatesOnlyFilter(sess)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
2024-11-11 08:38:30 +08:00
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
if opts.ListOptions.PageSize > 0 {
|
|
|
|
sess = db.SetSessionPagination(sess, opts)
|
|
|
|
|
|
|
|
ous := make([]*OrgUser, 0, opts.PageSize)
|
|
|
|
return ous, sess.Find(&ous)
|
|
|
|
}
|
|
|
|
|
|
|
|
var ous []*OrgUser
|
|
|
|
return ous, sess.Find(&ous)
|
|
|
|
}
|
|
|
|
|
|
|
|
// ChangeOrgUserStatus changes public or private membership status.
|
2023-10-03 18:30:41 +08:00
|
|
|
func ChangeOrgUserStatus(ctx context.Context, orgID, uid int64, public bool) error {
|
2022-03-29 14:29:02 +08:00
|
|
|
ou := new(OrgUser)
|
2023-10-03 18:30:41 +08:00
|
|
|
has, err := db.GetEngine(ctx).
|
2022-03-29 14:29:02 +08:00
|
|
|
Where("uid=?", uid).
|
|
|
|
And("org_id=?", orgID).
|
|
|
|
Get(ou)
|
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
} else if !has {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
ou.IsPublic = public
|
2023-10-03 18:30:41 +08:00
|
|
|
_, err = db.GetEngine(ctx).ID(ou.ID).Cols("is_public").Update(ou)
|
2022-03-29 14:29:02 +08:00
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
// AddOrgUser adds new user to given organization.
|
2023-10-03 18:30:41 +08:00
|
|
|
func AddOrgUser(ctx context.Context, orgID, uid int64) error {
|
|
|
|
isAlreadyMember, err := IsOrganizationMember(ctx, orgID, uid)
|
2022-03-29 14:29:02 +08:00
|
|
|
if err != nil || isAlreadyMember {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2023-10-03 18:30:41 +08:00
|
|
|
ctx, committer, err := db.TxContext(ctx)
|
2022-03-29 14:29:02 +08:00
|
|
|
if err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
defer committer.Close()
|
|
|
|
|
2022-05-14 21:30:19 +08:00
|
|
|
// check in transaction
|
|
|
|
isAlreadyMember, err = IsOrganizationMember(ctx, orgID, uid)
|
|
|
|
if err != nil || isAlreadyMember {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
ou := &OrgUser{
|
|
|
|
UID: uid,
|
|
|
|
OrgID: orgID,
|
|
|
|
IsPublic: setting.Service.DefaultOrgMemberVisible,
|
|
|
|
}
|
|
|
|
|
|
|
|
if err := db.Insert(ctx, ou); err != nil {
|
|
|
|
return err
|
|
|
|
} else if _, err = db.Exec(ctx, "UPDATE `user` SET num_members = num_members + 1 WHERE id = ?", orgID); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
return committer.Commit()
|
|
|
|
}
|
|
|
|
|
2022-05-20 22:08:52 +08:00
|
|
|
// GetOrgByID returns the user object by given ID if exists.
|
|
|
|
func GetOrgByID(ctx context.Context, id int64) (*Organization, error) {
|
2022-03-29 14:29:02 +08:00
|
|
|
u := new(Organization)
|
|
|
|
has, err := db.GetEngine(ctx).ID(id).Get(u)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
} else if !has {
|
|
|
|
return nil, user_model.ErrUserNotExist{
|
2024-02-15 01:50:10 +08:00
|
|
|
UID: id,
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
}
|
|
|
|
return u, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// RemoveOrgRepo removes all team-repository relations of organization.
|
|
|
|
func RemoveOrgRepo(ctx context.Context, orgID, repoID int64) error {
|
|
|
|
teamRepos := make([]*TeamRepo, 0, 10)
|
|
|
|
e := db.GetEngine(ctx)
|
|
|
|
if err := e.Find(&teamRepos, &TeamRepo{OrgID: orgID, RepoID: repoID}); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
if len(teamRepos) == 0 {
|
|
|
|
return nil
|
|
|
|
}
|
|
|
|
|
|
|
|
if _, err := e.Delete(&TeamRepo{
|
|
|
|
OrgID: orgID,
|
|
|
|
RepoID: repoID,
|
|
|
|
}); err != nil {
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
|
|
|
teamIDs := make([]int64, len(teamRepos))
|
|
|
|
for i, teamRepo := range teamRepos {
|
|
|
|
teamIDs[i] = teamRepo.TeamID
|
|
|
|
}
|
|
|
|
|
|
|
|
_, err := e.Decr("num_repos").In("id", teamIDs).Update(new(Team))
|
|
|
|
return err
|
|
|
|
}
|
|
|
|
|
2022-05-20 22:08:52 +08:00
|
|
|
func (org *Organization) getUserTeams(ctx context.Context, userID int64, cols ...string) ([]*Team, error) {
|
2022-03-29 14:29:02 +08:00
|
|
|
teams := make([]*Team, 0, org.NumTeams)
|
2022-05-20 22:08:52 +08:00
|
|
|
return teams, db.GetEngine(ctx).
|
2022-03-29 14:29:02 +08:00
|
|
|
Where("`team_user`.org_id = ?", org.ID).
|
|
|
|
Join("INNER", "team_user", "`team_user`.team_id = team.id").
|
|
|
|
Join("INNER", "`user`", "`user`.id=team_user.uid").
|
|
|
|
And("`team_user`.uid = ?", userID).
|
|
|
|
Asc("`user`.name").
|
|
|
|
Cols(cols...).
|
|
|
|
Find(&teams)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (org *Organization) getUserTeamIDs(ctx context.Context, userID int64) ([]int64, error) {
|
|
|
|
teamIDs := make([]int64, 0, org.NumTeams)
|
|
|
|
return teamIDs, db.GetEngine(ctx).
|
|
|
|
Table("team").
|
|
|
|
Cols("team.id").
|
|
|
|
Where("`team_user`.org_id = ?", org.ID).
|
|
|
|
Join("INNER", "team_user", "`team_user`.team_id = team.id").
|
|
|
|
And("`team_user`.uid = ?", userID).
|
|
|
|
Find(&teamIDs)
|
|
|
|
}
|
|
|
|
|
2024-11-12 11:44:24 +08:00
|
|
|
func getUserTeamIDsQueryBuilder(orgID, userID int64) *builder.Builder {
|
|
|
|
return builder.Select("team.id").From("team").
|
|
|
|
InnerJoin("team_user", "team_user.team_id = team.id").
|
|
|
|
Where(builder.Eq{
|
|
|
|
"team_user.org_id": orgID,
|
|
|
|
"team_user.uid": userID,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
|
2022-03-29 14:29:02 +08:00
|
|
|
// TeamsWithAccessToRepo returns all teams that have given access level to the repository.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) TeamsWithAccessToRepo(ctx context.Context, repoID int64, mode perm.AccessMode) ([]*Team, error) {
|
|
|
|
return GetTeamsWithAccessToRepo(ctx, org.ID, repoID, mode)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetUserTeamIDs returns of all team IDs of the organization that user is member of.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) GetUserTeamIDs(ctx context.Context, userID int64) ([]int64, error) {
|
|
|
|
return org.getUserTeamIDs(ctx, userID)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// GetUserTeams returns all teams that belong to user,
|
|
|
|
// and that the user has joined.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) GetUserTeams(ctx context.Context, userID int64) ([]*Team, error) {
|
|
|
|
return org.getUserTeams(ctx, userID)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
// AccessibleReposEnvironment operations involving the repositories that are
|
|
|
|
// accessible to a particular user
|
|
|
|
type AccessibleReposEnvironment interface {
|
|
|
|
CountRepos() (int64, error)
|
|
|
|
RepoIDs(page, pageSize int) ([]int64, error)
|
2023-06-05 15:25:47 +08:00
|
|
|
Repos(page, pageSize int) (repo_model.RepositoryList, error)
|
|
|
|
MirrorRepos() (repo_model.RepositoryList, error)
|
2022-03-29 14:29:02 +08:00
|
|
|
AddKeyword(keyword string)
|
|
|
|
SetSort(db.SearchOrderBy)
|
|
|
|
}
|
|
|
|
|
|
|
|
type accessibleReposEnv struct {
|
|
|
|
org *Organization
|
|
|
|
user *user_model.User
|
|
|
|
team *Team
|
|
|
|
teamIDs []int64
|
2022-06-20 20:38:58 +08:00
|
|
|
ctx context.Context
|
2022-03-29 14:29:02 +08:00
|
|
|
keyword string
|
|
|
|
orderBy db.SearchOrderBy
|
|
|
|
}
|
|
|
|
|
|
|
|
// AccessibleReposEnv builds an AccessibleReposEnvironment for the repositories in `org`
|
|
|
|
// that are accessible to the specified user.
|
|
|
|
func AccessibleReposEnv(ctx context.Context, org *Organization, userID int64) (AccessibleReposEnvironment, error) {
|
|
|
|
var user *user_model.User
|
|
|
|
|
|
|
|
if userID > 0 {
|
2022-12-03 10:48:26 +08:00
|
|
|
u, err := user_model.GetUserByID(ctx, userID)
|
2022-03-29 14:29:02 +08:00
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
user = u
|
|
|
|
}
|
|
|
|
|
|
|
|
teamIDs, err := org.getUserTeamIDs(ctx, userID)
|
|
|
|
if err != nil {
|
|
|
|
return nil, err
|
|
|
|
}
|
|
|
|
return &accessibleReposEnv{
|
|
|
|
org: org,
|
|
|
|
user: user,
|
|
|
|
teamIDs: teamIDs,
|
2022-06-20 20:38:58 +08:00
|
|
|
ctx: ctx,
|
2022-03-29 14:29:02 +08:00
|
|
|
orderBy: db.SearchOrderByRecentUpdated,
|
|
|
|
}, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
// AccessibleTeamReposEnv an AccessibleReposEnvironment for the repositories in `org`
|
|
|
|
// that are accessible to the specified team.
|
2023-10-03 18:30:41 +08:00
|
|
|
func (org *Organization) AccessibleTeamReposEnv(ctx context.Context, team *Team) AccessibleReposEnvironment {
|
2022-03-29 14:29:02 +08:00
|
|
|
return &accessibleReposEnv{
|
|
|
|
org: org,
|
|
|
|
team: team,
|
2023-10-03 18:30:41 +08:00
|
|
|
ctx: ctx,
|
2022-03-29 14:29:02 +08:00
|
|
|
orderBy: db.SearchOrderByRecentUpdated,
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
func (env *accessibleReposEnv) cond() builder.Cond {
|
|
|
|
cond := builder.NewCond()
|
|
|
|
if env.team != nil {
|
|
|
|
cond = cond.And(builder.Eq{"team_repo.team_id": env.team.ID})
|
|
|
|
} else {
|
|
|
|
if env.user == nil || !env.user.IsRestricted {
|
|
|
|
cond = cond.Or(builder.Eq{
|
|
|
|
"`repository`.owner_id": env.org.ID,
|
|
|
|
"`repository`.is_private": false,
|
|
|
|
})
|
|
|
|
}
|
|
|
|
if len(env.teamIDs) > 0 {
|
|
|
|
cond = cond.Or(builder.In("team_repo.team_id", env.teamIDs))
|
|
|
|
}
|
|
|
|
}
|
|
|
|
if env.keyword != "" {
|
|
|
|
cond = cond.And(builder.Like{"`repository`.lower_name", strings.ToLower(env.keyword)})
|
|
|
|
}
|
|
|
|
return cond
|
|
|
|
}
|
|
|
|
|
|
|
|
func (env *accessibleReposEnv) CountRepos() (int64, error) {
|
2022-06-20 20:38:58 +08:00
|
|
|
repoCount, err := db.GetEngine(env.ctx).
|
2022-03-29 14:29:02 +08:00
|
|
|
Join("INNER", "team_repo", "`team_repo`.repo_id=`repository`.id").
|
|
|
|
Where(env.cond()).
|
|
|
|
Distinct("`repository`.id").
|
|
|
|
Count(&repo_model.Repository{})
|
|
|
|
if err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return 0, fmt.Errorf("count user repositories in organization: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
return repoCount, nil
|
|
|
|
}
|
|
|
|
|
|
|
|
func (env *accessibleReposEnv) RepoIDs(page, pageSize int) ([]int64, error) {
|
|
|
|
if page <= 0 {
|
|
|
|
page = 1
|
|
|
|
}
|
|
|
|
|
|
|
|
repoIDs := make([]int64, 0, pageSize)
|
2022-06-20 20:38:58 +08:00
|
|
|
return repoIDs, db.GetEngine(env.ctx).
|
2022-03-29 14:29:02 +08:00
|
|
|
Table("repository").
|
|
|
|
Join("INNER", "team_repo", "`team_repo`.repo_id=`repository`.id").
|
|
|
|
Where(env.cond()).
|
|
|
|
GroupBy("`repository`.id,`repository`."+strings.Fields(string(env.orderBy))[0]).
|
|
|
|
OrderBy(string(env.orderBy)).
|
|
|
|
Limit(pageSize, (page-1)*pageSize).
|
|
|
|
Cols("`repository`.id").
|
|
|
|
Find(&repoIDs)
|
|
|
|
}
|
|
|
|
|
2023-06-05 15:25:47 +08:00
|
|
|
func (env *accessibleReposEnv) Repos(page, pageSize int) (repo_model.RepositoryList, error) {
|
2022-03-29 14:29:02 +08:00
|
|
|
repoIDs, err := env.RepoIDs(page, pageSize)
|
|
|
|
if err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return nil, fmt.Errorf("GetUserRepositoryIDs: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
repos := make([]*repo_model.Repository, 0, len(repoIDs))
|
|
|
|
if len(repoIDs) == 0 {
|
|
|
|
return repos, nil
|
|
|
|
}
|
|
|
|
|
2022-06-20 20:38:58 +08:00
|
|
|
return repos, db.GetEngine(env.ctx).
|
2022-03-29 14:29:02 +08:00
|
|
|
In("`repository`.id", repoIDs).
|
|
|
|
OrderBy(string(env.orderBy)).
|
|
|
|
Find(&repos)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (env *accessibleReposEnv) MirrorRepoIDs() ([]int64, error) {
|
|
|
|
repoIDs := make([]int64, 0, 10)
|
2022-06-20 20:38:58 +08:00
|
|
|
return repoIDs, db.GetEngine(env.ctx).
|
2022-03-29 14:29:02 +08:00
|
|
|
Table("repository").
|
|
|
|
Join("INNER", "team_repo", "`team_repo`.repo_id=`repository`.id AND `repository`.is_mirror=?", true).
|
|
|
|
Where(env.cond()).
|
|
|
|
GroupBy("`repository`.id, `repository`.updated_unix").
|
|
|
|
OrderBy(string(env.orderBy)).
|
|
|
|
Cols("`repository`.id").
|
|
|
|
Find(&repoIDs)
|
|
|
|
}
|
|
|
|
|
2023-06-05 15:25:47 +08:00
|
|
|
func (env *accessibleReposEnv) MirrorRepos() (repo_model.RepositoryList, error) {
|
2022-03-29 14:29:02 +08:00
|
|
|
repoIDs, err := env.MirrorRepoIDs()
|
|
|
|
if err != nil {
|
2022-10-25 03:29:17 +08:00
|
|
|
return nil, fmt.Errorf("MirrorRepoIDs: %w", err)
|
2022-03-29 14:29:02 +08:00
|
|
|
}
|
|
|
|
|
|
|
|
repos := make([]*repo_model.Repository, 0, len(repoIDs))
|
|
|
|
if len(repoIDs) == 0 {
|
|
|
|
return repos, nil
|
|
|
|
}
|
|
|
|
|
2022-06-20 20:38:58 +08:00
|
|
|
return repos, db.GetEngine(env.ctx).
|
2022-03-29 14:29:02 +08:00
|
|
|
In("`repository`.id", repoIDs).
|
|
|
|
Find(&repos)
|
|
|
|
}
|
|
|
|
|
|
|
|
func (env *accessibleReposEnv) AddKeyword(keyword string) {
|
|
|
|
env.keyword = keyword
|
|
|
|
}
|
|
|
|
|
|
|
|
func (env *accessibleReposEnv) SetSort(orderBy db.SearchOrderBy) {
|
|
|
|
env.orderBy = orderBy
|
|
|
|
}
|