gitlab-org--gitlab-foss/app/controllers/concerns/renders_notes.rb

37 lines
1.2 KiB
Ruby
Raw Normal View History

# frozen_string_literal: true
2017-03-30 23:06:09 -04:00
module RendersNotes
# rubocop:disable Gitlab/ModuleWithInstanceVariables
2017-08-02 10:06:28 -04:00
def prepare_notes_for_rendering(notes, noteable = nil)
2017-03-30 23:06:09 -04:00
preload_noteable_for_regular_notes(notes)
preload_max_access_for_authors(notes, @project)
preload_author_status(notes)
Notes::RenderService.new(current_user).execute(notes)
2017-03-30 23:06:09 -04:00
notes
end
# rubocop:enable Gitlab/ModuleWithInstanceVariables
2017-03-30 23:06:09 -04:00
private
def preload_max_access_for_authors(notes, project)
return unless project
2017-04-27 06:41:26 -04:00
2017-03-30 23:06:09 -04:00
user_ids = notes.map(&:author_id)
access = project.team.max_member_access_for_user_ids(user_ids).select { |k, v| v == Gitlab::Access::NO_ACCESS }.keys
project.team.contribution_check_for_user_ids(access)
2017-03-30 23:06:09 -04:00
end
# rubocop: disable CodeReuse/ActiveRecord
2017-03-30 23:06:09 -04:00
def preload_noteable_for_regular_notes(notes)
ActiveRecord::Associations::Preloader.new.preload(notes.reject(&:for_commit?), :noteable)
end
# rubocop: enable CodeReuse/ActiveRecord
# rubocop: disable CodeReuse/ActiveRecord
def preload_author_status(notes)
ActiveRecord::Associations::Preloader.new.preload(notes, { author: :status })
end
# rubocop: enable CodeReuse/ActiveRecord
2017-03-30 23:06:09 -04:00
end