module Text.Pandoc.Lua
( LuaException (..)
, runLuaFilter
, runPandocLua
, pushPandocModule
) where
import Control.Monad (when, (>=>))
import Foreign.Lua (FromLuaStack (peek), Lua, LuaException (..),
Status (OK), ToLuaStack (push))
import Text.Pandoc.Class (PandocIO)
import Text.Pandoc.Definition (Pandoc)
import Text.Pandoc.Lua.Filter (LuaFilter, walkMWithLuaFilter)
import Text.Pandoc.Lua.Init (runPandocLua)
import Text.Pandoc.Lua.Module.Pandoc (pushModule)
import qualified Foreign.Lua as Lua
runLuaFilter :: FilePath -> String
-> Pandoc -> PandocIO (Either LuaException Pandoc)
runLuaFilter filterPath format doc =
runPandocLua (runLuaFilter' filterPath format doc)
runLuaFilter' :: FilePath -> String
-> Pandoc -> Lua Pandoc
runLuaFilter' filterPath format pd = do
registerFormat
top <- Lua.gettop
stat <- Lua.dofile filterPath
if stat /= OK
then do
luaErrMsg <- peek (1) <* Lua.pop 1
Lua.throwLuaError luaErrMsg
else do
newtop <- Lua.gettop
when (newtop top < 1) pushGlobalFilter
luaFilters <- peek (1)
runAll luaFilters pd
where
registerFormat = do
push format
Lua.setglobal "FORMAT"
pushGlobalFilter :: Lua ()
pushGlobalFilter = do
Lua.newtable
Lua.getglobal' "pandoc.global_filter"
Lua.call 0 1
Lua.rawseti (2) 1
runAll :: [LuaFilter] -> Pandoc -> Lua Pandoc
runAll = foldr ((>=>) . walkMWithLuaFilter) return
pushPandocModule :: Maybe FilePath -> Lua Lua.NumResults
pushPandocModule = pushModule