@@ -4,9 +4,12 @@
44{-# OPTIONS_GHC -Wno-orphans #-}
55
66module Data.Id
7− ( Id, mkId, unId, randomId, unsafeIdTagConvert, coerceId, _Id, nilId
7+ ( Id, mkId, unId, unsafeIdTagConvert, coerceId, _Id, nilId
88 , IntId, mkIntId, unIntId, unsafeIntIdTagConvert, coerceIntId, _IntId
99 , Name, mkName, unName, unsafeNameTagConvert, coerceName, _Name
10+#ifdef USE_UUID
11+ , randomId
12+#endif
1013 )
1114 where
1215
@@ -18,33 +21,52 @@ import Data.Binary (Binary)
1821import Data.Coerce (coerce)
1922import Data.Data
2023import Data.Hashable (Hashable)
21−import Data.OpenApi
2224import Data.String
2325import Data.Text as T
24−import Data.UUID (UUID)
25−import qualified Data.UUID as UUID
26+import Data.UUID.Types (UUID)
27+import qualified Data.UUID.Types as UUID
28+import GHC.TypeLits
29+import Prelude as P
30+import Text.ParserCombinators.ReadP
31+import Web.HttpApiData
32+import Web.PathPieces (PathPiece(..))
33+
34+#ifdef USE_UUID
2635import qualified Data.UUID.V4 as UUID.V4
27−#ifndef ghcjs_HOST_OS
36+#endif
37+#ifdef BACKEND
38+import Data.OpenApi
39+import Test.QuickCheck
40+#ifdef USE_CASSAVA
41+import Data.Csv as Csv hiding(Name)
42+#endif
43+#ifdef USE_POSTGRES
2844import Database.PostgreSQL.Simple.FromField as PG (FromField)
2945import Database.PostgreSQL.Simple.ToField as PG (ToField)
30−import Data.Csv as Csv hiding(Name)
3146#endif
47+#endif
48+
49+#ifdef USE_FLAT
3250import Flat as F
33−import GHC.TypeLits
34−import Test.QuickCheck
35−import Text.ParserCombinators.ReadP
36−import Web.HttpApiData
37−import Web.PathPieces (PathPiece(..))
51+#endif
52+#ifdef USE_STORE
53+import qualified Data.Store
54+#endif
3855
3956newtype Id t = Id { unId :: UUID }
4057 deriving stock Data
4158 deriving newtype
4259 ( Eq, Ord, Binary
4360#ifndef ghcjs_HOST_OS
61+#ifdef USE_POSTGRES
4462 , PG.ToField, PG.FromField
4563#endif
46− , FromJSON, ToJSON, NFData, Hashable, FromJSONKey, ToJSONKey, ToSchema
47− , ToParamSchema, FromHttpApiData, ToHttpApiData )
64+#endif
65+ , FromJSON, ToJSON, NFData, Hashable, FromJSONKey, ToJSONKey
66+#ifdef BACKEND
67+ , ToSchema, ToParamSchema
68+#endif
69+ , FromHttpApiData, ToHttpApiData )
4870
4971type role Id nominal
5072
@@ -54,29 +76,42 @@ mkId :: forall s. UUID -> Id s
5476mkId = coerce
5577{-# INLINE mkId #-}
5678
79+#ifdef USE_UUID
5780randomId :: forall s. IO (Id s)
5881randomId = coerce <$> UUID.V4.nextRandom
5982{-# INLINE randomId #-}
83+#endif
6084
6185instance KnownSymbol s => Show (Id s) where
62− show (Id v) = "Id-" <> symbolVal (Proxy @s) <> "-" <> show v
86+ show (Id v) = "Id-" <> symbolVal (Proxy @s) <> "-" <> P.show v
6387
6488instance KnownSymbol s => Read (Id s) where
6589 readsPrec _ = readP_to_S $ do
6690 _ <- string ("Id-" <> symbolVal (Proxy @s) <> "-")
6791 mkId <$> readS_to_P reads
6892
93+#ifdef USE_FLAT
6994instance Flat (Id s) where
7095 encode = F.encode . UUID.toWords . coerce
7196 decode = coerce . (\(a,b,c,d) -> UUID.fromWords a b c d) <$> F.decode
7297 size = F.size . UUID.toWords . coerce
98+#endif
7399
100+#ifdef USE_STORE
101+instance Data.Store.Store (Id s) where
102+ poke = Data.Store.poke . UUID.toWords . coerce
103+ peek = coerce . (\(a,b,c,d) -> UUID.fromWords a b c d) <$> Data.Store.peek
104+ size = Data.Store.ConstSize 16
105+#endif
106+
107+#ifdef BACKEND
74108instance Arbitrary (Id t) where
75109 arbitrary = fmap Id $ UUID.fromWords
76110 <$> arbitrary
77111 <*> arbitrary
78112 <*> arbitrary
79113 <*> arbitrary
114+#endif
80115
81116instance PathPiece (Id t) where
82117 fromPathPiece = coerce . UUID.fromText
@@ -84,12 +119,14 @@ instance PathPiece (Id t) where
84119
85120#ifndef ghcjs_HOST_OS
86121
122+#ifdef USE_CASSAVA
87123instance Csv.FromField (Id tag) where
88124 parseField = parseField
89125 >=> fmap mkId . maybe (fail "invalid UUID in CSV") pure . UUID.fromText
90126
91127instance Csv.ToField (Id tag) where
92128 toField = Csv.toField . UUID.toText . coerce
129+#endif
93130
94131#endif
95132
@@ -116,11 +153,24 @@ newtype IntId t = IntId { unIntId :: Integer }
116153 deriving newtype
117154 ( Eq, Ord, Binary
118155#ifndef ghcjs_HOST_OS
119− , PG.ToField, PG.FromField, Csv.ToField, Csv.FromField
156+#ifdef USE_POSTGRES
157+ , PG.ToField, PG.FromField
158+#endif
159+#ifdef USE_CASSAVA
160+ , Csv.ToField, Csv.FromField
120161#endif
121− , FromJSON, ToJSON, NFData, Hashable, FromJSONKey, ToJSONKey, ToSchema
122− , ToParamSchema, FromHttpApiData, ToHttpApiData, PathPiece, Flat, Arbitrary
123− , Num, Integral, Real, Enum )
162+#endif
163+#ifdef USE_FLAT
164+ , Flat
165+#endif
166+#ifdef USE_STORE
167+ , Data.Store.Store
168+#endif
169+#ifdef BACKEND
170+ , ToSchema, ToParamSchema, Arbitrary
171+#endif
172+ , FromJSON, ToJSON, NFData, Hashable, FromJSONKey, ToJSONKey, FromHttpApiData
173+ , ToHttpApiData, PathPiece, Num, Integral, Real, Enum )
124174
125175type role IntId nominal
126176
@@ -131,7 +181,7 @@ mkIntId = coerce
131181{-# INLINE mkIntId #-}
132182
133183instance KnownSymbol s => Show (IntId s) where
134− show (IntId v) = "IntId-" <> symbolVal (Proxy @s) <> "-" <> show v
184+ show (IntId v) = "IntId-" <> symbolVal (Proxy @s) <> "-" <> P.show v
135185
136186instance KnownSymbol s => Read (IntId s) where
137187 readsPrec _ = readP_to_S $ do
@@ -154,11 +204,24 @@ newtype Name t = Name { unName :: Text }
154204 deriving newtype
155205 ( Eq, Ord, Binary
156206#ifndef ghcjs_HOST_OS
157− , PG.ToField, PG.FromField, Csv.ToField, Csv.FromField
207+#ifdef USE_POSTGRES
208+ , PG.ToField, PG.FromField
209+#endif
210+#ifdef USE_CASSAVA
211+ , Csv.ToField, Csv.FromField
158212#endif
159− , FromJSON, ToJSON, NFData, Hashable, FromJSONKey, ToJSONKey, ToSchema
160− , ToParamSchema, FromHttpApiData, ToHttpApiData, PathPiece, Flat
161− , IsString, Semigroup, Monoid )
213+#endif
214+#ifdef USE_FLAT
215+ , Flat
216+#endif
217+#ifdef USE_STORE
218+ , Data.Store.Store
219+#endif
220+#ifdef BACKEND
221+ , ToSchema, ToParamSchema
222+#endif
223+ , FromJSON, ToJSON, NFData, Hashable, FromJSONKey, ToJSONKey, FromHttpApiData
224+ , ToHttpApiData, PathPiece, IsString, Semigroup, Monoid )
162225
163226type role Name nominal
164227
@@ -169,15 +232,17 @@ mkName = coerce
169232{-# INLINE mkName #-}
170233
171234instance KnownSymbol s => Show (Name s) where
172− show (Name v) = "Name-" <> symbolVal (Proxy @s) <> "-" <> show v
235+ show (Name v) = "Name-" <> symbolVal (Proxy @s) <> "-" <> P.show v
173236
174237instance KnownSymbol s => Read (Name s) where
175238 readsPrec _ = readP_to_S $ do
176239 _ <- string ("Name-" <> symbolVal (Proxy @s) <> "-")
177240 mkName <$> readS_to_P reads
178241
242+#ifdef BACKEND
179243instance Arbitrary (Name s) where
180244 arbitrary = coerce . T.pack . getPrintableString <$> arbitrary
245+#endif
181246
182247-- | This is a more \"explicit\" 'coerce' specifically for 'Name'.
183248-- You are forced to explicitly specify the phantom types you are converting