hid-examples-0.1.0.0: stockquotes/Main.hs

{-# LANGUAGE RecordWildCards #-}

module Main where

import Data.Bool (bool)
import Control.Monad (when, unless)
import qualified Data.Text.IO as TIO
import qualified Data.ByteString.Lazy as BL (readFile, writeFile)
import Data.Csv (decodeByName)

import QuoteData
import Statistics
import StatReport
import Charts
import HtmlReport
import Params

generateReports :: (Functor t, Foldable t) => Params -> t QuoteData -> IO ()
generateReports Params {..} quotes = do
  unless no_text $ TIO.putStr $ statReport statInfo'
  when prices $ plotChart title quotes [Open, Close, High, Low] fname_prices
  when volumes $ plotChart title quotes [Volume] fname_volumes
  when html $ BL.writeFile fname_html $ htmlReport title quotes statInfo' images
 where
   statInfo' = statInfo quotes
   withCompany pref  = if company /= "" then pref ++ company else ""
   img_suffix = withCompany "_" ++ ".svg"
   fname_prices = "prices" ++ img_suffix
   fname_volumes = "volumes" ++ img_suffix
   images = concat $ zipWith (bool []) [[fname_prices],[fname_volumes]]
                                       [prices, volumes]
   fname_html = "report" ++ withCompany "_" ++ ".html"
   title = "Historical Quotes" ++ withCompany " for "

work :: Params -> IO ()
work params = do 
  csvData <- BL.readFile (fname params)
  case decodeByName csvData of
    Left err -> putStrLn err
    Right (_, quotes) -> generateReports params quotes

main :: IO ()
main = cmdLineParser >>= work