{-# LANGUAGE DuplicateRecordFields, DeriveGeneric #-}
-- |
-- Module: Network.Greskell.WebSocket.Request
-- Description: Request to Gremlin Server
-- Maintainer: Toshio Ito <debug.ito@gmail.com>
--
-- 
module Network.Greskell.WebSocket.Request
       ( -- * RequestMessage
         RequestMessage(..),
         Operation(..),
         toRequestMessage,
         makeRequestMessage
       ) where

import Control.Applicative ((<$>), (<*>))
import Data.Aeson (Object, ToJSON(..), FromJSON(..))
import Data.Text (Text)
import Data.UUID (UUID)
import Data.UUID.V4 (nextRandom)
import GHC.Generics (Generic)

import qualified Network.Greskell.WebSocket.Request.Aeson as GAeson
import Network.Greskell.WebSocket.Request.Common (Operation(..))


-- | RequestMessage to a Gremlin Server. See
-- <http://tinkerpop.apache.org/docs/current/dev/provider/>.
data RequestMessage =
  RequestMessage
  { RequestMessage -> UUID
requestId :: !UUID,
    RequestMessage -> Text
op :: !Text,
    RequestMessage -> Text
processor :: !Text,
    RequestMessage -> Object
args :: !Object
  }
  deriving (Int -> RequestMessage -> ShowS
[RequestMessage] -> ShowS
RequestMessage -> String
(Int -> RequestMessage -> ShowS)
-> (RequestMessage -> String)
-> ([RequestMessage] -> ShowS)
-> Show RequestMessage
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [RequestMessage] -> ShowS
$cshowList :: [RequestMessage] -> ShowS
show :: RequestMessage -> String
$cshow :: RequestMessage -> String
showsPrec :: Int -> RequestMessage -> ShowS
$cshowsPrec :: Int -> RequestMessage -> ShowS
Show,RequestMessage -> RequestMessage -> Bool
(RequestMessage -> RequestMessage -> Bool)
-> (RequestMessage -> RequestMessage -> Bool) -> Eq RequestMessage
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: RequestMessage -> RequestMessage -> Bool
$c/= :: RequestMessage -> RequestMessage -> Bool
== :: RequestMessage -> RequestMessage -> Bool
$c== :: RequestMessage -> RequestMessage -> Bool
Eq,(forall x. RequestMessage -> Rep RequestMessage x)
-> (forall x. Rep RequestMessage x -> RequestMessage)
-> Generic RequestMessage
forall x. Rep RequestMessage x -> RequestMessage
forall x. RequestMessage -> Rep RequestMessage x
forall a.
(forall x. a -> Rep a x) -> (forall x. Rep a x -> a) -> Generic a
$cto :: forall x. Rep RequestMessage x -> RequestMessage
$cfrom :: forall x. RequestMessage -> Rep RequestMessage x
Generic)

instance ToJSON RequestMessage where
  toJSON :: RequestMessage -> Value
toJSON = Options -> RequestMessage -> Value
forall a.
(Generic a, GToJSON' Value Zero (Rep a)) =>
Options -> a -> Value
GAeson.genericToJSON Options
GAeson.opt
  toEncoding :: RequestMessage -> Encoding
toEncoding = Options -> RequestMessage -> Encoding
forall a.
(Generic a, GToJSON' Encoding Zero (Rep a)) =>
Options -> a -> Encoding
GAeson.genericToEncoding Options
GAeson.opt

instance FromJSON RequestMessage where
  parseJSON :: Value -> Parser RequestMessage
parseJSON = Options -> Value -> Parser RequestMessage
forall a.
(Generic a, GFromJSON Zero (Rep a)) =>
Options -> Value -> Parser a
GAeson.genericParseJSON Options
GAeson.opt

-- | Convert an 'Operation' object to 'RequestMessage'.
toRequestMessage :: Operation o => UUID -> o -> RequestMessage
toRequestMessage :: UUID -> o -> RequestMessage
toRequestMessage UUID
rid o
o =
  RequestMessage :: UUID -> Text -> Text -> Object -> RequestMessage
RequestMessage { $sel:requestId:RequestMessage :: UUID
requestId = UUID
rid,
                   $sel:op:RequestMessage :: Text
op = o -> Text
forall o. Operation o => o -> Text
opName o
o,
                   $sel:processor:RequestMessage :: Text
processor = o -> Text
forall o. Operation o => o -> Text
opProcessor o
o,
                   $sel:args:RequestMessage :: Object
args = o -> Object
forall o. Operation o => o -> Object
opArgs o
o
                 }

-- | Create a 'RequestMessage' from an 'Operation' object. The
-- 'requestId' is generated by the random number generator of the
-- system.
makeRequestMessage :: Operation o => o -> IO RequestMessage
makeRequestMessage :: o -> IO RequestMessage
makeRequestMessage o
o = UUID -> o -> RequestMessage
forall o. Operation o => UUID -> o -> RequestMessage
toRequestMessage (UUID -> o -> RequestMessage)
-> IO UUID -> IO (o -> RequestMessage)
forall (f :: * -> *) a b. Functor f => (a -> b) -> f a -> f b
<$> IO UUID
nextRandom IO (o -> RequestMessage) -> IO o -> IO RequestMessage
forall (f :: * -> *) a b. Applicative f => f (a -> b) -> f a -> f b
<*> o -> IO o
forall (f :: * -> *) a. Applicative f => a -> f a
pure o
o