{-# LANGUAGE DerivingVia #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE FunctionalDependencies #-}
{-# LANGUAGE KindSignatures #-}
{-# LANGUAGE StandaloneDeriving #-}
{-# LANGUAGE Trustworthy #-}
{-# LANGUAGE UndecidableInstances #-}
-- Later GHCs infer DerivingVia as not Safe
-- We just downgrade to Trustworthy and go fish
{-# OPTIONS_GHC -Wno-trustworthy-safe #-}

-- | Module: Control.Monad.Select
-- Copyright: (C) Koz Ross 2022
-- License: BSD-3-Clause (see the LICENSE file)
-- Maintainer: koz.ross@retro-freedom.nz
-- Stability: Experimental
-- Portability: GHC only
--
-- [Computation type:] Backtracking search, with @r@ as a \'ranking\' or
-- \'evaluation\' type.
--
-- [Binding strategy:] Binding a function to a monadic value \'chains together\'
-- strategies; having seen the result of one search, decide which policy to use
-- to continue.
--
-- [Useful for:] Search problems.
--
-- [Zero and plus:] None.
--
-- [Example type:] @'Control.Monad.Trans.Select.Select' r a@
--
-- = A note on commutativity
--
-- Some effects are /commutative/: it doesn't matter which you resolve first, as
-- all possible orderings of commutative effects are isomorphic. Consider, for
-- example, the reader and state effects, as exemplified by 'ReaderT' and
-- 'StrictState.StateT' respectively. If we have
-- @'ReaderT' r ('StrictState.State' s) a@, this is
-- effectively @r -> 'StrictState.State' s a ~ r -> s -> (a, s)@; if we instead have
-- @'StrictState.StateT' s ('Control.Monad.Trans.Reader.Reader' r) a@, this is effectively
-- @s -> 'Control.Monad.Trans.Reader' r (a, s) ~ s -> r -> (a, s)@. Since we
-- can always reorder function arguments (for example, using 'flip', as in
-- this case) without changing the result, these are
-- isomorphic, showing that reader and state are /commutative/, or, more
-- precisely, /commute with each other/.
--
-- However, this isn't generally the case. Consider instead the error and state
-- effects, as exemplified by 'MaybeT' and 'StrictState.StateT' respectively.
-- If we have @'MaybeT' ('Control.Monad.Trans.State.Strict.State' s) a@, this
-- is effectively @'State' s ('Maybe' a) ~ s -> ('Maybe' a, s)@: put simply,
-- the error can occur only in the /result/, but
-- not the state, which always \'survives\'. On the other hand, if we have
-- @'StrictState.StateT' s 'Maybe' a@, this is instead @s -> 'Maybe' (a, s)@: here,
-- if we error, we lose /both/ the state and the result! Thus, error and state effects
-- do /not/ commute with each other.
--
-- As the MTL is capability-based, we support any ordering of non-commutative
-- effects on an equal footing. Indeed, if you wish to use
-- 'Control.Monad.State.Class.MonadState', for
-- example, whether your final monadic stack ends up being @'MaybeT'
-- ('Control.Monad.Trans.State.Strict.State' s)
-- a@, @'StrictState.StateT' s 'Maybe' a@, or anything else, you will be able to write your
-- desired code without having to consider such differences. However, the way we
-- /implement/ these capabilities for any given transformer (or rather, any
-- given transformed stack) /is/ affected by this ordering unless the effects in
-- question are commutative.
--
-- We note in this module which effects the accumulation effect does and doesn't
-- commute with; we also note on implementations with non-commutative
-- transformers what the outcome will be. Note that, depending on how the
-- \'inner monad\' is structured, this may be more complex than we note: we
-- describe only what impact the \'outer effect\' has, not what else might be in
-- the stack.
--
-- = Commutativity of selection
--
-- The selection effect commutes with the identity effect ('IdentityT'), but
-- nothing else.
module Control.Monad.Select
  ( -- * Type class
    MonadSelect (..),

    -- * Lifting helper type
    LiftingSelect (..),
  )
where

import Control.Monad.Trans.Accum (AccumT)
import Control.Monad.Trans.Class (MonadTrans (lift))
import Control.Monad.Trans.Cont (ContT)
import Control.Monad.Trans.Except (ExceptT)
import Control.Monad.Trans.Identity (IdentityT)
import Control.Monad.Trans.Maybe (MaybeT)
import qualified Control.Monad.Trans.RWS.CPS as CPSRWS
import qualified Control.Monad.Trans.RWS.Lazy as LazyRWS
import qualified Control.Monad.Trans.RWS.Strict as StrictRWS
import Control.Monad.Trans.Reader (ReaderT)
import Control.Monad.Trans.Select (SelectT)
import qualified Control.Monad.Trans.Select as Select
import qualified Control.Monad.Trans.State.Lazy as LazyState
import qualified Control.Monad.Trans.State.Strict as StrictState
import qualified Control.Monad.Trans.Writer.CPS as CPSWriter
import qualified Control.Monad.Trans.Writer.Lazy as LazyWriter
import qualified Control.Monad.Trans.Writer.Strict as StrictWriter
import Data.Functor.Identity (Identity)
import Data.Kind (Type)

-- | The capability to search with backtracking. Essentially describes a
-- \'policy function\': given the state of the search (and a \'ranking\' or
-- \'evaluation\' of each possible result so far), pick the result that's
-- currently best.
--
-- = Laws
--
-- Any instance of 'MonadSelect' must follow these laws:
--
-- * @'select' ('const' x)@ @=@ @'pure' x@
-- * @'select' f '*>' 'select' g@ @=@ @'select' g@
--
-- @since 2.3
class (Monad m) => MonadSelect r m | m -> r where
  select :: ((a -> r) -> a) -> m a

-- | @since 2.3
instance MonadSelect r (SelectT r Identity) where
  select :: forall a. ((a -> r) -> a) -> SelectT r Identity a
select = forall a r. ((a -> r) -> a) -> Select r a
Select.select

-- | \'Extends\' the possibilities considered by @m@ to include 'Nothing'; this
-- means that 'Nothing' gains a \'rank\' (namely, a value of @r@), and the
-- potential result could also be 'Nothing'.
--
-- @since 2.3
deriving via
  (LiftingSelect MaybeT m)
  instance
    (MonadSelect r m) =>
    MonadSelect r (MaybeT m)

-- | The continuation describes a way of choosing a \'search\' or \'ranking\'
-- strategy for @r@, based on a \'ranking\' using @r'@, given any @a@. We then
-- get a \'search\' strategy for @r@.
--
-- @since 2.3
deriving via
  (LiftingSelect (ContT r) m)
  instance
    (MonadSelect r' m) =>
    MonadSelect r' (ContT r m)

-- | \'Extends\' the possibilities considered by @m@ to include every value of
-- @e@; this means that the potential result could be either a 'Left' (making it
-- a choice of type @e@) or a 'Right' (making it a choice of type @a@).
--
-- @since 2.3
deriving via
  (LiftingSelect (ExceptT e) m)
  instance
    (MonadSelect r m) =>
    MonadSelect r (ExceptT e m)

-- | @since 2.3
deriving via
  (LiftingSelect IdentityT m)
  instance
    (MonadSelect r m) =>
    MonadSelect r (IdentityT m)

-- | Provides a read-only environment of type @r@ to the \'strategy\' function.
-- However, the \'ranking\' function (or more accurately, representation) has no
-- access to @r@. Put another way, you can influence what values get chosen by
-- changing @r@, but not how solutions are ranked.
--
-- @since 2.3
deriving via
  (LiftingSelect (ReaderT r) m)
  instance
    (MonadSelect r' m) =>
    MonadSelect r' (ReaderT r m)

-- | \'Readerizes\' the state: the \'ranking\' function can /see/ a value of
-- type @s@, but not modify it. Effectively, can be thought of as \'extending\'
-- the \'ranking\' by all values in @s@, but /which/ @s@ gets given to any rank
-- calls is predetermined by the \'outer state\' (and cannot change).
--
-- @since 2.3
deriving via
  (LiftingSelect (LazyState.StateT s) m)
  instance
    (MonadSelect w m) =>
    MonadSelect w (LazyState.StateT s m)

-- | \'Readerizes\' the state: the \'ranking\' function can /see/ a value of
-- type @s@, but not modify it. Effectively, can be thought of as \'extending\'
-- the \'ranking\' by all values in @s@, but /which/ @s@ gets given to any rank
-- calls is predetermined by the \'outer state\' (and cannot change).
--
-- @since 2.3
deriving via
  (LiftingSelect (StrictState.StateT s) m)
  instance
    (MonadSelect w m) =>
    MonadSelect w (StrictState.StateT s m)

-- | \'Readerizes\' the writer: the \'ranking\' function can see the value
-- that's been accumulated (of type @w@), but can't add anything to the log.
-- Effectively, can be thought of as \'extending\' the \'ranking\' by all values
-- of @w@, but /which/ @w@ gets given to any rank calls is predetermined by the
-- \'outer writer\' (and cannot change).
--
-- @since 2.3
deriving via
  (LiftingSelect (CPSWriter.WriterT w) m)
  instance
    (MonadSelect w' m) =>
    MonadSelect w' (CPSWriter.WriterT w m)

-- | \'Readerizes\' the writer: the \'ranking\' function can see the value
-- that's been accumulated (of type @w@), but can't add anything to the log.
-- Effectively, can be thought of as \'extending\' the \'ranking\' by all values
-- of @w@, but /which/ @w@ gets given to any rank calls is predetermined by the
-- \'outer writer\' (and cannot change).
--
-- @since 2.3
deriving via
  (LiftingSelect (LazyWriter.WriterT w) m)
  instance
    (MonadSelect w' m, Monoid w) =>
    MonadSelect w' (LazyWriter.WriterT w m)

-- | \'Readerizes\' the writer: the \'ranking\' function can see the value
-- that's been accumulated (of type @w@), but can't add anything to the log.
-- Effectively, can be thought of as \'extending\' the \'ranking\' by all values
-- of @w@, but /which/ @w@ gets given to any rank calls is predetermined by the
-- \'outer writer\' (and cannot change).
--
-- @since 2.3
deriving via
  (LiftingSelect (StrictWriter.WriterT w) m)
  instance
    (MonadSelect w' m, Monoid w) =>
    MonadSelect w' (StrictWriter.WriterT w m)

-- | A combination of an \'outer\' 'ReaderT', 'WriterT' and 'StateT'. In short,
-- you get a value of type @r@ which can influence what gets picked, but not how
-- anything is ranked, and the \'ranking\' function gets access to an @s@ and a
-- @w@, but can modify neither.
--
-- @since 2.3
deriving via
  (LiftingSelect (CPSRWS.RWST r w s) m)
  instance
    (MonadSelect w' m) =>
    MonadSelect w' (CPSRWS.RWST r w s m)

-- | A combination of an \'outer\' 'ReaderT', 'WriterT' and 'StateT'. In short,
-- you get a value of type @r@ which can influence what gets picked, but not how
-- anything is ranked, and the \'ranking\' function gets access to an @s@ and a
-- @w@, but can modify neither.
--
-- @since 2.3
deriving via
  (LiftingSelect (LazyRWS.RWST r w s) m)
  instance
    (MonadSelect w' m, Monoid w) =>
    MonadSelect w' (LazyRWS.RWST r w s m)

-- | A combination of an \'outer\' 'ReaderT', 'WriterT' and 'StateT'. In short,
-- you get a value of type @r@ which can influence what gets picked, but not how
-- anything is ranked, and the \'ranking\' function gets access to an @s@ and a
-- @w@, but can modify neither.
--
-- @since 2.3
deriving via
  (LiftingSelect (StrictRWS.RWST r w s) m)
  instance
    (MonadSelect w' m, Monoid w) =>
    MonadSelect w' (StrictRWS.RWST r w s m)

-- | \'Readerizes\' the accumulator: the \'ranking\' function can see the value
-- that has been accumulated (of type @w@), but can't add anything to it.
-- Effectively, can be thought of as \'extending\' the \'ranking\' by all values
-- of @w@, but /which/ @w@ gets given to any rank calls is predetermined by the
-- \'outer accumulation\' (and cannot change).
--
-- @since 2.3
deriving via
  (LiftingSelect (AccumT w) m)
  instance
    (MonadSelect r m, Monoid w) =>
    MonadSelect r (AccumT w m)

-- | A helper type to decrease boilerplate when defining new transformer
-- instances of 'MonadSelect'.
--
-- Most of the instances in this module are derived using this method; for
-- example, our instance of 'ExceptT' is derived as follows:
--
-- > deriving via (LiftingSelect (ExceptT e) m) instance (MonadSelect r m) =>
-- >  MonadSelect r (ExceptT e m)
--
-- @since 2.3
newtype LiftingSelect (t :: (Type -> Type) -> Type -> Type) (m :: Type -> Type) (a :: Type)
  = LiftingSelect (t m a)
  deriving
    ( -- | @since 2.3
      forall a b. a -> LiftingSelect t m b -> LiftingSelect t m a
forall a b. (a -> b) -> LiftingSelect t m a -> LiftingSelect t m b
forall (f :: * -> *).
(forall a b. (a -> b) -> f a -> f b)
-> (forall a b. a -> f b -> f a) -> Functor f
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Functor (t m) =>
a -> LiftingSelect t m b -> LiftingSelect t m a
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Functor (t m) =>
(a -> b) -> LiftingSelect t m a -> LiftingSelect t m b
<$ :: forall a b. a -> LiftingSelect t m b -> LiftingSelect t m a
$c<$ :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Functor (t m) =>
a -> LiftingSelect t m b -> LiftingSelect t m a
fmap :: forall a b. (a -> b) -> LiftingSelect t m a -> LiftingSelect t m b
$cfmap :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Functor (t m) =>
(a -> b) -> LiftingSelect t m a -> LiftingSelect t m b
Functor,
      -- | @since 2.3
      forall a. a -> LiftingSelect t m a
forall a b.
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m a
forall a b.
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m b
forall a b.
LiftingSelect t m (a -> b)
-> LiftingSelect t m a -> LiftingSelect t m b
forall a b c.
(a -> b -> c)
-> LiftingSelect t m a
-> LiftingSelect t m b
-> LiftingSelect t m c
forall (f :: * -> *).
Functor f
-> (forall a. a -> f a)
-> (forall a b. f (a -> b) -> f a -> f b)
-> (forall a b c. (a -> b -> c) -> f a -> f b -> f c)
-> (forall a b. f a -> f b -> f b)
-> (forall a b. f a -> f b -> f a)
-> Applicative f
forall {t :: (* -> *) -> * -> *} {m :: * -> *}.
Applicative (t m) =>
Functor (LiftingSelect t m)
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
Applicative (t m) =>
a -> LiftingSelect t m a
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Applicative (t m) =>
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m a
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Applicative (t m) =>
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m b
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Applicative (t m) =>
LiftingSelect t m (a -> b)
-> LiftingSelect t m a -> LiftingSelect t m b
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b c.
Applicative (t m) =>
(a -> b -> c)
-> LiftingSelect t m a
-> LiftingSelect t m b
-> LiftingSelect t m c
<* :: forall a b.
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m a
$c<* :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Applicative (t m) =>
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m a
*> :: forall a b.
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m b
$c*> :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Applicative (t m) =>
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m b
liftA2 :: forall a b c.
(a -> b -> c)
-> LiftingSelect t m a
-> LiftingSelect t m b
-> LiftingSelect t m c
$cliftA2 :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b c.
Applicative (t m) =>
(a -> b -> c)
-> LiftingSelect t m a
-> LiftingSelect t m b
-> LiftingSelect t m c
<*> :: forall a b.
LiftingSelect t m (a -> b)
-> LiftingSelect t m a -> LiftingSelect t m b
$c<*> :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Applicative (t m) =>
LiftingSelect t m (a -> b)
-> LiftingSelect t m a -> LiftingSelect t m b
pure :: forall a. a -> LiftingSelect t m a
$cpure :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
Applicative (t m) =>
a -> LiftingSelect t m a
Applicative,
      -- | @since 2.3
      forall a. a -> LiftingSelect t m a
forall a b.
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m b
forall a b.
LiftingSelect t m a
-> (a -> LiftingSelect t m b) -> LiftingSelect t m b
forall (m :: * -> *).
Applicative m
-> (forall a b. m a -> (a -> m b) -> m b)
-> (forall a b. m a -> m b -> m b)
-> (forall a. a -> m a)
-> Monad m
forall {t :: (* -> *) -> * -> *} {m :: * -> *}.
Monad (t m) =>
Applicative (LiftingSelect t m)
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
Monad (t m) =>
a -> LiftingSelect t m a
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Monad (t m) =>
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m b
forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Monad (t m) =>
LiftingSelect t m a
-> (a -> LiftingSelect t m b) -> LiftingSelect t m b
return :: forall a. a -> LiftingSelect t m a
$creturn :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
Monad (t m) =>
a -> LiftingSelect t m a
>> :: forall a b.
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m b
$c>> :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Monad (t m) =>
LiftingSelect t m a -> LiftingSelect t m b -> LiftingSelect t m b
>>= :: forall a b.
LiftingSelect t m a
-> (a -> LiftingSelect t m b) -> LiftingSelect t m b
$c>>= :: forall (t :: (* -> *) -> * -> *) (m :: * -> *) a b.
Monad (t m) =>
LiftingSelect t m a
-> (a -> LiftingSelect t m b) -> LiftingSelect t m b
Monad
    )
    via (t m)

-- | @since 2.3
instance (MonadTrans t, MonadSelect r m, Monad (t m)) => MonadSelect r (LiftingSelect t m) where
  select :: forall a. ((a -> r) -> a) -> LiftingSelect t m a
select (a -> r) -> a
f = forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
t m a -> LiftingSelect t m a
LiftingSelect forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall (t :: (* -> *) -> * -> *) (m :: * -> *) a.
(MonadTrans t, Monad m) =>
m a -> t m a
lift forall a b. (a -> b) -> a -> b
$ forall r (m :: * -> *) a. MonadSelect r m => ((a -> r) -> a) -> m a
select (a -> r) -> a
f