-- 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 Network.Wai.Utilities.Headers where

import Data.ByteString
import Data.ByteString.Conversion (FromByteString (..), ToByteString (..), fromByteString', toByteString')
import Data.OpenApi.ParamSchema (ToParamSchema (..))
import Data.Text as T hiding (show)
import Data.Text.Encoding
import Data.Text.Encoding.Error
import Imports
import Servant (FromHttpApiData (..), Proxy (Proxy), ToHttpApiData (..))

data CacheControl = NoStore
  deriving (Eq, Show, Generic)

instance ToByteString CacheControl where
  builder NoStore = "no-store"

instance FromByteString CacheControl where
  parser = do
    t :: Text <- parser
    case t & T.toLower of
      "no-store" -> pure NoStore
      _ -> fail $ "Invalid CacheControl type: " ++ show t

instance ToHttpApiData CacheControl where
  toQueryParam = decodeUtf8With lenientDecode . toByteString'

instance FromHttpApiData CacheControl where
  parseQueryParam =
    maybe (Left "Invalid CacheControl") Right
      . fromByteString'
      . fromStrict
      . encodeUtf8

instance ToParamSchema CacheControl where
  toParamSchema _ = toParamSchema (Proxy @Text)
