module Main where

import Control.DeepSeq
import Data.Int
import Test.Tasty.Bench

import Database.PostgreSQL.PQTypes.Deriving

main :: IO ()
main = defaultMain
  [ bgroup "enum"
    [ bench "encode" $ nf (encodeEnum @T) T42
    , bench "decode" $ nf (decodeEnum @T) 42
    ]
  , bgroup "enum-text"
    [ bench "encode" $ nf (encodeEnumAsText @S) S42
    , bench "decode" $ nf (decodeEnumAsText @S) "text_42"
    ]
  ]

data T = T01 | T02 | T03 | T04 | T05 | T06 | T07 | T08 | T09 | T10
       | T11 | T12 | T13 | T14 | T15 | T16 | T17 | T18 | T19 | T20
       | T21 | T22 | T23 | T24 | T25 | T26 | T27 | T28 | T29 | T30
       | T31 | T32 | T33 | T34 | T35 | T36 | T37 | T38 | T39 | T40
       | T41 | T42 | T43 | T44 | T45 | T46 | T47 | T48 | T49 | T50
  deriving (Eq, Show, Enum, Bounded)

instance NFData T where
  rnf = (`seq` ())

-- | Enum encoding for 'T'.
--
-- >>> isInjective (encodeEnum @T)
-- True
instance EnumEncoding T where
   type EnumBase T = Int16
   encodeEnum = \case
     T01 -> 1;  T02 -> 2;  T03 -> 3;  T04 -> 4;  T05 -> 5
     T06 -> 6;  T07 -> 7;  T08 -> 8;  T09 -> 9;  T10 -> 10
     T11 -> 11; T12 -> 12; T13 -> 13; T14 -> 14; T15 -> 15
     T16 -> 16; T17 -> 17; T18 -> 18; T19 -> 19; T20 -> 20
     T21 -> 21; T22 -> 22; T23 -> 23; T24 -> 24; T25 -> 25
     T26 -> 26; T27 -> 27; T28 -> 28; T29 -> 29; T30 -> 30
     T31 -> 31; T32 -> 32; T33 -> 33; T34 -> 34; T35 -> 35
     T36 -> 36; T37 -> 37; T38 -> 38; T39 -> 39; T40 -> 40
     T41 -> 41; T42 -> 42; T43 -> 43; T44 -> 44; T45 -> 45
     T46 -> 46; T47 -> 47; T48 -> 48; T49 -> 49; T50 -> 50

----------------------------------------

data S = S01 | S02 | S03 | S04 | S05 | S06 | S07 | S08 | S09 | S10
       | S11 | S12 | S13 | S14 | S15 | S16 | S17 | S18 | S19 | S20
       | S21 | S22 | S23 | S24 | S25 | S26 | S27 | S28 | S29 | S30
       | S31 | S32 | S33 | S34 | S35 | S36 | S37 | S38 | S39 | S40
       | S41 | S42 | S43 | S44 | S45 | S46 | S47 | S48 | S49 | S50
  deriving (Eq, Show, Enum, Bounded)

instance NFData S where
  rnf = (`seq` ())

-- | Enum encoding for 'S'.
--
-- >>> isInjective (encodeEnumAsText @S)
-- True
instance EnumAsTextEncoding S where
  encodeEnumAsText = \case
    S01 -> "text_01"; S02 -> "text_02"; S03 -> "text_03"; S04 -> "text_04"
    S05 -> "text_05"; S06 -> "text_06"; S07 -> "text_07"; S08 -> "text_08"
    S09 -> "text_09"; S10 -> "text_10"; S11 -> "text_11"; S12 -> "text_12"
    S13 -> "text_13"; S14 -> "text_14"; S15 -> "text_15"; S16 -> "text_16"
    S17 -> "text_17"; S18 -> "text_18"; S19 -> "text_19"; S20 -> "text_20"
    S21 -> "text_21"; S22 -> "text_22"; S23 -> "text_23"; S24 -> "text_24"
    S25 -> "text_25"; S26 -> "text_26"; S27 -> "text_27"; S28 -> "text_28"
    S29 -> "text_29"; S30 -> "text_30"; S31 -> "text_31"; S32 -> "text_32"
    S33 -> "text_33"; S34 -> "text_34"; S35 -> "text_35"; S36 -> "text_36"
    S37 -> "text_37"; S38 -> "text_38"; S39 -> "text_39"; S40 -> "text_40"
    S41 -> "text_41"; S42 -> "text_42"; S43 -> "text_43"; S44 -> "text_44"
    S45 -> "text_45"; S46 -> "text_46"; S47 -> "text_47"; S48 -> "text_48"
    S49 -> "text_49"; S50 -> "text_50";