diff options
-rw-r--r-- | lib/gitlab/project_search_results.rb | 62 | ||||
-rw-r--r-- | lib/gitlab/search_results.rb | 75 |
2 files changed, 137 insertions, 0 deletions
diff --git a/lib/gitlab/project_search_results.rb b/lib/gitlab/project_search_results.rb new file mode 100644 index 00000000000..b0a0bee1c81 --- /dev/null +++ b/lib/gitlab/project_search_results.rb @@ -0,0 +1,62 @@ +module Gitlab + class ProjectSearchResults < SearchResults + attr_reader :project, :repository_ref + + def initialize(project_id, query, scope = nil, page = nil, repository_ref = nil) + @project = Project.find(project_id) + @repository_ref = repository_ref + @page = page + @query = Shellwords.shellescape(query) if query.present? + @scope = scope + + unless %w(blobs notes issues merge_requests).include?(@scope) + @scope = default_scope + end + end + + def objects + case scope + when 'notes' + notes.page(page).per(per_page) + when 'blobs' + Kaminari.paginate_array(blobs).page(page).per(per_page) + else + super + end + end + + def total_count + @total_count ||= issues_count + merge_requests_count + blobs_count + notes_count + end + + def blobs_count + @blobs_count ||= blobs.count + end + + def notes_count + @notes_count ||= notes.count + end + + private + + def blobs + if project.empty_repo? + [] + else + project.repository.search_files(query, repository_ref) + end + end + + def notes + Note.where(project_id: limit_project_ids).search(query).order('updated_at DESC') + end + + def default_scope + 'blobs' + end + + def limit_project_ids + [project.id] + end + end +end diff --git a/lib/gitlab/search_results.rb b/lib/gitlab/search_results.rb new file mode 100644 index 00000000000..1325d542a0f --- /dev/null +++ b/lib/gitlab/search_results.rb @@ -0,0 +1,75 @@ +module Gitlab + class SearchResults + attr_reader :scope, :objects, :query, :page + + # Limit search results by passed project ids + # It allows us to search only for projects user has access to + attr_reader :limit_project_ids + + def initialize(limit_project_ids, query, scope = nil, page = nil) + @limit_project_ids = limit_project_ids || Project.all + @page = page + @query = Shellwords.shellescape(query) if query.present? + @scope = scope + + unless %w(projects issues merge_requests).include?(@scope) + @scope = default_scope + end + end + + def objects + case scope + when 'projects' + projects.page(page).per(per_page) + when 'issues' + issues.page(page).per(per_page) + when 'merge_requests' + merge_requests.page(page).per(per_page) + else + Kaminari.paginate_array([]).page(page).per(per_page) + end + end + + def total_count + @total_count ||= projects_count + issues_count + merge_requests_count + end + + def projects_count + @projects_count ||= projects.count + end + + def issues_count + @issues_count ||= issues.count + end + + def merge_requests_count + @merge_requests_count ||= merge_requests.count + end + + def empty? + total_count.zero? + end + + private + + def projects + Project.where(id: limit_project_ids).search(query) + end + + def issues + Issue.where(project_id: limit_project_ids).search(query).order('updated_at DESC') + end + + def merge_requests + MergeRequest.in_projects(limit_project_ids).search(query).order('updated_at DESC') + end + + def default_scope + 'projects' + end + + def per_page + 20 + end + end +end |