debian-mirror-gitlab/lib/banzai/object_renderer.rb

94 lines
3.2 KiB
Ruby
Raw Normal View History

2016-08-24 12:49:21 +05:30
module Banzai
2016-11-03 12:29:30 +05:30
# Class for rendering multiple objects (e.g. Note instances) in a single pass,
# using +render_field+ to benefit from caching in the database. Rendering and
# redaction are both performed.
2016-08-24 12:49:21 +05:30
#
2016-11-03 12:29:30 +05:30
# The unredacted HTML is generated according to the usual +render_field+
# policy, so specify the pipeline and any other context options on the model.
#
# The *redacted* (i.e., suitable for use) HTML is placed in an attribute
# named "redacted_<foo>", where <foo> is the name of the cache field for the
# chosen attribute.
#
# As an example, rendering the attribute `note` would place the unredacted
# HTML into `note_html` and the redacted HTML into `redacted_note_html`.
2016-08-24 12:49:21 +05:30
class ObjectRenderer
attr_reader :project, :user
2016-11-03 12:29:30 +05:30
# project - A Project to use for redacting Markdown.
2016-08-24 12:49:21 +05:30
# user - The user viewing the Markdown/HTML documents, if any.
2018-03-17 18:26:18 +05:30
# redaction_context - A Hash containing extra attributes to use during redaction
2016-11-03 12:29:30 +05:30
def initialize(project, user = nil, redaction_context = {})
2016-08-24 12:49:21 +05:30
@project = project
@user = user
2018-03-17 18:26:18 +05:30
@redaction_context = base_context.merge(redaction_context)
2016-08-24 12:49:21 +05:30
end
# Renders and redacts an Array of objects.
#
2016-11-03 12:29:30 +05:30
# objects - The objects to render.
2016-08-24 12:49:21 +05:30
# attribute - The attribute containing the raw Markdown to render.
#
# Returns the same input objects.
def render(objects, attribute)
2017-08-17 22:00:37 +05:30
documents = render_documents(objects, attribute)
documents = post_process_documents(documents, objects, attribute)
2016-08-24 12:49:21 +05:30
redacted = redact_documents(documents)
objects.each_with_index do |object, index|
redacted_data = redacted[index]
2018-03-17 18:26:18 +05:30
object.__send__("redacted_#{attribute}_html=", redacted_data[:document].to_html(save_options).html_safe) # rubocop:disable GitlabSecurity/PublicSend
object.user_visible_reference_count = redacted_data[:visible_reference_count] if object.respond_to?(:user_visible_reference_count)
2016-08-24 12:49:21 +05:30
end
end
2017-08-17 22:00:37 +05:30
private
def render_documents(objects, attribute)
pipeline = HTML::Pipeline.new([])
objects.map do |object|
pipeline.to_document(Banzai.render_field(object, attribute))
end
end
def post_process_documents(documents, objects, attribute)
# Called here to populate cache, refer to IssuableExtractor docs
IssuableExtractor.new(project, user).extract(documents)
documents.zip(objects).map do |document, object|
context = context_for(object, attribute)
Banzai::Pipeline[:post_process].to_document(document, context)
end
2016-08-24 12:49:21 +05:30
end
# Redacts the list of documents.
#
# Returns an Array containing the redacted documents.
def redact_documents(documents)
redactor = Redactor.new(project, user)
redactor.redact(documents)
end
# Returns a Banzai context for the given object and attribute.
def context_for(object, attribute)
2018-03-17 18:26:18 +05:30
@redaction_context.merge(object.banzai_render_context(attribute))
2016-08-24 12:49:21 +05:30
end
def base_context
2018-03-17 18:26:18 +05:30
{
2017-08-17 22:00:37 +05:30
current_user: user,
project: project,
skip_redaction: true
2018-03-17 18:26:18 +05:30
}
end
def save_options
return {} unless @redaction_context[:xhtml]
{ save_with: Nokogiri::XML::Node::SaveOptions::AS_XHTML }
2016-08-24 12:49:21 +05:30
end
end
end