criterion-to-html-0.0.0.3: src/Criterion/ToHtml.hs

-- | Main module
--
module Main
    ( main
    ) where

import Control.Applicative ((<$>))
import System.Environment (getArgs, getProgName)
import System.FilePath (replaceExtension)

import Text.Blaze.Html.Renderer.Utf8 (renderHtml)
import qualified Data.ByteString as B
import qualified Data.ByteString.Lazy as BL

import Criterion.ToHtml.Html
import Criterion.ToHtml.Result
import Paths_criterion_to_html (getDataFileName)

toHtml' :: FilePath -> FilePath -> IO ()
toHtml' csv html = do
    js <- B.readFile =<< getDataFileName "criterion-to-html.js"
    putStrLn $ "Parsing " ++ csv
    csv' <- parseCriterionCsv <$> readFile csv
    BL.writeFile html $ renderHtml $ report js csv'
    putStrLn $ "Wrote " ++ html

main :: IO ()
main = do
    args <- getArgs
    progName <- getProgName
    case args of
        [csv, html] -> toHtml' csv html
        [csv]       -> toHtml' csv (replaceExtension csv "html")
        _           -> putStrLn $
            "Usage: " ++ progName ++ " <csv-file> [out-file]"