1 files changed, 67 insertions, 0 deletions
diff --git a/app/services/draft_notes/publish_service.rb b/app/services/draft_notes/publish_service.rb
new file mode 100644
index 00000000000..a9a7304e5ed
--- /dev/null
+++ b/app/services/draft_notes/publish_service.rb
@@ -0,0 +1,67 @@
+# frozen_string_literal: true
+
+module DraftNotes
+  class PublishService < DraftNotes::BaseService
+    def execute(draft = nil)
+      return error('Not allowed to create notes') unless can?(current_user, :create_note, merge_request)
+
+      if draft
+        publish_draft_note(draft)
+      else
+        publish_draft_notes
+      end
+
+      success
+    rescue ActiveRecord::RecordInvalid => e
+      message = "Unable to save #{e.record.class.name}: #{e.record.errors.full_messages.join(", ")} "
+      error(message)
+    end
+
+    private
+
+    def publish_draft_note(draft)
+      create_note_from_draft(draft)
+      draft.delete
+
+      MergeRequests::ResolvedDiscussionNotificationService.new(project, current_user).execute(merge_request)
+    end
+
+    def publish_draft_notes
+      return if draft_notes.empty?
+
+      review = Review.create!(author: current_user, merge_request: merge_request, project: project)
+
+      draft_notes.map do |draft_note|
+        draft_note.review = review
+        create_note_from_draft(draft_note)
+      end
+      draft_notes.delete_all
+
+      notification_service.async.new_review(review)
+      MergeRequests::ResolvedDiscussionNotificationService.new(project, current_user).execute(merge_request)
+    end
+
+    def create_note_from_draft(draft)
+      # Make sure the diff file is unfolded in order to find the correct line
+      # codes.
+      draft.diff_file&.unfold_diff_lines(draft.original_position)
+
+      note = Notes::CreateService.new(draft.project, draft.author, draft.publish_params).execute
+      set_discussion_resolve_status(note, draft)
+
+      note
+    end
+
+    def set_discussion_resolve_status(note, draft_note)
+      return unless draft_note.discussion_id.present?
+
+      discussion = note.discussion
+
+      if draft_note.resolve_discussion && discussion.can_resolve?(current_user)
+        discussion.resolve!(current_user)
+      else
+        discussion.unresolve!
+      end
+    end
+  end
+end