339 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			Ruby
		
	
	
	
			
		
		
	
	
			339 lines
		
	
	
		
			9.0 KiB
		
	
	
	
		
			Ruby
		
	
	
	
# frozen_string_literal: true
 | 
						|
 | 
						|
module NotesActions
 | 
						|
  include RendersNotes
 | 
						|
  include Gitlab::Utils::StrongMemoize
 | 
						|
  extend ActiveSupport::Concern
 | 
						|
 | 
						|
  # last_fetched_at is an integer number of microseconds, which is the same
 | 
						|
  # precision as PostgreSQL "timestamp" fields.
 | 
						|
  MICROSECOND = 1_000_000
 | 
						|
 | 
						|
  included do
 | 
						|
    before_action :set_polling_interval_header, only: [:index]
 | 
						|
    before_action :require_last_fetched_at_header!, only: [:index]
 | 
						|
    before_action :require_noteable!, only: [:index, :create]
 | 
						|
    before_action :authorize_admin_note!, only: [:update, :destroy]
 | 
						|
    before_action :note_project, only: [:create]
 | 
						|
    before_action -> {
 | 
						|
      check_rate_limit!(:notes_create,
 | 
						|
        scope: current_user,
 | 
						|
        users_allowlist: Gitlab::CurrentSettings.current_application_settings.notes_create_limit_allowlist)
 | 
						|
    }, only: [:create]
 | 
						|
  end
 | 
						|
 | 
						|
  def index
 | 
						|
    notes, meta = gather_all_notes
 | 
						|
    notes = prepare_notes_for_rendering(notes)
 | 
						|
    notes = notes.select { |n| n.readable_by?(current_user) }
 | 
						|
    notes =
 | 
						|
      if use_note_serializer?
 | 
						|
        note_serializer.represent(notes)
 | 
						|
      else
 | 
						|
        notes.map { |note| note_json(note) }
 | 
						|
      end
 | 
						|
 | 
						|
    render json: meta.merge(notes: notes)
 | 
						|
  end
 | 
						|
 | 
						|
  # rubocop:disable Gitlab/ModuleWithInstanceVariables
 | 
						|
  def create
 | 
						|
    @note = Notes::CreateService.new(note_project, current_user, create_note_params).execute
 | 
						|
 | 
						|
    respond_to do |format|
 | 
						|
      format.json do
 | 
						|
        json = {
 | 
						|
          commands_changes: @note.commands_changes&.slice(:emoji_award, :time_estimate, :spend_time)
 | 
						|
        }
 | 
						|
 | 
						|
        if @note.persisted? && return_discussion?
 | 
						|
          json[:valid] = true
 | 
						|
 | 
						|
          discussion = @note.discussion
 | 
						|
          prepare_notes_for_rendering(discussion.notes)
 | 
						|
          json[:discussion] = discussion_serializer.represent(discussion, context: self)
 | 
						|
        else
 | 
						|
          prepare_notes_for_rendering([@note])
 | 
						|
 | 
						|
          json.merge!(note_json(@note))
 | 
						|
        end
 | 
						|
 | 
						|
        quick_actions = @note.quick_actions_status
 | 
						|
        json[:quick_actions_status] = quick_actions.to_h if quick_actions
 | 
						|
 | 
						|
        if @note.errors.present?
 | 
						|
          render json: { errors: errors_on_create(@note) }, status: :unprocessable_entity
 | 
						|
        elsif quick_actions&.error?
 | 
						|
          render json: { quick_actions_status: quick_actions.to_h }, status: :unprocessable_entity
 | 
						|
        else
 | 
						|
          render json: json
 | 
						|
        end
 | 
						|
      end
 | 
						|
      format.html { redirect_back_or_default }
 | 
						|
    end
 | 
						|
  end
 | 
						|
  # rubocop:enable Gitlab/ModuleWithInstanceVariables
 | 
						|
 | 
						|
  # rubocop:disable Gitlab/ModuleWithInstanceVariables
 | 
						|
  def update
 | 
						|
    @note = Notes::UpdateService.new(project, current_user, update_note_params).execute(note)
 | 
						|
    if @note.destroyed?
 | 
						|
      head :gone
 | 
						|
      return
 | 
						|
    end
 | 
						|
 | 
						|
    respond_to do |format|
 | 
						|
      format.json do
 | 
						|
        if @note.errors.present?
 | 
						|
          render json: { errors: @note.errors.full_messages.to_sentence }, status: :unprocessable_entity
 | 
						|
        else
 | 
						|
          prepare_notes_for_rendering([@note])
 | 
						|
          render json: note_json(@note)
 | 
						|
        end
 | 
						|
      end
 | 
						|
 | 
						|
      format.html { redirect_back_or_default }
 | 
						|
    end
 | 
						|
  end
 | 
						|
  # rubocop:enable Gitlab/ModuleWithInstanceVariables
 | 
						|
 | 
						|
  def destroy
 | 
						|
    Notes::DestroyService.new(project, current_user).execute(note) if note.editable?
 | 
						|
 | 
						|
    respond_to do |format|
 | 
						|
      format.js { head :ok }
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  private
 | 
						|
 | 
						|
  def gather_all_notes
 | 
						|
    now = Time.current
 | 
						|
    notes = merge_resource_events(notes_finder.execute.inc_relations_for_view(noteable))
 | 
						|
 | 
						|
    [notes, { last_fetched_at: (now.to_i * MICROSECOND) + now.usec }]
 | 
						|
  end
 | 
						|
 | 
						|
  def merge_resource_events(notes)
 | 
						|
    return notes if notes_filter == UserPreference::NOTES_FILTERS[:only_comments]
 | 
						|
 | 
						|
    ResourceEvents::MergeIntoNotesService
 | 
						|
      .new(noteable, current_user, last_fetched_at: last_fetched_at)
 | 
						|
      .execute(notes)
 | 
						|
  end
 | 
						|
 | 
						|
  def note_html(note)
 | 
						|
    render_to_string(
 | 
						|
      "shared/notes/_note",
 | 
						|
      layout: false,
 | 
						|
      formats: [:html],
 | 
						|
      locals: { note: note }
 | 
						|
    )
 | 
						|
  end
 | 
						|
 | 
						|
  def note_json(note)
 | 
						|
    attrs = {}
 | 
						|
 | 
						|
    if note.persisted?
 | 
						|
      attrs[:valid] = true
 | 
						|
 | 
						|
      if return_discussion?
 | 
						|
        discussion = note.discussion
 | 
						|
        prepare_notes_for_rendering(discussion.notes)
 | 
						|
 | 
						|
        attrs[:discussion] = discussion_serializer.represent(discussion, context: self)
 | 
						|
      elsif use_note_serializer?
 | 
						|
        attrs.merge!(note_serializer.represent(note))
 | 
						|
      else
 | 
						|
        attrs.merge!(
 | 
						|
          id: note.id,
 | 
						|
          discussion_id: note.discussion_id(noteable),
 | 
						|
          html: note_html(note),
 | 
						|
          note: note.note,
 | 
						|
          on_image: note.try(:on_image?)
 | 
						|
        )
 | 
						|
 | 
						|
        discussion = note.to_discussion(noteable)
 | 
						|
        unless discussion.individual_note?
 | 
						|
          attrs.merge!(
 | 
						|
            discussion_resolvable: discussion.resolvable?,
 | 
						|
 | 
						|
            diff_discussion_html: diff_discussion_html(discussion),
 | 
						|
            discussion_html: discussion_html(discussion)
 | 
						|
          )
 | 
						|
 | 
						|
          attrs[:discussion_line_code] = discussion.line_code if discussion.diff_discussion?
 | 
						|
        end
 | 
						|
      end
 | 
						|
    else
 | 
						|
      attrs.merge!(
 | 
						|
        valid: false,
 | 
						|
        errors: note.errors
 | 
						|
      )
 | 
						|
    end
 | 
						|
 | 
						|
    attrs
 | 
						|
  end
 | 
						|
 | 
						|
  def diff_discussion_html(discussion)
 | 
						|
    return unless discussion.diff_discussion?
 | 
						|
 | 
						|
    on_image = discussion.on_image?
 | 
						|
 | 
						|
    if params[:view] == 'parallel' && !on_image
 | 
						|
      template = "discussions/_parallel_diff_discussion"
 | 
						|
      locals =
 | 
						|
        if params[:line_type] == 'old'
 | 
						|
          { discussions_left: [discussion], discussions_right: nil }
 | 
						|
        else
 | 
						|
          { discussions_left: nil, discussions_right: [discussion] }
 | 
						|
        end
 | 
						|
    else
 | 
						|
      template = "discussions/_diff_discussion"
 | 
						|
      @fresh_discussion = true # rubocop:disable Gitlab/ModuleWithInstanceVariables
 | 
						|
 | 
						|
      locals = { discussions: [discussion], on_image: on_image }
 | 
						|
    end
 | 
						|
 | 
						|
    render_to_string(
 | 
						|
      template,
 | 
						|
      layout: false,
 | 
						|
      formats: [:html],
 | 
						|
      locals: locals
 | 
						|
    )
 | 
						|
  end
 | 
						|
 | 
						|
  def discussion_html(discussion)
 | 
						|
    return if discussion.individual_note?
 | 
						|
 | 
						|
    render_to_string(
 | 
						|
      "discussions/_discussion",
 | 
						|
      layout: false,
 | 
						|
      formats: [:html],
 | 
						|
      locals: { discussion: discussion }
 | 
						|
    )
 | 
						|
  end
 | 
						|
 | 
						|
  def authorize_admin_note!
 | 
						|
    access_denied! unless can?(current_user, :admin_note, note)
 | 
						|
  end
 | 
						|
 | 
						|
  def create_note_params
 | 
						|
    params.require(:note).permit(
 | 
						|
      :type,
 | 
						|
      :note,
 | 
						|
      :line_code, # LegacyDiffNote
 | 
						|
      :position, # DiffNote
 | 
						|
      :confidential,
 | 
						|
      :internal
 | 
						|
    ).tap do |create_params|
 | 
						|
      create_params.merge!(
 | 
						|
        params.permit(:merge_request_diff_head_sha, :in_reply_to_discussion_id)
 | 
						|
      )
 | 
						|
 | 
						|
      # These params are also sent by the client but we need to set these based on
 | 
						|
      # target_type and target_id because we're checking permissions based on that
 | 
						|
      create_params[:noteable_type] = noteable.class.name
 | 
						|
 | 
						|
      case noteable
 | 
						|
      when Commit
 | 
						|
        create_params[:commit_id] = noteable.id
 | 
						|
      when MergeRequest
 | 
						|
        create_params[:noteable_id] = noteable.id
 | 
						|
        # Notes on MergeRequest can have an extra `commit_id` context
 | 
						|
        create_params[:commit_id] = params.dig(:note, :commit_id)
 | 
						|
      else
 | 
						|
        create_params[:noteable_id] = noteable.id
 | 
						|
      end
 | 
						|
    end
 | 
						|
  end
 | 
						|
 | 
						|
  def update_note_params
 | 
						|
    params.require(:note).permit(:note, :position)
 | 
						|
  end
 | 
						|
 | 
						|
  def set_polling_interval_header(interval: 6000)
 | 
						|
    Gitlab::PollingInterval.set_header(response, interval: interval)
 | 
						|
  end
 | 
						|
 | 
						|
  def noteable
 | 
						|
    @noteable ||= notes_finder.target || @note&.noteable # rubocop:disable Gitlab/ModuleWithInstanceVariables
 | 
						|
  end
 | 
						|
 | 
						|
  def require_noteable!
 | 
						|
    render_404 unless noteable
 | 
						|
  end
 | 
						|
 | 
						|
  def require_last_fetched_at_header!
 | 
						|
    return if request.headers['X-Last-Fetched-At'].present?
 | 
						|
 | 
						|
    render json: { message: 'X-Last-Fetched-At header is required' }, status: :bad_request
 | 
						|
  end
 | 
						|
 | 
						|
  def last_fetched_at
 | 
						|
    microseconds = request.headers['X-Last-Fetched-At'].to_i
 | 
						|
 | 
						|
    seconds = microseconds / MICROSECOND
 | 
						|
    frac = microseconds % MICROSECOND
 | 
						|
 | 
						|
    Time.zone.at(seconds, frac)
 | 
						|
  end
 | 
						|
  strong_memoize_attr :last_fetched_at
 | 
						|
 | 
						|
  def notes_filter
 | 
						|
    current_user&.notes_filter_for(params[:target_type])
 | 
						|
  end
 | 
						|
 | 
						|
  def notes_finder
 | 
						|
    @notes_finder ||= NotesFinder.new(current_user, finder_params)
 | 
						|
  end
 | 
						|
 | 
						|
  def note_serializer
 | 
						|
    ProjectNoteSerializer.new(project: project, noteable: noteable, current_user: current_user)
 | 
						|
  end
 | 
						|
  strong_memoize_attr :note_serializer
 | 
						|
 | 
						|
  def discussion_serializer
 | 
						|
    DiscussionSerializer.new(
 | 
						|
      project: project,
 | 
						|
      noteable: noteable,
 | 
						|
      current_user: current_user,
 | 
						|
      note_entity: ProjectNoteEntity
 | 
						|
    )
 | 
						|
  end
 | 
						|
 | 
						|
  def note_project
 | 
						|
    return unless project
 | 
						|
 | 
						|
    note_project_id = params[:note_project_id]
 | 
						|
 | 
						|
    the_project =
 | 
						|
      if note_project_id.present?
 | 
						|
        Project.find(note_project_id)
 | 
						|
      else
 | 
						|
        project
 | 
						|
      end
 | 
						|
 | 
						|
    return access_denied! unless can?(current_user, :create_note, the_project)
 | 
						|
 | 
						|
    the_project
 | 
						|
  end
 | 
						|
  strong_memoize_attr :note_project
 | 
						|
 | 
						|
  def return_discussion?
 | 
						|
    Gitlab::Utils.to_boolean(params[:return_discussion])
 | 
						|
  end
 | 
						|
 | 
						|
  def use_note_serializer?
 | 
						|
    return false if params['html']
 | 
						|
 | 
						|
    noteable.discussions_rendered_on_frontend?
 | 
						|
  end
 | 
						|
 | 
						|
  def errors_on_create(note)
 | 
						|
    note.errors.full_messages.to_sentence
 | 
						|
  end
 | 
						|
end
 | 
						|
 | 
						|
NotesActions.prepend_mod_with('NotesActions')
 |