@@ -10,14 +10,14 @@ 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
18 , deriveApiFromJSON
19 , deriveApiFromJSON
19 , deriveApiToJSON
20 , deriveApiToJSON
21 , generateParseJSON
21 , generateParseJSON
@@ -25,21 +25,22 @@ 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
28import Data.Maybe
29import Data.Maybe
29import Data.Typeable
30import Data.Typeable
30import Language.Haskell.TH
31import Language.Haskell.TH
31import Language.Haskell.TH.Datatype
32import Language.Haskell.TH.Datatype
32import Language.Haskell.TH.Syntax
33import Language.Haskell.TH.Syntax
33−import Test.QuickCheck
34−import Test.QuickCheck.Arbitrary.Generic
36−#ifndef ghcjs_HOST_OS
35+#ifdef BACKEND
37import Data.OpenApi hiding (prefix)
36import Data.OpenApi hiding (prefix)
37+import Test.QuickCheck
38+import Test.QuickCheck.Arbitrary.Generic
40-- | derive JSON and openapi3 (ToSchema) instances
41-- | derive JSON and openapi3 (ToSchema) instances
41deriveApiInstances :: DerivingOptions -> Name -> Q [Dec]
42deriveApiInstances :: DerivingOptions -> Name -> Q [Dec]
42−#ifndef ghcjs_HOST_OS
43+#ifdef BACKEND
43deriveApiInstances opts tname = liftM2 (<>)
44deriveApiInstances opts tname = liftM2 (<>)
44 (deriveJSON (derivingOptionsToJsonOptions opts) tname)
45 (deriveJSON (derivingOptionsToJsonOptions opts) tname)
45 (deriveToSchema opts tname)
46 (deriveToSchema opts tname)
@@ -50,7 +51,7 @@ deriveApiInstances opts tname =
51-- | derive JSON and openapi3 (ToSchema) and Arbitrary instances
52-- | derive JSON and openapi3 (ToSchema) and Arbitrary instances
52deriveApiAndArbitraryInstances :: DerivingOptions -> Name -> Q [Dec]
53deriveApiAndArbitraryInstances :: DerivingOptions -> Name -> Q [Dec]
53−#ifndef ghcjs_HOST_OS
54+#ifdef BACKEND
54deriveApiAndArbitraryInstances opts tname = liftM2 (<>)
55deriveApiAndArbitraryInstances opts tname = liftM2 (<>)
55 (deriveApiInstances opts tname)
56 (deriveApiInstances opts tname)
56 (deriveArbitrary tname)
57 (deriveArbitrary tname)
@@ -59,14 +60,29 @@ deriveApiAndArbitraryInstances opts tname =
59 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)|]
62-- We can't directly 'lift' Options into TH (there is no instance for function),
76-- We can't directly 'lift' Options into TH (there is no instance for function),
63-- so we use this type.
77-- so we use this type.
64data DerivingOptions = DerivingOptions
78data DerivingOptions = DerivingOptions
65 { prefix :: Maybe String
79 { prefix :: Maybe String
66− -- ^ Optional prefix to strip from constructors and field labels
80+ -- ^ Optional prefix to strip from constructors and field labels
67 , snake :: Bool
81 , snake :: Bool
68 -- ^ Whether to snake constructors and field labels
82 -- ^ Whether to snake constructors and field labels
69 -- 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
70 , unwrapUnaryRecords :: Bool
86 , unwrapUnaryRecords :: Bool
71 , omitNothingFields :: Bool
87 , omitNothingFields :: Bool
72 , sumEncoding :: SumEncoding
88 , sumEncoding :: SumEncoding
@@ -78,27 +94,29 @@ defaultDerivingOptions :: DerivingOptions
78defaultDerivingOptions = DerivingOptions
94defaultDerivingOptions = DerivingOptions
79 { prefix = Nothing
95 { prefix = Nothing
80 , snake = True
96 , snake = True
97+ , convertMap = mempty
81 , unwrapUnaryRecords = False
98 , unwrapUnaryRecords = False
82 , omitNothingFields = 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 , omitNothingFields
107 , omitNothingFields
91 , unwrapUnaryRecords }
108 , unwrapUnaryRecords }
110+ convert s = fromMaybe s $ M.lookup s convertMap
93 snaked = if snake then camelTo2 '_' else id
111 snaked = if snake then camelTo2 '_' else id
94 stripped = maybe id stripPrefix' prefix
112 stripped = maybe id stripPrefix' prefix
96−#ifndef ghcjs_HOST_OS
114+#ifdef BACKEND
97datatypeVarsTypes :: DatatypeInfo -> [Type]
115datatypeVarsTypes :: DatatypeInfo -> [Type]
98#if MIN_VERSION_th_abstraction(0,3,0)
116#if MIN_VERSION_th_abstraction(0,3,0)
99−datatypeVarsTypes = fmap (VarT . tvName) . datatypeVars
117+datatypeVarsTypes = fmap (VarT . tvName) . datatypeVars
101−datatypeVarsTypes = datatypeVars
119+datatypeVarsTypes = datatypeVars
104derivingOptionsToSchemaOptions :: DerivingOptions -> SchemaOptions
122derivingOptionsToSchemaOptions :: DerivingOptions -> SchemaOptions
@@ -106,7 +124,7 @@ derivingOptionsToSchemaOptions = fromAesonOptions . derivingOptionsToJsonOptions
107deriveToParamSchema :: DerivingOptions -> Name -> Q [Dec]
125deriveToParamSchema :: DerivingOptions -> Name -> Q [Dec]
108deriveToParamSchema opts tname = do
126deriveToParamSchema opts tname = do
109− body <- AppE (VarE 'genericToParamSchema)
127+ body <- AppE (VarE 'genericToParamSchema)
110 . AppE (VarE 'derivingOptionsToSchemaOptions) <$> lift opts
128 . AppE (VarE 'derivingOptionsToSchemaOptions) <$> lift opts
111 pure [InstanceD Nothing []
129 pure [InstanceD Nothing []
112 (ConT ''ToParamSchema `AppT` ConT tname)
130 (ConT ''ToParamSchema `AppT` ConT tname)
@@ -126,11 +144,11 @@ deriveToSchema opts tname = do
126 [ FunD 'declareNamedSchema [Clause [] (NormalB body) []] ]]
144 [ FunD 'declareNamedSchema [Clause [] (NormalB body) []] ]]
128deriveArbitrary :: Name -> Q [Dec]
146deriveArbitrary :: Name -> Q [Dec]
129−deriveArbitrary t = do
147+deriveArbitrary t = do
130 tinfo <- reifyDatatype t
148 tinfo <- reifyDatatype t
132 typ = datatypeType tinfo
150 typ = datatypeType tinfo
133− ctx = datatypeVarsTypes tinfo >>= \tv ->
151+ ctx = datatypeVarsTypes tinfo >>= \tv ->
134 [ ConT ''Arbitrary `AppT` tv
152 [ ConT ''Arbitrary `AppT` tv
135 , ConT ''Arg `AppT` typ `AppT` tv ]
153 , ConT ''Arg `AppT` typ `AppT` tv ]
136 pure [InstanceD Nothing ctx (ConT ''Arbitrary `AppT` typ)
154 pure [InstanceD Nothing ctx (ConT ''Arbitrary `AppT` typ)
@@ -149,4 +167,3 @@ generateParseJSON = mkParseJSON . derivingOptionsToJsonOptions
150stripPrefix' :: String -> String -> String
168stripPrefix' :: String -> String -> String
151stripPrefix' pfx = fromMaybe <*> stripPrefix pfx
169stripPrefix' pfx = fromMaybe <*> stripPrefix pfx