Class: Gitlab::Gfm::UploadsRewriter

Inherits:
Object
  • Object
show all
Includes:
Utils::StrongMemoize
Defined in:
lib/gitlab/gfm/uploads_rewriter.rb

Overview

Class that rewrites markdown links for uploads

Using a pattern defined in ‘FileUploader` it copies files to a new project and rewrites all links to uploads in a given text.

Instance Method Summary collapse

Constructor Details

#initialize(text, _text_html, source_project, _current_user) ⇒ UploadsRewriter

Returns a new instance of UploadsRewriter.



17
18
19
20
21
 
# File 'lib/gitlab/gfm/uploads_rewriter.rb', line 17

def initialize(text, _text_html, source_project, _current_user)
  @text = text
  @source_project = source_project
  @pattern = FileUploader::MARKDOWN_PATTERN
end

Instance Method Details

#needs_rewrite? ⇒ Boolean

Returns:

  • (Boolean) 


36
37
38
39
40
 
# File 'lib/gitlab/gfm/uploads_rewriter.rb', line 36

def needs_rewrite?
  strong_memoize(:needs_rewrite) do
    @pattern.match?(@text)
  end
end

#rewrite(target_parent) ⇒ Object 



23
24
25
26
27
28
29
30
31
32
33
34
 
# File 'lib/gitlab/gfm/uploads_rewriter.rb', line 23

def rewrite(target_parent)
  return @text unless needs_rewrite?

  @target_parent = target_parent

  rewritten_text = Gitlab::StringRegexMarker.new(@text).mark(@pattern) do |markdown, left:, right:, mode:|
    transform_markdown(markdown)
  end

  # MarkdownContentRewriterService relies on the text being changed _in place_.
  @text.gsub!(@text, rewritten_text)
end