summaryrefslogtreecommitdiff
path: root/lib/banzai/filter/kroki_filter.rb
diff options
context:
space:
mode:
Diffstat (limited to 'lib/banzai/filter/kroki_filter.rb')
-rw-r--r--lib/banzai/filter/kroki_filter.rb16
1 files changed, 12 insertions, 4 deletions
diff --git a/lib/banzai/filter/kroki_filter.rb b/lib/banzai/filter/kroki_filter.rb
index 9aa2afce5a8..845c7f2bc0a 100644
--- a/lib/banzai/filter/kroki_filter.rb
+++ b/lib/banzai/filter/kroki_filter.rb
@@ -25,11 +25,19 @@ module Banzai
diagram_type = node.parent['lang']
diagram_src = node.content
image_src = create_image_src(diagram_type, diagram_format, diagram_src)
- lazy_load = diagram_src.length > MAX_CHARACTER_LIMIT
- other_attrs = lazy_load ? "hidden" : ""
+ img_tag = Nokogiri::HTML::DocumentFragment.parse(%(<img src="#{image_src}" />))
+ img_tag = img_tag.children.first
- img_tag = Nokogiri::HTML::DocumentFragment.parse(%(<img class="js-render-kroki" src="#{image_src}" #{other_attrs} />))
- node.parent.replace(img_tag)
+ unless img_tag.nil?
+ lazy_load = diagram_src.length > MAX_CHARACTER_LIMIT
+ img_tag.set_attribute('hidden', '') if lazy_load
+ img_tag.set_attribute('class', 'js-render-kroki')
+
+ img_tag.set_attribute('data-diagram', node.parent['lang'])
+ img_tag.set_attribute('data-diagram-src', "data:text/plain;base64,#{Base64.strict_encode64(node.content)}")
+
+ node.parent.replace(img_tag)
+ end
end
doc