-- This file is part of the Wire Server implementation.
--
-- Copyright (C) 2022 Wire Swiss GmbH <opensource@wire.com>
--
-- This program is free software: you can redistribute it and/or modify it under
-- the terms of the GNU Affero General Public License as published by the Free
-- Software Foundation, either version 3 of the License, or (at your option) any
-- later version.
--
-- This program is distributed in the hope that it will be useful, but WITHOUT
-- ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
-- FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License for more
-- details.
--
-- You should have received a copy of the GNU Affero General Public License along
-- with this program. If not, see <https://www.gnu.org/licenses/>.

module Wire.API.MLS.Servant (MLS, mimeUnrenderMLSWith) where

import Data.Bifunctor
import Data.Binary
import Data.Text qualified as T
import Imports
import Network.HTTP.Media ((//))
import Servant.API hiding (Get)
import Wire.API.MLS.Serialisation

data MLS

instance Accept MLS where
  contentType :: Proxy MLS -> MediaType
contentType Proxy MLS
_ = ByteString
"message" ByteString -> ByteString -> MediaType
// ByteString
"mls"

instance {-# OVERLAPPABLE #-} (ParseMLS a) => MimeUnrender MLS a where
  mimeUnrender :: Proxy MLS -> ByteString -> Either String a
mimeUnrender Proxy MLS
_ = Get a -> ByteString -> Either String a
forall a. Get a -> ByteString -> Either String a
mimeUnrenderMLSWith Get a
forall a. ParseMLS a => Get a
parseMLS

instance {-# OVERLAPPABLE #-} (SerialiseMLS a) => MimeRender MLS a where
  mimeRender :: Proxy MLS -> a -> ByteString
mimeRender Proxy MLS
_ = a -> ByteString
forall a. SerialiseMLS a => a -> ByteString
encodeMLS

mimeUnrenderMLSWith :: Get a -> LByteString -> Either String a
mimeUnrenderMLSWith :: forall a. Get a -> ByteString -> Either String a
mimeUnrenderMLSWith Get a
p = (Text -> String) -> Either Text a -> Either String a
forall a b c. (a -> b) -> Either a c -> Either b c
forall (p :: * -> * -> *) a b c.
Bifunctor p =>
(a -> b) -> p a c -> p b c
first Text -> String
T.unpack (Either Text a -> Either String a)
-> (ByteString -> Either Text a) -> ByteString -> Either String a
forall b c a. (b -> c) -> (a -> b) -> a -> c
. Get a -> ByteString -> Either Text a
forall a. Get a -> ByteString -> Either Text a
decodeMLSWith Get a
p