summaryrefslogtreecommitdiff
path: root/lib/gitlab/project_search_results.rb
blob: 54389f7d662ddea31bae623e547b77ff1e53c12d (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
module Gitlab
  class ProjectSearchResults < SearchResults
    attr_reader :project, :repository_ref

    def initialize(project_id, query, repository_ref = nil)
      @project = Project.find(project_id)
      @repository_ref = if repository_ref.present?
                          repository_ref
                        else
                          nil
                        end
      @query = query
    end

    def objects(scope, page = nil)
      case scope
      when 'notes'
        notes.page(page).per(per_page)
      when 'blobs'
        Kaminari.paginate_array(blobs).page(page).per(per_page)
      when 'wiki_blobs'
        Kaminari.paginate_array(wiki_blobs).page(page).per(per_page)
      when 'commits'
        Kaminari.paginate_array(commits).page(page).per(per_page)
      else
        super
      end
    end

    def total_count
      @total_count ||= issues_count + merge_requests_count + blobs_count +
                       notes_count + wiki_blobs_count + commits_count
    end

    def blobs_count
      @blobs_count ||= blobs.count
    end

    def notes_count
      @notes_count ||= notes.count
    end

    def wiki_blobs_count
      @wiki_blobs_count ||= wiki_blobs.count
    end

    def commits_count
      @commits_count ||= commits.count
    end

    private

    def blobs
      if project.empty_repo? || query.blank?
        []
      else
        project.repository.search_files(query, repository_ref)
      end
    end

    def wiki_blobs
      if project.wiki_enabled? && query.present?
        project_wiki = ProjectWiki.new(project)

        unless project_wiki.empty?
          project_wiki.search_files(query)
        else
          []
        end
      else
        []
      end
    end

    def notes
      Note.where(project_id: limit_project_ids).user.search(query).order('updated_at DESC')
    end

    def commits
      project.repository.find_commits_with_matching_log(query)
    end

    def limit_project_ids
      [project.id]
    end
  end
end