@@ -1,4 +1,5 @@
11{-# LANGUAGE CPP #-}
2+{-# LANGUAGE ConstraintKinds #-}
23{-# LANGUAGE DefaultSignatures #-}
34{-# LANGUAGE FlexibleContexts #-}
45{-# LANGUAGE FlexibleInstances #-}
@@ -13,6 +14,13 @@
1314{-# LANGUAGE TypeFamilies #-}
1415{-# LANGUAGE TypeOperators #-}
1516{-# LANGUAGE TypeSynonymInstances #-}
17+{-# LANGUAGE UndecidableInstances #-}
18+#if __GLASGOW_HASKELL__ >= 800
19+-- Few generics related redundant constraints
20+{-# OPTIONS_GHC -Wno-redundant-constraints #-}
21+-- For TypeErrors
22+{-# OPTIONS_GHC -Wno-unticked-promoted-constructors #-}
23+#endif
1624#include "overlapping-compat.h"
1725module Data.Swagger.Internal.Schema where
1826
@@ -22,27 +30,25 @@ import Prelude.Compat
2230import Control.Lens
2331import Data.Data.Lens (template)
2432
25−import Control.Applicative
2633import Control.Monad
2734import Control.Monad.Writer
28−import Data.Aeson
29−import qualified Data.Aeson.Types as Aeson
35+import Data.Aeson (ToJSON (..), ToJSONKey (..), ToJSONKeyFunction (..), Value (..), Object(..))
3036import Data.Char
3137import Data.Data (Data)
3238import Data.Foldable (traverse_)
33−import Data.Function (on)
3439import Data.HashMap.Strict (HashMap)
3540import qualified Data.HashMap.Strict as HashMap
3641import "unordered-containers" Data.HashSet (HashSet)
3742import qualified "unordered-containers" Data.HashSet as HashSet
38−import Data.HashMap.Strict.InsOrd (InsOrdHashMap)
3943import qualified Data.HashMap.Strict.InsOrd as InsOrdHashMap
4044import Data.Int
4145import Data.IntSet (IntSet)
4246import Data.IntMap (IntMap)
47+import Data.List.NonEmpty.Compat (NonEmpty)
4348import Data.Map (Map)
4449import Data.Proxy
4550import Data.Scientific (Scientific)
51+import Data.Fixed (Fixed, HasResolution, Pico)
4652import Data.Set (Set)
4753import qualified Data.Text as T
4854import qualified Data.Text.Lazy as TL
@@ -51,8 +57,11 @@ import qualified Data.Vector as V
5157import qualified Data.Vector.Primitive as VP
5258import qualified Data.Vector.Storable as VS
5359import qualified Data.Vector.Unboxed as VU
60+import Data.Version (Version)
61+import Numeric.Natural.Compat (Natural)
5462import Data.Word
5563import GHC.Generics
64+import qualified Data.UUID.Types as UUID
5665
5766import Data.Swagger.Declare
5867import Data.Swagger.Internal
@@ -60,9 +69,13 @@ import Data.Swagger.Internal.ParamSchema (ToParamSchema(..))
6069import Data.Swagger.Lens hiding (name, schema)
6170import qualified Data.Swagger.Lens as Swagger
6271import Data.Swagger.SchemaOptions
72+import Data.Swagger.Internal.TypeShape
6373
64−#ifdef __DOCTEST__
65−import Data.Swagger.Lens (name, schema)
74+#if __GLASGOW_HASKELL__ < 800
75+#else
76+import qualified Data.ByteString as BS
77+import qualified Data.ByteString.Lazy as BSL
78+import GHC.TypeLits (TypeError, ErrorMessage(..))
6679#endif
6780
6881unnamed :: Schema -> NamedSchema
@@ -89,14 +102,16 @@ rename name (NamedSchema _ schema) = NamedSchema name schema
89102-- {-\# LANGUAGE OverloadedLists \#-} -- allows to write 'Map' and 'HashMap' as lists
90103--
91104-- import Control.Lens
105+-- import Data.Proxy
106+-- import Data.Swagger
92107--
93108-- data Coord = Coord { x :: Double, y :: Double }
94109--
95110-- instance ToSchema Coord where
96−-- declareNamedSchema = do
111+-- declareNamedSchema _ = do
97112-- doubleSchema <- declareSchemaRef (Proxy :: Proxy Double)
98−-- return $ NamedSchema (Just \"Coord\")) $ mempty
99−-- & type_ .~ SwaggerObject
113+-- return $ NamedSchema (Just \"Coord\") $ mempty
114+-- & type_ ?~ SwaggerObject
100115-- & properties .~
101116-- [ (\"x\", doubleSchema)
102117-- , (\"y\", doubleSchema)
@@ -127,12 +142,17 @@ class ToSchema a where
127142 -- together with all used definitions.
128143 -- Note that the schema itself is included in definitions
129144 -- only if it is recursive (and thus needs its definition in scope).
130− declareNamedSchema :: proxy a -> Declare (Definitions Schema) NamedSchema
131− default declareNamedSchema :: (Generic a, GToSchema (Rep a)) => proxy a -> Declare (Definitions Schema) NamedSchema
145+ declareNamedSchema :: Proxy a -> Declare (Definitions Schema) NamedSchema
146+ default declareNamedSchema :: (Generic a, GToSchema (Rep a), TypeHasSimpleShape a "genericDeclareNamedSchemaUnrestricted") =>
147+ Proxy a -> Declare (Definitions Schema) NamedSchema
132148 declareNamedSchema = genericDeclareNamedSchema defaultSchemaOptions
133149
150+instance ToSchema TimeOfDay where
151+ declareNamedSchema _ = pure $ named "TimeOfDay" $ timeSchema "hh:MM:ss"
152+ & example ?~ toJSON (TimeOfDay 12 33 15)
153+
134154-- | Convert a type into a schema and declare all used schema definitions.
135−declareSchema :: ToSchema a => proxy a -> Declare (Definitions Schema) Schema
155+declareSchema :: ToSchema a => Proxy a -> Declare (Definitions Schema) Schema
136156declareSchema = fmap _namedSchemaSchema . declareNamedSchema
137157
138158-- | Convert a type into an optionally named schema.
@@ -146,7 +166,7 @@ declareSchema = fmap _namedSchemaSchema . declareNamedSchema
146166-- Just "Day"
147167-- >>> encode (toNamedSchema (Proxy :: Proxy Day) ^. schema)
148168-- "{\"example\":\"2016-07-22\",\"format\":\"date\",\"type\":\"string\"}"
149−toNamedSchema :: ToSchema a => proxy a -> NamedSchema
169+toNamedSchema :: ToSchema a => Proxy a -> NamedSchema
150170toNamedSchema = undeclare . declareNamedSchema
151171
152172-- | Get type's schema name according to its @'ToSchema'@ instance.
@@ -156,7 +176,7 @@ toNamedSchema = undeclare . declareNamedSchema
156176--
157177-- >>> schemaName (Proxy :: Proxy UTCTime)
158178-- Just "UTCTime"
159−schemaName :: ToSchema a => proxy a -> Maybe T.Text
179+schemaName :: ToSchema a => Proxy a -> Maybe T.Text
160180schemaName = _namedSchemaName . toNamedSchema
161181
162182-- | Convert a type into a schema.
@@ -166,7 +186,7 @@ schemaName = _namedSchemaName . toNamedSchema
166186--
167187-- >>> encode $ toSchema (Proxy :: Proxy [Day])
168188-- "{\"items\":{\"$ref\":\"#/definitions/Day\"},\"type\":\"array\"}"
169−toSchema :: ToSchema a => proxy a -> Schema
189+toSchema :: ToSchema a => Proxy a -> Schema
170190toSchema = _namedSchemaSchema . toNamedSchema
171191
172192-- | Convert a type into a referenced schema if possible.
@@ -177,7 +197,7 @@ toSchema = _namedSchemaSchema . toNamedSchema
177197--
178198-- >>> encode $ toSchemaRef (Proxy :: Proxy Day)
179199-- "{\"$ref\":\"#/definitions/Day\"}"
180−toSchemaRef :: ToSchema a => proxy a -> Referenced Schema
200+toSchemaRef :: ToSchema a => Proxy a -> Referenced Schema
181201toSchemaRef = undeclare . declareSchemaRef
182202
183203-- | Convert a type into a referenced schema if possible
@@ -187,7 +207,7 @@ toSchemaRef = undeclare . declareSchemaRef
187207-- Schema definitions are typically declared for every referenced schema.
188208-- If @'declareSchemaRef'@ returns a reference, a corresponding schema
189209-- will be declared (regardless of whether it is recusive or not).
190−declareSchemaRef :: ToSchema a => proxy a -> Declare (Definitions Schema) (Referenced Schema)
210+declareSchemaRef :: ToSchema a => Proxy a -> Declare (Definitions Schema) (Referenced Schema)
191211declareSchemaRef proxy = do
192212 case toNamedSchema proxy of
193213 NamedSchema (Just name) schema -> do
@@ -249,7 +269,7 @@ inlineAllSchemas = inlineSchemasWhen (const True)
249269--
250270-- __WARNING:__ @'toInlinedSchema'@ will produce infinite schema
251271-- when inlining recursive schemas.
252−toInlinedSchema :: ToSchema a => proxy a -> Schema
272+toInlinedSchema :: ToSchema a => Proxy a -> Schema
253273toInlinedSchema proxy = inlineAllSchemas defs schema
254274 where
255275 (defs, schema) = runDeclare (declareSchema proxy) mempty
@@ -278,20 +298,20 @@ inlineNonRecursiveSchemas defs = inlineSchemasWhen nonRecursive defs
278298-- | Default schema for binary data (any sequence of octets).
279299binarySchema :: Schema
280300binarySchema = mempty
281− & type_ .~ SwaggerString
301+ & type_ ?~ SwaggerString
282302 & format ?~ "binary"
283303
284304-- | Default schema for binary data (base64 encoded).
285305byteSchema :: Schema
286306byteSchema = mempty
287− & type_ .~ SwaggerString
307+ & type_ ?~ SwaggerString
288308 & format ?~ "byte"
289309
290310-- | Default schema for password string.
291311-- @"password"@ format is used to hint UIs the input needs to be obscured.
292312passwordSchema :: Schema
293313passwordSchema = mempty
294− & type_ .~ SwaggerString
314+ & type_ ?~ SwaggerString
295315 & format ?~ "password"
296316
297317-- | Make an unrestrictive sketch of a @'Schema'@ based on a @'ToJSON'@ instance.
@@ -317,12 +337,12 @@ sketchSchema = sketch . toJSON
317337 sketch js@(Bool _) = go js
318338 sketch js = go js & example ?~ js
319339
320− go Null = mempty & type_ .~ SwaggerNull
321− go js@(Bool _) = mempty & type_ .~ SwaggerBoolean
322− go js@(String s) = mempty & type_ .~ SwaggerString
323− go js@(Number n) = mempty & type_ .~ SwaggerNumber
324− go js@(Array xs) = mempty
325− & type_ .~ SwaggerArray
340+ go Null = mempty & type_ ?~ SwaggerNull
341+ go (Bool _) = mempty & type_ ?~ SwaggerBoolean
342+ go (String _) = mempty & type_ ?~ SwaggerString
343+ go (Number _) = mempty & type_ ?~ SwaggerNumber
344+ go (Array xs) = mempty
345+ & type_ ?~ SwaggerArray
326346 & items ?~ case ischema of
327347 Just s -> SwaggerItemsObject (Inline s)
328348 _ -> SwaggerItemsArray (map Inline ys)
@@ -331,10 +351,10 @@ sketchSchema = sketch . toJSON
331351 allSame = and ((zipWith (==)) ys (tail ys))
332352
333353 ischema = case ys of
334− (z:zs) | allSame -> Just z
335− _ -> Nothing
336− go js@(Object o) = mempty
337− & type_ .~ SwaggerObject
354+ (z:_) | allSame -> Just z
355+ _ -> Nothing
356+ go (Object o) = mempty
357+ & type_ ?~ SwaggerObject
338358 & required .~ HashMap.keys o
339359 & properties .~ fmap (Inline . go) (InsOrdHashMap.fromHashMap o)
340360
@@ -357,24 +377,24 @@ sketchSchema = sketch . toJSON
357377sketchStrictSchema :: ToJSON a => a -> Schema
358378sketchStrictSchema = go . toJSON
359379 where
360− go Null = mempty & type_ .~ SwaggerNull
380+ go Null = mempty & type_ ?~ SwaggerNull
361381 go js@(Bool _) = mempty
362− & type_ .~ SwaggerBoolean
382+ & type_ ?~ SwaggerBoolean
363383 & enum_ ?~ [js]
364384 go js@(String s) = mempty
365− & type_ .~ SwaggerString
385+ & type_ ?~ SwaggerString
366386 & maxLength ?~ fromIntegral (T.length s)
367387 & minLength ?~ fromIntegral (T.length s)
368388 & pattern ?~ s
369389 & enum_ ?~ [js]
370390 go js@(Number n) = mempty
371− & type_ .~ SwaggerNumber
391+ & type_ ?~ SwaggerNumber
372392 & maximum_ ?~ n
373393 & minimum_ ?~ n
374394 & multipleOf ?~ n
375395 & enum_ ?~ [js]
376396 go js@(Array xs) = mempty
377− & type_ .~ SwaggerArray
397+ & type_ ?~ SwaggerArray
378398 & maxItems ?~ fromIntegral sz
379399 & minItems ?~ fromIntegral sz
380400 & items ?~ SwaggerItemsArray (map (Inline . go) (V.toList xs))
@@ -384,7 +404,7 @@ sketchStrictSchema = go . toJSON
384404 sz = length xs
385405 allUnique = sz == HashSet.size (HashSet.fromList (V.toList xs))
386406 go js@(Object o) = mempty
387− & type_ .~ SwaggerObject
407+ & type_ ?~ SwaggerObject
388408 & required .~ names
389409 & properties .~ fmap (Inline . go) (InsOrdHashMap.fromHashMap o)
390410 & maxProperties ?~ fromIntegral (length names)
@@ -394,18 +414,19 @@ sketchStrictSchema = go . toJSON
394414 names = HashMap.keys o
395415
396416class GToSchema (f :: * -> *) where
397− gdeclareNamedSchema :: SchemaOptions -> proxy f -> Schema -> Declare (Definitions Schema) NamedSchema
417+ gdeclareNamedSchema :: SchemaOptions -> Proxy f -> Schema -> Declare (Definitions Schema) NamedSchema
398418
399419instance OVERLAPPABLE_ ToSchema a => ToSchema [a] where
400420 declareNamedSchema _ = do
401421 ref <- declareSchemaRef (Proxy :: Proxy a)
402422 return $ unnamed $ mempty
403− & type_ .~ SwaggerArray
423+ & type_ ?~ SwaggerArray
404424 & items ?~ SwaggerItemsObject ref
405425
406426instance OVERLAPPING_ ToSchema String where declareNamedSchema = plain . paramSchemaToSchema
407427instance ToSchema Bool where declareNamedSchema = plain . paramSchemaToSchema
408428instance ToSchema Integer where declareNamedSchema = plain . paramSchemaToSchema
429+instance ToSchema Natural where declareNamedSchema = plain . paramSchemaToSchema
409430instance ToSchema Int where declareNamedSchema = plain . paramSchemaToSchema
410431instance ToSchema Int8 where declareNamedSchema = plain . paramSchemaToSchema
411432instance ToSchema Int16 where declareNamedSchema = plain . paramSchemaToSchema
@@ -425,6 +446,8 @@ instance ToSchema Scientific where declareNamedSchema = plain . paramSchemaToSc
425446instance ToSchema Double where declareNamedSchema = plain . paramSchemaToSchema
426447instance ToSchema Float where declareNamedSchema = plain . paramSchemaToSchema
427448
449+instance HasResolution a => ToSchema (Fixed a) where declareNamedSchema = plain . paramSchemaToSchema
450+
428451instance ToSchema a => ToSchema (Maybe a) where
429452 declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy a)
430453
@@ -433,6 +456,11 @@ instance (ToSchema a, ToSchema b) => ToSchema (Either a b)
433456instance ToSchema () where
434457 declareNamedSchema _ = pure (NamedSchema Nothing nullarySchema)
435458
459+-- | For 'ToJSON' instance, see <http://hackage.haskell.org/package/uuid-aeson uuid-aeson> package.
460+instance ToSchema UUID.UUID where
461+ declareNamedSchema p = pure $ named "UUID" $ paramSchemaToSchema p
462+ & example ?~ toJSON (UUID.toText UUID.nil)
463+
436464instance (ToSchema a, ToSchema b) => ToSchema (a, b)
437465instance (ToSchema a, ToSchema b, ToSchema c) => ToSchema (a, b, c)
438466instance (ToSchema a, ToSchema b, ToSchema c, ToSchema d) => ToSchema (a, b, c, d)
@@ -442,7 +470,7 @@ instance (ToSchema a, ToSchema b, ToSchema c, ToSchema d, ToSchema e, ToSchema f
442470
443471timeSchema :: T.Text -> Schema
444472timeSchema fmt = mempty
445− & type_ .~ SwaggerString
473+ & type_ ?~ SwaggerString
446474 & format ?~ fmt
447475
448476-- | Format @"date"@ corresponds to @yyyy-mm-dd@ format.
@@ -463,7 +491,7 @@ instance ToSchema ZonedTime where
463491 & example ?~ toJSON (ZonedTime (LocalTime (fromGregorian 2016 7 22) (TimeOfDay 7 40 0)) (hoursToTimeZone 3))
464492
465493instance ToSchema NominalDiffTime where
466− declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy Integer)
494+ declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy Pico)
467495
468496-- |
469497-- >>> toSchema (Proxy :: Proxy UTCTime) ^. format
@@ -475,6 +503,20 @@ instance ToSchema UTCTime where
475503instance ToSchema T.Text where declareNamedSchema = plain . paramSchemaToSchema
476504instance ToSchema TL.Text where declareNamedSchema = plain . paramSchemaToSchema
477505
506+instance ToSchema Version where declareNamedSchema = plain . paramSchemaToSchema
507+
508+#if __GLASGOW_HASKELL__ < 800
509+#else
510+type family ToSchemaByteStringError bs where
511+ ToSchemaByteStringError bs = TypeError
512+ ( Text "Impossible to have an instance " :<>: ShowType (ToSchema bs) :<>: Text "."
513+ :$$: Text "Please, use a newtype wrapper around " :<>: ShowType bs :<>: Text " instead."
514+ :$$: Text "Consider using byteSchema or binarySchema templates." )
515+
516+instance ToSchemaByteStringError BS.ByteString => ToSchema BS.ByteString where declareNamedSchema = error "impossible"
517+instance ToSchemaByteStringError BSL.ByteString => ToSchema BSL.ByteString where declareNamedSchema = error "impossible"
518+#endif
519+
478520instance ToSchema IntSet where declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy (Set Int))
479521
480522-- | NOTE: This schema does not account for the uniqueness of keys.
@@ -490,8 +532,8 @@ instance (ToJSONKey k, ToSchema k, ToSchema v) => ToSchema (Map k v) where
490532 declareObjectMapSchema = do
491533 schema <- declareSchemaRef (Proxy :: Proxy v)
492534 return $ unnamed $ mempty
493− & type_ .~ SwaggerObject
494− & additionalProperties ?~ schema
535+ & type_ ?~ SwaggerObject
536+ & additionalProperties ?~ AdditionalPropertiesSchema schema
495537
496538instance (ToJSONKey k, ToSchema k, ToSchema v) => ToSchema (HashMap k v) where
497539 declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy (Map k v))
@@ -502,7 +544,7 @@ instance ToSchema a => ToSchema (Map String a) where
502544 declareNamedSchema _ = do
503545 schema <- declareSchemaRef (Proxy :: Proxy a)
504546 return $ unnamed $ mempty
505− & type_ .~ SwaggerObject
547+ & type_ ?~ SwaggerObject
506548 & additionalProperties ?~ schema
507549
508550instance ToSchema a => ToSchema (Map T.Text a) where declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy (Map String a))
@@ -514,6 +556,12 @@ instance ToSchema a => ToSchema (HashMap TL.Text a) where declareNamedSchema _ =
514556
515557#endif
516558
559+instance OVERLAPPING_ ToSchema Object where
560+ declareNamedSchema _ = pure $ NamedSchema (Just "Object") $ mempty
561+ & type_ ?~ SwaggerObject
562+ & description ?~ "Arbitrary JSON object."
563+ & additionalProperties ?~ AdditionalPropertiesAllowed True
564+
517565instance ToSchema a => ToSchema (V.Vector a) where declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy [a])
518566instance ToSchema a => ToSchema (VU.Vector a) where declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy [a])
519567instance ToSchema a => ToSchema (VS.Vector a) where declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy [a])
@@ -527,6 +575,13 @@ instance ToSchema a => ToSchema (Set a) where
527575
528576instance ToSchema a => ToSchema (HashSet a) where declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy (Set a))
529577
578+-- | @since 2.2.1
579+instance ToSchema a => ToSchema (NonEmpty a) where
580+ declareNamedSchema _ = do
581+ schema <- declareSchema (Proxy :: Proxy [a])
582+ return $ unnamed $ schema
583+ & minItems .~ Just 1
584+
530585instance ToSchema All where declareNamedSchema = plain . paramSchemaToSchema
531586instance ToSchema Any where declareNamedSchema = plain . paramSchemaToSchema
532587
@@ -536,61 +591,141 @@ instance ToSchema a => ToSchema (First a) where declareNamedSchema _ = unname
536591instance ToSchema a => ToSchema (Last a) where declareNamedSchema _ = unname <$> declareNamedSchema (Proxy :: Proxy a)
537592instance ToSchema a => ToSchema (Dual a) where declareNamedSchema _ = unname <$> declareNamedSchema (Proxy :: Proxy a)
538593
594+instance ToSchema a => ToSchema (Identity a) where declareNamedSchema _ = declareNamedSchema (Proxy :: Proxy a)
595+
539596-- | Default schema for @'Bounded'@, @'Integral'@ types.
540597--
541598-- >>> encode $ toSchemaBoundedIntegral (Proxy :: Proxy Int16)
542599-- "{\"maximum\":32767,\"minimum\":-32768,\"type\":\"integer\"}"
543−toSchemaBoundedIntegral :: forall a proxy. (Bounded a, Integral a) => proxy a -> Schema
600+toSchemaBoundedIntegral :: forall a. (Bounded a, Integral a) => Proxy a -> Schema
544601toSchemaBoundedIntegral _ = mempty
545− & type_ .~ SwaggerInteger
602+ & type_ ?~ SwaggerInteger
546603 & minimum_ ?~ fromInteger (toInteger (minBound :: a))
547604 & maximum_ ?~ fromInteger (toInteger (maxBound :: a))
548605
549606-- | Default generic named schema for @'Bounded'@, @'Integral'@ types.
550−genericToNamedSchemaBoundedIntegral :: forall a d f proxy.
607+genericToNamedSchemaBoundedIntegral :: forall a d f.
551608 ( Bounded a, Integral a
552609 , Generic a, Rep a ~ D1 d f, Datatype d)
553− => SchemaOptions -> proxy a -> NamedSchema
610+ => SchemaOptions -> Proxy a -> NamedSchema
554611genericToNamedSchemaBoundedIntegral opts proxy
555− = NamedSchema (gdatatypeSchemaName opts (Proxy :: Proxy d)) (toSchemaBoundedIntegral proxy)
612+ = genericNameSchema opts proxy (toSchemaBoundedIntegral proxy)
613+
614+-- | Declare a named schema for a @newtype@ wrapper.
615+genericDeclareNamedSchemaNewtype :: forall a d c s i inner.
616+ (Generic a, Datatype d, Rep a ~ D1 d (C1 c (S1 s (K1 i inner))))
617+ => SchemaOptions -- ^ How to derive the name.
618+ -> (Proxy inner -> Declare (Definitions Schema) Schema) -- ^ How to create a schema for the wrapped type.
619+ -> Proxy a
620+ -> Declare (Definitions Schema) NamedSchema
621+genericDeclareNamedSchemaNewtype opts f proxy = genericNameSchema opts proxy <$> f (Proxy :: Proxy inner)
622+
623+-- | Declare 'Schema' for a mapping with 'Bounded' 'Enum' keys.
624+-- This makes a much more useful schema when there aren't many options for key values.
625+--
626+-- >>> data ButtonState = Neutral | Focus | Active | Hover | Disabled deriving (Show, Bounded, Enum, Generic)
627+-- >>> instance ToJSON ButtonState
628+-- >>> instance ToSchema ButtonState
629+-- >>> instance ToJSONKey ButtonState where toJSONKey = toJSONKeyText (T.pack . show)
630+-- >>> type ImageUrl = T.Text
631+-- >>> encode $ toSchemaBoundedEnumKeyMapping (Proxy :: Proxy (Map ButtonState ImageUrl))
632+-- "{\"properties\":{\"Neutral\":{\"type\":\"string\"},\"Focus\":{\"type\":\"string\"},\"Active\":{\"type\":\"string\"},\"Hover\":{\"type\":\"string\"},\"Disabled\":{\"type\":\"string\"}},\"type\":\"object\"}"
633+--
634+-- Note: this is only useful when @key@ is encoded with 'ToJSONKeyText'.
635+-- If it is encoded with 'ToJSONKeyValue' then a regular schema for @[(key, value)]@ is used.
636+declareSchemaBoundedEnumKeyMapping :: forall map key value.
637+ (Bounded key, Enum key, ToJSONKey key, ToSchema key, ToSchema value)
638+ => Proxy (map key value) -> Declare (Definitions Schema) Schema
639+declareSchemaBoundedEnumKeyMapping _ = case toJSONKey :: ToJSONKeyFunction key of
640+ ToJSONKeyText keyToText _ -> objectSchema keyToText
641+ ToJSONKeyValue _ _ -> declareSchema (Proxy :: Proxy [(key, value)])
642+ where
643+ objectSchema keyToText = do
644+ valueRef <- declareSchemaRef (Proxy :: Proxy value)
645+ let allKeys = [minBound..maxBound :: key]
646+ mkPair k = (keyToText k, valueRef)
647+ return $ mempty
648+ & type_ ?~ SwaggerObject
649+ & properties .~ InsOrdHashMap.fromList (map mkPair allKeys)
650+
651+-- | A 'Schema' for a mapping with 'Bounded' 'Enum' keys.
652+-- This makes a much more useful schema when there aren't many options for key values.
653+--
654+-- >>> data ButtonState = Neutral | Focus | Active | Hover | Disabled deriving (Show, Bounded, Enum, Generic)
655+-- >>> instance ToJSON ButtonState
656+-- >>> instance ToSchema ButtonState
657+-- >>> instance ToJSONKey ButtonState where toJSONKey = toJSONKeyText (T.pack . show)
658+-- >>> type ImageUrl = T.Text
659+-- >>> encode $ toSchemaBoundedEnumKeyMapping (Proxy :: Proxy (Map ButtonState ImageUrl))
660+-- "{\"properties\":{\"Neutral\":{\"type\":\"string\"},\"Focus\":{\"type\":\"string\"},\"Active\":{\"type\":\"string\"},\"Hover\":{\"type\":\"string\"},\"Disabled\":{\"type\":\"string\"}},\"type\":\"object\"}"
661+--
662+-- Note: this is only useful when @key@ is encoded with 'ToJSONKeyText'.
663+-- If it is encoded with 'ToJSONKeyValue' then a regular schema for @[(key, value)]@ is used.
664+toSchemaBoundedEnumKeyMapping :: forall map key value.
665+ (Bounded key, Enum key, ToJSONKey key, ToSchema key, ToSchema value)
666+ => Proxy (map key value) -> Schema
667+toSchemaBoundedEnumKeyMapping = flip evalDeclare mempty . declareSchemaBoundedEnumKeyMapping
556668
557669-- | A configurable generic @'Schema'@ creator.
558−genericDeclareSchema :: (Generic a, GToSchema (Rep a)) => SchemaOptions -> proxy a -> Declare (Definitions Schema) Schema
559−genericDeclareSchema opts proxy = _namedSchemaSchema <$> genericDeclareNamedSchema opts proxy
670+genericDeclareSchema :: (Generic a, GToSchema (Rep a), TypeHasSimpleShape a "genericDeclareSchemaUnrestricted") =>
671+ SchemaOptions -> Proxy a -> Declare (Definitions Schema) Schema
672+genericDeclareSchema = genericDeclareSchemaUnrestricted
560673
561674-- | A configurable generic @'NamedSchema'@ creator.
562675-- This function applied to @'defaultSchemaOptions'@
563676-- is used as the default for @'declareNamedSchema'@
564677-- when the type is an instance of @'Generic'@.
565−genericDeclareNamedSchema :: forall a proxy. (Generic a, GToSchema (Rep a)) => SchemaOptions -> proxy a -> Declare (Definitions Schema) NamedSchema
566−genericDeclareNamedSchema opts _ = gdeclareNamedSchema opts (Proxy :: Proxy (Rep a)) mempty
678+genericDeclareNamedSchema :: (Generic a, GToSchema (Rep a), TypeHasSimpleShape a "genericDeclareNamedSchemaUnrestricted") =>
679+ SchemaOptions -> Proxy a -> Declare (Definitions Schema) NamedSchema
680+genericDeclareNamedSchema = genericDeclareNamedSchemaUnrestricted
681+
682+-- | A configurable generic @'Schema'@ creator.
683+--
684+-- Unlike 'genericDeclareSchema' also works for mixed sum types.
685+-- Use with care since some Swagger tools do not support well schemas for mixed sum types.
686+genericDeclareSchemaUnrestricted :: (Generic a, GToSchema (Rep a)) => SchemaOptions -> Proxy a -> Declare (Definitions Schema) Schema
687+genericDeclareSchemaUnrestricted opts proxy = _namedSchemaSchema <$> genericDeclareNamedSchemaUnrestricted opts proxy
688+
689+-- | A configurable generic @'NamedSchema'@ creator.
690+--
691+-- Unlike 'genericDeclareNamedSchema' also works for mixed sum types.
692+-- Use with care since some Swagger tools do not support well schemas for mixed sum types.
693+genericDeclareNamedSchemaUnrestricted :: forall a. (Generic a, GToSchema (Rep a)) =>
694+ SchemaOptions -> Proxy a -> Declare (Definitions Schema) NamedSchema
695+genericDeclareNamedSchemaUnrestricted opts _ = gdeclareNamedSchema opts (Proxy :: Proxy (Rep a)) mempty
696+
697+-- | Derive a 'Generic'-based name for a datatype and assign it to a given 'Schema'.
698+genericNameSchema :: forall a d f.
699+ (Generic a, Rep a ~ D1 d f, Datatype d)
700+ => SchemaOptions -> Proxy a -> Schema -> NamedSchema
701+genericNameSchema opts _ = NamedSchema (gdatatypeSchemaName opts (Proxy :: Proxy d))
567702
568−gdatatypeSchemaName :: forall proxy d. Datatype d => SchemaOptions -> proxy d -> Maybe T.Text
569−gdatatypeSchemaName opts _ = case name of
703+gdatatypeSchemaName :: forall d. Datatype d => SchemaOptions -> Proxy d -> Maybe T.Text
704+gdatatypeSchemaName opts _ = case orig of
570705 (c:_) | isAlpha c && isUpper c -> Just (T.pack name)
571706 _ -> Nothing
572707 where
573− name = datatypeNameModifier opts (datatypeName (Proxy3 :: Proxy3 d f a))
708+ orig = datatypeName (Proxy3 :: Proxy3 d f a)
709+ name = datatypeNameModifier opts orig
574710
575711-- | Lift a plain @'ParamSchema'@ into a model @'NamedSchema'@.
576−paramSchemaToNamedSchema :: forall a d f proxy.
577− (ToParamSchema a, Generic a, Rep a ~ D1 d f, Datatype d)
578− => SchemaOptions -> proxy a -> NamedSchema
579−paramSchemaToNamedSchema opts proxy = NamedSchema (gdatatypeSchemaName opts (Proxy :: Proxy d)) (paramSchemaToSchema proxy)
712+paramSchemaToNamedSchema :: (ToParamSchema a, Generic a, Rep a ~ D1 d f, Datatype d) =>
713+ SchemaOptions -> Proxy a -> NamedSchema
714+paramSchemaToNamedSchema opts proxy = genericNameSchema opts proxy (paramSchemaToSchema proxy)
580715
581716-- | Lift a plain @'ParamSchema'@ into a model @'Schema'@.
582−paramSchemaToSchema :: forall a proxy. ToParamSchema a => proxy a -> Schema
583−paramSchemaToSchema _ = mempty & paramSchema .~ toParamSchema (Proxy :: Proxy a)
717+paramSchemaToSchema :: ToParamSchema a => Proxy a -> Schema
718+paramSchemaToSchema proxy = mempty & paramSchema .~ toParamSchema proxy
584719
585720nullarySchema :: Schema
586721nullarySchema = mempty
587− & type_ .~ SwaggerArray
722+ & type_ ?~ SwaggerArray
588723 & items ?~ SwaggerItemsArray []
589724
590−gtoNamedSchema :: GToSchema f => SchemaOptions -> proxy f -> NamedSchema
725+gtoNamedSchema :: GToSchema f => SchemaOptions -> Proxy f -> NamedSchema
591726gtoNamedSchema opts proxy = undeclare $ gdeclareNamedSchema opts proxy mempty
592727
593−gdeclareSchema :: GToSchema f => SchemaOptions -> proxy f -> Declare (Definitions Schema) Schema
728+gdeclareSchema :: GToSchema f => SchemaOptions -> Proxy f -> Declare (Definitions Schema) Schema
594729gdeclareSchema opts proxy = _namedSchemaSchema <$> gdeclareNamedSchema opts proxy mempty
595730
596731instance (GToSchema f, GToSchema g) => GToSchema (f :*: g) where
@@ -610,7 +745,7 @@ instance OVERLAPPING_ Constructor c => GToSchema (C1 c U1) where
610745 gdeclareNamedSchema = gdeclareNamedSumSchema
611746
612747-- | Single field constructor.
613−instance (Selector s, GToSchema f) => GToSchema (C1 c (S1 s f)) where
748+instance (Selector s, GToSchema f, GToSchema (S1 s f)) => GToSchema (C1 c (S1 s f)) where
614749 gdeclareNamedSchema opts _ s
615750 | unwrapUnaryRecords opts = fieldSchema
616751 | otherwise =
@@ -624,7 +759,7 @@ instance (Selector s, GToSchema f) => GToSchema (C1 c (S1 s f)) where
624759 recordSchema = gdeclareNamedSchema opts (Proxy :: Proxy (S1 s f)) s
625760 fieldSchema = gdeclareNamedSchema opts (Proxy :: Proxy f) s
626761
627−gdeclareSchemaRef :: GToSchema a => SchemaOptions -> proxy a -> Declare (Definitions Schema) (Referenced Schema)
762+gdeclareSchemaRef :: GToSchema a => SchemaOptions -> Proxy a -> Declare (Definitions Schema) (Referenced Schema)
628763gdeclareSchemaRef opts proxy = do
629764 case gtoNamedSchema opts proxy of
630765 NamedSchema (Just name) schema -> do
@@ -643,7 +778,7 @@ gdeclareSchemaRef opts proxy = do
643778 return $ Ref (Reference name)
644779 _ -> Inline <$> gdeclareSchema opts proxy
645780
646−appendItem :: Referenced Schema -> Maybe (SwaggerItems SwaggerKindSchema) -> Maybe (SwaggerItems SwaggerKindSchema)
781+appendItem :: Referenced Schema -> Maybe (SwaggerItems 'SwaggerKindSchema) -> Maybe (SwaggerItems 'SwaggerKindSchema)
647782appendItem x Nothing = Just (SwaggerItemsArray [x])
648783appendItem x (Just (SwaggerItemsArray xs)) = Just (SwaggerItemsArray (xs ++ [x]))
649784appendItem _ _ = error "GToSchema.appendItem: cannot append to SwaggerItemsObject"
@@ -655,10 +790,12 @@ withFieldSchema opts _ isRequiredField schema = do
655790 return $
656791 if T.null fname
657792 then schema
658− & type_ .~ SwaggerArray
793+ & type_ ?~ SwaggerArray
659794 & items %~ appendItem ref
795+ & maxItems %~ Just . maybe 1 (+1) -- increment maxItems
796+ & minItems %~ Just . maybe 1 (+1) -- increment minItems
660797 else schema
661− & type_ .~ SwaggerObject
798+ & type_ ?~ SwaggerObject
662799 & properties . at fname ?~ ref
663800 & if isRequiredField
664801 then required %~ (++ [fname])
@@ -680,10 +817,13 @@ instance OVERLAPPING_ ToSchema c => GToSchema (K1 i (Maybe c)) where
680817instance OVERLAPPABLE_ ToSchema c => GToSchema (K1 i c) where
681818 gdeclareNamedSchema _ _ _ = declareNamedSchema (Proxy :: Proxy c)
682819
683−instance (GSumToSchema f, GSumToSchema g) => GToSchema (f :+: g) where
820+instance ( GSumToSchema f
821+ , GSumToSchema g
822+ ) => GToSchema (f :+: g)
823+ where
684824 gdeclareNamedSchema = gdeclareNamedSumSchema
685825
686−gdeclareNamedSumSchema :: GSumToSchema f => SchemaOptions -> proxy f -> Schema -> Declare (Definitions Schema) NamedSchema
826+gdeclareNamedSumSchema :: GSumToSchema f => SchemaOptions -> Proxy f -> Schema -> Declare (Definitions Schema) NamedSchema
687827gdeclareNamedSumSchema opts proxy s
688828 | allNullaryToStringTag opts && allNullary = pure $ unnamed (toStringTag sumSchema)
689829 | otherwise = (unnamed . fst) <$> runWriterT declareSumSchema
@@ -692,29 +832,29 @@ gdeclareNamedSumSchema opts proxy s
692832 (sumSchema, All allNullary) = undeclare (runWriterT declareSumSchema)
693833
694834 toStringTag schema = mempty
695− & type_ .~ SwaggerString
835+ & type_ ?~ SwaggerString
696836 & enum_ ?~ map toJSON (schema ^.. properties.ifolded.asIndex)
697837
698838type AllNullary = All
699839
700−class GSumToSchema f where
701− gsumToSchema :: SchemaOptions -> proxy f -> Schema -> WriterT AllNullary (Declare (Definitions Schema)) Schema
840+class GSumToSchema (f :: * -> *) where
841+ gsumToSchema :: SchemaOptions -> Proxy f -> Schema -> WriterT AllNullary (Declare (Definitions Schema)) Schema
702842
703843instance (GSumToSchema f, GSumToSchema g) => GSumToSchema (f :+: g) where
704844 gsumToSchema opts _ = gsumToSchema opts (Proxy :: Proxy f) >=> gsumToSchema opts (Proxy :: Proxy g)
705845
706−gsumConToSchemaWith :: forall c f proxy. (GToSchema (C1 c f), Constructor c) =>
707− Referenced Schema -> SchemaOptions -> proxy (C1 c f) -> Schema -> Schema
846+gsumConToSchemaWith :: forall c f. (GToSchema (C1 c f), Constructor c) =>
847+ Referenced Schema -> SchemaOptions -> Proxy (C1 c f) -> Schema -> Schema
708848gsumConToSchemaWith ref opts _ schema = schema
709− & type_ .~ SwaggerObject
849+ & type_ ?~ SwaggerObject
710850 & properties . at tag ?~ ref
711851 & maxProperties ?~ 1
712852 & minProperties ?~ 1
713853 where
714854 tag = T.pack (constructorTagModifier opts (conName (Proxy3 :: Proxy3 c f p)))
715855
716−gsumConToSchema :: forall c f proxy. (GToSchema (C1 c f), Constructor c) =>
717− SchemaOptions -> proxy (C1 c f) -> Schema -> Declare (Definitions Schema) Schema
856+gsumConToSchema :: (GToSchema (C1 c f), Constructor c) =>
857+ SchemaOptions -> Proxy (C1 c f) -> Schema -> Declare (Definitions Schema) Schema
718858gsumConToSchema opts proxy schema = do
719859 ref <- gdeclareSchemaRef opts proxy
720860 return $ gsumConToSchemaWith ref opts proxy schema
@@ -736,3 +876,7 @@ data Proxy2 a b = Proxy2
736876
737877data Proxy3 a b c = Proxy3
738878
879+-- $setup
880+-- >>> import Data.Swagger
881+-- >>> import Data.Aeson (encode)
882+-- >>> import Data.Aeson.Types (toJSONKeyText)