296 lines
10 KiB
Ruby
296 lines
10 KiB
Ruby
# frozen_string_literal: true
|
|
|
|
module Banzai
|
|
module Filter
|
|
module References
|
|
# Issues, merge requests, Snippets, Commits and Commit Ranges share
|
|
# similar functionality in reference filtering.
|
|
class AbstractReferenceFilter < ReferenceFilter
|
|
include CrossProjectReference
|
|
|
|
def initialize(doc, context = nil, result = nil)
|
|
super
|
|
|
|
@reference_cache = ReferenceCache.new(self, context)
|
|
end
|
|
|
|
# REFERENCE_PLACEHOLDER is used for re-escaping HTML text except found
|
|
# reference (which we replace with placeholder during re-scaping). The
|
|
# random number helps ensure it's pretty close to unique. Since it's a
|
|
# transitory value (it never gets saved) we can initialize once, and it
|
|
# doesn't matter if it changes on a restart.
|
|
REFERENCE_PLACEHOLDER = "_reference_#{SecureRandom.hex(16)}_"
|
|
REFERENCE_PLACEHOLDER_PATTERN = %r{#{REFERENCE_PLACEHOLDER}(\d+)}.freeze
|
|
|
|
# Public: Find references in text (like `!123` for merge requests)
|
|
#
|
|
# references_in(text) do |match, id, project_ref, matches|
|
|
# object = find_object(project_ref, id)
|
|
# "<a href=...>#{object.to_reference}</a>"
|
|
# end
|
|
#
|
|
# text - String text to search.
|
|
#
|
|
# Yields the String match, the Integer referenced object ID, an optional String
|
|
# of the external project reference, and all of the matchdata.
|
|
#
|
|
# Returns a String replaced with the return of the block.
|
|
def references_in(text, pattern = object_class.reference_pattern)
|
|
text.gsub(pattern) do |match|
|
|
if ident = identifier($~)
|
|
yield match, ident, $~[:project], $~[:namespace], $~
|
|
else
|
|
match
|
|
end
|
|
end
|
|
end
|
|
|
|
def identifier(match_data)
|
|
symbol = symbol_from_match(match_data)
|
|
|
|
parse_symbol(symbol, match_data) if object_class.reference_valid?(symbol)
|
|
end
|
|
|
|
def symbol_from_match(match)
|
|
key = object_sym
|
|
match[key] if match.names.include?(key.to_s)
|
|
end
|
|
|
|
# Transform a symbol extracted from the text to a meaningful value
|
|
# In most cases these will be integers, so we call #to_i by default
|
|
#
|
|
# This method has the contract that if a string `ref` refers to a
|
|
# record `record`, then `parse_symbol(ref) == record_identifier(record)`.
|
|
def parse_symbol(symbol, match_data)
|
|
symbol.to_i
|
|
end
|
|
|
|
# We assume that most classes are identifying records by ID.
|
|
#
|
|
# This method has the contract that if a string `ref` refers to a
|
|
# record `record`, then `class.parse_symbol(ref) == record_identifier(record)`.
|
|
def record_identifier(record)
|
|
record.id
|
|
end
|
|
|
|
# Implement in child class
|
|
# Example: project.merge_requests.find
|
|
def find_object(parent_object, id)
|
|
raise NotImplementedError, "#{self.class} must implement method: #{__callee__}"
|
|
end
|
|
|
|
# Override if the link reference pattern produces a different ID (global
|
|
# ID vs internal ID, for instance) to the regular reference pattern.
|
|
def find_object_from_link(parent_object, id)
|
|
find_object(parent_object, id)
|
|
end
|
|
|
|
# Implement in child class
|
|
# Example: project_merge_request_url
|
|
def url_for_object(object, parent_object)
|
|
raise NotImplementedError, "#{self.class} must implement method: #{__callee__}"
|
|
end
|
|
|
|
def find_object_cached(parent_object, id)
|
|
cached_call(:banzai_find_object, id, path: [object_class, parent_object.id]) do
|
|
find_object(parent_object, id)
|
|
end
|
|
end
|
|
|
|
def find_object_from_link_cached(parent_object, id)
|
|
cached_call(:banzai_find_object_from_link, id, path: [object_class, parent_object.id]) do
|
|
find_object_from_link(parent_object, id)
|
|
end
|
|
end
|
|
|
|
def from_ref_cached(ref)
|
|
cached_call("banzai_#{parent_type}_refs".to_sym, ref) do
|
|
parent_from_ref(ref)
|
|
end
|
|
end
|
|
|
|
def url_for_object_cached(object, parent_object)
|
|
cached_call(:banzai_url_for_object, object, path: [object_class, parent_object.id]) do
|
|
url_for_object(object, parent_object)
|
|
end
|
|
end
|
|
|
|
def call
|
|
return doc unless project || group || user
|
|
|
|
reference_cache.load_reference_cache(nodes) if respond_to?(:parent_records)
|
|
|
|
ref_pattern = object_reference_pattern
|
|
link_pattern = object_class.link_reference_pattern
|
|
|
|
# Compile often used regexps only once outside of the loop
|
|
ref_pattern_anchor = /\A#{ref_pattern}\z/
|
|
link_pattern_start = /\A#{link_pattern}/
|
|
link_pattern_anchor = /\A#{link_pattern}\z/
|
|
|
|
nodes.each_with_index do |node, index|
|
|
if text_node?(node) && ref_pattern
|
|
replace_text_when_pattern_matches(node, index, ref_pattern) do |content|
|
|
object_link_filter(content, ref_pattern)
|
|
end
|
|
|
|
elsif element_node?(node)
|
|
yield_valid_link(node) do |link, inner_html|
|
|
if ref_pattern && link =~ ref_pattern_anchor
|
|
replace_link_node_with_href(node, index, link) do
|
|
object_link_filter(link, ref_pattern, link_content: inner_html)
|
|
end
|
|
|
|
next
|
|
end
|
|
|
|
next unless link_pattern
|
|
|
|
if link == inner_html && inner_html =~ link_pattern_start
|
|
replace_link_node_with_text(node, index) do
|
|
object_link_filter(inner_html, link_pattern, link_reference: true)
|
|
end
|
|
|
|
next
|
|
end
|
|
|
|
if link =~ link_pattern_anchor
|
|
replace_link_node_with_href(node, index, link) do
|
|
object_link_filter(link, link_pattern, link_content: inner_html, link_reference: true)
|
|
end
|
|
|
|
next
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
doc
|
|
end
|
|
|
|
# Replace references (like `!123` for merge requests) in text with links
|
|
# to the referenced object's details page.
|
|
#
|
|
# text - String text to replace references in.
|
|
# pattern - Reference pattern to match against.
|
|
# link_content - Original content of the link being replaced.
|
|
# link_reference - True if this was using the link reference pattern,
|
|
# false otherwise.
|
|
#
|
|
# Returns a String with references replaced with links. All links
|
|
# have `gfm` and `gfm-OBJECT_NAME` class names attached for styling.
|
|
def object_link_filter(text, pattern, link_content: nil, link_reference: false)
|
|
references_in(text, pattern) do |match, id, project_ref, namespace_ref, matches|
|
|
parent_path = if parent_type == :group
|
|
reference_cache.full_group_path(namespace_ref)
|
|
else
|
|
reference_cache.full_project_path(namespace_ref, project_ref)
|
|
end
|
|
|
|
parent = from_ref_cached(parent_path)
|
|
|
|
if parent
|
|
object =
|
|
if link_reference
|
|
find_object_from_link_cached(parent, id)
|
|
else
|
|
find_object_cached(parent, id)
|
|
end
|
|
end
|
|
|
|
if object
|
|
title = object_link_title(object, matches)
|
|
klass = reference_class(object_sym)
|
|
|
|
data_attributes = data_attributes_for(link_content || match, parent, object,
|
|
link_content: !!link_content,
|
|
link_reference: link_reference)
|
|
data = data_attribute(data_attributes)
|
|
|
|
url =
|
|
if matches.names.include?("url") && matches[:url]
|
|
matches[:url]
|
|
else
|
|
url_for_object_cached(object, parent)
|
|
end
|
|
|
|
content = link_content || object_link_text(object, matches)
|
|
|
|
link = %(<a href="#{url}" #{data}
|
|
title="#{escape_once(title)}"
|
|
class="#{klass}">#{content}</a>)
|
|
|
|
wrap_link(link, object)
|
|
else
|
|
match
|
|
end
|
|
end
|
|
end
|
|
|
|
def wrap_link(link, object)
|
|
link
|
|
end
|
|
|
|
def data_attributes_for(text, parent, object, link_content: false, link_reference: false)
|
|
object_parent_type = parent.is_a?(Group) ? :group : :project
|
|
|
|
{
|
|
original: escape_html_entities(text),
|
|
link: link_content,
|
|
link_reference: link_reference,
|
|
object_parent_type => parent.id,
|
|
object_sym => object.id
|
|
}
|
|
end
|
|
|
|
def object_link_text_extras(object, matches)
|
|
extras = []
|
|
|
|
if matches.names.include?("anchor") && matches[:anchor] && matches[:anchor] =~ /\A\#note_(\d+)\z/
|
|
extras << "comment #{Regexp.last_match(1)}"
|
|
end
|
|
|
|
extension = matches[:extension] if matches.names.include?("extension")
|
|
|
|
extras << extension if extension
|
|
|
|
extras
|
|
end
|
|
|
|
def object_link_title(object, matches)
|
|
object.title
|
|
end
|
|
|
|
def object_link_text(object, matches)
|
|
parent = project || group || user
|
|
text = object.reference_link_text(parent)
|
|
|
|
extras = object_link_text_extras(object, matches)
|
|
text += " (#{extras.join(", ")})" if extras.any?
|
|
|
|
text
|
|
end
|
|
|
|
def parent_type
|
|
:project
|
|
end
|
|
|
|
def parent
|
|
parent_type == :project ? project : group
|
|
end
|
|
|
|
private
|
|
|
|
attr_accessor :reference_cache
|
|
|
|
def escape_with_placeholders(text, placeholder_data)
|
|
escaped = escape_html_entities(text)
|
|
|
|
escaped.gsub(REFERENCE_PLACEHOLDER_PATTERN) do |match|
|
|
placeholder_data[Regexp.last_match(1).to_i]
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|