Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Reproduce project problem #39

Open
wants to merge 6 commits into
base: master
Choose a base branch
from
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
79 changes: 52 additions & 27 deletions src/SuperRecord.hs
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,6 @@ import GHC.Generics
import GHC.Exts
import GHC.TypeLits
import qualified Control.Monad.State as S
import qualified Data.Text as T
import Data.Semigroup as Sem (Semigroup(..))

#ifdef JS_RECORD
Expand All @@ -98,6 +97,25 @@ import qualified JavaScript.Object.Internal as JS
import GHC.ST ( ST(..) , runST)
#endif

import Data.Kind (Type)

#if MIN_VERSION_aeson(2, 0, 0)
import qualified Data.Aeson.Key as Key
import Text.Show (showListWith)
#else
import qualified Data.Text as T
#endif

#if MIN_VERSION_aeson(2, 0, 0)
jsonKey :: String -> Key.Key
jsonKey = Key.fromString
#else
jsonKey :: String -> T.Text
jsonKey = T.pack
#endif
{-# INLINE jsonKey #-}


-- | Sort a list of fields using merge sort, alias to 'FieldListSort'
type Sort xs = FieldListSort xs

Expand All @@ -108,7 +126,7 @@ type Record lts = Rec (Sort lts)
-- | Internal record type. When manually writing an explicit type signature for
-- a record, use 'Record' instead. For abstract type signatures 'Rec' will work
-- well.
data Rec (lts :: [*])
data Rec (lts :: [Type])
= Rec
{
#ifndef JS_RECORD
Expand Down Expand Up @@ -143,7 +161,7 @@ class ( c1 k a b, c2 k a b ) => Tuple222C c1 c2 k a b
instance ( c1 k a b, c2 k a b ) => Tuple222C c1 c2 k a b

instance (RecApply lts lts (ConstC Show)) => Show (Rec lts) where
show = show . showRec
showsPrec = showsPrecRec

instance RecApply lts lts (Tuple22C (ConstC Eq) (Has lts)) => Eq (Rec lts) where
r1 == r2 = recApply @lts @lts @(Tuple22C (ConstC Eq) (Has lts)) ( \lbl v b -> get lbl r2 == v && b ) r1 True
Expand Down Expand Up @@ -241,7 +259,7 @@ rcons (lbl := val) (Rec obj) =

infixr 5 `rcons`

class RecCopy (pts :: [*]) (lts :: [*]) (rts :: [*]) where
class RecCopy (pts :: [Type]) (lts :: [Type]) (rts :: [Type]) where
recCopyInto ::
Proxy pts -> Rec lts -> Proxy rts
-> SmallMutableArray# s Any
Expand Down Expand Up @@ -325,7 +343,7 @@ type family RecAll (c :: u -> Constraint) (rs :: [u]) :: Constraint where
RecAll c '[] = ()
RecAll c (r ': rs) = (c r, RecAll c rs)

type family KeyDoesNotExist (l :: Symbol) (lts :: [*]) :: Constraint where
type family KeyDoesNotExist (l :: Symbol) (lts :: [Type]) :: Constraint where
KeyDoesNotExist l '[] = 'True ~ 'True
KeyDoesNotExist l (l := t ': lts) =
TypeError
Expand All @@ -335,27 +353,27 @@ type family KeyDoesNotExist (l :: Symbol) (lts :: [*]) :: Constraint where

type RecAppend lhs rhs = RecAppendH lhs rhs rhs '[]

type family ListConcat (xs :: [*]) (ys :: [*]) :: [*] where
type family ListConcat (xs :: [Type]) (ys :: [Type]) :: [Type] where
ListConcat '[] ys = ys
ListConcat xs '[] = xs
ListConcat (x ': xs) ys = x ': (ListConcat xs ys)

type family ListReverse (xs :: [*]) :: [*] where
type family ListReverse (xs :: [Type]) :: [Type] where
ListReverse (x ': xs) = ListConcat (ListReverse xs) '[x]
ListReverse '[] = '[]

type family RecAppendH (lhs ::[*]) (rhs :: [*]) (rhsall :: [*]) (accum :: [*]) :: [*] where
type family RecAppendH (lhs ::[Type]) (rhs :: [Type]) (rhsall :: [Type]) (accum :: [Type]) :: [Type] where
RecAppendH (l := t ': lhs) (m := u ': rhs) rhsall acc = RecAppendH (l := t ': lhs) rhs rhsall acc
RecAppendH (l := t ': lhs) '[] rhsall acc = RecAppendH lhs rhsall rhsall (l := t ': acc)
RecAppendH '[] rhs rhsall acc = ListConcat (ListReverse acc) rhsall

type family RecSize (lts :: [*]) :: Nat where
type family RecSize (lts :: [Type]) :: Nat where
RecSize '[] = 0
RecSize (l := t ': lts) = 1 + RecSize lts

type RecVecIdxPos l lts = RecSize lts - RecTyIdxH 0 l lts - 1

type family RecTyIdxH (i :: Nat) (l :: Symbol) (lts :: [*]) :: Nat where
type family RecTyIdxH (i :: Nat) (l :: Symbol) (lts :: [Type]) :: Nat where
RecTyIdxH idx l (l := t ': lts) = idx
RecTyIdxH idx m (l := t ': lts) = RecTyIdxH (1 + idx) m lts
RecTyIdxH idx m '[] =
Expand All @@ -364,25 +382,25 @@ type family RecTyIdxH (i :: Nat) (l :: Symbol) (lts :: [*]) :: Nat where
':<>: 'Text m
)

type family RecTy (l :: Symbol) (lts :: [*]) :: Maybe * where
type family RecTy (l :: Symbol) (lts :: [Type]) :: Maybe Type where
RecTy l '[] = 'Nothing
RecTy l (l := t ': lts) = 'Just t
RecTy q (l := t ': lts) = RecTy q lts

-- | Require a record to contain at least the listed labels
type family HasOf (req :: [*]) (lts :: [*]) :: Constraint where
type family HasOf (req :: [Type]) (lts :: [Type]) :: Constraint where
HasOf (l := t ': req) lts = (Has lts l t, HasOf req lts)
HasOf '[] lts = 'True ~ 'True

-- | Intersect two sets of record fields.
--
-- Retains the order of fields in the *first* argument.
-- Throw a type error if a label is associated with distinct types in each of the arguments.
type family Intersect (as :: [*]) (bs :: [*]) :: [*] where
type family Intersect (as :: [Type]) (bs :: [Type]) :: [Type] where
Intersect '[] _ = '[]
Intersect (k := a ': as) bs = IntersectHelper (RecTy k bs) k a as bs

type family IntersectHelper (lk :: Maybe *) (k :: Symbol) (a :: *) (as :: [*]) (bs :: [*]) :: [*] where
type family IntersectHelper (lk :: Maybe Type) (k :: Symbol) (a :: Type) (as :: [Type]) (bs :: [Type]) :: [Type] where
IntersectHelper 'Nothing _ _ as bs = Intersect as bs
IntersectHelper ( 'Just a ) k a as bs = ( k := a ) ': Intersect as bs
IntersectHelper ( 'Just b ) k a _ bs =
Expand Down Expand Up @@ -501,7 +519,7 @@ infixr 8 &:-
fld :: FldProxy l -> FldProxy l
fld = id

type family RecDeepTy (ps :: r) (lts :: [*]) :: * where
type family RecDeepTy (ps :: r) (lts :: [Type]) :: Type where
RecDeepTy (l :& more) (l := Rec t ': lts) = RecDeepTy more t
RecDeepTy (l :& more) (l := t ': lts) = t
RecDeepTy (l :& more) (q := t ': lts) = RecDeepTy (l :& more) lts
Expand Down Expand Up @@ -626,7 +644,7 @@ inject small
class (a ~ b, Lookup kvs k a (RecTy k kvs)) => Inject kvs k a b where
instance (a ~ b, Lookup kvs k a (RecTy k kvs)) => Inject kvs k a b where

class ( r ~ RecTy k kvs ) => Lookup (kvs :: [*]) (k :: Symbol) (a :: *) (r :: Maybe *) where
class ( r ~ RecTy k kvs ) => Lookup (kvs :: [Type]) (k :: Symbol) (a :: Type) (r :: Maybe Type) where
lookupWithDefault :: FldProxy k -> a -> Rec kvs -> a
instance (RecTy k kvs ~ 'Nothing)
=> Lookup kvs k a 'Nothing
Expand All @@ -642,7 +660,7 @@ data RecFields (flds :: [Symbol]) where
RFNil :: RecFields '[]
RFCons :: KnownSymbol f => FldProxy f -> RecFields xs -> RecFields (f ': xs)

recKeys :: forall t (lts :: [*]). RecKeys lts => t lts -> [String]
recKeys :: forall t (lts :: [Type]). RecKeys lts => t lts -> [String]
recKeys = recKeys' . recFields

recKeys' :: RecFields lts -> [String]
Expand All @@ -652,7 +670,7 @@ recKeys' x =
RFCons q qs -> symbolVal q : recKeys' qs

-- | Get keys of a record on value and type level
class RecKeys (lts :: [*]) where
class RecKeys (lts :: [Type]) where
type RecKeysT lts :: [Symbol]
recFields :: t lts -> RecFields (RecKeysT lts)

Expand Down Expand Up @@ -695,11 +713,18 @@ reflectRecFold f r =
showRec :: forall lts. (RecApply lts lts (ConstC Show)) => Rec lts -> [(String, String)]
showRec = reflectRec @(ConstC Show) (\(_ :: FldProxy lbl) v -> (symbolVal' (proxy# :: Proxy# lbl), show v))

showsPrecRec :: forall lts. (RecApply lts lts (ConstC Show)) => Int -> Rec lts -> ShowS
showsPrecRec d r =
showListWith id $
reflectRec
@(ConstC Show) (\(lbl :: FldProxy lbl) v -> showsPrec (d+1) (lbl := v))
r

recToValue :: forall lts. (RecApply lts lts (ConstC ToJSON)) => Rec lts -> Value
recToValue r = object $ reflectRec @(ConstC ToJSON) (\(_ :: FldProxy lbl) v -> (T.pack $ symbolVal' (proxy# :: Proxy# lbl), toJSON v)) r
recToValue r = object $ reflectRec @(ConstC ToJSON) (\(_ :: FldProxy lbl) v -> (jsonKey $ symbolVal' (proxy# :: Proxy# lbl), toJSON v)) r

recToEncoding :: forall lts. (RecApply lts lts (ConstC ToJSON)) => Rec lts -> Encoding
recToEncoding r = pairs $ mconcat $ reflectRec @(ConstC ToJSON) (\(_ :: FldProxy lbl) v -> (T.pack (symbolVal' (proxy# :: Proxy# lbl)) .= v)) r
recToEncoding r = pairs $ mconcat $ reflectRec @(ConstC ToJSON) (\(_ :: FldProxy lbl) v -> (jsonKey (symbolVal' (proxy# :: Proxy# lbl))) .= v) r

recJsonParser :: forall lts s. (RecSize lts ~ s, KnownNat s, RecJsonParse lts) => Value -> Parser (Rec lts)
recJsonParser =
Expand All @@ -709,7 +734,7 @@ recJsonParser =
initSize = fromIntegral $ natVal' (proxy# :: Proxy# s)

-- | Machinery needed to implement 'reflectRec'
class RecApply (rts :: [*]) (lts :: [*]) c where
class RecApply (rts :: [Type]) (lts :: [Type]) c where
recApply :: (forall (l :: Symbol) a. (KnownSymbol l, c l a) => FldProxy l -> a -> b -> b) -> Rec rts -> b -> b

instance RecApply rts '[] c where
Expand All @@ -729,7 +754,7 @@ instance
in recApply @rts @(RemoveAccessTo l lts) @c f r res


class ( KnownNat ( RecSize bs ) ) => TraversalCHelper (bs_acc ::[*]) (as :: [*]) (bs :: [*]) c where
class ( KnownNat ( RecSize bs ) ) => TraversalCHelper (bs_acc ::[Type]) (as :: [Type]) (bs :: [Type]) c where
traversalCHelper :: forall f. Applicative f => ( forall (l :: Symbol) a b. (KnownSymbol l, c l a b) => FldProxy l -> a -> f b ) -> Rec as -> f ( Rec bs_acc )

instance ( RecSize bs ~ s, KnownNat s )
Expand Down Expand Up @@ -762,17 +787,17 @@ instance TraversalCHelper bs as bs c => TraversalC c as bs where
--
-- Effects are performed in the same order as the fields.
traverseC ::
forall c f as bs. ( TraversalC c as bs, Applicative f ) =>
forall c f as bs. ( TraversalC c as bs, Applicative f ) =>
( forall (l :: Symbol) a b. (KnownSymbol l, c l a b) => FldProxy l -> a -> f b ) -> Rec as -> f ( Rec bs )
traverseC = traversalCHelper @bs @as @bs @c @f


type family RemoveAccessTo (l :: Symbol) (lts :: [*]) :: [*] where
type family RemoveAccessTo (l :: Symbol) (lts :: [Type]) :: [Type] where
RemoveAccessTo l (l := t ': lts) = RemoveAccessTo l lts
RemoveAccessTo q (l := t ': lts) = (l := t ': RemoveAccessTo l lts)
RemoveAccessTo q '[] = '[]

class UnsafeRecBuild (rts :: [*]) (lts :: [*]) c where
class UnsafeRecBuild (rts :: [Type]) (lts :: [Type]) c where
-- | Build a record from a constrained applicative function.
--
-- Effects are performed in order of the given (potentially unsorted) fields.
Expand Down Expand Up @@ -818,7 +843,7 @@ recBuildPure f = runIdentity $ recBuild @c @Identity @lts @sortedLts ( \ k v ->


-- | Machinery to implement parseJSON
class RecJsonParse (lts :: [*]) where
class RecJsonParse (lts :: [Type]) where
recJsonParse :: Int -> Object -> Parser (Rec lts)

instance RecJsonParse '[] where
Expand All @@ -835,7 +860,7 @@ instance
do let lbl :: FldProxy l
lbl = FldProxy
rest <- recJsonParse initSize obj
(v :: t) <- obj .: T.pack (symbolVal lbl)
(v :: t) <- obj .: jsonKey (symbolVal lbl)
pure $ unsafeRCons (lbl := v) rest

-- | Conversion helper to bring a Haskell type to a record. Note that the
Expand Down
8 changes: 6 additions & 2 deletions src/SuperRecord/Field.hs
Original file line number Diff line number Diff line change
Expand Up @@ -29,8 +29,12 @@ instance (Ord value) => Ord (label := value) where

instance (Show t) =>
Show (l := t) where
showsPrec p (l := t) =
showParen (p > 10) (showString ("#" ++ symbolVal l ++ " := " ++ show t))
showsPrec d (l := t) =
showParen (d > labelPrec) $
showString ("#" ++ symbolVal l ++ " := ")
. showsPrec (labelPrec+1) t
where
labelPrec = 6

-- | A proxy witness for a label. Very similar to 'Proxy', but needed to implement
-- a non-orphan 'IsLabel' instance
Expand Down
11 changes: 6 additions & 5 deletions src/SuperRecord/Variant.hs
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE RoleAnnotations #-}
{-# LANGUAGE UndecidableInstances #-}
Expand All @@ -20,16 +21,17 @@ where
import Control.Applicative
import Control.DeepSeq
import Data.Aeson
import Data.Aeson.Types (Parser)
import Data.Aeson.Types (Parser, parseFail)
import Data.Maybe
import Data.Proxy
import Data.Kind (Type)
import GHC.Base (Any)
import GHC.TypeLits
import Unsafe.Coerce

-- | A variant is used to express that a values type is of any of
-- the types tracked in the type level list.
data Variant (opts :: [*])
data Variant (opts :: [Type])
= Variant {-# UNPACK #-} !Word Any

type role Variant representational
Expand All @@ -53,9 +55,8 @@ instance (ToJSON t, ToJSON (Variant ts)) => ToJSON (Variant (t ': ts)) where
in fromMaybe (toJSON $ shrinkVariant v1) $ toJSON <$> w1

instance FromJSON (Variant '[]) where
parseJSON r =
do () <- parseJSON r
pure emptyVariant
parseJSON _ =
parseFail "There is no JSON value devoid of a value, so no way to represent an emptyVariant"

instance ( FromJSON t, FromJSON (Variant ts)
) => FromJSON (Variant (t ': ts)) where
Expand Down
27 changes: 21 additions & 6 deletions src/SuperRecord/Variant/Tagged.hs
Original file line number Diff line number Diff line change
@@ -1,5 +1,7 @@
{-# LANGUAGE CPP #-}
{-# LANGUAGE MultiParamTypeClasses #-}
{-# LANGUAGE FlexibleInstances #-}
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE GADTs #-}
{-# LANGUAGE RoleAnnotations #-}
{-# LANGUAGE UndecidableInstances #-}
Expand All @@ -19,10 +21,24 @@ import SuperRecord.Variant

import Control.Applicative
import Data.Aeson
import Data.Aeson.Types (Parser)
import Data.Aeson.Types (Parser, parseFail)
import Data.Maybe
import GHC.TypeLits

#if MIN_VERSION_aeson(2, 0, 0)
import qualified Data.Aeson.Key as Key
#else
import qualified Data.Text as T
#endif

#if MIN_VERSION_aeson(2, 0, 0)
jsonKey :: String -> Key.Key
jsonKey = Key.fromString
#else
jsonKey :: String -> T.Text
jsonKey = T.pack
#endif
{-# INLINE jsonKey #-}

-- | Just a type alias vor 'Variant'
type TaggedVariant opts = Variant opts
Expand All @@ -40,22 +56,21 @@ instance (KnownSymbol lbl, ToJSON t, ToJSON (JsonTaggedVariant ts)) => ToJSON (J
toJSON (JsonTaggedVariant v1) =
let w1 :: Maybe t
w1 = fromTaggedVariant (FldProxy :: FldProxy lbl) v1
tag = T.pack $ symbolVal (FldProxy :: FldProxy lbl)
tag = jsonKey $ symbolVal (FldProxy :: FldProxy lbl)
in let val =
fromMaybe (toJSON $ JsonTaggedVariant $ shrinkVariant v1) $
(\x -> object [tag .= x]) <$> w1
in val

instance FromJSON (JsonTaggedVariant '[]) where
parseJSON r =
do () <- parseJSON r
pure $ JsonTaggedVariant emptyVariant
parseJSON _ =
parseFail "There is no JSON value devoid of a value, so no way to represent an emptyVariant"

instance ( FromJSON t, FromJSON (JsonTaggedVariant ts)
, KnownSymbol lbl
) => FromJSON (JsonTaggedVariant (lbl := t ': ts)) where
parseJSON r =
do let tag = T.pack $ symbolVal (FldProxy :: FldProxy lbl)
do let tag = jsonKey $ symbolVal (FldProxy :: FldProxy lbl)
myParser :: Parser t
myParser = withObject ("Tagged " ++ show tag) (\o -> o .: tag) r
myPackedParser :: Parser (JsonTaggedVariant (lbl := t ': ts))
Expand Down
4 changes: 3 additions & 1 deletion superrecord.cabal
Original file line number Diff line number Diff line change
Expand Up @@ -53,7 +53,9 @@ test-suite superrecord-test
, aeson
, mtl
, text
ghc-options: -threaded -rtsopts -with-rtsopts=-N
-- we test with -O2 in order to provoke bugs caused by optimizations,
-- e.g. https://github.com/agrafix/superrecord/issues/38
ghc-options: -threaded -rtsopts -with-rtsopts=-N -O2
default-language: Haskell2010

benchmark superrecord-bench
Expand Down
Loading