module Language.PureScript.Kinds where
import Prelude ()
import Prelude.Compat
import qualified Data.Aeson.TH as A
data Kind
= KUnknown Int
| Star
| Bang
| Row Kind
| FunKind Kind Kind
deriving (Show, Read, Eq, Ord)
$(A.deriveJSON A.defaultOptions ''Kind)
everywhereOnKinds :: (Kind -> Kind) -> Kind -> Kind
everywhereOnKinds f = go
where
go (Row k1) = f (Row (go k1))
go (FunKind k1 k2) = f (FunKind (go k1) (go k2))
go other = f other
everywhereOnKindsM :: (Functor m, Applicative m, Monad m) => (Kind -> m Kind) -> Kind -> m Kind
everywhereOnKindsM f = go
where
go (Row k1) = (Row <$> go k1) >>= f
go (FunKind k1 k2) = (FunKind <$> go k1 <*> go k2) >>= f
go other = f other
everythingOnKinds :: (r -> r -> r) -> (Kind -> r) -> Kind -> r
everythingOnKinds (<>) f = go
where
go k@(Row k1) = f k <> go k1
go k@(FunKind k1 k2) = f k <> go k1 <> go k2
go other = f other