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