--  yammat - Yet Another MateMAT
--  Copyright (C) 2015  Amedeo Molnár
--
--  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/>.
module Handler.Statistics where

import Import
import Handler.Common
import Data.List hiding (length)
import Data.Maybe (fromMaybe)
import Data.Time.Calendar (addDays)

getStatisticsR :: Handler RepJson
getStatisticsR = do
  today <- liftIO $ utctDay <$> getCurrentTime
  users <- runDB $ selectList [] [Asc UserId]
  positiveBalance <- return $ foldl (\acc (Entity _ u) -> if userBalance u >= 0
    then acc + (fromIntegral $ userBalance u) / 100
    else acc
    ) 0 users
  negativeBalance <- return $ foldl (\acc (Entity _ u) -> if userBalance u < 0
    then acc + (fromIntegral $ userBalance u) / 100
    else acc
    ) 0 users
  aUsers <- runDB $ selectList [UserTimestamp >=. addDays (-30) today] [Asc UserId]
  aPositiveBalance <- return $ foldl (\acc (Entity _ u) -> if userBalance u >= 0
    then acc + (fromIntegral $ userBalance u) / 100
    else acc
    ) 0 aUsers
  aNegativeBalance <- return $ foldl (\acc (Entity _ u) -> if userBalance u < 0
    then acc + (fromIntegral $ userBalance u) / 100
    else acc
    ) 0 aUsers
  dUsers <- runDB $ selectList [UserTimestamp <. addDays (-30) today] [Asc UserId]
  dPositiveBalance <- return $ foldl (\acc (Entity _ u) -> if userBalance u >= 0
    then acc + (fromIntegral $ userBalance u) / 100
    else acc
    ) 0 dUsers
  dNegativeBalance <- return $ foldl (\acc (Entity _ u) -> if userBalance u < 0
    then acc + (fromIntegral $ userBalance u) / 100
    else acc
    ) 0 dUsers
  totalBalance <- (/100) . fromIntegral <$> getCashierBalance
  goodUsers <- runDB $ selectList [UserBalance >=. 0] []
  noobAngels <- runDB $ selectList [UserBalance >=. 0, UserBalance <=. 1000] []
  noobDevils <- runDB $ selectList [UserBalance <=. 0, UserBalance >=. -1000] []
  archangels <- runDB $ selectList [UserBalance >. 5000] []
  archdevils <- runDB $ selectList [UserBalance <. -5000] []
  bevs <- runDB $ selectList [] [Asc BeverageId]
  -- let totalLossPrime = foldl (\acc (Entity _ bev) -> let primePrice = (fromIntegral $ fromMaybe 0 (beveragePricePerCrate bev)) / (fromIntegral $ fromMaybe 1 (beveragePerCrate bev)) in acc + (((fromIntegral $ abs $ beverageCorrectedAmount bev) * primePrice) / 100)) 0 bevs
  totalLossRetail <- return $ foldl (\acc (Entity _ bev) ->
    acc + ((fromIntegral $ abs $ beverageCorrectedAmount bev) * (fromIntegral $ beveragePrice bev) / 100)
    ) 0 bevs
  return $ repJson $ toJSON $ Statistics
    (length users)
    (length aUsers)
    (length dUsers)
    positiveBalance
    negativeBalance
    totalBalance
    (length goodUsers)
    (length users - length goodUsers)
    (length noobAngels)
    (length noobDevils)
    (length archangels)
    (length archdevils)
    aPositiveBalance
    aNegativeBalance
    dPositiveBalance
    dNegativeBalance
    -- totalLossPrime
    totalLossRetail

data Statistics = Statistics
  { totalUsers :: Int
  , activeUsers :: Int
  , deadUsers :: Int
  , positiveBalance :: Double
  , negativeBalance :: Double
  , totalBalance :: Double
  , goodUsers :: Int
  , evilUsers :: Int
  , noobAngels :: Int
  , noobDevils :: Int
  , archangels :: Int
  , archdevils :: Int
  , activeUsersPositiveBalance :: Double
  , activeUsersNegativeBalance :: Double
  , deadUsersPositiveBalance :: Double
  , deadUsersNegativeBalance :: Double
  -- , totalLossPrime :: Double
  , totalLossRetail :: Double
  }

instance ToJSON Statistics where
  toJSON (Statistics tu au du pb nb tb gu eu na nd aa ad aupb aunb dupb dunb tlr) =
    object
      [ "total_users" .= tu
      , "active_users" .= au
      , "inactive_users" .= du
      , "positive_balance" .= pb
      , "negative_balance" .= nb
      , "total_balance" .= tb
      , "good_users" .= gu
      , "evil_users" .= eu
      , "noob_angels" .= na
      , "noob_devils" .= nd
      , "archangels" .= aa
      , "archdevils" .= ad
      , "active_users_positive_balance" .= aupb
      , "active_users_negative_balance" .= aunb
      , "inactive_users_positive_balance" .= dupb
      , "inactive_users_negative_balance" .= dunb
      -- , "total_loss_prime_price" .= tlp
      , "total_loss_retail_price" .= tlr
      ]