-- sflc-ledger-reports: web interface to hledger-based reports
--  Copyright (C) 2013  Clint Adams
--
--  This program is free software: you can redistribute it and/or modify
--  it under the terms of the GNU Affero General Public License as
--  published by the Free Software Foundation, either version 3 of the
--  License, or (at your option) any later version.
--
--  This program is distributed in the hope that it will be useful,
--  but WITHOUT ANY WARRANTY; without even the implied warranty of
--  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
--  GNU Affero General Public License for more details.
--
--  You should have received a copy of the GNU Affero General Public License
--  along with this program.  If not, see <http://www.gnu.org/licenses/>.

{-# LANGUAGE OverloadedStrings #-}
module Handler.Register where

import Import
import qualified Data.Text as T
import Hledger.RegisterCSV
import qualified Data.Map as Map

typeCsv :: ContentType
typeCsv = "text/csv; charset=utf-8"

getRegisterR :: Text -> Handler TypedContent
getRegisterR ledger = do
    ledgers <- fmap hledgerConfig getYesod
    case Map.lookup ledger ledgers of
        Nothing -> notFound
	Just fn -> do
            csv <- liftIO $ convertJournalToRegisterCSV fn
            addHeader "Content-Disposition" (T.concat ["attachment; filename=", ledger, ".csv"])
            sendResponse (typeCsv, toContent csv)