Handle Figures in Lua

Author: argent0

data/pandoc.lua

@@ -469,6 +469,22 @@ M.Div = M.Block:create_constructor(
   {{attr = {"identifier", "classes", "attributes"}}, "content"}
 )
 
+
+--- Creates a figure element.
+-- @function Figure
+-- @tparam      {Block,...}  content     figure block contents
+-- @tparam      Caption      caption     figure caption
+-- @tparam[opt] Attr         attr        element attributes
+-- @treturn     Block                    figure element
+M.Figure = M.Block:create_constructor(
+  "Figure",
+  function(content, caption, attr)
+    return {c = {ensureAttr(attr), caption, ensureList(content)}}
+  end,
+  {{attr = {"identifier", "classes", "attributes"}}, "caption", "content"}
+)
+
+
 --- Creates a header element.
 -- @function Header
 -- @tparam      int          level       header level

src/Text/Pandoc/Lua/Marshaling/AST.hs

@@ -160,6 +160,7 @@ pushBlock = \case
   CodeBlock attr code      -> pushViaConstructor "CodeBlock" code (LuaAttr attr)
   DefinitionList items     -> pushViaConstructor "DefinitionList" items
   Div attr blcks           -> pushViaConstructor "Div" blcks (LuaAttr attr)
+  Figure attr capt blcks   -> pushViaConstructor "Figure" blcks capt (LuaAttr attr)
   Header lvl attr inlns    -> pushViaConstructor "Header" lvl inlns (LuaAttr attr)
   HorizontalRule           -> pushViaConstructor "HorizontalRule"
   LineBlock blcks          -> pushViaConstructor "LineBlock" blcks
@@ -171,7 +172,6 @@ pushBlock = \case
   RawBlock f cs            -> pushViaConstructor "RawBlock" f cs
   Table attr blkCapt specs thead tbody tfoot ->
     pushViaConstructor "Table" blkCapt specs thead tbody tfoot attr
-  Figure {} -> undefined -- pushViaConstructor "Figure" capt body attr
 
 -- | Return the value at the given index as block if possible.
 peekBlock :: StackIndex -> Lua Block
@@ -183,6 +183,8 @@ peekBlock idx = defineHowTo "get Block value" $! do
       "CodeBlock"      -> withAttr CodeBlock <$!> elementContent
       "DefinitionList" -> DefinitionList <$!> elementContent
       "Div"            -> withAttr Div <$!> elementContent
+      "Figure"         -> (\(LuaAttr attr, capt, bs) -> Figure attr capt bs)
+                          <$!> elementContent
       "Header"         -> (\(lvl, LuaAttr attr, lst) -> Header lvl attr lst)
                           <$!> elementContent
       "HorizontalRule" -> return HorizontalRule