2014-08-26 20:32:41 +00:00
|
|
|
module Gitlab
|
|
|
|
class ProjectSearchResults < SearchResults
|
|
|
|
attr_reader :project, :repository_ref
|
|
|
|
|
2014-08-26 21:04:14 +00:00
|
|
|
def initialize(project_id, query, repository_ref = nil)
|
2014-08-26 20:32:41 +00:00
|
|
|
@project = Project.find(project_id)
|
2014-09-25 10:56:23 +00:00
|
|
|
@repository_ref = if repository_ref.present?
|
|
|
|
repository_ref
|
|
|
|
else
|
|
|
|
nil
|
|
|
|
end
|
2014-08-26 20:32:41 +00:00
|
|
|
@query = Shellwords.shellescape(query) if query.present?
|
|
|
|
end
|
|
|
|
|
2014-08-27 12:26:35 +00:00
|
|
|
def objects(scope, page = nil)
|
2014-08-26 20:32:41 +00:00
|
|
|
case scope
|
|
|
|
when 'notes'
|
|
|
|
notes.page(page).per(per_page)
|
|
|
|
when 'blobs'
|
|
|
|
Kaminari.paginate_array(blobs).page(page).per(per_page)
|
2014-09-05 10:33:05 +00:00
|
|
|
when 'wiki_blobs'
|
|
|
|
Kaminari.paginate_array(wiki_blobs).page(page).per(per_page)
|
2014-08-26 20:32:41 +00:00
|
|
|
else
|
|
|
|
super
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
|
|
|
def total_count
|
2014-09-06 18:42:11 +00:00
|
|
|
@total_count ||= issues_count + merge_requests_count + blobs_count +
|
2014-09-06 09:46:14 +00:00
|
|
|
notes_count + wiki_blobs_count
|
2014-08-26 20:32:41 +00:00
|
|
|
end
|
|
|
|
|
|
|
|
def blobs_count
|
|
|
|
@blobs_count ||= blobs.count
|
|
|
|
end
|
|
|
|
|
|
|
|
def notes_count
|
|
|
|
@notes_count ||= notes.count
|
|
|
|
end
|
|
|
|
|
2014-09-05 10:33:05 +00:00
|
|
|
def wiki_blobs_count
|
|
|
|
@wiki_blobs_count ||= wiki_blobs.count
|
|
|
|
end
|
|
|
|
|
2014-08-26 20:32:41 +00:00
|
|
|
private
|
|
|
|
|
|
|
|
def blobs
|
2014-09-25 12:01:02 +00:00
|
|
|
if project.empty_repo? || query.blank?
|
2014-08-26 20:32:41 +00:00
|
|
|
[]
|
|
|
|
else
|
|
|
|
project.repository.search_files(query, repository_ref)
|
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2014-09-05 10:33:05 +00:00
|
|
|
def wiki_blobs
|
2014-09-25 12:01:02 +00:00
|
|
|
if project.wiki_enabled? && query.present?
|
2014-09-25 10:56:23 +00:00
|
|
|
project_wiki = ProjectWiki.new(project)
|
2014-09-09 14:56:33 +00:00
|
|
|
|
2014-09-25 10:56:23 +00:00
|
|
|
unless project_wiki.empty?
|
|
|
|
project_wiki.search_files(query)
|
2014-09-09 14:56:33 +00:00
|
|
|
else
|
|
|
|
[]
|
|
|
|
end
|
2014-09-05 10:33:05 +00:00
|
|
|
else
|
2014-09-09 14:56:33 +00:00
|
|
|
[]
|
2014-09-05 10:33:05 +00:00
|
|
|
end
|
|
|
|
end
|
|
|
|
|
2014-08-26 20:32:41 +00:00
|
|
|
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
|