{-# LANGUAGE OverloadedStrings #-}

module Controller.Payment
  ( getPayments
  , getMonthlyPayments
  , createPayment
  , deletePayment
  , getTotalPayments
  , getPaymentsCount
  ) where

import Web.Scotty

import Network.HTTP.Types.Status (ok200, badRequest400)

import Database.Persist

import Control.Monad.IO.Class (liftIO)

import Data.Text (Text)
import qualified Data.Aeson.Types as Json

import qualified Secure

import Json (jsonObject)

import Model.Database
import qualified Model.Payment as P
import Model.Frequency
import Model.Json.Number
import qualified Model.Json.PaymentId as JP
import Model.Message
import Model.Message.Key (Key(PaymentNotDeleted))

getPayments :: Int -> Int -> ActionM ()
getPayments page perPage =
  Secure.loggedAction (\_ -> do
    (liftIO $ runDb (P.getPunctualPayments page perPage)) >>= json
  )

getMonthlyPayments :: ActionM ()
getMonthlyPayments =
  Secure.loggedAction (\user -> do
    (liftIO $ runDb (P.getUserMonthlyPayments (entityKey user))) >>= json
  )

createPayment :: Text -> Int -> Frequency -> ActionM ()
createPayment name cost frequency =
  Secure.loggedAction (\user -> do
    paymentId <- liftIO . runDb $ P.createPayment (entityKey user) name cost frequency
    json (JP.PaymentId paymentId)
  )

deletePayment :: Text -> ActionM ()
deletePayment paymentId =
  Secure.loggedAction (\user -> do
    deleted <- liftIO . runDb $ P.deleteOwnPayment user (textToKey paymentId)
    if deleted
      then
        status ok200
      else do
        status badRequest400
        jsonObject [("error", Json.String $ getMessage PaymentNotDeleted)]
  )

getTotalPayments :: ActionM ()
getTotalPayments =
  Secure.loggedAction (\_ -> do
    (liftIO . runDb $ P.getTotalPayments) >>= json
  )

getPaymentsCount :: ActionM ()
getPaymentsCount =
  Secure.loggedAction (\_ -> do
    Number <$> (liftIO . runDb $ P.getPaymentsCount) >>= json
  )