{-# LANGUAGE StrictData #-}

-- This is required due to the use of 'Void' in a constructor slot in
-- combination with 'deriveNoun' which generates an unreachable pattern.
{-# OPTIONS_GHC -Wno-overlapping-patterns #-}

{-|
    Event Types and Noun Conversion
-}
module Urbit.Arvo.Event where

import Urbit.Prelude

import Control.Monad.Fail (fail)
import Urbit.Arvo.Common (KingId(..), ServId(..), Vere(..))
import Urbit.Arvo.Common (Desk, Mime)
import Urbit.Arvo.Common (Header(..), HttpEvent)
import Urbit.Arvo.Common (AmesDest, Ipv4, Ipv6, Port, Turf)
import Urbit.Arvo.Common (ReOrg(..), reorgThroughNoun)

import qualified Crypto.Sign.Ed25519       as Ed
import qualified Data.ByteString           as BS
import qualified Data.Char                 as C
import qualified Data.ByteString.Char8     as C
import qualified Network.HTTP.Types.Method as H

-- Misc Types ------------------------------------------------------------------

type Rift = Atom -- Continuity number
type Life = Word -- Number of Azimoth key revs.
type Bloq = Atom -- TODO
type Oath = Atom -- Signature


-- Parsed URLs -----------------------------------------------------------------

type Host = Each Turf Ipv4
type Hart = (Bool, Maybe Atom, Host)
type Pork = (Maybe Knot, [Cord])
type Quay = [(Cord, Cord)]

data PUrl = PUrl Hart Pork Quay
  deriving (Eq, Ord, Show)

deriveNoun ''PUrl


-- Dawn Records ----------------------------------------------------------------

padByteString :: BS.ByteString -> Int -> BS.ByteString
padByteString bs length | remaining > 0 = bs <> (BS.replicate remaining 0)
                        | otherwise = bs
  where remaining = (length - (BS.length bs))

-- A Pass is the Atom concatenation of 'b', the public encryption key, and the
-- public authentication key. (see +pass-from-eth.)
data Pass = Pass { passSign :: Ed.PublicKey, passCrypt :: Ed.PublicKey }
  deriving (Eq, Ord, Show)

passToBS :: Pass -> BS.ByteString
passToBS Pass{..} = C.singleton 'b' <>
                    (Ed.unPublicKey passSign) <>
                    (Ed.unPublicKey passCrypt)

instance ToNoun Pass where
  toNoun = Atom . bytesAtom . passToBS

instance FromNoun Pass where
  parseNoun n = named "Pass" $ do
    MkBytes unpadded <- parseNoun n
    let bs = padByteString unpadded 65
    when ((C.head bs) /= 'b') $ do
      fail "Expecting 'b' prefix in public key structure"
    let removedPrefix = C.tail bs
    let passSign = Ed.PublicKey (take 32 removedPrefix)
    let passCrypt = Ed.PublicKey (drop 32 removedPrefix)
    unless ((length $ Ed.unPublicKey passSign) == 32) $
      error "Sign pubkey not 32 bytes"
    unless ((length $ Ed.unPublicKey passCrypt) == 32) $
      error "Crypt pubkey not 32 bytes"
    pure $ Pass{..}

-- A Ring isn't the secret keys: it's the ByteString input which generates both
-- the public key and the secret key. A Ring is the concatenation of 'B', the
-- encryption key derivation seed, and the authentication key derivation
-- seed. These aren't actually private keys, but public/private keypairs which
-- can be derived from these seeds.
data Ring = Ring { ringSign :: BS.ByteString, ringCrypt :: BS.ByteString }
  deriving (Eq, Ord)

instance ToNoun Ring where
  toNoun Ring{..} =
    Atom $ bytesAtom (C.singleton 'B' <> ringSign <> ringCrypt)

instance FromNoun Ring where
  parseNoun n = named "Ring" $ do
      MkBytes unpadded <- parseNoun n
      let bs = padByteString unpadded 65
      when ((C.head bs) /= 'B') $ do
        fail "Expecting 'B' prefix in public key structure"
      let removedPrefix = C.tail bs
      let ringSign = (take 32 removedPrefix)
      let ringCrypt = (drop 32 removedPrefix)
      unless ((length ringSign) == 32) $
        error "Sign seed not 32 bytes"
      unless ((length ringCrypt) == 32) $
        error "Crypt seed not 32 bytes"
      pure $ Ring ringSign ringCrypt

instance Show Ring where
  show r = "(Ring <<seed>> <<seed>>)"

data Seed = Seed
    { sShip :: Ship
    , sLife :: Life
    , sRing :: Ring
    , sOath :: (Maybe Oath)
    }
  deriving (Eq, Show)

data Germs = Germs
    { gShip :: Ship
    , gFeed :: [Germ]
    }
  deriving (Eq, Show)

data Germ = Germ
    { gLife :: Life
    , gRing :: Ring
    }
  deriving (Eq, Ord, Show)

data Feed
  = Feed0 Seed
  | Feed1 Germs
  deriving (Eq, Show)

-- NOTE reify type environment
$(pure [])

instance ToNoun Feed where
  toNoun = \case
    Feed0 s -> $(deriveToNounFunc ''Seed) s
    Feed1 s -> C (C (A 1) (A 0)) $ $(deriveToNounFunc ''Germs) s

instance FromNoun Feed where
  parseNoun = \case
    (C (C (A 1) (A 0)) s) -> Feed1 <$> $(deriveFromNounFunc ''Germs) s
    n                     -> Feed0 <$> $(deriveFromNounFunc ''Seed) n

type Public = (Life, HoonMap Life Pass)

data Dnses = Dnses { dPri::Cord, dSec::Cord, dTer::Cord }
  deriving (Eq, Ord, Show)

type EthAddr = Atom --Bytes -- 20 bytes
type ContNum = Word

data EthPoint = EthPoint
    { epOwn :: (EthAddr, EthAddr, EthAddr, EthAddr)
    , epNet :: Maybe (Life, Pass, ContNum, (Bool, Ship), Maybe Ship)
    , epKid :: Maybe (EthAddr, HoonSet Ship)
    }
  deriving (Eq, Show)

data Dawn = MkDawn
    { dSeed    :: Seed
    , dSponsor :: [(Ship, EthPoint)]
    , dCzar    :: HoonMap Ship (Rift, Life, Pass)
    , dTurf    :: [Turf]
    , dBloq    :: Bloq
    , dNode    :: (Maybe PUrl)
    }
  deriving (Eq, Show)

deriveNoun ''Dnses
deriveNoun ''EthPoint
deriveNoun ''Seed
deriveNoun ''Germ
deriveNoun ''Dawn


-- HTTP ------------------------------------------------------------------------

type ServerId = Atom

data Address
    = AIpv4 Ipv4
    | AIpv6 Ipv6
    | AAmes Ship
  deriving (Eq, Ord, Show)

data HttpRequest = HttpRequest
    { reqMeth :: H.StdMethod
    , reqUrl  :: Cord
    , reqHead :: [Header]
    , reqBody :: Maybe File
    }
  deriving (Eq, Ord, Show)

data HttpServerReq = HttpServerReq
    { hsrSecure  :: Bool
    , hsrAddress :: Address
    , hsrRequest :: HttpRequest
    }
  deriving (Eq, Ord, Show)

data HttpClientEv
    = HttpClientEvReceive (KingId, ()) ServerId HttpEvent
    | HttpClientEvBorn    (KingId, ()) ()
    | HttpClientEvCrud    Path         Noun
  deriving (Eq, Ord, Show)

data HttpServerEv
    = HttpServerEvRequest       (ServId, UD, UD, ()) HttpServerReq
    | HttpServerEvCancelRequest (ServId, UD, UD, ()) ()
    | HttpServerEvRequestLocal  (ServId, UD, UD, ()) HttpServerReq
    | HttpServerEvLive          (ServId, ())         Port (Maybe Port)
    | HttpServerEvBorn          (KingId, ())         ()
    | HttpServerEvCrud          Path                 Noun
  deriving (Eq, Ord, Show)

deriveNoun ''Address
deriveNoun ''HttpClientEv
deriveNoun ''HttpRequest
deriveNoun ''HttpServerEv
deriveNoun ''HttpServerReq


-- Ames ------------------------------------------------------------------------

data AmesEv
    = AmesEvHear ()   AmesDest Bytes
    | AmesEvHole ()   AmesDest Bytes
    | AmesEvCrud Path Noun
  deriving (Eq, Ord, Show)

deriveNoun ''AmesEv


-- Arvo Events -----------------------------------------------------------------

newtype Entropy = Entropy { entropyBits :: Word512 }
 deriving newtype (Eq, Ord, FromNoun, ToNoun)

instance Show Entropy where
  show = const "\"ENTROPY (secret)\""


data ArvoEv
    = ArvoEvWhom ()   Ship
    | ArvoEvWack ()   Entropy
    | ArvoEvWyrd ()   Vere
    | ArvoEvCrud Path Noun
    | ArvoEvTrim UD
    | ArvoEvWhat [Noun]
    | ArvoEvWhey ()
    | ArvoEvVerb (Maybe Bool)
  deriving (Eq, Ord, Show)

deriveNoun ''ArvoEv


-- Boat Events -----------------------------------------------------------------

data BoatEv
    = BoatEvBoat ()   ()
    | BoatEvCrud Path Noun
  deriving (Eq, Ord, Show)

deriveNoun ''BoatEv


-- Jael Events -----------------------------------------------------------------

data JaelEv
    = JaelEvRekey () (Life, Ring)
    | JaelEvCrud Path Noun
  deriving (Eq, Show)

deriveNoun ''JaelEv


-- Timer Events ----------------------------------------------------------------

data BehnEv
    = BehnEvWake ()           ()
    | BehnEvBorn (KingId, ()) ()
    | BehnEvCrud Path         Noun
  deriving (Eq, Ord, Show)

deriveNoun ''BehnEv


-- Newt Events -----------------------------------------------------------------

data NewtEv
    = NewtEvBorn (KingId, ()) ()
    | NewtEvCrud Path         Noun
  deriving (Eq, Ord, Show)

deriveNoun ''NewtEv


-- FileSystem Events -----------------------------------------------------------

data SyncEv
    = SyncEvInto (Nullable (KingId, ())) Desk Bool [(Path, Maybe Mime)]
    | SyncEvCrud Path                    Noun
  deriving (Eq, Ord, Show)

deriveNoun ''SyncEv


-- Terminal Events -------------------------------------------------------------

data LegacyBootEvent
    = Fake Ship
    | Dawn Dawn
  deriving (Eq, Show)

data Bolt
    = Key Char
    | Aro ArrowKey
    | Bac ()
    | Del ()
    | Hit Word64 Word64
    | Ret ()
  deriving (Eq, Ord, Show)

data Belt
    = Bol Bolt
    | Mod Modifier Bolt
    | Txt Tour
  deriving (Eq, Ord, Show)

data ArrowKey = D | L | R | U
  deriving (Eq, Ord, Show)

data Modifier = Ctl | Met | Hyp
  deriving (Eq, Ord, Show)

--NOTE  required to get the above declarations into reify's type environment
--      see also ghc/ghc#9813
$(pure [])

instance FromNoun Bolt where
  parseNoun = \case
    A c             -> pure $ Key $ C.chr $ fromIntegral c
    C (A 7955819) _ -> fail "%key not valid bolt tag"
    n               -> $(deriveFromNounFunc ''Bolt) n

instance FromNoun Belt where
  parseNoun = \case
    C (A 7106402) _ -> fail "%bol not valid belt tag"
    n               -> Bol <$> parseNoun n <|> $(deriveFromNounFunc ''Belt) n

instance ToNoun Bolt where
  toNoun = \case
    Key c -> A $ fromIntegral $ C.ord c
    n     -> $(deriveToNounFunc ''Bolt) n

instance ToNoun Belt where
  toNoun = \case
    Bol b -> toNoun b
    n     -> $(deriveToNounFunc ''Belt) n

data TermEv
    = TermEvBelt (UD, ()) Belt
    | TermEvBlew (UD, ()) Word Word
    | TermEvBoot (UD, ()) Bool LegacyBootEvent
    | TermEvHail (UD, ()) ()
    | TermEvCrud Path     Noun
  deriving (Eq, Show)

deriveNoun ''LegacyBootEvent
deriveNoun ''ArrowKey
deriveNoun ''Modifier
deriveNoun ''TermEv


-- Events for Device Drivers ---------------------------------------------------

data BlipEv
    = BlipEvAmes       AmesEv
    | BlipEvArvo       ArvoEv
    | BlipEvBehn       BehnEv
    | BlipEvBoat       BoatEv
    | BlipEvHttpClient HttpClientEv
    | BlipEvHttpServer HttpServerEv
    | BlipEvJael       JaelEv
    | BlipEvNewt       NewtEv
    | BlipEvSync       SyncEv
    | BlipEvTerm       TermEv
  deriving (Eq, Show)

deriveNoun ''BlipEv


-- The Main Event Type ---------------------------------------------------------

data Ev
    = EvBlip BlipEv
  deriving (Eq, Show)

instance ToNoun Ev where
  toNoun = toNoun . \case
    EvBlip v@BlipEvAmes{}       -> reorgThroughNoun ("ames", v)
    EvBlip v@BlipEvArvo{}       -> reorgThroughNoun ("",  v)
    EvBlip v@BlipEvBehn{}       -> reorgThroughNoun ("behn", v)
    EvBlip v@BlipEvBoat{}       -> reorgThroughNoun ("clay", v)
    EvBlip v@BlipEvHttpClient{} -> reorgThroughNoun ("iris", v)
    EvBlip v@BlipEvHttpServer{} -> reorgThroughNoun ("eyre", v)
    EvBlip v@BlipEvJael{}       -> reorgThroughNoun ("jael", v)
    EvBlip v@BlipEvNewt{}       -> reorgThroughNoun ("ames", v)
    EvBlip v@BlipEvSync{}       -> reorgThroughNoun ("clay", v)
    EvBlip v@BlipEvTerm{}       -> reorgThroughNoun ("dill", v)

-- XX We really should check the first path element, but since this is used only
-- in the event browser, which otherwise is broken, I don't care right now.
instance FromNoun Ev where
  parseNoun = parseNoun >=> \case
    ReOrg _ s t p v -> fmap EvBlip $ parseNoun $ toNoun (s,t,p,v)


-- Short Event Names -----------------------------------------------------------

{-
    In the case of user input, the cause is technically a terminal event,
    but we don't display any name because the cause is really the user.
-}
getSpinnerNameForEvent :: Ev -> Maybe Text
getSpinnerNameForEvent = \case
    EvBlip b -> case b of
        BlipEvAmes _                -> Just "ames"
        BlipEvArvo _                -> Just "arvo"
        BlipEvBehn _                -> Just "behn"
        BlipEvBoat _                -> Just "boat"
        BlipEvHttpClient _          -> Just "iris"
        BlipEvHttpServer _          -> Just "eyre"
        BlipEvJael _                -> Just "jael"
        BlipEvNewt _                -> Just "newt"
        BlipEvSync _                -> Just "clay"
        BlipEvTerm (TermEvBelt _ _) -> Nothing
        BlipEvTerm t                -> Just "term"

summarizeEvent :: Ev -> Text
summarizeEvent ev =
  fromNoun (toNoun ev) & \case
    Nothing -> "//invalid %event"
    Just (pax :: [Cord], tag :: Cord, val :: Noun) ->
      "/" <> intercalate "/" (unCord <$> pax) <> " %" <> unCord tag