summaryrefslogtreecommitdiff
path: root/app/services/issuable/clone/content_rewriter.rb
blob: e1e0b75085d103c8bed31a29ac3f4143c47db98c (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
# frozen_string_literal: true

module Issuable
  module Clone
    class ContentRewriter < ::Issuable::Clone::BaseService
      def initialize(current_user, original_entity, new_entity)
        @current_user = current_user
        @original_entity = original_entity
        @new_entity = new_entity
        @project = original_entity.project
      end

      def execute
        rewrite_description
        rewrite_award_emoji(original_entity, new_entity)
        rewrite_notes
      end

      private

      def rewrite_description
        new_entity.update(description: rewrite_content(original_entity.description))
      end

      def rewrite_notes
        original_entity.notes_with_associations.find_each do |note|
          new_note = note.dup
          new_params = {
            project: new_entity.project, noteable: new_entity,
            note: rewrite_content(new_note.note),
            created_at: note.created_at,
            updated_at: note.updated_at
          }

          if note.system_note_metadata
            new_params[:system_note_metadata] = note.system_note_metadata.dup
          end

          new_note.update(new_params)

          rewrite_award_emoji(note, new_note)
        end
      end

      def rewrite_content(content)
        return unless content

        rewriters = [Gitlab::Gfm::ReferenceRewriter, Gitlab::Gfm::UploadsRewriter]

        rewriters.inject(content) do |text, klass|
          rewriter = klass.new(text, old_project, current_user)
          rewriter.rewrite(new_parent)
        end
      end

      def rewrite_award_emoji(old_awardable, new_awardable)
        old_awardable.award_emoji.each do |award|
          new_award = award.dup
          new_award.awardable = new_awardable
          new_award.save
        end
      end
    end
  end
end