summaryrefslogtreecommitdiff
path: root/lib/gitlab/snippet_search_results.rb
blob: ac3b219e0c7320f9fdaa98aa8b81656f4f235c78 (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
# frozen_string_literal: true

module Gitlab
  class SnippetSearchResults < SearchResults
    include SnippetsHelper

    attr_reader :limit_snippets

    def initialize(limit_snippets, query)
      @limit_snippets = limit_snippets
      @query = query
    end

    def objects(scope, page = nil)
      case scope
      when 'snippet_titles'
        snippet_titles.page(page).per(per_page)
      when 'snippet_blobs'
        snippet_blobs.page(page).per(per_page)
      else
        super(scope, nil, false)
      end
    end

    def formatted_count(scope)
      case scope
      when 'snippet_titles'
        snippet_titles_count.to_s
      when 'snippet_blobs'
        snippet_blobs_count.to_s
      else
        super
      end
    end

    def snippet_titles_count
      @snippet_titles_count ||= snippet_titles.count
    end

    def snippet_blobs_count
      @snippet_blobs_count ||= snippet_blobs.count
    end

    private

    # rubocop: disable CodeReuse/ActiveRecord
    def snippet_titles
      limit_snippets.search(query).order('updated_at DESC').includes(:author)
    end
    # rubocop: enable CodeReuse/ActiveRecord

    # rubocop: disable CodeReuse/ActiveRecord
    def snippet_blobs
      limit_snippets.search_code(query).order('updated_at DESC').includes(:author)
    end
    # rubocop: enable CodeReuse/ActiveRecord

    def default_scope
      'snippet_blobs'
    end
  end
end