123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302 |
- // 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 (
- "fmt"
- "strings"
-
- "github.com/go-xorm/builder"
- )
-
- // RepositoryList contains a list of repositories
- type RepositoryList []*Repository
-
- // RepositoryListOfMap make list from values of map
- func RepositoryListOfMap(repoMap map[int64]*Repository) RepositoryList {
- return RepositoryList(valuesRepository(repoMap))
- }
-
- func (repos RepositoryList) loadAttributes(e Engine) error {
- if len(repos) == 0 {
- return nil
- }
-
- // Load owners.
- set := make(map[int64]struct{})
- for i := range repos {
- set[repos[i].OwnerID] = struct{}{}
- }
- users := make(map[int64]*User, len(set))
- if err := e.
- Where("id > 0").
- In("id", keysInt64(set)).
- Find(&users); err != nil {
- return fmt.Errorf("find users: %v", err)
- }
- for i := range repos {
- repos[i].Owner = users[repos[i].OwnerID]
- }
- return nil
- }
-
- // LoadAttributes loads the attributes for the given RepositoryList
- func (repos RepositoryList) LoadAttributes() error {
- return repos.loadAttributes(x)
- }
-
- // MirrorRepositoryList contains the mirror repositories
- type MirrorRepositoryList []*Repository
-
- func (repos MirrorRepositoryList) loadAttributes(e Engine) error {
- if len(repos) == 0 {
- return nil
- }
-
- // Load mirrors.
- repoIDs := make([]int64, 0, len(repos))
- for i := range repos {
- if !repos[i].IsMirror {
- continue
- }
-
- repoIDs = append(repoIDs, repos[i].ID)
- }
- mirrors := make([]*Mirror, 0, len(repoIDs))
- if err := e.
- Where("id > 0").
- In("repo_id", repoIDs).
- Find(&mirrors); err != nil {
- return fmt.Errorf("find mirrors: %v", err)
- }
-
- set := make(map[int64]*Mirror)
- for i := range mirrors {
- set[mirrors[i].RepoID] = mirrors[i]
- }
- for i := range repos {
- repos[i].Mirror = set[repos[i].ID]
- }
- return nil
- }
-
- // LoadAttributes loads the attributes for the given MirrorRepositoryList
- func (repos MirrorRepositoryList) LoadAttributes() error {
- return repos.loadAttributes(x)
- }
-
- // SearchRepoOptions holds the search options
- // swagger:parameters repoSearch
- type SearchRepoOptions struct {
- // Keyword to search
- //
- // in: query
- Keyword string `json:"q"`
- // Owner in we search search
- //
- // in: query
- OwnerID int64 `json:"uid"`
- Searcher *User `json:"-"` //ID of the person who's seeking
- OrderBy SearchOrderBy `json:"-"`
- Private bool `json:"-"` // Include private repositories in results
- Collaborate bool `json:"-"` // Include collaborative repositories
- Starred bool `json:"-"`
- Page int `json:"-"`
- IsProfile bool `json:"-"`
- // Limit of result
- //
- // maximum: setting.ExplorePagingNum
- // in: query
- PageSize int `json:"limit"` // Can be smaller than or equal to setting.ExplorePagingNum
- }
-
- //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 = "name DESC"
- SearchOrderByLeastUpdated = "updated_unix ASC"
- SearchOrderByRecentUpdated = "updated_unix DESC"
- SearchOrderByOldest = "created_unix ASC"
- SearchOrderByNewest = "created_unix DESC"
- SearchOrderBySize = "size ASC"
- SearchOrderBySizeReverse = "size DESC"
- )
-
- // 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) (repos RepositoryList, count int64, err error) {
- var cond = builder.NewCond()
- if opts.Page <= 0 {
- opts.Page = 1
- }
-
- var starJoin bool
- if opts.Starred && opts.OwnerID > 0 {
- cond = builder.Eq{
- "star.uid": opts.OwnerID,
- }
- starJoin = true
- }
-
- opts.Keyword = strings.ToLower(opts.Keyword)
- if opts.Keyword != "" {
- cond = cond.And(builder.Like{"lower_name", opts.Keyword})
- }
-
- // Append conditions
- if !opts.Starred && opts.OwnerID > 0 {
- var searcherReposCond builder.Cond = builder.Eq{"owner_id": opts.OwnerID}
- if opts.Searcher != nil {
- var ownerIds []int64
-
- ownerIds = append(ownerIds, opts.Searcher.ID)
- err = opts.Searcher.GetOrganizations(true)
-
- if err != nil {
- return nil, 0, fmt.Errorf("Organization: %v", err)
- }
-
- for _, org := range opts.Searcher.Orgs {
- ownerIds = append(ownerIds, org.ID)
- }
-
- searcherReposCond = searcherReposCond.Or(builder.In("owner_id", ownerIds))
- if opts.Collaborate {
- searcherReposCond = searcherReposCond.Or(builder.Expr("id IN (SELECT repo_id FROM `access` WHERE access.user_id = ? AND owner_id != ?)",
- opts.Searcher.ID, opts.Searcher.ID))
- }
- }
- cond = cond.And(searcherReposCond)
- }
-
- if !opts.Private {
- cond = cond.And(builder.Eq{"is_private": false})
- }
-
- if len(opts.OrderBy) == 0 {
- opts.OrderBy = SearchOrderByAlphabetically
- }
-
- sess := x.NewSession()
- defer sess.Close()
-
- if starJoin {
- count, err = sess.
- Join("INNER", "star", "star.repo_id = repository.id").
- Where(cond).
- Count(new(Repository))
- if err != nil {
- return nil, 0, fmt.Errorf("Count: %v", err)
- }
-
- sess.Join("INNER", "star", "star.repo_id = repository.id")
- } else {
- count, err = sess.
- Where(cond).
- Count(new(Repository))
- if err != nil {
- return nil, 0, fmt.Errorf("Count: %v", err)
- }
- }
-
- repos = make([]*Repository, 0, opts.PageSize)
- if err = sess.
- Where(cond).
- Limit(opts.PageSize, (opts.Page-1)*opts.PageSize).
- OrderBy(opts.OrderBy.String()).
- Find(&repos); err != nil {
- return nil, 0, fmt.Errorf("Repo: %v", err)
- }
-
- if !opts.IsProfile {
- if err = repos.loadAttributes(sess); err != nil {
- return nil, 0, fmt.Errorf("LoadAttributes: %v", err)
- }
- }
-
- return
- }
-
- // Repositories returns all repositories
- func Repositories(opts *SearchRepoOptions) (_ RepositoryList, count int64, err error) {
- if len(opts.OrderBy) == 0 {
- opts.OrderBy = "id ASC"
- }
-
- repos := make(RepositoryList, 0, opts.PageSize)
-
- if err = x.
- Limit(opts.PageSize, (opts.Page-1)*opts.PageSize).
- OrderBy(opts.OrderBy.String()).
- Find(&repos); err != nil {
- return nil, 0, fmt.Errorf("Repo: %v", err)
- }
-
- if err = repos.loadAttributes(x); err != nil {
- return nil, 0, fmt.Errorf("LoadAttributes: %v", err)
- }
-
- count = countRepositories(-1, opts.Private)
-
- return repos, count, nil
- }
-
- // GetRecentUpdatedRepositories returns the list of repositories that are recently updated.
- func GetRecentUpdatedRepositories(opts *SearchRepoOptions) (repos RepositoryList, _ int64, _ error) {
- var cond = builder.NewCond()
-
- if len(opts.OrderBy) == 0 {
- opts.OrderBy = SearchOrderByRecentUpdated
- }
-
- if !opts.Private {
- cond = builder.Eq{
- "is_private": false,
- }
- }
-
- if opts.Searcher != nil && !opts.Searcher.IsAdmin {
- var ownerIds []int64
-
- ownerIds = append(ownerIds, opts.Searcher.ID)
- err := opts.Searcher.GetOrganizations(true)
-
- if err != nil {
- return nil, 0, fmt.Errorf("Organization: %v", err)
- }
-
- for _, org := range opts.Searcher.Orgs {
- ownerIds = append(ownerIds, org.ID)
- }
-
- cond = cond.Or(builder.In("owner_id", ownerIds))
- }
-
- count, err := x.Where(cond).Count(new(Repository))
- if err != nil {
- return nil, 0, fmt.Errorf("Count: %v", err)
- }
-
- if err = x.Where(cond).
- Limit(opts.PageSize, (opts.Page-1)*opts.PageSize).
- Limit(opts.PageSize).
- OrderBy(opts.OrderBy.String()).
- Find(&repos); err != nil {
- return nil, 0, fmt.Errorf("Repo: %v", err)
- }
-
- if err = repos.loadAttributes(x); err != nil {
- return nil, 0, fmt.Errorf("LoadAttributes: %v", err)
- }
-
- return repos, count, nil
- }
|