project_search_results.rb 1.5 KB
Newer Older
1 2 3 4
module Gitlab
  class ProjectSearchResults < SearchResults
    attr_reader :project, :repository_ref

5
    def initialize(project_id, query, repository_ref = nil)
6 7 8 9 10
      @project = Project.find(project_id)
      @repository_ref = repository_ref
      @query = Shellwords.shellescape(query) if query.present?
    end

Dmitriy Zaporozhets's avatar
Dmitriy Zaporozhets committed
11
    def objects(scope, page = nil)
12 13 14 15 16
      case scope
      when 'notes'
        notes.page(page).per(per_page)
      when 'blobs'
        Kaminari.paginate_array(blobs).page(page).per(per_page)
Ralf Seidler's avatar
Ralf Seidler committed
17 18
      when 'wiki_blobs'
        Kaminari.paginate_array(wiki_blobs).page(page).per(per_page)
19 20 21 22 23 24
      else
        super
      end
    end

    def total_count
Ralf Seidler's avatar
Ralf Seidler committed
25
      @total_count ||= issues_count + merge_requests_count + blobs_count +
26
                       notes_count + wiki_blobs_count
27 28 29 30 31 32 33 34 35 36
    end

    def blobs_count
      @blobs_count ||= blobs.count
    end

    def notes_count
      @notes_count ||= notes.count
    end

Ralf Seidler's avatar
Ralf Seidler committed
37 38 39 40
    def wiki_blobs_count
      @wiki_blobs_count ||= wiki_blobs.count
    end

41 42 43 44 45 46 47 48 49 50
    private

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

Ralf Seidler's avatar
Ralf Seidler committed
51 52 53 54
    def wiki_blobs
      if !project.wiki_enabled?
        []
      else
55 56
        Repository.new(ProjectWiki.new(project).path_with_namespace).
           search_files(query)
Ralf Seidler's avatar
Ralf Seidler committed
57 58 59
      end
    end

60 61 62 63 64 65 66 67 68
    def notes
      Note.where(project_id: limit_project_ids).search(query).order('updated_at DESC')
    end

    def limit_project_ids
      [project.id]
    end
  end
end