@@ -36,9 +36,10 @@ module Text.Pandoc.Lua.StackInstances () where
37import Prelude
37import Prelude
38import Control.Applicative ((<|>))
38import Control.Applicative ((<|>))
39−import Control.Monad (when)
40import Data.Data (showConstr, toConstr)
39import Data.Data (showConstr, toConstr)
41import Foreign.Lua (Lua, Peekable, Pushable, StackIndex)
40import Foreign.Lua (Lua, Peekable, Pushable, StackIndex)
41+import Foreign.Lua.Userdata ( ensureUserdataMetatable, pushAnyWithMetatable
42import Text.Pandoc.Definition
43import Text.Pandoc.Definition
43import Text.Pandoc.Extensions (Extensions)
44import Text.Pandoc.Extensions (Extensions)
44import Text.Pandoc.Lua.Util (defineHowTo, pushViaConstructor)
45import Text.Pandoc.Lua.Util (defineHowTo, pushViaConstructor)
@@ -308,24 +309,28 @@ instance Peekable LuaListAttributes where
309instance Pushable Element where
310instance Pushable Element where
310 push (Blk blk) = Lua.push blk
311 push (Blk blk) = Lua.push blk
312+ push sec = pushAnyWithMetatable pushElementMetatable sec
314+ pushElementMetatable = ensureUserdataMetatable (metatableName sec) $
315+ LuaUtil.addFunction "__index" indexElement
317+instance Peekable Element where
318+ peek idx = Lua.ltype idx >>= \case
319+ Lua.TypeUserdata -> Lua.peekAny idx
320+ _ -> Blk <$> Lua.peek idx
322+indexElement :: Element -> String -> Lua Lua.NumResults
324+ (Blk _) -> const (1 <$ Lua.pushnil) -- this shouldn't happen
311− push (Sec lvl num attr label contents) = do
325+ (Sec lvl num attr label contents) -> fmap (return 1) . \case
313− LuaUtil.addField "level" lvl
326+ "level" -> Lua.push lvl
314− LuaUtil.addField "numbering" num
327+ "numbering" -> Lua.push num
315− LuaUtil.addField "attr" (LuaAttr attr)
328+ "attr" -> Lua.push (LuaAttr attr)
316− LuaUtil.addField "label" label
329+ "label" -> Lua.push label
317− LuaUtil.addField "contents" contents
330+ "contents" -> Lua.push contents
318− pushSecMetaTable
331+ "tag" -> Lua.push "Sec"
319− Lua.setmetatable (-2)
321− pushSecMetaTable :: Lua ()
322− pushSecMetaTable = do
323− inexistant <- Lua.newmetatable "PandocElementSec"
324− when inexistant $ do
325− LuaUtil.addField "t" "Sec"
326− Lua.push "__index"
332+ "t" -> Lua.push "Sec"
327− Lua.pushvalue (-2)
333+ _ -> Lua.pushnil