module Effectful.Zoo.Environment
  ( -- * Effect
    Environment,

    -- ** Handlers
    E.runEnvironment,

    -- * Querying the environment
    E.getArgs,
    E.getProgName,
    E.getExecutablePath,
    E.getEnv,
    E.getEnvironment,
    lookupEnv,
    lookupEnvMaybe,
    lookupMapEnv,
    lookupParseMaybeEnv,
    lookupParseEitherEnv,

    -- * Modifying the environment
    E.setEnv,
    E.unsetEnv,
    E.withArgs,
    E.withProgName,
  )
  where

import Data.Text qualified as T
import Effectful
import Effectful.Environment (Environment)
import Effectful.Environment qualified as E
import Effectful.Zoo.Core
import Effectful.Zoo.Error.Static
import Effectful.Zoo.Errors.EnvironmentVariableInvalid
import Effectful.Zoo.Errors.EnvironmentVariableMissing
import HaskellWorks.Prelude

lookupEnv :: ()
  => r <: Environment
  => r <: Error EnvironmentVariableMissing
  => Text
  -> Eff r Text
lookupEnv envName =
  lookupEnvMaybe envName
    & onNothingM (throw $ EnvironmentVariableMissing envName)

lookupMapEnv :: ()
  => r <: Environment
  => r <: Error EnvironmentVariableMissing
  => Text
  -> (Text -> a)
  -> Eff r a
lookupMapEnv envName f =
  f <$> lookupEnv envName

lookupParseMaybeEnv :: ()
  => r <: Environment
  => r <: Error EnvironmentVariableInvalid
  => r <: Error EnvironmentVariableMissing
  => Text
  -> (Text -> Maybe a)
  -> Eff r a
lookupParseMaybeEnv envName parse = do
  text <- lookupEnv envName

  parse text
    & onNothing (throw $ EnvironmentVariableInvalid envName text Nothing)

lookupParseEitherEnv :: ()
  => r <: Environment
  => r <: Error EnvironmentVariableInvalid
  => r <: Error EnvironmentVariableMissing
  => Text
  -> (Text -> Either Text a)
  -> Eff r a
lookupParseEitherEnv envName parse = do
  text <- lookupEnv envName

  parse text
    & onLeft (throw . EnvironmentVariableInvalid envName text . Just)

lookupEnvMaybe :: ()
  => r <: Environment
  => Text
  -> Eff r (Maybe Text)
lookupEnvMaybe envName = do
  value <- E.lookupEnv $ T.unpack envName
  pure $ T.pack <$> value