{-# LANGUAGE AllowAmbiguousTypes #-}
{-# LANGUAGE ConstraintKinds #-}
{-# LANGUAGE DataKinds #-}
{-# LANGUAGE ExistentialQuantification #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE FunctionalDependencies #-}
{-# LANGUAGE GADTSyntax #-}
{-# LANGUAGE ImportQualifiedPost #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE PolyKinds #-}
{-# LANGUAGE RankNTypes #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE StandaloneKindSignatures #-}
{-# LANGUAGE TypeApplications #-}
{-# LANGUAGE TypeFamilies #-}
{-# LANGUAGE TypeOperators #-}
{-# LANGUAGE UndecidableInstances #-}
{-# LANGUAGE UndecidableSuperClasses #-}
{-# LANGUAGE BlockArguments #-}
{-# LANGUAGE InstanceSigs #-}
module Dep.IOAdvice
(
Advice,
makeAdvice,
makeArgsAdvice,
makeExecutionAdvice,
advise,
restrictArgs,
adviseRecord,
Top,
And,
All,
NP (..),
I (..),
cfoldMap_NP,
Dict (..)
)
where
import Dep.Has
import Dep.Env
import Data.Functor.Identity
import Data.Kind
import Data.List.NonEmpty qualified as N
import Data.List.NonEmpty (NonEmpty)
import Data.SOP
import Data.SOP.Dict
import Data.SOP.NP
import Data.Typeable
import GHC.Generics qualified as G
import GHC.TypeLits
import Data.Coerce
import Data.Bifunctor (first)
type Advice ::
(Type -> Constraint) ->
Type ->
Type
data Advice (ca :: Type -> Constraint) r where
Advice ::
forall ca r.
( forall as.
All ca as =>
NP I as ->
IO (IO r -> IO r, NP I as)
) ->
Advice ca r
instance Semigroup (Advice ca r) where
Advice forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as)
outer <> :: Advice ca r -> Advice ca r -> Advice ca r
<> Advice forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as)
inner = forall (ca :: * -> Constraint) r.
(forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as))
-> Advice ca r
Advice \NP I as
args -> do
(IO r -> IO r
tweakOuter, NP I as
argsOuter) <- forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as)
outer NP I as
args
(IO r -> IO r
tweakInner, NP I as
argsInner) <- forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as)
inner NP I as
argsOuter
forall (f :: * -> *) a. Applicative f => a -> f a
pure (IO r -> IO r
tweakOuter forall b c a. (b -> c) -> (a -> b) -> a -> c
. IO r -> IO r
tweakInner, NP I as
argsInner)
instance Monoid (Advice ca r) where
mappend :: Advice ca r -> Advice ca r -> Advice ca r
mappend = forall a. Semigroup a => a -> a -> a
(<>)
mempty :: Advice ca r
mempty = forall (ca :: * -> Constraint) r.
(forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as))
-> Advice ca r
Advice \NP I as
args -> forall (f :: * -> *) a. Applicative f => a -> f a
pure (forall a. a -> a
id, NP I as
args)
makeAdvice ::
forall ca r.
( forall as.
All ca as =>
NP I as ->
IO (IO r -> IO r, NP I as)
) ->
Advice ca r
makeAdvice :: forall (ca :: * -> Constraint) r.
(forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as))
-> Advice ca r
makeAdvice = forall (ca :: * -> Constraint) r.
(forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as))
-> Advice ca r
Advice
makeArgsAdvice ::
forall ca r.
( forall as.
All ca as =>
NP I as ->
IO (NP I as)
) ->
Advice ca r
makeArgsAdvice :: forall (ca :: * -> Constraint) r.
(forall (as :: [*]). All ca as => NP I as -> IO (NP I as))
-> Advice ca r
makeArgsAdvice forall (as :: [*]). All ca as => NP I as -> IO (NP I as)
tweakArgs =
forall (ca :: * -> Constraint) r.
(forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as))
-> Advice ca r
makeAdvice forall a b. (a -> b) -> a -> b
$ \NP I as
args -> do
NP I as
args' <- forall (as :: [*]). All ca as => NP I as -> IO (NP I as)
tweakArgs NP I as
args
forall (f :: * -> *) a. Applicative f => a -> f a
pure (forall a. a -> a
id, NP I as
args')
makeExecutionAdvice ::
forall ca r.
( IO r ->
IO r
) ->
Advice ca r
makeExecutionAdvice :: forall (ca :: * -> Constraint) r. (IO r -> IO r) -> Advice ca r
makeExecutionAdvice IO r -> IO r
tweakExecution = forall (ca :: * -> Constraint) r.
(forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as))
-> Advice ca r
makeAdvice \NP I as
args -> forall (f :: * -> *) a. Applicative f => a -> f a
pure (IO r -> IO r
tweakExecution, NP I as
args)
data Pair a b = Pair !a !b
advise ::
forall ca r as advisee.
(Multicurryable as r advisee, All ca as) =>
Advice ca r ->
advisee ->
advisee
advise :: forall (ca :: * -> Constraint) r (as :: [*]) advisee.
(Multicurryable as r advisee, All ca as) =>
Advice ca r -> advisee -> advisee
advise (Advice forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as)
f) advisee
advisee =
let uncurried :: NP I as -> IO r
uncurried = forall (as :: [*]) r curried.
Multicurryable as r curried =>
curried -> NP I as -> IO r
multiuncurry @as @r advisee
advisee
uncurried' :: NP I as -> IO r
uncurried' NP I as
args = do
(IO r -> IO r
tweakExecution, NP I as
args') <- forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as)
f NP I as
args
IO r -> IO r
tweakExecution (NP I as -> IO r
uncurried NP I as
args')
in forall (as :: [*]) r curried.
Multicurryable as r curried =>
(NP I as -> IO r) -> curried
multicurry @as @r NP I as -> IO r
uncurried'
type Multicurryable ::
[Type] ->
Type ->
Type ->
Constraint
class Multicurryable as r curried | curried -> as r where
multiuncurry :: curried -> NP I as -> IO r
multicurry :: (NP I as -> IO r) -> curried
instance Multicurryable '[] r (IO r) where
multiuncurry :: IO r -> NP I '[] -> IO r
multiuncurry :: IO r -> NP I '[] -> IO r
multiuncurry IO r
action NP I '[]
Nil = IO r
action
multicurry :: (NP I '[] -> IO r) -> IO r
multicurry NP I '[] -> IO r
f = NP I '[] -> IO r
f forall {k} (a :: k -> *). NP a '[]
Nil
instance (Multicurryable as r curried) => Multicurryable (a ': as) r (a -> curried) where
multiuncurry :: (a -> curried) -> NP I (a : as) -> IO r
multiuncurry a -> curried
f (I x
a :* NP I xs
as) = forall (as :: [*]) r curried.
Multicurryable as r curried =>
curried -> NP I as -> IO r
multiuncurry @as @r @curried (a -> curried
f x
a) NP I xs
as
multicurry :: (NP I (a : as) -> IO r) -> a -> curried
multicurry NP I (a : as) -> IO r
f a
a = forall (as :: [*]) r curried.
Multicurryable as r curried =>
(NP I as -> IO r) -> curried
multicurry @as @r @curried (NP I (a : as) -> IO r
f forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall {k} (a :: k -> *) (x :: k) (xs :: [k]).
a x -> NP a xs -> NP a (x : xs)
(:*) (forall a. a -> I a
I a
a))
restrictArgs ::
forall more less r.
(forall x. Dict more x -> Dict less x) ->
Advice less r ->
Advice more r
restrictArgs :: forall (more :: * -> Constraint) (less :: * -> Constraint) r.
(forall x. Dict more x -> Dict less x)
-> Advice less r -> Advice more r
restrictArgs forall x. Dict more x -> Dict less x
evidence (Advice forall (as :: [*]).
All less as =>
NP I as -> IO (IO r -> IO r, NP I as)
advice) = forall (ca :: * -> Constraint) r.
(forall (as :: [*]).
All ca as =>
NP I as -> IO (IO r -> IO r, NP I as))
-> Advice ca r
Advice \NP I as
args ->
let advice' :: forall as. All more as => NP I as -> IO (IO r -> IO r, NP I as)
advice' :: forall (as :: [*]).
All more as =>
NP I as -> IO (IO r -> IO r, NP I as)
advice' NP I as
args' =
case forall {k} (c :: k -> Constraint) (d :: k -> Constraint)
(xs :: [k]).
(forall (a :: k). Dict c a -> Dict d a)
-> Dict (All c) xs -> Dict (All d) xs
Data.SOP.Dict.mapAll @more @less forall x. Dict more x -> Dict less x
evidence of
Dict (All more) as -> Dict (All less) as
f -> case Dict (All more) as -> Dict (All less) as
f (forall {k} (c :: k -> Constraint) (a :: k). c a => Dict c a
Dict @(All more) @as) of
Dict (All less) as
Dict -> forall (as :: [*]).
All less as =>
NP I as -> IO (IO r -> IO r, NP I as)
advice NP I as
args'
in forall (as :: [*]).
All more as =>
NP I as -> IO (IO r -> IO r, NP I as)
advice' NP I as
args
data RecordComponent
= Terminal
| IWrapped
| Recurse
type AdvisedRecord :: (Type -> Constraint) -> (Type -> Constraint) -> ((Type -> Type) -> Type) -> Constraint
class AdvisedRecord ca cr advised where
_adviseRecord :: [(TypeRep, String)] -> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r) -> advised IO -> advised IO
type AdvisedProduct :: (Type -> Constraint) -> (Type -> Constraint) -> (k -> Type) -> Constraint
class AdvisedProduct ca cr advised_ where
_adviseProduct :: TypeRep -> [(TypeRep, String)] -> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r) -> advised_ k -> advised_ k
instance
( G.Generic (advised IO),
G.Rep (advised IO) ~ G.D1 x (G.C1 y advised_),
Typeable advised,
AdvisedProduct ca cr advised_
) =>
AdvisedRecord ca cr advised
where
_adviseRecord :: [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised IO
-> advised IO
_adviseRecord [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advised IO
unadvised =
let G.M1 (G.M1 advised_ Any
unadvised_) = forall a x. Generic a => a -> Rep a x
G.from advised IO
unadvised
advised_ :: advised_ Any
advised_ = forall k (ca :: * -> Constraint) (cr :: * -> Constraint)
(advised_ :: k -> *) (k :: k).
AdvisedProduct ca cr advised_ =>
TypeRep
-> [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised_ k
-> advised_ k
_adviseProduct @_ @ca @cr (forall {k} (proxy :: k -> *) (a :: k).
Typeable a =>
proxy a -> TypeRep
typeRep (forall {k} (t :: k). Proxy t
Proxy @advised)) [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advised_ Any
unadvised_
in forall a x. Generic a => Rep a x -> a
G.to (forall k i (c :: Meta) (f :: k -> *) (p :: k). f p -> M1 i c f p
G.M1 (forall k i (c :: Meta) (f :: k -> *) (p :: k). f p -> M1 i c f p
G.M1 advised_ Any
advised_))
instance
( AdvisedProduct ca cr advised_left,
AdvisedProduct ca cr advised_right
) =>
AdvisedProduct ca cr (advised_left G.:*: advised_right)
where
_adviseProduct :: forall (k :: k).
TypeRep
-> [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> (:*:) advised_left advised_right k
-> (:*:) advised_left advised_right k
_adviseProduct TypeRep
tr [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f (advised_left k
unadvised_left G.:*: advised_right k
unadvised_right) = forall k (ca :: * -> Constraint) (cr :: * -> Constraint)
(advised_ :: k -> *) (k :: k).
AdvisedProduct ca cr advised_ =>
TypeRep
-> [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised_ k
-> advised_ k
_adviseProduct @_ @ca @cr TypeRep
tr [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advised_left k
unadvised_left forall k (f :: k -> *) (g :: k -> *) (p :: k).
f p -> g p -> (:*:) f g p
G.:*: forall k (ca :: * -> Constraint) (cr :: * -> Constraint)
(advised_ :: k -> *) (k :: k).
AdvisedProduct ca cr advised_ =>
TypeRep
-> [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised_ k
-> advised_ k
_adviseProduct @_ @ca @cr TypeRep
tr [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advised_right k
unadvised_right
type DiscriminateAdvisedComponent :: Type -> RecordComponent
type family DiscriminateAdvisedComponent c where
DiscriminateAdvisedComponent (_ -> _) = 'Terminal
DiscriminateAdvisedComponent (IO _) = 'Terminal
DiscriminateAdvisedComponent (Identity _) = 'IWrapped
DiscriminateAdvisedComponent (I _) = 'IWrapped
DiscriminateAdvisedComponent _ = 'Recurse
type AdvisedComponent :: RecordComponent -> (Type -> Constraint) -> (Type -> Constraint) -> Type -> Constraint
class AdvisedComponent component_type ca cr advised where
_adviseComponent :: [(TypeRep, String)] -> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r) -> advised -> advised
instance
( AdvisedComponent (DiscriminateAdvisedComponent advised) ca cr advised,
KnownSymbol fieldName
) =>
AdvisedProduct ca cr (G.S1 ( 'G.MetaSel ( 'Just fieldName) su ss ds) (G.Rec0 advised))
where
_adviseProduct :: forall (k :: k).
TypeRep
-> [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> S1 ('MetaSel ('Just fieldName) su ss ds) (Rec0 advised) k
-> S1 ('MetaSel ('Just fieldName) su ss ds) (Rec0 advised) k
_adviseProduct TypeRep
tr [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f (G.M1 (G.K1 advised
advised)) =
let acc' :: [(TypeRep, String)]
acc' = (TypeRep
tr, forall (n :: Symbol) (proxy :: Symbol -> *).
KnownSymbol n =>
proxy n -> String
symbolVal (forall {k} (t :: k). Proxy t
Proxy @fieldName)) forall a. a -> [a] -> [a]
: [(TypeRep, String)]
acc
in forall k i (c :: Meta) (f :: k -> *) (p :: k). f p -> M1 i c f p
G.M1 (forall k i c (p :: k). c -> K1 i c p
G.K1 (forall (component_type :: RecordComponent) (ca :: * -> Constraint)
(cr :: * -> Constraint) advised.
AdvisedComponent component_type ca cr advised =>
[(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised
-> advised
_adviseComponent @(DiscriminateAdvisedComponent advised) @ca @cr [(TypeRep, String)]
acc' forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advised
advised))
instance
(Multicurryable as r advised, All ca as, cr r) =>
AdvisedComponent 'Terminal ca cr advised
where
_adviseComponent :: [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised
-> advised
_adviseComponent [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advised
advised = forall (ca :: * -> Constraint) r (as :: [*]) advisee.
(Multicurryable as r advisee, All ca as) =>
Advice ca r -> advisee -> advisee
advise @ca (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f (forall a. [a] -> NonEmpty a
N.fromList [(TypeRep, String)]
acc)) advised
advised
instance
AdvisedComponent (DiscriminateAdvisedComponent advised) ca cr advised =>
AdvisedComponent 'IWrapped ca cr (Identity advised)
where
_adviseComponent :: [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> Identity advised
-> Identity advised
_adviseComponent [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f (Identity advised
advised) = forall a. a -> Identity a
Identity (forall (component_type :: RecordComponent) (ca :: * -> Constraint)
(cr :: * -> Constraint) advised.
AdvisedComponent component_type ca cr advised =>
[(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised
-> advised
_adviseComponent @(DiscriminateAdvisedComponent advised) @ca @cr [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advised
advised)
instance
AdvisedComponent (DiscriminateAdvisedComponent advised) ca cr advised =>
AdvisedComponent 'IWrapped ca cr (I advised)
where
_adviseComponent :: [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> I advised
-> I advised
_adviseComponent [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f (I advised
advised) = forall a. a -> I a
I (forall (component_type :: RecordComponent) (ca :: * -> Constraint)
(cr :: * -> Constraint) advised.
AdvisedComponent component_type ca cr advised =>
[(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised
-> advised
_adviseComponent @(DiscriminateAdvisedComponent advised) @ca @cr [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advised
advised)
instance
AdvisedRecord ca cr advisable =>
AdvisedComponent 'Recurse ca cr (advisable IO)
where
_adviseComponent :: [(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advisable IO
-> advisable IO
_adviseComponent [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advisable IO
advised = forall (ca :: * -> Constraint) (cr :: * -> Constraint)
(advisable :: (* -> *) -> *).
AdvisedRecord ca cr advisable =>
[(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advisable IO
-> advisable IO
_adviseRecord @ca @cr [(TypeRep, String)]
acc forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r
f advisable IO
advised
adviseRecord ::
forall ca cr advised.
AdvisedRecord ca cr advised =>
(forall r . cr r => NonEmpty (TypeRep, String) -> Advice ca r) ->
advised IO ->
advised IO
adviseRecord :: forall (ca :: * -> Constraint) (cr :: * -> Constraint)
(advised :: (* -> *) -> *).
AdvisedRecord ca cr advised =>
(forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advised IO -> advised IO
adviseRecord = forall (ca :: * -> Constraint) (cr :: * -> Constraint)
(advisable :: (* -> *) -> *).
AdvisedRecord ca cr advisable =>
[(TypeRep, String)]
-> (forall r. cr r => NonEmpty (TypeRep, String) -> Advice ca r)
-> advisable IO
-> advisable IO
_adviseRecord @ca @cr []