diff --git a/app/contexts/search/global_context.rb b/app/contexts/search/global_context.rb new file mode 100644 index 00000000000..c0966bb5220 --- /dev/null +++ b/app/contexts/search/global_context.rb @@ -0,0 +1,45 @@ +module Search + class GlobalContext + attr_accessor :current_user, :params + + def initialize(user, params) + @current_user, @params = user, params.dup + end + + def execute + query = params[:search] + query = Shellwords.shellescape(query) if query.present? + return result unless query.present? + + + projects = current_user.authorized_projects + + if params[:group_id].present? + group = Group.find_by_id(params[:group_id]) + projects = projects.where(namespace_id: group.id) if group + end + + project_ids = projects.pluck(:id) + + visibility_levels = if current_user + [Gitlab::VisibilityLevel::INTERNAL, Gitlab::VisibilityLevel::PUBLIC] + else + [Gitlab::VisibilityLevel::PUBLIC] + end + result[:projects] = Project.where("projects.id in (?) OR projects.visibility_level in (?)", project_ids, visibility_levels).search(query).limit(20) + result[:merge_requests] = MergeRequest.in_projects(project_ids).search(query).order('updated_at DESC').limit(20) + result[:issues] = Issue.where(project_id: project_ids).search(query).order('updated_at DESC').limit(20) + result[:total_results] = %w(projects issues merge_requests).sum { |items| result[items.to_sym].size } + result + end + + def result + @result ||= { + projects: [], + merge_requests: [], + issues: [], + total_results: 0, + } + end + end +end diff --git a/app/contexts/search/project_context.rb b/app/contexts/search/project_context.rb new file mode 100644 index 00000000000..690652b2cdb --- /dev/null +++ b/app/contexts/search/project_context.rb @@ -0,0 +1,37 @@ +module Search + class ProjectContext + attr_accessor :project, :current_user, :params + + def initialize(project, user, params) + @project, @current_user, @params = project, user, params.dup + end + + def execute + query = params[:search] + query = Shellwords.shellescape(query) if query.present? + return result unless query.present? + + if params[:search_code].present? + blobs = project.repository.search_files(query, params[:repository_ref]) unless project.empty_repo? + blobs = Kaminari.paginate_array(blobs).page(params[:page]).per(20) + result[:blobs] = blobs + result[:total_results] = blobs.total_count + else + result[:merge_requests] = project.merge_requests.search(query).order('updated_at DESC').limit(20) + result[:issues] = project.issues.search(query).order('updated_at DESC').limit(20) + result[:total_results] = %w(issues merge_requests).sum { |items| result[items.to_sym].size } + end + + result + end + + def result + @result ||= { + merge_requests: [], + issues: [], + blobs: [], + total_results: 0, + } + end + end +end