{-# LANGUAGE ScopedTypeVariables #-}

module Cachix.Client.Retry
  ( retryAll,
    retryAllWithLogging,
    endlessRetryPolicy,
    endlessConstantRetryPolicy,
  )
where

import Control.Exception.Safe (Handler (..), MonadMask, isSyncException)
import Control.Retry (RetryPolicy, RetryPolicyM, RetryStatus, constantDelay, exponentialBackoff, limitRetries, logRetries, recoverAll, recovering, skipAsyncExceptions)
import Protolude hiding (Handler (..))

retryAll :: (MonadIO m, MonadMask m) => (RetryStatus -> m a) -> m a
retryAll =
  recoverAll defaultRetryPolicy

-- Catches all exceptions except async exceptions with logging support
retryAllWithLogging :: (MonadIO m, MonadMask m) => RetryPolicyM m -> (Bool -> SomeException -> RetryStatus -> m ()) -> m a -> m a
retryAllWithLogging retryPolicy logger action = recovering retryPolicy handlers $ const action
  where
    handlers = skipAsyncExceptions ++ [exitSuccessHandler, loggingHandler]
    exitSuccessHandler :: MonadIO m => RetryStatus -> Handler m Bool
    exitSuccessHandler _ = Handler $ \(_ :: ExitCode) -> return False
    loggingHandler = logRetries exceptionPredicate logger
    exceptionPredicate = return . isSyncException

defaultRetryPolicy :: RetryPolicy
defaultRetryPolicy =
  exponentialBackoff (1000 * 1000) <> limitRetries 5

endlessConstantRetryPolicy :: RetryPolicy
endlessConstantRetryPolicy =
  constantDelay (1000 * 1000)

endlessRetryPolicy :: RetryPolicy
endlessRetryPolicy =
  exponentialBackoff (1000 * 1000)