Render figure element when images have title.

CHANGELOG.md

@@ -13,6 +13,7 @@
   in any system.
 - Update all templates to use h1 as the root heading.
 - Remove inline alignment from tables.
+- Render `<figure>` for images with title.
 
 ## v3.1.0
 

lib/kitabu/markdown.rb

@@ -61,6 +61,27 @@ def table(header, body)
           </table>
         HTML
       end
+
+      def image(src, title, alt)
+        html = Nokogiri::HTML.fragment("<img />")
+        img = html.css("img").first
+        img.set_attribute(:src, src)
+        img.set_attribute(:alt, alt)
+        img.set_attribute(:title, title) if title
+
+        return html.to_s unless title
+
+        html = Nokogiri::HTML.fragment <<~HTML
+          <figure>
+            #{img}
+            <figcaption></figcaption>
+          </figure>
+        HTML
+
+        html.css("figcaption").first.content = title
+
+        html.to_s
+      end
     end
 
     class << self

spec/kitabu/markdown_spec.rb

@@ -118,4 +118,29 @@ class User
     expect(html).to have_tag("table>tbody>tr>td.align-center", "Positive")
     expect(html).to have_tag("table>tbody>tr>td:not([class*='align'])", "N/A")
   end
+
+  it "replaces images with titles with figure" do
+    html = Kitabu::Markdown.render <<~TEXT
+      ![ALT](image.png "TITLE")
+    TEXT
+
+    html = Nokogiri::HTML(html)
+
+    expect(html).to have_tag(
+      "figure>img[alt='ALT'][title='TITLE']+figcaption",
+      "TITLE"
+    )
+  end
+
+  it "keeps images without titles as it is" do
+    html = Kitabu::Markdown.render <<~TEXT
+      ![This is the alt](image.png)
+    TEXT
+
+    html = Nokogiri::HTML(html)
+
+    expect(html).to have_tag("img[alt='This is the alt']")
+    expect(html).not_to have_tag("figure")
+    expect(html).not_to have_tag("figcaption")
+  end
 end