{-# OPTIONS_GHC -Wno-orphans #-}

-- | OpenApi implementation of 'Header' trait.
module WebGear.OpenApi.Trait.Header () where

import Control.Lens ((&), (.~), (?~))
import Data.OpenApi hiding (Response)
import Data.Proxy (Proxy (Proxy))
import Data.String (fromString)
import Data.Text (Text)
import GHC.TypeLits (KnownSymbol, symbolVal)
import WebGear.Core.Modifiers (Existence (..))
import WebGear.Core.Request (Request)
import WebGear.Core.Response (Response)
import WebGear.Core.Trait (Get (..), Set (..), TraitAbsence)
import qualified WebGear.Core.Trait.Header as WG
import WebGear.OpenApi.Handler (DocNode (..), OpenApiHandler (..), nullNode, singletonNode)

mkParam ::
  forall name val.
  (KnownSymbol name, ToSchema val) =>
  Proxy name ->
  Proxy val ->
  Bool ->
  Param
mkParam :: forall (name :: Symbol) val.
(KnownSymbol name, ToSchema val) =>
Proxy name -> Proxy val -> Bool -> Param
mkParam Proxy name
_ Proxy val
_ Bool
isRequired =
  (Param
forall a. Monoid a => a
mempty :: Param)
    Param -> (Param -> Param) -> Param
forall a b. a -> (a -> b) -> b
& (Text -> Identity Text) -> Param -> Identity Param
forall s a. HasName s a => Lens' s a
Lens' Param Text
name ((Text -> Identity Text) -> Param -> Identity Param)
-> Text -> Param -> Param
forall s t a b. ASetter s t a b -> b -> s -> t
.~ forall a. IsString a => String -> a
fromString @Text (Proxy name -> String
forall (n :: Symbol) (proxy :: Symbol -> *).
KnownSymbol n =>
proxy n -> String
symbolVal (Proxy name -> String) -> Proxy name -> String
forall a b. (a -> b) -> a -> b
$ forall {k} (t :: k). Proxy t
forall (t :: Symbol). Proxy t
Proxy @name)
    Param -> (Param -> Param) -> Param
forall a b. a -> (a -> b) -> b
& (ParamLocation -> Identity ParamLocation)
-> Param -> Identity Param
forall s a. HasIn s a => Lens' s a
Lens' Param ParamLocation
in_ ((ParamLocation -> Identity ParamLocation)
 -> Param -> Identity Param)
-> ParamLocation -> Param -> Param
forall s t a b. ASetter s t a b -> b -> s -> t
.~ ParamLocation
ParamHeader
    Param -> (Param -> Param) -> Param
forall a b. a -> (a -> b) -> b
& (Maybe Bool -> Identity (Maybe Bool)) -> Param -> Identity Param
forall s a. HasRequired s a => Lens' s a
Lens' Param (Maybe Bool)
required ((Maybe Bool -> Identity (Maybe Bool)) -> Param -> Identity Param)
-> Bool -> Param -> Param
forall s t a b. ASetter s t a (Maybe b) -> b -> s -> t
?~ Bool
isRequired
    Param -> (Param -> Param) -> Param
forall a b. a -> (a -> b) -> b
& (Maybe (Referenced Schema) -> Identity (Maybe (Referenced Schema)))
-> Param -> Identity Param
forall s a. HasSchema s a => Lens' s a
Lens' Param (Maybe (Referenced Schema))
schema ((Maybe (Referenced Schema)
  -> Identity (Maybe (Referenced Schema)))
 -> Param -> Identity Param)
-> Referenced Schema -> Param -> Param
forall s t a b. ASetter s t a (Maybe b) -> b -> s -> t
?~ Schema -> Referenced Schema
forall a. a -> Referenced a
Inline (Proxy val -> Schema
forall a. ToSchema a => Proxy a -> Schema
toSchema (Proxy val -> Schema) -> Proxy val -> Schema
forall a b. (a -> b) -> a -> b
$ forall t. Proxy t
forall {k} (t :: k). Proxy t
Proxy @val)

instance (KnownSymbol name, ToSchema val, TraitAbsence (WG.RequestHeader Required ps name val) Request) => Get (OpenApiHandler m) (WG.RequestHeader Required ps name val) Request where
  {-# INLINE getTrait #-}
  getTrait :: forall (ts :: [*]).
Prerequisite (RequestHeader 'Required ps name val) ts Request =>
RequestHeader 'Required ps name val
-> OpenApiHandler
     m
     (With Request ts)
     (Either
        (Absence (RequestHeader 'Required ps name val) Request)
        (Attribute (RequestHeader 'Required ps name val) Request))
getTrait RequestHeader 'Required ps name val
WG.RequestHeader =
    Tree DocNode
-> OpenApiHandler
     m
     (With Request ts)
     (Either
        (Absence (RequestHeader 'Required ps name val) Request)
        (Attribute (RequestHeader 'Required ps name val) Request))
forall {k} {k1} {k2} (m :: k) (a :: k1) (b :: k2).
Tree DocNode -> OpenApiHandler m a b
OpenApiHandler (Tree DocNode
 -> OpenApiHandler
      m
      (With Request ts)
      (Either
         (Absence (RequestHeader 'Required ps name val) Request)
         (Attribute (RequestHeader 'Required ps name val) Request)))
-> Tree DocNode
-> OpenApiHandler
     m
     (With Request ts)
     (Either
        (Absence (RequestHeader 'Required ps name val) Request)
        (Attribute (RequestHeader 'Required ps name val) Request))
forall a b. (a -> b) -> a -> b
$ DocNode -> Tree DocNode
forall a. a -> Tree a
singletonNode (Param -> DocNode
DocRequestHeader (Param -> DocNode) -> Param -> DocNode
forall a b. (a -> b) -> a -> b
$ Proxy name -> Proxy val -> Bool -> Param
forall (name :: Symbol) val.
(KnownSymbol name, ToSchema val) =>
Proxy name -> Proxy val -> Bool -> Param
mkParam (forall {k} (t :: k). Proxy t
forall (t :: Symbol). Proxy t
Proxy @name) (forall t. Proxy t
forall {k} (t :: k). Proxy t
Proxy @val) Bool
True)

instance (KnownSymbol name, ToSchema val, TraitAbsence (WG.RequestHeader Optional ps name val) Request) => Get (OpenApiHandler m) (WG.RequestHeader Optional ps name val) Request where
  {-# INLINE getTrait #-}
  getTrait :: forall (ts :: [*]).
Prerequisite (RequestHeader 'Optional ps name val) ts Request =>
RequestHeader 'Optional ps name val
-> OpenApiHandler
     m
     (With Request ts)
     (Either
        (Absence (RequestHeader 'Optional ps name val) Request)
        (Attribute (RequestHeader 'Optional ps name val) Request))
getTrait RequestHeader 'Optional ps name val
WG.RequestHeader =
    Tree DocNode
-> OpenApiHandler
     m
     (With Request ts)
     (Either
        (Absence (RequestHeader 'Optional ps name val) Request)
        (Attribute (RequestHeader 'Optional ps name val) Request))
forall {k} {k1} {k2} (m :: k) (a :: k1) (b :: k2).
Tree DocNode -> OpenApiHandler m a b
OpenApiHandler (Tree DocNode
 -> OpenApiHandler
      m
      (With Request ts)
      (Either
         (Absence (RequestHeader 'Optional ps name val) Request)
         (Attribute (RequestHeader 'Optional ps name val) Request)))
-> Tree DocNode
-> OpenApiHandler
     m
     (With Request ts)
     (Either
        (Absence (RequestHeader 'Optional ps name val) Request)
        (Attribute (RequestHeader 'Optional ps name val) Request))
forall a b. (a -> b) -> a -> b
$ DocNode -> Tree DocNode
forall a. a -> Tree a
singletonNode (Param -> DocNode
DocRequestHeader (Param -> DocNode) -> Param -> DocNode
forall a b. (a -> b) -> a -> b
$ Proxy name -> Proxy val -> Bool -> Param
forall (name :: Symbol) val.
(KnownSymbol name, ToSchema val) =>
Proxy name -> Proxy val -> Bool -> Param
mkParam (forall {k} (t :: k). Proxy t
forall (t :: Symbol). Proxy t
Proxy @name) (forall t. Proxy t
forall {k} (t :: k). Proxy t
Proxy @val) Bool
False)

instance (KnownSymbol name, ToSchema val) => Set (OpenApiHandler m) (WG.ResponseHeader Required name val) Response where
  {-# INLINE setTrait #-}
  setTrait :: forall (ts :: [*]).
ResponseHeader 'Required name val
-> (With Response ts
    -> Response
    -> Attribute (ResponseHeader 'Required name val) Response
    -> With Response (ResponseHeader 'Required name val : ts))
-> OpenApiHandler
     m
     (With Response ts,
      Attribute (ResponseHeader 'Required name val) Response)
     (With Response (ResponseHeader 'Required name val : ts))
setTrait ResponseHeader 'Required name val
WG.ResponseHeader With Response ts
-> Response
-> Attribute (ResponseHeader 'Required name val) Response
-> With Response (ResponseHeader 'Required name val : ts)
_ =
    let headerName :: Text
headerName = String -> Text
forall a. IsString a => String -> a
fromString (String -> Text) -> String -> Text
forall a b. (a -> b) -> a -> b
$ Proxy name -> String
forall (n :: Symbol) (proxy :: Symbol -> *).
KnownSymbol n =>
proxy n -> String
symbolVal (Proxy name -> String) -> Proxy name -> String
forall a b. (a -> b) -> a -> b
$ forall {k} (t :: k). Proxy t
forall (t :: Symbol). Proxy t
Proxy @name
        header :: Header
header =
          forall a. Monoid a => a
mempty @Header
            Header -> (Header -> Header) -> Header
forall a b. a -> (a -> b) -> b
& (Maybe Bool -> Identity (Maybe Bool)) -> Header -> Identity Header
forall s a. HasRequired s a => Lens' s a
Lens' Header (Maybe Bool)
required ((Maybe Bool -> Identity (Maybe Bool))
 -> Header -> Identity Header)
-> Bool -> Header -> Header
forall s t a b. ASetter s t a (Maybe b) -> b -> s -> t
?~ Bool
True
            Header -> (Header -> Header) -> Header
forall a b. a -> (a -> b) -> b
& (Maybe (Referenced Schema) -> Identity (Maybe (Referenced Schema)))
-> Header -> Identity Header
forall s a. HasSchema s a => Lens' s a
Lens' Header (Maybe (Referenced Schema))
schema ((Maybe (Referenced Schema)
  -> Identity (Maybe (Referenced Schema)))
 -> Header -> Identity Header)
-> Referenced Schema -> Header -> Header
forall s t a b. ASetter s t a (Maybe b) -> b -> s -> t
?~ Schema -> Referenced Schema
forall a. a -> Referenced a
Inline (Proxy val -> Schema
forall a. ToSchema a => Proxy a -> Schema
toSchema (Proxy val -> Schema) -> Proxy val -> Schema
forall a b. (a -> b) -> a -> b
$ forall t. Proxy t
forall {k} (t :: k). Proxy t
Proxy @val)
     in if Text
headerName Text -> Text -> Bool
forall a. Eq a => a -> a -> Bool
== Text
"Content-Type"
          then Tree DocNode
-> OpenApiHandler
     m
     (With Response ts, val)
     (With Response (ResponseHeader 'Required name val : ts))
forall {k} {k1} {k2} (m :: k) (a :: k1) (b :: k2).
Tree DocNode -> OpenApiHandler m a b
OpenApiHandler Tree DocNode
forall a. Tree a
nullNode
          else Tree DocNode
-> OpenApiHandler
     m
     (With Response ts,
      Attribute (ResponseHeader 'Required name val) Response)
     (With Response (ResponseHeader 'Required name val : ts))
forall {k} {k1} {k2} (m :: k) (a :: k1) (b :: k2).
Tree DocNode -> OpenApiHandler m a b
OpenApiHandler (Tree DocNode
 -> OpenApiHandler
      m
      (With Response ts,
       Attribute (ResponseHeader 'Required name val) Response)
      (With Response (ResponseHeader 'Required name val : ts)))
-> Tree DocNode
-> OpenApiHandler
     m
     (With Response ts,
      Attribute (ResponseHeader 'Required name val) Response)
     (With Response (ResponseHeader 'Required name val : ts))
forall a b. (a -> b) -> a -> b
$ DocNode -> Tree DocNode
forall a. a -> Tree a
singletonNode (Text -> Header -> DocNode
DocResponseHeader Text
headerName Header
header)

instance (KnownSymbol name, ToSchema val) => Set (OpenApiHandler m) (WG.ResponseHeader Optional name val) Response where
  {-# INLINE setTrait #-}
  setTrait :: forall (ts :: [*]).
ResponseHeader 'Optional name val
-> (With Response ts
    -> Response
    -> Attribute (ResponseHeader 'Optional name val) Response
    -> With Response (ResponseHeader 'Optional name val : ts))
-> OpenApiHandler
     m
     (With Response ts,
      Attribute (ResponseHeader 'Optional name val) Response)
     (With Response (ResponseHeader 'Optional name val : ts))
setTrait ResponseHeader 'Optional name val
WG.ResponseHeader With Response ts
-> Response
-> Attribute (ResponseHeader 'Optional name val) Response
-> With Response (ResponseHeader 'Optional name val : ts)
_ =
    let headerName :: Text
headerName = String -> Text
forall a. IsString a => String -> a
fromString (String -> Text) -> String -> Text
forall a b. (a -> b) -> a -> b
$ Proxy name -> String
forall (n :: Symbol) (proxy :: Symbol -> *).
KnownSymbol n =>
proxy n -> String
symbolVal (Proxy name -> String) -> Proxy name -> String
forall a b. (a -> b) -> a -> b
$ forall {k} (t :: k). Proxy t
forall (t :: Symbol). Proxy t
Proxy @name
        header :: Header
header =
          forall a. Monoid a => a
mempty @Header
            Header -> (Header -> Header) -> Header
forall a b. a -> (a -> b) -> b
& (Maybe Bool -> Identity (Maybe Bool)) -> Header -> Identity Header
forall s a. HasRequired s a => Lens' s a
Lens' Header (Maybe Bool)
required ((Maybe Bool -> Identity (Maybe Bool))
 -> Header -> Identity Header)
-> Bool -> Header -> Header
forall s t a b. ASetter s t a (Maybe b) -> b -> s -> t
?~ Bool
True
            Header -> (Header -> Header) -> Header
forall a b. a -> (a -> b) -> b
& (Maybe (Referenced Schema) -> Identity (Maybe (Referenced Schema)))
-> Header -> Identity Header
forall s a. HasSchema s a => Lens' s a
Lens' Header (Maybe (Referenced Schema))
schema ((Maybe (Referenced Schema)
  -> Identity (Maybe (Referenced Schema)))
 -> Header -> Identity Header)
-> Referenced Schema -> Header -> Header
forall s t a b. ASetter s t a (Maybe b) -> b -> s -> t
?~ Schema -> Referenced Schema
forall a. a -> Referenced a
Inline (Proxy val -> Schema
forall a. ToSchema a => Proxy a -> Schema
toSchema (Proxy val -> Schema) -> Proxy val -> Schema
forall a b. (a -> b) -> a -> b
$ forall t. Proxy t
forall {k} (t :: k). Proxy t
Proxy @val)
     in Tree DocNode
-> OpenApiHandler
     m
     (With Response ts,
      Attribute (ResponseHeader 'Optional name val) Response)
     (With Response (ResponseHeader 'Optional name val : ts))
forall {k} {k1} {k2} (m :: k) (a :: k1) (b :: k2).
Tree DocNode -> OpenApiHandler m a b
OpenApiHandler (Tree DocNode
 -> OpenApiHandler
      m
      (With Response ts,
       Attribute (ResponseHeader 'Optional name val) Response)
      (With Response (ResponseHeader 'Optional name val : ts)))
-> Tree DocNode
-> OpenApiHandler
     m
     (With Response ts,
      Attribute (ResponseHeader 'Optional name val) Response)
     (With Response (ResponseHeader 'Optional name val : ts))
forall a b. (a -> b) -> a -> b
$ DocNode -> Tree DocNode
forall a. a -> Tree a
singletonNode (Text -> Header -> DocNode
DocResponseHeader Text
headerName Header
header)