forked from filipg/gonito
59 lines
1.6 KiB
Haskell
59 lines
1.6 KiB
Haskell
module Handler.Extract where
|
|
|
|
import Import
|
|
|
|
import qualified Data.Text as T
|
|
import Text.Pandoc
|
|
import Text.Pandoc.Walk (walk)
|
|
import Text.Pandoc.Shared (stringify)
|
|
|
|
import Data.Maybe
|
|
|
|
import System.IO (withFile, IOMode(..), readFile)
|
|
|
|
extractHeaders :: Block -> [String]
|
|
extractHeaders (Header 1 _ x) = [stringify x]
|
|
extractHeaders _ = []
|
|
|
|
extractFirstHeader :: Pandoc -> Maybe String
|
|
extractFirstHeader doc = case queryWith extractHeaders doc of
|
|
(s:_) -> Just s
|
|
[] -> Nothing
|
|
|
|
extractParas :: Block -> [String]
|
|
extractParas (Para x) = [stringify x]
|
|
extractParas _ = []
|
|
|
|
extractFirstPara :: Pandoc -> Maybe String
|
|
extractFirstPara doc = case queryWith extractParas doc of
|
|
(s:_) -> Just s
|
|
[] -> Nothing
|
|
|
|
readDoc :: String -> Pandoc
|
|
readDoc s = case runPure $ readMarkdown def (T.pack s) of
|
|
Right doc -> doc
|
|
Left err -> error (show err)
|
|
|
|
defaultTitle :: String
|
|
defaultTitle = "[???]"
|
|
|
|
defaultDescription :: String
|
|
defaultDescription = ""
|
|
|
|
readmeFile :: FilePath
|
|
readmeFile = "README.md"
|
|
|
|
imageFile :: FilePath
|
|
imageFile = ".seeme.png"
|
|
|
|
getTitleAndDescription :: String -> (String, String)
|
|
getTitleAndDescription contents = (title, description)
|
|
where title = fromMaybe defaultTitle $ extractFirstHeader doc
|
|
description = fromMaybe defaultDescription $ extractFirstPara doc
|
|
doc = readDoc contents
|
|
|
|
extractTitleAndDescription :: FilePath -> IO (String, String)
|
|
extractTitleAndDescription fp = do
|
|
contents <- System.IO.readFile fp
|
|
return $ getTitleAndDescription contents
|