asciidoctor-kroki 0.3.0 → 0.4.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: c68a706bac6918c95daef5ada0de0d6a8471cbd69cc83ced875119ca69b90f58
-  data.tar.gz: c2683ddf3c70e743faff32a73442ef651fd17145191148e4a5930ac7d16852f9
+  metadata.gz: dfa0c461bdb74571b618747d791e059125dc09a5e67dd6886861101fe972237f
+  data.tar.gz: d882ff8e33c787b3a1578eb7d11b590411e18f594eaec188d7f06ffc182a3b11
 SHA512:
-  metadata.gz: 45ea1d0252ebfab9fc9dfba56b13a3497ddf1e5ca58cb05777c941829ec2316f0f194a87bef490483c800d43cd7cf0e008804cb4f92ad4955cc864203ad26dfb
-  data.tar.gz: 3fec5d036175f848b0fa26df6c47439ee2f61a776393c097ecb0e79e0f407939e99df233bb1461360feaf4aae4224377aadff7bbc9d93ffa6fad6af62f7fc40e
+  metadata.gz: f72b7293f3fb8c6d37b456c39d4e2691c0b65e1cb908d5f3f29d23466f0c51135b9977dd0977a2ccac7adb667b4fc5071a4a5fc59bed6a3304154bddc8efda3d
+  data.tar.gz: 1e93fbf9df1cab9282f7743f76331ea6caddfdce150fff36fb7aa7222d561d089c31101ec0f16a939f5aa2c0c7dd5a09206059700974b76ae6fadcfc0150acd3

data/.rubocop.yml

@@ -1,8 +1,14 @@
 Style/Encoding:
   Enabled: false
 
+Layout/EndOfLine:
+  EnforcedStyle: lf
+
 Metrics/LineLength:
-  Max: 160
+  Max: 180
+
+Metrics/ClassLength:
+  Max: 150
 
 Metrics/MethodLength:
   Max: 50
@@ -15,3 +21,6 @@ Metrics/PerceivedComplexity:
 
 Metrics/AbcSize:
   Max: 30
+
+Metrics/ParameterLists:
+  Max: 7

data/Gemfile.lock

@@ -1,13 +1,13 @@
 PATH
   remote: .
   specs:
-    asciidoctor-kroki (0.3.0)
+    asciidoctor-kroki (0.4.0)
       asciidoctor (~> 2.0)
 
 GEM
   remote: https://rubygems.org/
   specs:
-    asciidoctor (2.0.10)
+    asciidoctor (2.0.11)
     ast (2.4.1)
     diff-lcs (1.3)
     jaro_winkler (1.5.4)

data/asciidoctor-kroki.gemspec

@@ -2,7 +2,7 @@
 
 Gem::Specification.new do |s|
   s.name = 'asciidoctor-kroki'
-  s.version = '0.3.0'
+  s.version = '0.4.0'
   s.summary = 'Asciidoctor extension to convert diagrams to images using Kroki'
   s.description = 'An extension for Asciidoctor to convert diagrams to images using https://kroki.io'
 

data/lib/asciidoctor/extensions/asciidoctor_kroki/extension.rb

@@ -15,11 +15,24 @@ module AsciidoctorExtensions
     on_context :listing, :literal
     name_positional_attributes 'target', 'format'
 
+    # @param name [String] name of the block macro (optional)
+    # @param config [Hash] a config hash (optional)
+    # @param logger [Logger] a logger used to log warning and errors (optional)
+    #
+    def initialize(name = nil, config = {}, logger: ::Asciidoctor::LoggerManager.logger)
+      super(name, config)
+      @logger = logger
+    end
+
     def process(parent, reader, attrs)
       diagram_type = @name
       diagram_text = reader.string
-      KrokiProcessor.process(self, parent, attrs, diagram_type, diagram_text)
+      KrokiProcessor.process(self, parent, attrs, diagram_type, diagram_text, @logger)
     end
+
+    protected
+
+    attr_reader :logger
   end
 
   # A block macro extension that converts a diagram into an image.
@@ -29,11 +42,48 @@ module AsciidoctorExtensions
 
     name_positional_attributes 'format'
 
+    # @param name [String] name of the block macro (optional)
+    # @param config [Hash] a config hash (optional)
+    # @param logger [Logger] a logger used to log warning and errors (optional)
+    #
+    def initialize(name = nil, config = {}, logger: ::Asciidoctor::LoggerManager.logger)
+      super(name, config)
+      @logger = logger
+    end
+
     def process(parent, target, attrs)
       diagram_type = @name
       target = parent.apply_subs(target, [:attributes])
-      diagram_text = read(target)
-      KrokiProcessor.process(self, parent, attrs, diagram_type, diagram_text)
+
+      unless read_allowed?(target)
+        link = create_inline(parent, :anchor, target, type: :link, target: target)
+        return create_block(parent, :paragraph, link.convert, {}, content_model: :raw)
+      end
+
+      unless (path = resolve_target_path(target))
+        logger.error "#{diagram_type} block macro not found: #{target}."
+        create_block(parent, 'paragraph', unresolved_block_macro_message(diagram_type, target), {})
+      end
+
+      begin
+        diagram_text = read(path)
+      rescue => e # rubocop:disable RescueStandardError
+        logger.error "Failed to read #{diagram_type} file: #{path}. #{e}."
+        return create_block(parent, 'paragraph', unresolved_block_macro_message(diagram_type, path), {})
+      end
+      KrokiProcessor.process(self, parent, attrs, diagram_type, diagram_text, @logger)
+    end
+
+    protected
+
+    attr_reader :logger
+
+    def resolve_target_path(target)
+      target
+    end
+
+    def read_allowed?(_target)
+      true
     end
 
     def read(target)
@@ -44,6 +94,10 @@ module AsciidoctorExtensions
         File.open(target, &:read)
       end
     end
+
+    def unresolved_block_macro_message(name, target)
+      "Unresolved block macro - #{name}::#{target}[]"
+    end
   end
 
   # Kroki API
@@ -70,7 +124,7 @@ module AsciidoctorExtensions
       umlet
       vega
       vegalite
-      wavedrow
+      wavedrom
     ].freeze
   end
 
@@ -80,9 +134,9 @@ module AsciidoctorExtensions
     TEXT_FORMATS = %w[txt atxt utxt].freeze
 
     class << self
-      def process(processor, parent, attrs, diagram_type, diagram_text)
+      def process(processor, parent, attrs, diagram_type, diagram_text, logger)
         doc = parent.document
-        diagram_text = prepend_plantuml_config(diagram_text, diagram_type, doc)
+        diagram_text = prepend_plantuml_config(diagram_text, diagram_type, doc, logger)
         # If "subs" attribute is specified, substitute accordingly.
         # Be careful not to specify "specialcharacters" or your diagram code won't be valid anymore!
         if (subs = attrs['subs'])
@@ -96,7 +150,7 @@ module AsciidoctorExtensions
         attrs['role'] = get_role(format, role)
         attrs['format'] = format
         kroki_diagram = KrokiDiagram.new(diagram_type, format, diagram_text)
-        kroki_client = KrokiClient.new(server_url(doc), http_method(doc), KrokiHttpClient)
+        kroki_client = KrokiClient.new(server_url(doc), http_method(doc), KrokiHttpClient, logger, max_uri_length(doc))
         if TEXT_FORMATS.include?(format)
           text_content = kroki_client.text_content(kroki_diagram)
           block = processor.create_block(parent, 'literal', text_content, attrs)
@@ -112,12 +166,17 @@ module AsciidoctorExtensions
 
       private
 
-      def prepend_plantuml_config(diagram_text, diagram_type, doc)
-        if diagram_type == :plantuml && doc.attr?('kroki-plantuml-include')
-          # TODO: this behaves different than the JS version
-          # The file should be added by !include #{plantuml_include}" once we have a preprocessor for ruby
-          config = File.read(doc.attr('kroki-plantuml-include'))
-          diagram_text = config + '\n' + diagram_text
+      def prepend_plantuml_config(diagram_text, diagram_type, doc, logger)
+        if diagram_type == :plantuml && doc.safe < ::Asciidoctor::SafeMode::SECURE && doc.attr?('kroki-plantuml-include')
+          # REMIND: this behaves different than the JS version
+          # Once we have a preprocessor for Ruby, the value should be added in the diagram source as "!include #{plantuml_include}"
+          plantuml_include_path = doc.normalize_system_path(doc.attr('kroki-plantuml-include'))
+          if ::File.readable? plantuml_include_path
+            config = File.read(plantuml_include_path)
+            diagram_text = config + "\n" + diagram_text
+          else
+            logger.warn "Unable to read plantuml-include. File not found or not readable: #{plantuml_include_path}."
+          end
         end
         diagram_text
       end
@@ -157,7 +216,7 @@ module AsciidoctorExtensions
       end
 
       def create_image_src(doc, kroki_diagram, kroki_client)
-        if doc.attr('kroki-fetch-diagram')
+        if doc.attr('kroki-fetch-diagram') && doc.safe < ::Asciidoctor::SafeMode::SECURE
           kroki_diagram.save(output_dir_path(doc), kroki_client)
         else
           kroki_diagram.get_diagram_uri(server_url(doc))
@@ -172,20 +231,20 @@ module AsciidoctorExtensions
         doc.attr('kroki-http-method', 'adaptive').downcase
       end
 
+      def max_uri_length(doc)
+        doc.attr('kroki-max-uri-length', '4000').to_i
+      end
+
       def output_dir_path(doc)
-        images_output_dir = doc.attr('imagesoutdir')
-        out_dir = doc.attr('outdir')
-        to_dir = doc.attr('to_dir')
-        base_dir = doc.base_dir
         images_dir = doc.attr('imagesdir', '')
-        if images_output_dir
+        if (images_output_dir = doc.attr('imagesoutdir'))
           images_output_dir
-        elsif out_dir
+        elsif (out_dir = doc.attr('outdir'))
           File.join(out_dir, images_dir)
-        elsif to_dir
+        elsif (to_dir = doc.attr('to_dir'))
           File.join(to_dir, images_dir)
         else
-          File.join(base_dir, images_dir)
+          File.join(doc.base_dir, images_dir)
         end
       end
     end
@@ -259,18 +318,19 @@ module AsciidoctorExtensions
   class KrokiClient
     attr_reader :server_url
     attr_reader :method
+    attr_reader :max_uri_length
 
     SUPPORTED_HTTP_METHODS = %w[get post adaptive].freeze
 
-    def initialize(server_url, http_method, http_client)
+    def initialize(server_url, http_method, http_client, logger = ::Asciidoctor::LoggerManager.logger, max_uri_length = 4000)
       @server_url = server_url
-      @max_uri_length = 4096
+      @max_uri_length = max_uri_length
       @http_client = http_client
       method = (http_method || 'adaptive').downcase
       if SUPPORTED_HTTP_METHODS.include?(method)
         @method = method
       else
-        puts "Invalid value '#{method}' for kroki-http-method attribute. The value must be either: 'get', 'post' or 'adaptive'. Proceeding using: 'adaptive'."
+        logger.warn "Invalid value '#{method}' for kroki-http-method attribute. The value must be either: 'get', 'post' or 'adaptive'. Proceeding using: 'adaptive'."
         @method = 'adaptive'
       end
     end
@@ -286,7 +346,7 @@ module AsciidoctorExtensions
       if @method == 'adaptive' || @method == 'get'
         uri = kroki_diagram.get_diagram_uri(server_url)
         if uri.length > @max_uri_length
-          # The request URI is longer than 4096.
+          # The request URI is longer than the max URI length.
           if @method == 'get'
             # The request might be rejected by the server with a 414 Request-URI Too Large.
             # Consider using the attribute kroki-http-method with the value 'adaptive'.

data/spec/asciidoctor_kroki_block_macro_spec.rb

@@ -0,0 +1,128 @@
+# frozen_string_literal: true
+
+require 'rspec_helper'
+require 'asciidoctor'
+require_relative '../lib/asciidoctor/extensions/asciidoctor_kroki'
+require_relative '../lib/asciidoctor/extensions/asciidoctor_kroki/extension'
+
+describe ::AsciidoctorExtensions::KrokiBlockMacroProcessor do
+  context 'convert to html5' do
+    it 'should catch exception if target is not readable' do
+      input = <<~'ADOC'
+        plantuml::spec/fixtures/missing.puml[svg,role=sequence]
+      ADOC
+      output = Asciidoctor.convert(input, standalone: false)
+      (expect output).to eql %(<div class="paragraph">
+<p>Unresolved block macro - plantuml::spec/fixtures/missing.puml[]</p>
+</div>)
+    end
+  end
+  context 'using a custom block macro' do
+    it 'should disallow read' do
+      # noinspection RubyClassModuleNamingConvention
+      class DisallowReadKrokiBlockMacroProcessor < ::AsciidoctorExtensions::KrokiBlockMacroProcessor
+        def read_allowed?(_target)
+          false
+        end
+      end
+      registry = Asciidoctor::Extensions.create do
+        block_macro DisallowReadKrokiBlockMacroProcessor, 'plantuml'
+      end
+      input = <<~'ADOC'
+        plantuml::spec/fixtures/alice.puml[svg,role=sequence]
+      ADOC
+      output = Asciidoctor.convert(input, standalone: false, extension_registry: registry)
+      (expect output).to eql %(<div class="paragraph">
+<p><a href="spec/fixtures/alice.puml">spec/fixtures/alice.puml</a></p>
+</div>)
+    end
+    it 'should allow read if target is not a URI' do
+      # noinspection RubyClassModuleNamingConvention
+      class DisallowUriReadKrokiBlockMacroProcessor < ::AsciidoctorExtensions::KrokiBlockMacroProcessor
+        def read_allowed?(target)
+          return false if ::Asciidoctor::Helpers.uriish?(target)
+
+          true
+        end
+      end
+      registry = Asciidoctor::Extensions.create do
+        block_macro DisallowUriReadKrokiBlockMacroProcessor, 'plantuml'
+      end
+      input = <<~'ADOC'
+        plantuml::https://domain.org/alice.puml[svg,role=sequence]
+
+        plantuml::file://path/to/alice.puml[svg,role=sequence]
+
+        plantuml::spec/fixtures/alice.puml[svg,role=sequence]
+      ADOC
+      output = Asciidoctor.convert(input, standalone: false, extension_registry: registry)
+      (expect output).to eql %(<div class="paragraph">
+<p><a href="https://domain.org/alice.puml">https://domain.org/alice.puml</a></p>
+</div>
+<div class="paragraph">
+<p><a href="file://path/to/alice.puml">file://path/to/alice.puml</a></p>
+</div>
+<div class="imageblock sequence kroki-format-svg kroki">
+<div class="content">
+<img src="https://kroki.io/plantuml/svg/eNpLzMlMTlXQtVNIyk-yUshIzcnJ5wIAQ-AGVQ==" alt="Diagram">
+</div>
+</div>)
+    end
+    it 'should override the resolve target method' do
+      # noinspection RubyClassModuleNamingConvention
+      class FixtureResolveTargetKrokiBlockMacroProcessor < ::AsciidoctorExtensions::KrokiBlockMacroProcessor
+        def resolve_target_path(target)
+          "spec/fixtures/#{target}"
+        end
+      end
+      registry = Asciidoctor::Extensions.create do
+        block_macro FixtureResolveTargetKrokiBlockMacroProcessor, 'plantuml'
+      end
+      input = <<~'ADOC'
+        plantuml::alice.puml[svg,role=sequence]
+      ADOC
+      output = Asciidoctor.convert(input, standalone: false, extension_registry: registry)
+      (expect output).to eql %(<div class="imageblock sequence kroki-format-svg kroki">
+<div class="content">
+<img src="https://kroki.io/plantuml/svg/eNpLzMlMTlXQtVNIyk-yUshIzcnJ5wIAQ-AGVQ==" alt="Diagram">
+</div>
+</div>)
+    end
+    it 'should display unresolved block macro message when the traget cannot be resolved' do
+      # noinspection RubyClassModuleNamingConvention
+      class UnresolvedTargetKrokiBlockMacroProcessor < ::AsciidoctorExtensions::KrokiBlockMacroProcessor
+        def resolve_target_path(_target)
+          nil
+        end
+      end
+      registry = Asciidoctor::Extensions.create do
+        block_macro UnresolvedTargetKrokiBlockMacroProcessor, 'plantuml'
+      end
+      input = <<~'ADOC'
+        plantuml::alice.puml[svg,role=sequence]
+      ADOC
+      output = Asciidoctor.convert(input, standalone: false, extension_registry: registry)
+      (expect output).to eql %(<div class="paragraph">
+<p>Unresolved block macro - plantuml::[]</p>
+</div>)
+    end
+    it 'should override the unresolved block macro message' do
+      # noinspection RubyClassModuleNamingConvention
+      class CustomUnresolvedTargetMessageKrokiBlockMacroProcessor < ::AsciidoctorExtensions::KrokiBlockMacroProcessor
+        def unresolved_block_macro_message(name, target)
+          "*[ERROR: #{name}::#{target}[] - unresolved block macro]*"
+        end
+      end
+      registry = Asciidoctor::Extensions.create do
+        block_macro CustomUnresolvedTargetMessageKrokiBlockMacroProcessor, 'plantuml'
+      end
+      input = <<~'ADOC'
+        plantuml::spec/fixtures/missing.puml[svg,role=sequence]
+      ADOC
+      output = Asciidoctor.convert(input, standalone: false, extension_registry: registry)
+      (expect output).to eql %(<div class="paragraph">
+<p><strong>[ERROR: plantuml::spec/fixtures/missing.puml[] - unresolved block macro]</strong></p>
+</div>)
+    end
+  end
+end

data/spec/asciidoctor_kroki_client_spec.rb

@@ -20,4 +20,72 @@ describe ::AsciidoctorExtensions::KrokiClient do
     kroki_client = ::AsciidoctorExtensions::KrokiClient.new('http://localhost:8000', 'get', kroki_http_client)
     expect(kroki_client.method).to eq('get')
   end
+  it 'should use 4000 as the default max URI length' do
+    kroki_http_client = ::AsciidoctorExtensions::KrokiHttpClient
+    kroki_client = ::AsciidoctorExtensions::KrokiClient.new('http://localhost:8000', 'get', kroki_http_client)
+    expect(kroki_client.max_uri_length).to eq(4000)
+  end
+  it 'should use a custom value as max URI length' do
+    kroki_http_client = ::AsciidoctorExtensions::KrokiHttpClient
+    kroki_client = ::AsciidoctorExtensions::KrokiClient.new('http://localhost:8000', 'get', kroki_http_client, nil, 8000)
+    expect(kroki_client.max_uri_length).to eq(8000)
+  end
+  it 'should get an image with POST request if the URI length is greater than the value configured' do
+    kroki_http_client = Class.new do
+      class << self
+        def get(uri, _)
+          "GET #{uri}"
+        end
+
+        def post(uri, data, _)
+          "POST #{uri} - #{data}"
+        end
+      end
+    end
+    kroki_diagram = Class.new do
+      attr_reader :type, :text, :format
+
+      def initialize(type, format, text)
+        @text = text
+        @type = type
+        @format = format
+      end
+
+      def get_diagram_uri(_)
+        'diagram-uri'
+      end
+    end.new('type', 'format', 'text')
+    kroki_client = ::AsciidoctorExtensions::KrokiClient.new('http://localhost:8000', 'adaptive', kroki_http_client, nil, 10)
+    result = kroki_client.get_image(kroki_diagram, 'utf8')
+    expect(result).to eq('POST http://localhost:8000/type/format - text')
+  end
+  it 'should get an image with GET request if the URI length is lower or equals than the value configured' do
+    kroki_http_client = Class.new do
+      class << self
+        def get(uri, _)
+          "GET #{uri}"
+        end
+
+        def post(uri, data, _)
+          "POST #{uri} - #{data}"
+        end
+      end
+    end
+    kroki_diagram = Class.new do
+      attr_reader :type, :text, :format
+
+      def initialize(type, format, text)
+        @text = text
+        @type = type
+        @format = format
+      end
+
+      def get_diagram_uri(_)
+        'diagram-uri'
+      end
+    end.new('type', 'format', 'text')
+    kroki_client = ::AsciidoctorExtensions::KrokiClient.new('http://localhost:8000', 'adaptive', kroki_http_client, nil, 11)
+    result = kroki_client.get_image(kroki_diagram, 'utf8')
+    expect(result).to eq('GET diagram-uri')
+  end
 end

data/spec/asciidoctor_kroki_spec.rb

@@ -34,17 +34,59 @@ describe ::AsciidoctorExtensions::KrokiBlockProcessor do
 </div>
 </div>)
     end
-    it 'should include the plantuml-include file' do
+    it 'should include the plantuml-include file when safe mode is safe' do
       input = <<~'ADOC'
         [plantuml]
         ....
         alice -> bob: hello
         ....
       ADOC
-      output = Asciidoctor.convert(input, attributes: { 'env-idea' => '', 'kroki-plantuml-include' => 'spec/fixtures/config.puml' }, standalone: false)
+      output = Asciidoctor.convert(input, attributes: { 'env-idea' => '', 'kroki-plantuml-include' => 'spec/fixtures/config.puml' }, standalone: false, safe: :safe)
       (expect output).to eql %(<div class="imageblock kroki">
 <div class="content">
-<img src="https://kroki.io/plantuml/png/eNorzs7MK0gsSsxVyM3Py0_OKMrPTVUoKSpN5YrJS8zJTE5V0LVTSMpPslLISM3JyQcArVsRHA==" alt="Diagram">
+<img src="https://kroki.io/plantuml/png/eNorzs7MK0gsSsxVyM3Py0_OKMrPTVUoKSpN5eJKzMlMTlXQtVNIyk-yUshIzcnJBwCT9xBc" alt="Diagram">
+</div>
+</div>)
+    end
+    it 'should normalize plantuml-include path when safe mode is safe' do
+      input = <<~'ADOC'
+        [plantuml]
+        ....
+        alice -> bob: hello
+        ....
+      ADOC
+      output = Asciidoctor.convert(input, attributes: { 'env-idea' => '', 'kroki-plantuml-include' => '../../../spec/fixtures/config.puml' }, standalone: false, safe: :safe)
+      (expect output).to eql %(<div class="imageblock kroki">
+<div class="content">
+<img src="https://kroki.io/plantuml/png/eNorzs7MK0gsSsxVyM3Py0_OKMrPTVUoKSpN5eJKzMlMTlXQtVNIyk-yUshIzcnJBwCT9xBc" alt="Diagram">
+</div>
+</div>)
+    end
+    it 'should not include file which reside outside of the parent directory of the source when safe mode is safe' do
+      input = <<~'ADOC'
+        [plantuml]
+        ....
+        alice -> bob: hello
+        ....
+      ADOC
+      output = Asciidoctor.convert(input, attributes: { 'env-idea' => '', 'kroki-plantuml-include' => '/etc/passwd' }, standalone: false, safe: :safe)
+      (expect output).to eql %(<div class="imageblock kroki">
+<div class="content">
+<img src="https://kroki.io/plantuml/png/eNpLzMlMTlXQtVNIyk-yUshIzcnJBwA9iwZL" alt="Diagram">
+</div>
+</div>)
+    end
+    it 'should not include file when safe mode is secure' do
+      input = <<~'ADOC'
+        [plantuml]
+        ....
+        alice -> bob: hello
+        ....
+      ADOC
+      output = Asciidoctor.convert(input, attributes: { 'env-idea' => '', 'kroki-plantuml-include' => 'spec/fixtures/config.puml' }, standalone: false, safe: :secure)
+      (expect output).to eql %(<div class="imageblock kroki">
+<div class="content">
+<img src="https://kroki.io/plantuml/png/eNpLzMlMTlXQtVNIyk-yUshIzcnJBwA9iwZL" alt="Diagram">
 </div>
 </div>)
     end
@@ -54,11 +96,24 @@ describe ::AsciidoctorExtensions::KrokiBlockProcessor do
 
         plantuml::spec/fixtures/alice.puml[svg,role=sequence]
       ADOC
-      output = Asciidoctor.convert(input, attributes: { 'kroki-fetch-diagram' => '' }, standalone: false)
+      output = Asciidoctor.convert(input, attributes: { 'kroki-fetch-diagram' => '' }, standalone: false, safe: :safe)
       (expect output).to eql %(<div class="imageblock sequence kroki-format-svg kroki">
 <div class="content">
 <img src=".asciidoctor/kroki/diag-f6acdc206506b6ca7badd3fe722f252af992871426e580c8361ff4d47c2c7d9b.svg" alt="Diagram">
 </div>
+</div>)
+    end
+    it 'should not fetch diagram when safe mode is secure' do
+      input = <<~'ADOC'
+        :imagesdir: .asciidoctor/kroki
+
+        plantuml::spec/fixtures/alice.puml[svg,role=sequence]
+      ADOC
+      output = Asciidoctor.convert(input, attributes: { 'kroki-fetch-diagram' => '' }, standalone: false)
+      (expect output).to eql %(<div class="imageblock sequence kroki-format-svg kroki">
+<div class="content">
+<img src="https://kroki.io/plantuml/svg/eNpLzMlMTlXQtVNIyk-yUshIzcnJ5wIAQ-AGVQ==" alt="Diagram">
+</div>
 </div>)
     end
     it 'should create PNG diagram in imagesdir if kroki-fetch-diagram is set' do
@@ -67,7 +122,7 @@ describe ::AsciidoctorExtensions::KrokiBlockProcessor do
 
         plantuml::spec/fixtures/alice.puml[png,role=sequence]
       ADOC
-      output = Asciidoctor.convert(input, attributes: { 'kroki-fetch-diagram' => '' }, standalone: false)
+      output = Asciidoctor.convert(input, attributes: { 'kroki-fetch-diagram' => '' }, standalone: false, safe: :safe)
       (expect output).to eql %(<div class="imageblock sequence kroki-format-png kroki">
 <div class="content">
 <img src=".asciidoctor/kroki/diag-d4f314b2d4e75cc08aa4f8c2c944f7bf78321895d8ec5f665b42476d4e67e610.png" alt="Diagram">
@@ -80,7 +135,7 @@ end
 describe ::AsciidoctorExtensions::Kroki do
   it 'should return the list of supported diagrams' do
     diagram_names = ::AsciidoctorExtensions::Kroki::SUPPORTED_DIAGRAM_NAMES
-    expect(diagram_names).to include('vegalite', 'plantuml', 'bytefield', 'bpmn', 'excalidraw')
+    expect(diagram_names).to include('vegalite', 'plantuml', 'bytefield', 'bpmn', 'excalidraw', 'wavedrom')
   end
   it 'should register the extension for the list of supported diagrams' do
     doc = Asciidoctor::Document.new

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: asciidoctor-kroki
 version: !ruby/object:Gem::Version
-  version: 0.3.0
+  version: 0.4.0
 platform: ruby
 authors:
 - Guillaume Grossetie
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2021-01-12 00:00:00.000000000 Z
+date: 2021-02-16 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: asciidoctor
@@ -85,6 +85,7 @@ files:
 - lib/asciidoctor/extensions/asciidoctor_kroki.rb
 - lib/asciidoctor/extensions/asciidoctor_kroki/extension.rb
 - spec/.rubocop.yml
+- spec/asciidoctor_kroki_block_macro_spec.rb
 - spec/asciidoctor_kroki_client_spec.rb
 - spec/asciidoctor_kroki_diagram_spec.rb
 - spec/asciidoctor_kroki_spec.rb
@@ -122,6 +123,7 @@ specification_version: 4
 summary: Asciidoctor extension to convert diagrams to images using Kroki
 test_files:
 - spec/.rubocop.yml
+- spec/asciidoctor_kroki_block_macro_spec.rb
 - spec/asciidoctor_kroki_client_spec.rb
 - spec/asciidoctor_kroki_diagram_spec.rb
 - spec/asciidoctor_kroki_spec.rb