@@ -10,14 +10,14 @@ module Derive.API.TH
10 , derivingOptionsToJsonOptions
10 , derivingOptionsToJsonOptions
11 , deriveApiInstances
11 , deriveApiInstances
12 , deriveApiAndArbitraryInstances
12 , deriveApiAndArbitraryInstances
13−#ifdef BACKEND
13+#ifndef ghcjs_HOST_OS
14 , derivingOptionsToSchemaOptions
14 , derivingOptionsToSchemaOptions
15 , deriveToParamSchema
15 , deriveToParamSchema
16 , deriveToSchema
16 , deriveToSchema
19 , deriveApiFromJSON
18 , deriveApiFromJSON
20 , deriveApiToJSON
19 , deriveApiToJSON
21 , generateParseJSON
21 , generateParseJSON
@@ -25,22 +25,21 @@ import Control.Monad
25import Data.Aeson.TH as A
25import Data.Aeson.TH as A
26import Data.Aeson.Types
26import Data.Aeson.Types
27import Data.List
27import Data.List
29import Data.Maybe
28import Data.Maybe
30import Data.Typeable
29import Data.Typeable
31import Language.Haskell.TH
30import Language.Haskell.TH
32import Language.Haskell.TH.Datatype
31import Language.Haskell.TH.Datatype
33import Language.Haskell.TH.Syntax
32import Language.Haskell.TH.Syntax
36−import Data.OpenApi hiding (prefix)
37import Test.QuickCheck
33import Test.QuickCheck
38import Test.QuickCheck.Arbitrary.Generic
34import Test.QuickCheck.Arbitrary.Generic
37+import Data.OpenApi hiding (prefix)
41-- | derive JSON and openapi3 (ToSchema) instances
40-- | derive JSON and openapi3 (ToSchema) instances
42deriveApiInstances :: DerivingOptions -> Name -> Q [Dec]
41deriveApiInstances :: DerivingOptions -> Name -> Q [Dec]
43−#ifdef BACKEND
42+#ifndef ghcjs_HOST_OS
44deriveApiInstances opts tname = liftM2 (<>)
43deriveApiInstances opts tname = liftM2 (<>)
45 (deriveJSON (derivingOptionsToJsonOptions opts) tname)
44 (deriveJSON (derivingOptionsToJsonOptions opts) tname)
46 (deriveToSchema opts tname)
45 (deriveToSchema opts tname)
@@ -49,9 +48,9 @@ deriveApiInstances opts tname =
49 deriveJSON (derivingOptionsToJsonOptions opts) tname
48 deriveJSON (derivingOptionsToJsonOptions opts) tname
52−-- | derive JSON and openapi3 (ToSchema) and Arbitrary instances
51+-- | derive JSON and openapi3 (ToSchema) instances
53deriveApiAndArbitraryInstances :: DerivingOptions -> Name -> Q [Dec]
52deriveApiAndArbitraryInstances :: DerivingOptions -> Name -> Q [Dec]
54−#ifdef BACKEND
53+#ifndef ghcjs_HOST_OS
55deriveApiAndArbitraryInstances opts tname = liftM2 (<>)
54deriveApiAndArbitraryInstances opts tname = liftM2 (<>)
56 (deriveApiInstances opts tname)
55 (deriveApiInstances opts tname)
57 (deriveArbitrary tname)
56 (deriveArbitrary tname)
@@ -60,31 +59,14 @@ deriveApiAndArbitraryInstances opts tname =
60 deriveApiInstances opts tname
59 deriveApiInstances opts tname
63−#if MIN_VERSION_containers(0,6,6)
64−#elif MIN_VERSION_template_haskell(2,17,0)
65−instance (Lift k, Lift a) => Lift (Map k a) where
66− liftTyped x = unsafeCodeCoerce (lift x)
67− lift m = [|M.fromList (M.toList m)|]
68−#elif MIN_VERSION_template_haskell(2,16,0)
69−instance (Lift k, Lift a) => Lift (Map k a) where
70− liftTyped x = unsafeTExpCoerce (lift x)
71− lift m = [|M.fromList (M.toList m)|]
73−instance (Lift k, Lift a) => Lift (Map k a) where
74− lift m = [|M.fromList (M.toList m)|]
76−-- We can't directly 'lift' Options into TH (there is no instance for function),
62+-- We can't directly 'lift' Options into TH, so have to use this
77−-- so we use this type.
78data DerivingOptions = DerivingOptions
63data DerivingOptions = DerivingOptions
79 { prefix :: Maybe String
64 { prefix :: Maybe String
80 -- ^ Optional prefix to strip from constructors and field labels
65 -- ^ Optional prefix to strip from constructors and field labels
81 , snake :: Bool
66 , snake :: Bool
82 -- ^ Whether to snake constructors and field labels
67 -- ^ Whether to snake constructors and field labels
83 -- Applied after stripping prefix
68 -- Applied after stripping prefix
84− , convertMap :: M.Map String String
85− -- ^ convert names (constructors and fields) before snake and strip prefix
86 , unwrapUnaryRecords :: Bool
69 , unwrapUnaryRecords :: Bool
87− , omitNothingFields :: Bool
88 , sumEncoding :: SumEncoding
70 , sumEncoding :: SumEncoding
89 } deriving (Lift)
71 } deriving (Lift)
@@ -94,66 +76,51 @@ defaultDerivingOptions :: DerivingOptions
94defaultDerivingOptions = DerivingOptions
76defaultDerivingOptions = DerivingOptions
95 { prefix = Nothing
77 { prefix = Nothing
96 , snake = True
78 , snake = True
97− , convertMap = mempty
98− , unwrapUnaryRecords = False
79+ , unwrapUnaryRecords = True
99− , omitNothingFields = False
100 , sumEncoding = defaultTaggedObject }
80 , sumEncoding = defaultTaggedObject }
102derivingOptionsToJsonOptions :: DerivingOptions -> Options
82derivingOptionsToJsonOptions :: DerivingOptions -> Options
103derivingOptionsToJsonOptions DerivingOptions{..} = defaultOptions
83derivingOptionsToJsonOptions DerivingOptions{..} = defaultOptions
104− { fieldLabelModifier = snaked . stripped . convert
84+ { fieldLabelModifier = snaked . stripped
105− , constructorTagModifier = snaked . stripped . convert
85+ , constructorTagModifier = snaked . stripped
106 , sumEncoding
86 , sumEncoding
108 , unwrapUnaryRecords }
87 , unwrapUnaryRecords }
110− convert s = fromMaybe s $ M.lookup s convertMap
111 snaked = if snake then camelTo2 '_' else id
89 snaked = if snake then camelTo2 '_' else id
112 stripped = maybe id stripPrefix' prefix
90 stripped = maybe id stripPrefix' prefix
114−#ifdef BACKEND
92+#ifndef ghcjs_HOST_OS
115−datatypeVarsTypes :: DatatypeInfo -> [Type]
116−#if MIN_VERSION_th_abstraction(0,3,0)
117−datatypeVarsTypes = fmap (VarT . tvName) . datatypeVars
119−datatypeVarsTypes = datatypeVars
122derivingOptionsToSchemaOptions :: DerivingOptions -> SchemaOptions
93derivingOptionsToSchemaOptions :: DerivingOptions -> SchemaOptions
123derivingOptionsToSchemaOptions = fromAesonOptions . derivingOptionsToJsonOptions
94derivingOptionsToSchemaOptions = fromAesonOptions . derivingOptionsToJsonOptions
125deriveToParamSchema :: DerivingOptions -> Name -> Q [Dec]
96deriveToParamSchema :: DerivingOptions -> Name -> Q [Dec]
126deriveToParamSchema opts tname = do
97deriveToParamSchema opts tname = do
127− body <- AppE (VarE 'genericToParamSchema)
99+ body = AppE (VarE 'genericToParamSchema) <$>
128− . AppE (VarE 'derivingOptionsToSchemaOptions) <$> lift opts
100+ appE (pure $ VarE 'derivingOptionsToSchemaOptions) (lift opts)
129− pure [InstanceD Nothing []
101+ pure <$> instanceD
130− (ConT ''ToParamSchema `AppT` ConT tname)
103+ (pure $ ConT ''ToParamSchema `AppT` ConT tname)
131− [FunD 'toParamSchema [Clause [] (NormalB body) []] ]]
104+ [ funD 'toParamSchema $ pure $ clause [] (normalB body) [] ]
133deriveToSchema :: DerivingOptions -> Name -> Q [Dec]
106deriveToSchema :: DerivingOptions -> Name -> Q [Dec]
134deriveToSchema opts tname = do
107deriveToSchema opts tname = do
135 tinfo <- reifyDatatype tname
108 tinfo <- reifyDatatype tname
110+#if MIN_VERSION_th_abstraction(0,3,0)
111+ tvars = VarT . tvName <$> datatypeVars tinfo
113+ tvars = datatypeVars tinfo
137 context = do
115 context = do
138− var <- datatypeVarsTypes tinfo
116+ var <- tvars
139 [ConT ''Typeable `AppT` var, ConT ''ToSchema `AppT` var]
117 [ConT ''Typeable `AppT` var, ConT ''ToSchema `AppT` var]
140− body <- AppE (VarE 'genericDeclareNamedSchema)
118+ body = AppE (VarE 'genericDeclareNamedSchema) <$>
141− . AppE (VarE 'derivingOptionsToSchemaOptions) <$> lift opts
119+ appE (pure $ VarE 'derivingOptionsToSchemaOptions) (lift opts)
142− pure [InstanceD Nothing context
120+ pure <$> instanceD
143− (ConT ''ToSchema `AppT` datatypeType tinfo)
122+ (pure $ ConT ''ToSchema `AppT` datatypeType tinfo)
144− [ FunD 'declareNamedSchema [Clause [] (NormalB body) []] ]]
123+ [ funD 'declareNamedSchema $ pure $ clause [] (normalB body) [] ]
146−deriveArbitrary :: Name -> Q [Dec]
147−deriveArbitrary t = do
148− tinfo <- reifyDatatype t
150− typ = datatypeType tinfo
151− ctx = datatypeVarsTypes tinfo >>= \tv ->
152− [ ConT ''Arbitrary `AppT` tv
153− , ConT ''Arg `AppT` typ `AppT` tv ]
154− pure [InstanceD Nothing ctx (ConT ''Arbitrary `AppT` typ)
155− [ FunD 'arbitrary [Clause [] (NormalB $ VarE 'genericArbitrary) []]
156− , FunD 'shrink [Clause [] (NormalB $ VarE 'genericShrink) []] ] ]
159deriveApiFromJSON :: DerivingOptions -> Name -> Q [Dec]
126deriveApiFromJSON :: DerivingOptions -> Name -> Q [Dec]
@@ -167,3 +134,10 @@ generateParseJSON = mkParseJSON . derivingOptionsToJsonOptions
168stripPrefix' :: String -> String -> String
135stripPrefix' :: String -> String -> String
169stripPrefix' pfx = fromMaybe <*> stripPrefix pfx
136stripPrefix' pfx = fromMaybe <*> stripPrefix pfx
138+deriveArbitrary :: Name -> Q [Dec]
139+deriveArbitrary t = [d|
140+ instance Arbitrary $(conT t) where
141+ arbitrary = genericArbitrary
142+ shrink = genericShrink