aboutsummaryrefslogtreecommitdiff
path: root/lib/Malodivo/Types/Ministry.hs
blob: f3878d119623b4aec8e636ac1e7c66d2a2181ae1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
{-|
Ministries can be encoded to and decoded from JSON:

>>> import Data.Aeson (decode, encode)
>>> import Data.ByteString.Lazy.Char8 (pack)

>>> encode Defense
"\"Defense\""

>>> encode [ Defense, Welfare ]
"[\"Defense\",\"Welfare\"]"

>>> decode . pack $ "[ \"Science\" ]" :: Maybe [Ministry]
Just [Science]
-}
{-# LANGUAGE DeriveAnyClass #-}
{-# LANGUAGE DeriveGeneric #-}

module Malodivo.Types.Ministry
  ( Ministry(..)
  , MinistryInfo
  , MinistryLimits
  , mi2ml
  ) where

import Control.Arrow ((&&&))
import GHC.Generics (Generic)

import Data.Aeson (FromJSON, ToJSON)
import qualified Data.HashMap.Strict as HM
import Data.Hashable (Hashable)

-- | Ministry of the Kingdom of Malodivo.
data Ministry
  = Defense
  | Science
  | Welfare
  deriving (Eq, Hashable, Show, Generic, FromJSON, ToJSON)

-- | Convenient type. Describes maximum money a ministry can get.
type MinistryLimits = HM.HashMap Ministry Integer

{-|
We use this data type instead of 'MinistryLimits' for JSON,
because maps other than @HashMap Text _@ are hard to decode/encode see
<https://github.com/bos/aeson/issues/79>.  Anyway, we still have type-checked
typos-proof structure.
-}
data MinistryInfo = MinistryInfo
  { name :: Ministry
  , amount :: Integer
  } deriving (Generic, FromJSON)

mi2ml :: [MinistryInfo] -> MinistryLimits
mi2ml = HM.fromListWith (+) . map (name &&& amount)