diff --git a/data/footer.htm b/data/footer.htm new file mode 100644 --- /dev/null +++ b/data/footer.htm @@ -0,0 +1,2 @@ + + diff --git a/data/header.htm b/data/header.htm new file mode 100644 --- /dev/null +++ b/data/header.htm @@ -0,0 +1,30 @@ + + + + + + + + +
diff --git a/lhs2html.cabal b/lhs2html.cabal --- a/lhs2html.cabal +++ b/lhs2html.cabal @@ -1,5 +1,5 @@ name: lhs2html -version: 0.999 +version: 0.9999 synopsis: Compile lhs in bird style to md, html, hs. license: PublicDomain license-file: LICENSE @@ -8,10 +8,11 @@ category: Development build-type: Simple cabal-version: >=1.10 +data-files: data/*.htm executable lhs2html main-is: lhs2html.hs - build-depends: base >=4.7 && <4.8, nicify + build-depends: base >=4.7 && <4.8, nicify, filepath hs-source-dirs: src default-language: Haskell2010 diff --git a/src/lhs2html.hs b/src/lhs2html.hs --- a/src/lhs2html.hs +++ b/src/lhs2html.hs @@ -2,54 +2,32 @@ module Main where -import Control.Applicative +import Data.Functor +import Control.Monad import System.Environment import Text.Nicify +import System.FilePath +import Paths_lhs2html + main = getArgs >>= \case [] -> lhs2html <$> getContents >>= putStrLn - args -> flip mapM_ args $ \f -> lhs2html <$> readFile f >>= writeFile (f ++ ".html") - -preamble = "\n\ - \\n\ - \\n\ - \ \n\ - \\n\ - \\n\ - \\n\ - \\n
\n" + args -> do + header <- getDataFileName ("data" "header.htm") >>= readFile + footer <- getDataFileName ("data" "footer.htm") >>= readFile + forM_ args $ \f -> do + let outfile = f ++ ".html" + + writeFile outfile header + lhs2html <$> readFile f >>= appendFile outfile + appendFile outfile footer + lhs2html :: String -> String -lhs2html = (preamble ++) . foldr toHTML "" . filter (\x -> x /= Empty && x /= Para []) . parse +lhs2html = foldr toHTML "" . parse parse :: String -> [Object] parse = process . map identify . lines @@ -154,5 +132,4 @@ codify True ('`' : xs) = "" ++ codify False xs codify flag (x : xs) = x : codify flag xs codify _ _ = [] -