Config.hs

module Network.HTTP3.Config where

import Network.HTTP2.Internal
import Network.HTTP3.Frame
import Network.QUIC (Stream)
import qualified System.TimeManager as T

-- | Hooks mainly for error testing.
data Hooks = Hooks {
    onControlFrameCreated :: [H3Frame] -> [H3Frame]
  , onHeadersFrameCreated :: [H3Frame] -> [H3Frame]
  , onControlStreamCreated :: Stream -> IO ()
  , onEncoderStreamCreated :: Stream -> IO ()
  , onDecoderStreamCreated :: Stream -> IO ()
  }

-- | Default hooks.
defaultHooks :: Hooks
defaultHooks = Hooks {
    onControlFrameCreated = id
  , onHeadersFrameCreated = id
  , onControlStreamCreated = \_ -> return ()
  , onEncoderStreamCreated = \_ -> return ()
  , onDecoderStreamCreated = \_ -> return ()
  }

-- | Configuration for HTTP\/3 or HQ.
data Config = Config {
    confHooks :: Hooks
  , confPositionReadMaker :: PositionReadMaker
  , confTimeoutManager :: T.Manager
  }

-- | Allocating a simple configuration with a handle-based position
--   reader and a locally allocated timeout manager.
allocSimpleConfig :: IO Config
allocSimpleConfig = Config defaultHooks defaultPositionReadMaker <$> T.initialize (30 * 1000000)

-- | Freeing a simple configration.
freeSimpleConfig :: Config -> IO ()
freeSimpleConfig conf = T.killManager $ confTimeoutManager conf