debian-mirror-gitlab/lib/gitlab/project_search_results.rb

92 lines
2 KiB
Ruby
Raw Normal View History

2015-04-26 12:48:37 +05:30
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
2015-11-26 14:37:03 +05:30
@query = query
2015-04-26 12:48:37 +05:30
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)
2015-11-26 14:37:03 +05:30
when 'commits'
Kaminari.paginate_array(commits).page(page).per(per_page)
2015-04-26 12:48:37 +05:30
else
super
end
end
def total_count
@total_count ||= issues_count + merge_requests_count + blobs_count +
2015-11-26 14:37:03 +05:30
notes_count + wiki_blobs_count + commits_count
2015-04-26 12:48:37 +05:30
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
2015-11-26 14:37:03 +05:30
def commits_count
@commits_count ||= commits.count
end
2015-04-26 12:48:37 +05:30
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
2015-11-26 14:37:03 +05:30
def commits
if project.empty_repo? || query.blank?
[]
else
project.repository.find_commits_by_message(query).compact
end
end
2015-04-26 12:48:37 +05:30
def limit_project_ids
[project.id]
end
end
end