{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE Unsafe #-}
module Text.Pandoc.Filter.Scripting
( runTempPythonScript
, hasBlockingShowCall
, PythonScript
, ScriptResult(..)
) where
import Data.Text (Text)
import qualified Data.Text as T
import qualified Data.Text.IO as T
import Data.Hashable (hash)
import System.Exit (ExitCode (..))
import System.FilePath ((</>))
import System.IO.Temp (getCanonicalTemporaryDirectory)
import System.Process.Typed (runProcess, shell)
import Data.Monoid (Any (..), (<>))
type PythonScript = Text
data ScriptResult
= ScriptSuccess
| ScriptFailure Int
runTempPythonScript ::
PythonScript
-> IO ScriptResult
runTempPythonScript script
= do
scriptPath <- (</> hashedPath) <$> getCanonicalTemporaryDirectory
T.writeFile scriptPath script
ec <- runProcess $ shell $ "python " <> (show scriptPath)
case ec of
ExitSuccess -> return ScriptSuccess
ExitFailure code -> return $ ScriptFailure code
where
hashedPath = show . hash $ script
hasBlockingShowCall :: PythonScript -> Bool
hasBlockingShowCall script =
anyOf
[ "plt.show()" `elem` scriptLines
, "pyplot.show()" `elem` scriptLines
, "matplotlib.pyplot.show()" `elem` scriptLines
]
where
scriptLines = T.lines script
anyOf xs = getAny $ mconcat $ Any <$> xs