123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370 |
- // Copyright 2014 The Gogs Authors. All rights reserved.
- // 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 (
- "context"
- "fmt"
- "time"
-
- _ "image/jpeg" // Needed for jpeg support
-
- "code.gitea.io/gitea/models/db"
- "code.gitea.io/gitea/models/unit"
- user_model "code.gitea.io/gitea/models/user"
- "code.gitea.io/gitea/modules/setting"
- "code.gitea.io/gitea/modules/structs"
-
- "xorm.io/builder"
- )
-
- // GetOrganizationCount returns count of membership of organization of the user.
- func GetOrganizationCount(ctx context.Context, u *user_model.User) (int64, error) {
- return db.GetEngine(ctx).
- Where("uid=?", u.ID).
- Count(new(OrgUser))
- }
-
- // GetRepositoryIDs returns repositories IDs where user owned and has unittypes
- // Caller shall check that units is not globally disabled
- func GetRepositoryIDs(u *user_model.User, units ...unit.Type) ([]int64, error) {
- var ids []int64
-
- sess := db.GetEngine(db.DefaultContext).Table("repository").Cols("repository.id")
-
- if len(units) > 0 {
- sess = sess.Join("INNER", "repo_unit", "repository.id = repo_unit.repo_id")
- sess = sess.In("repo_unit.type", units)
- }
-
- return ids, sess.Where("owner_id = ?", u.ID).Find(&ids)
- }
-
- // GetActiveRepositoryIDs returns non-archived repositories IDs where user owned and has unittypes
- // Caller shall check that units is not globally disabled
- func GetActiveRepositoryIDs(u *user_model.User, units ...unit.Type) ([]int64, error) {
- var ids []int64
-
- sess := db.GetEngine(db.DefaultContext).Table("repository").Cols("repository.id")
-
- if len(units) > 0 {
- sess = sess.Join("INNER", "repo_unit", "repository.id = repo_unit.repo_id")
- sess = sess.In("repo_unit.type", units)
- }
-
- sess.Where(builder.Eq{"is_archived": false})
-
- return ids, sess.Where("owner_id = ?", u.ID).GroupBy("repository.id").Find(&ids)
- }
-
- // GetOrgRepositoryIDs returns repositories IDs where user's team owned and has unittypes
- // Caller shall check that units is not globally disabled
- func GetOrgRepositoryIDs(u *user_model.User, units ...unit.Type) ([]int64, error) {
- var ids []int64
-
- if err := db.GetEngine(db.DefaultContext).Table("repository").
- Cols("repository.id").
- Join("INNER", "team_user", "repository.owner_id = team_user.org_id").
- Join("INNER", "team_repo", "(? != ? and repository.is_private != ?) OR (team_user.team_id = team_repo.team_id AND repository.id = team_repo.repo_id)", true, u.IsRestricted, true).
- Where("team_user.uid = ?", u.ID).
- GroupBy("repository.id").Find(&ids); err != nil {
- return nil, err
- }
-
- if len(units) > 0 {
- return FilterOutRepoIdsWithoutUnitAccess(u, ids, units...)
- }
-
- return ids, nil
- }
-
- // GetActiveOrgRepositoryIDs returns non-archived repositories IDs where user's team owned and has unittypes
- // Caller shall check that units is not globally disabled
- func GetActiveOrgRepositoryIDs(u *user_model.User, units ...unit.Type) ([]int64, error) {
- var ids []int64
-
- if err := db.GetEngine(db.DefaultContext).Table("repository").
- Cols("repository.id").
- Join("INNER", "team_user", "repository.owner_id = team_user.org_id").
- Join("INNER", "team_repo", "(? != ? and repository.is_private != ?) OR (team_user.team_id = team_repo.team_id AND repository.id = team_repo.repo_id)", true, u.IsRestricted, true).
- Where("team_user.uid = ?", u.ID).
- Where(builder.Eq{"is_archived": false}).
- GroupBy("repository.id").Find(&ids); err != nil {
- return nil, err
- }
-
- if len(units) > 0 {
- return FilterOutRepoIdsWithoutUnitAccess(u, ids, units...)
- }
-
- return ids, nil
- }
-
- // GetAccessRepoIDs returns all repositories IDs where user's or user is a team member organizations
- // Caller shall check that units is not globally disabled
- func GetAccessRepoIDs(u *user_model.User, units ...unit.Type) ([]int64, error) {
- ids, err := GetRepositoryIDs(u, units...)
- if err != nil {
- return nil, err
- }
- ids2, err := GetOrgRepositoryIDs(u, units...)
- if err != nil {
- return nil, err
- }
- return append(ids, ids2...), nil
- }
-
- // GetActiveAccessRepoIDs returns all non-archived repositories IDs where user's or user is a team member organizations
- // Caller shall check that units is not globally disabled
- func GetActiveAccessRepoIDs(u *user_model.User, units ...unit.Type) ([]int64, error) {
- ids, err := GetActiveRepositoryIDs(u, units...)
- if err != nil {
- return nil, err
- }
- ids2, err := GetActiveOrgRepositoryIDs(u, units...)
- if err != nil {
- return nil, err
- }
- return append(ids, ids2...), nil
- }
-
- // deleteBeans deletes all given beans, beans should contain delete conditions.
- func deleteBeans(e db.Engine, beans ...interface{}) (err error) {
- for i := range beans {
- if _, err = e.Delete(beans[i]); err != nil {
- return err
- }
- }
- return nil
- }
-
- // DeleteUser deletes models associated to an user.
- func DeleteUser(ctx context.Context, u *user_model.User) (err error) {
- e := db.GetEngine(ctx)
-
- // ***** START: Watch *****
- watchedRepoIDs := make([]int64, 0, 10)
- if err = e.Table("watch").Cols("watch.repo_id").
- Where("watch.user_id = ?", u.ID).And("watch.mode <>?", RepoWatchModeDont).Find(&watchedRepoIDs); err != nil {
- return fmt.Errorf("get all watches: %v", err)
- }
- if _, err = e.Decr("num_watches").In("id", watchedRepoIDs).NoAutoTime().Update(new(Repository)); err != nil {
- return fmt.Errorf("decrease repository num_watches: %v", err)
- }
- // ***** END: Watch *****
-
- // ***** START: Star *****
- starredRepoIDs := make([]int64, 0, 10)
- if err = e.Table("star").Cols("star.repo_id").
- Where("star.uid = ?", u.ID).Find(&starredRepoIDs); err != nil {
- return fmt.Errorf("get all stars: %v", err)
- } else if _, err = e.Decr("num_stars").In("id", starredRepoIDs).NoAutoTime().Update(new(Repository)); err != nil {
- return fmt.Errorf("decrease repository num_stars: %v", err)
- }
- // ***** END: Star *****
-
- // ***** START: Follow *****
- followeeIDs := make([]int64, 0, 10)
- if err = e.Table("follow").Cols("follow.follow_id").
- Where("follow.user_id = ?", u.ID).Find(&followeeIDs); err != nil {
- return fmt.Errorf("get all followees: %v", err)
- } else if _, err = e.Decr("num_followers").In("id", followeeIDs).Update(new(user_model.User)); err != nil {
- return fmt.Errorf("decrease user num_followers: %v", err)
- }
-
- followerIDs := make([]int64, 0, 10)
- if err = e.Table("follow").Cols("follow.user_id").
- Where("follow.follow_id = ?", u.ID).Find(&followerIDs); err != nil {
- return fmt.Errorf("get all followers: %v", err)
- } else if _, err = e.Decr("num_following").In("id", followerIDs).Update(new(user_model.User)); err != nil {
- return fmt.Errorf("decrease user num_following: %v", err)
- }
- // ***** END: Follow *****
-
- if err = deleteBeans(e,
- &AccessToken{UID: u.ID},
- &Collaboration{UserID: u.ID},
- &Access{UserID: u.ID},
- &Watch{UserID: u.ID},
- &Star{UID: u.ID},
- &user_model.Follow{UserID: u.ID},
- &user_model.Follow{FollowID: u.ID},
- &Action{UserID: u.ID},
- &IssueUser{UID: u.ID},
- &user_model.EmailAddress{UID: u.ID},
- &user_model.UserOpenID{UID: u.ID},
- &Reaction{UserID: u.ID},
- &TeamUser{UID: u.ID},
- &Collaboration{UserID: u.ID},
- &Stopwatch{UserID: u.ID},
- &user_model.Setting{UserID: u.ID},
- ); err != nil {
- return fmt.Errorf("deleteBeans: %v", err)
- }
-
- if setting.Service.UserDeleteWithCommentsMaxTime != 0 &&
- u.CreatedUnix.AsTime().Add(setting.Service.UserDeleteWithCommentsMaxTime).After(time.Now()) {
-
- // Delete Comments
- const batchSize = 50
- for start := 0; ; start += batchSize {
- comments := make([]*Comment, 0, batchSize)
- if err = e.Where("type=? AND poster_id=?", CommentTypeComment, u.ID).Limit(batchSize, start).Find(&comments); err != nil {
- return err
- }
- if len(comments) == 0 {
- break
- }
-
- for _, comment := range comments {
- if err = deleteComment(e, comment); err != nil {
- return err
- }
- }
- }
-
- // Delete Reactions
- if err = deleteReaction(e, &ReactionOptions{Doer: u}); err != nil {
- return err
- }
- }
-
- // ***** START: PublicKey *****
- if _, err = e.Delete(&PublicKey{OwnerID: u.ID}); err != nil {
- return fmt.Errorf("deletePublicKeys: %v", err)
- }
- // ***** END: PublicKey *****
-
- // ***** START: GPGPublicKey *****
- keys, err := listGPGKeys(e, u.ID, db.ListOptions{})
- if err != nil {
- return fmt.Errorf("ListGPGKeys: %v", err)
- }
- // Delete GPGKeyImport(s).
- for _, key := range keys {
- if _, err = e.Delete(&GPGKeyImport{KeyID: key.KeyID}); err != nil {
- return fmt.Errorf("deleteGPGKeyImports: %v", err)
- }
- }
- if _, err = e.Delete(&GPGKey{OwnerID: u.ID}); err != nil {
- return fmt.Errorf("deleteGPGKeys: %v", err)
- }
- // ***** END: GPGPublicKey *****
-
- // Clear assignee.
- if err = clearAssigneeByUserID(e, u.ID); err != nil {
- return fmt.Errorf("clear assignee: %v", err)
- }
-
- // ***** START: ExternalLoginUser *****
- if err = user_model.RemoveAllAccountLinks(ctx, u); err != nil {
- return fmt.Errorf("ExternalLoginUser: %v", err)
- }
- // ***** END: ExternalLoginUser *****
-
- if _, err = e.ID(u.ID).Delete(new(user_model.User)); err != nil {
- return fmt.Errorf("Delete: %v", err)
- }
-
- return nil
- }
-
- // 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<>?", RepoWatchModeDont).
- 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
- }
-
- // IsUserVisibleToViewer check if viewer is able to see user profile
- func IsUserVisibleToViewer(u *user_model.User, viewer *user_model.User) bool {
- return isUserVisibleToViewer(db.GetEngine(db.DefaultContext), u, viewer)
- }
-
- func isUserVisibleToViewer(e db.Engine, u *user_model.User, viewer *user_model.User) bool {
- if viewer != nil && viewer.IsAdmin {
- return true
- }
-
- switch u.Visibility {
- case structs.VisibleTypePublic:
- return true
- case structs.VisibleTypeLimited:
- if viewer == nil || viewer.IsRestricted {
- return false
- }
- return true
- case structs.VisibleTypePrivate:
- if viewer == nil || viewer.IsRestricted {
- return false
- }
-
- // If they follow - they see each over
- follower := user_model.IsFollowing(u.ID, viewer.ID)
- if follower {
- return true
- }
-
- // Now we need to check if they in some organization together
- count, err := e.Table("team_user").
- Where(
- builder.And(
- builder.Eq{"uid": viewer.ID},
- builder.Or(
- builder.Eq{"org_id": u.ID},
- builder.In("org_id",
- builder.Select("org_id").
- From("team_user", "t2").
- Where(builder.Eq{"uid": u.ID}))))).
- Count(new(TeamUser))
- if err != nil {
- return false
- }
-
- if count < 0 {
- // No common organization
- return false
- }
-
- // they are in an organization together
- return true
- }
- return false
- }
|