{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE DeriveAnyClass #-}

module Kubernetes.OpenAPI.CustomTypes where

import Data.Aeson (FromJSON, ToJSON)
import qualified Data.Aeson as A
import Data.Data (Typeable)
import Data.Text (Text)
import GHC.Base (mzero)

import GHC.Generics

{- | `IntOrString`
  IntOrString is a type that can hold an int32 or a string.  When used
  in JSON or YAML marshalling and unmarshalling, it produces or consumes
  the inner type.  This allows you to have, for example, a JSON field
  that can accept a name or number.
-}
data IntOrString
  = IntOrStringS Text
  | IntOrStringI Int
  deriving (Int -> IntOrString -> ShowS
[IntOrString] -> ShowS
IntOrString -> String
(Int -> IntOrString -> ShowS)
-> (IntOrString -> String)
-> ([IntOrString] -> ShowS)
-> Show IntOrString
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [IntOrString] -> ShowS
$cshowList :: [IntOrString] -> ShowS
show :: IntOrString -> String
$cshow :: IntOrString -> String
showsPrec :: Int -> IntOrString -> ShowS
$cshowsPrec :: Int -> IntOrString -> ShowS
Show, IntOrString -> IntOrString -> Bool
(IntOrString -> IntOrString -> Bool)
-> (IntOrString -> IntOrString -> Bool) -> Eq IntOrString
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: IntOrString -> IntOrString -> Bool
$c/= :: IntOrString -> IntOrString -> Bool
== :: IntOrString -> IntOrString -> Bool
$c== :: IntOrString -> IntOrString -> Bool
Eq, Typeable, (forall x. IntOrString -> Rep IntOrString x)
-> (forall x. Rep IntOrString x -> IntOrString)
-> Generic IntOrString
forall x. Rep IntOrString x -> IntOrString
forall x. IntOrString -> Rep IntOrString x
forall a.
(forall x. a -> Rep a x) -> (forall x. Rep a x -> a) -> Generic a
$cto :: forall x. Rep IntOrString x -> IntOrString
$cfrom :: forall x. IntOrString -> Rep IntOrString x
Generic)

instance FromJSON IntOrString where
  parseJSON :: Value -> Parser IntOrString
parseJSON (A.String Text
t) = IntOrString -> Parser IntOrString
forall (m :: * -> *) a. Monad m => a -> m a
return (IntOrString -> Parser IntOrString)
-> IntOrString -> Parser IntOrString
forall a b. (a -> b) -> a -> b
$ Text -> IntOrString
IntOrStringS Text
t
  parseJSON (A.Number Scientific
n) = IntOrString -> Parser IntOrString
forall (m :: * -> *) a. Monad m => a -> m a
return (IntOrString -> Parser IntOrString)
-> IntOrString -> Parser IntOrString
forall a b. (a -> b) -> a -> b
$ Int -> IntOrString
IntOrStringI (Scientific -> Int
forall a b. (RealFrac a, Integral b) => a -> b
round Scientific
n)
  parseJSON Value
_ = Parser IntOrString
forall (m :: * -> *) a. MonadPlus m => m a
mzero

instance ToJSON IntOrString where
  toJSON :: IntOrString -> Value
toJSON (IntOrStringS Text
t) = Text -> Value
A.String Text
t
  toJSON (IntOrStringI Int
n) = Scientific -> Value
A.Number (Int -> Scientific
forall a b. (Integral a, Num b) => a -> b
fromIntegral Int
n)

{- | `Quantity` is a fixed-point representation of a number.
  
  It provides convenient marshaling/unmarshaling in JSON and YAML, in
  addition to String() and Int64() accessors.
  
  The serialization format is:
  
  @
  \<quantity\>        ::= \<signedNumber\>\<suffix\>
  (Note that \<suffix\> may be empty, from the \"\" case in \<decimalSI\>.)
  \<digit\>           ::= 0 | 1 | ... | 9
  \<digits\>          ::= \<digit\> | \<digit\>\<digits\>
  \<number\>          ::= \<digits\> | \<digits\>.\<digits\> | \<digits\>. | .\<digits\>
  \<sign\>            ::= \"+\" | \"-\" \<signedNumber\>    ::= \<number\> | \<sign\>\<number\>
  \<suffix\>          ::= \<binarySI\> | \<decimalExponent\> | \<decimalSI\>
  \<binarySI\>        ::= Ki | Mi | Gi | Ti | Pi | Ei
  (International System of units; See: http://physics.nist.gov/cuu/Units/binary.html)
  \<decimalSI\>       ::= m | \"\" | k | M | G | T | P | E
  (Note that 1024 = 1Ki but 1000 = 1k; I didn't choose the capitalization.)
  \<decimalExponent\> ::= \"e\" \<signedNumber\> | \"E\" \<signedNumber\>
  @
  
  No matter which of the three exponent forms is used, no quantity may
  represent a number greater than 2^63-1 in magnitude, nor may it have
  more than 3 decimal places. Numbers larger or more precise will be
  capped or rounded up. (E.g.: 0.1m will rounded up to 1m.) This may be
  extended in the future if we require larger or smaller quantities.
  
  When a Quantity is parsed from a string, it will remember the type of
  suffix it had, and will use the same type again when it is serialized.
  
  Before serializing, Quantity will be put in "canonical form". This
  means that Exponent/suffix will be adjusted up or down (with a
  corresponding increase or decrease in Mantissa) such that:

    - No precision is lost
    - No fractional digits will be emitted
    - The exponent (or suffix) is as large as possible.

  The sign will be omitted unless the number is negative.
  
  Examples:

    - 1.5 will be serialized as "1500m"
    - 1.5Gi will be serialized as "1536Mi"
  
  Note that the quantity will NEVER be internally represented by a
  floating point number. That is the whole point of this exercise.
  
  Non-canonical values will still parse as long as they are well formed,
  but will be re-emitted in their canonical form. (So always use
  canonical form, or don't diff.)
  
  This format is intended to make it difficult to use these numbers
  without writing some sort of special handling code in the hopes that
  that will cause implementors to also use a fixed point implementation.
-}
newtype Quantity = Quantity { Quantity -> Text
unQuantity :: Text }
  deriving (Int -> Quantity -> ShowS
[Quantity] -> ShowS
Quantity -> String
(Int -> Quantity -> ShowS)
-> (Quantity -> String) -> ([Quantity] -> ShowS) -> Show Quantity
forall a.
(Int -> a -> ShowS) -> (a -> String) -> ([a] -> ShowS) -> Show a
showList :: [Quantity] -> ShowS
$cshowList :: [Quantity] -> ShowS
show :: Quantity -> String
$cshow :: Quantity -> String
showsPrec :: Int -> Quantity -> ShowS
$cshowsPrec :: Int -> Quantity -> ShowS
Show, Quantity -> Quantity -> Bool
(Quantity -> Quantity -> Bool)
-> (Quantity -> Quantity -> Bool) -> Eq Quantity
forall a. (a -> a -> Bool) -> (a -> a -> Bool) -> Eq a
/= :: Quantity -> Quantity -> Bool
$c/= :: Quantity -> Quantity -> Bool
== :: Quantity -> Quantity -> Bool
$c== :: Quantity -> Quantity -> Bool
Eq, Typeable, (forall x. Quantity -> Rep Quantity x)
-> (forall x. Rep Quantity x -> Quantity) -> Generic Quantity
forall x. Rep Quantity x -> Quantity
forall x. Quantity -> Rep Quantity x
forall a.
(forall x. a -> Rep a x) -> (forall x. Rep a x -> a) -> Generic a
$cto :: forall x. Rep Quantity x -> Quantity
$cfrom :: forall x. Quantity -> Rep Quantity x
Generic)

instance FromJSON Quantity where
  parseJSON :: Value -> Parser Quantity
parseJSON (A.String Text
t) = Quantity -> Parser Quantity
forall (m :: * -> *) a. Monad m => a -> m a
return (Quantity -> Parser Quantity) -> Quantity -> Parser Quantity
forall a b. (a -> b) -> a -> b
$ Text -> Quantity
Quantity Text
t
  parseJSON Value
_ = Parser Quantity
forall (m :: * -> *) a. MonadPlus m => m a
mzero

instance ToJSON Quantity where
  toJSON :: Quantity -> Value
toJSON (Quantity Text
t) = Text -> Value
A.String Text
t