Autolink the raw text and “transfer” the found links to the highlighted text

lib/gitlab/highlight.rb

@@ -22,7 +22,7 @@ module Gitlab
     def highlight(text, continue: true, plain: false)
       highlighted_text = highlight_text(text, continue: continue, plain: plain)
-      autolink_strings(highlighted_text)
+      autolink_strings(text, highlighted_text)
     end
     def lexer
@@ -61,9 +61,42 @@ module Gitlab
       highlight_plain(text)
     end
-    def autolink_strings(highlighted_text)
+    def autolink_strings(text, highlighted_text)
+      raw_lines = text.lines
+
       # TODO: Don't run pre-processing pipeline, because this may break the highlighting
-      Banzai.render(highlighted_text, pipeline: :autolink, autolink_emails: true).html_safe
+      linked_text = Banzai.render(
+        ERB::Util.html_escape(text),
+        pipeline: :autolink,
+        autolink_emails: true
+      ).html_safe
+
+      linked_lines = linked_text.lines
+
+      highlighted_lines = highlighted_text.lines
+
+      highlighted_lines.map!.with_index do |rich_line, i|
+        matches = []
+        linked_lines[i].scan(/(?<start><a[^>]+>)(?<content>[^<]+)(?<end><\/a>)/) { matches << Regexp.last_match }
+        next rich_line if matches.empty?
+
+        raw_line = raw_lines[i]
+        marked_line = rich_line.html_safe
+
+        matches.each do |match|
+          marker = StringRegexMarker.new(raw_line, marked_line)
+
+          regex = /#{Regexp.escape(match[:content])}/
+
+          marked_line = marker.mark(regex) do |text, left:, right:|
+            "#{match[:start]}#{text}#{match[:end]}"
+          end
+        end
+
+        marked_line
+      end
+
+      highlighted_lines.join.html_safe
     end
   end
 end