{-# LANGUAGE OverloadedStrings, RankNTypes, KindSignatures, GADTs #-}
module Types where

import Data.Aeson

import Crypto.Noise.Cipher
import Crypto.Noise.Cipher.ChaChaPoly1305
import Crypto.Noise.Cipher.AESGCM
import Crypto.Noise.DH
import Crypto.Noise.DH.Curve25519
import Crypto.Noise.DH.Curve448
import Crypto.Noise.HandshakePatterns
import Crypto.Noise.Hash
import Crypto.Noise.Hash.SHA256
import Crypto.Noise.Hash.SHA512
import Crypto.Noise.Hash.BLAKE2s
import Crypto.Noise.Hash.BLAKE2b

data HandshakeType = NoiseNN
                   | NoiseKN
                   | NoiseNK
                   | NoiseKK
                   | NoiseNX
                   | NoiseKX
                   | NoiseXN
                   | NoiseIN
                   | NoiseXK
                   | NoiseIK
                   | NoiseXX
                   | NoiseIX
                   | NoiseN
                   | NoiseK
                   | NoiseX
                   deriving Eq

instance FromJSON HandshakeType where
  parseJSON (String "NN") = pure NoiseNN
  parseJSON (String "KN") = pure NoiseKN
  parseJSON (String "NK") = pure NoiseNK
  parseJSON (String "KK") = pure NoiseKK
  parseJSON (String "NX") = pure NoiseNX
  parseJSON (String "KX") = pure NoiseKX
  parseJSON (String "XN") = pure NoiseXN
  parseJSON (String "IN") = pure NoiseIN
  parseJSON (String "XK") = pure NoiseXK
  parseJSON (String "IK") = pure NoiseIK
  parseJSON (String "XX") = pure NoiseXX
  parseJSON (String "IX") = pure NoiseIX
  parseJSON (String "N")  = pure NoiseN
  parseJSON (String "K")  = pure NoiseK
  parseJSON (String "X")  = pure NoiseX
  parseJSON _             = error "invalid pattern name"

data CipherType :: * -> * where
  CTChaChaPoly1305 :: CipherType ChaChaPoly1305
  CTAESGCM         :: CipherType AESGCM

data SomeCipherType where
  WrapCipherType :: forall c. Cipher c => CipherType c -> SomeCipherType

instance FromJSON SomeCipherType where
  parseJSON (String "ChaChaPoly") = pure . WrapCipherType $ CTChaChaPoly1305
  parseJSON (String "AESGCM")     = pure . WrapCipherType $ CTAESGCM
  parseJSON _                     = error "invalid cipher name"

instance ToJSON SomeCipherType where
  toJSON (WrapCipherType CTChaChaPoly1305) = "ChaChaPoly"
  toJSON (WrapCipherType CTAESGCM)         = "AESGCM"

instance Show (CipherType c) where
  show CTChaChaPoly1305 = "ChaChaPoly"
  show CTAESGCM = "AESGCM"

data DHType :: * -> * where
  DTCurve25519 :: DHType Curve25519
  DTCurve448   :: DHType Curve448

data SomeDHType where
  WrapDHType :: forall d. DH d => DHType d -> SomeDHType

instance FromJSON SomeDHType where
  parseJSON (String "25519") = pure . WrapDHType $ DTCurve25519
  parseJSON (String "448")   = pure . WrapDHType $ DTCurve448
  parseJSON _                = error "invalid DH name"

instance ToJSON SomeDHType where
  toJSON (WrapDHType DTCurve25519) = "25519"
  toJSON (WrapDHType DTCurve448)   = "448"

instance Show (DHType d) where
  show DTCurve25519 = "25519"
  show DTCurve448   = "448"

data HashType :: * -> * where
  HTSHA256  :: HashType SHA256
  HTSHA512  :: HashType SHA512
  HTBLAKE2s :: HashType BLAKE2s
  HTBLAKE2b :: HashType BLAKE2b

data SomeHashType where
  WrapHashType :: forall h. Hash h => HashType h -> SomeHashType

hsTypeToPattern :: HandshakeType
                -> HandshakePattern
hsTypeToPattern NoiseNN = noiseNN
hsTypeToPattern NoiseKN = noiseKN
hsTypeToPattern NoiseNK = noiseNK
hsTypeToPattern NoiseKK = noiseKK
hsTypeToPattern NoiseNX = noiseNX
hsTypeToPattern NoiseKX = noiseKX
hsTypeToPattern NoiseXN = noiseXN
hsTypeToPattern NoiseIN = noiseIN
hsTypeToPattern NoiseXK = noiseXK
hsTypeToPattern NoiseIK = noiseIK
hsTypeToPattern NoiseXX = noiseXX
hsTypeToPattern NoiseIX = noiseIX
hsTypeToPattern NoiseN  = noiseN
hsTypeToPattern NoiseK  = noiseK
hsTypeToPattern NoiseX  = noiseX

instance FromJSON SomeHashType where
  parseJSON (String "SHA256")  = pure . WrapHashType $ HTSHA256
  parseJSON (String "SHA512")  = pure . WrapHashType $ HTSHA512
  parseJSON (String "BLAKE2s") = pure . WrapHashType $ HTBLAKE2s
  parseJSON (String "BLAKE2b") = pure . WrapHashType $ HTBLAKE2b
  parseJSON _                  = error "invalid hash name"

instance ToJSON SomeHashType where
  toJSON (WrapHashType HTSHA256)  = "SHA256"
  toJSON (WrapHashType HTSHA512)  = "SHA512"
  toJSON (WrapHashType HTBLAKE2s) = "BLAKE2s"
  toJSON (WrapHashType HTBLAKE2b) = "BLAKE2b"

instance Show (HashType h) where
  show HTSHA256  = "SHA256"
  show HTSHA512  = "SHA512"
  show HTBLAKE2s = "BLAKE2s"
  show HTBLAKE2b = "BLAKE2b"

instance Show HandshakeType where
  show NoiseNN = "NN"
  show NoiseKN = "KN"
  show NoiseNK = "NK"
  show NoiseKK = "KK"
  show NoiseNX = "NX"
  show NoiseKX = "KX"
  show NoiseXN = "XN"
  show NoiseIN = "IN"
  show NoiseXK = "XK"
  show NoiseIK = "IK"
  show NoiseXX = "XX"
  show NoiseIX = "IX"
  show NoiseN  = "N"
  show NoiseK  = "K"
  show NoiseX  = "X"