module Network.AWS.Types
(
AccessKey (..)
, SecretKey (..)
, SessionToken (..)
, AuthEnv (..)
, Auth (..)
, withAuth
, LogLevel (..)
, Logger
, Algorithm
, Meta (..)
, Signer (..)
, Signed (..)
, Abbrev
, Service (..)
, serviceSigner
, serviceEndpoint
, serviceTimeout
, serviceCheck
, serviceRetry
, AWSRequest (..)
, Request (..)
, rqService
, rqMethod
, rqHeaders
, rqPath
, rqQuery
, rqBody
, rqSign
, rqPresign
, Response
, Retry (..)
, exponentBase
, exponentGrowth
, retryAttempts
, retryCheck
, AsError (..)
, Error (..)
, HttpException
, SerializeError (..)
, serializeAbbrev
, serializeStatus
, serializeMessage
, ServiceError (..)
, serviceAbbrev
, serviceStatus
, serviceHeaders
, serviceCode
, serviceMessage
, serviceRequestId
, ErrorCode (..)
, errorCode
, ErrorMessage (..)
, RequestId (..)
, Region (..)
, Endpoint (..)
, endpointHost
, endpointPort
, endpointSecure
, endpointScope
, ClientRequest
, ClientResponse
, ResponseBody
, clientRequest
, Seconds (..)
, seconds
, microseconds
, _Coerce
, _Default
) where
import Control.Applicative
import Control.Concurrent (ThreadId)
import Control.DeepSeq
import Control.Exception
import Control.Monad.IO.Class
import Control.Monad.Trans.Resource
import Data.Aeson hiding (Error)
import qualified Data.ByteString as BS
import Data.ByteString.Builder (Builder)
import Data.Coerce
import Data.Conduit
import Data.Data (Data, Typeable)
import Data.Hashable
import Data.IORef
import Data.Maybe
import Data.Monoid
import Data.Proxy
import Data.String
import qualified Data.Text as Text
import qualified Data.Text.Encoding as Text
import Data.Time
import GHC.Generics (Generic)
import Network.AWS.Data.Body
import Network.AWS.Data.ByteString
import Network.AWS.Data.JSON
import Network.AWS.Data.Log
import Network.AWS.Data.Path
import Network.AWS.Data.Query
import Network.AWS.Data.Text
import Network.AWS.Data.XML
import Network.AWS.Lens (Iso', Lens', Prism', Setter')
import Network.AWS.Lens (exception, iso, lens, prism,
sets)
import Network.HTTP.Conduit hiding (Proxy, Request, Response)
import qualified Network.HTTP.Conduit as Client
import Network.HTTP.Types.Header
import Network.HTTP.Types.Method
import Network.HTTP.Types.Status (Status)
import Text.XML (def)
type ClientRequest = Client.Request
type ClientResponse = Client.Response ResponseBody
type ResponseBody = ResumableSource (ResourceT IO) ByteString
newtype Abbrev = Abbrev Text
deriving (Eq, Ord, Show, IsString, FromXML, FromJSON, FromText, ToText, ToLog)
newtype ErrorCode = ErrorCode Text
deriving (Eq, Ord, Show, ToText, ToLog)
instance IsString ErrorCode where fromString = errorCode . fromString
instance FromJSON ErrorCode where parseJSON = parseJSONText "ErrorCode"
instance FromXML ErrorCode where parseXML = parseXMLText "ErrorCode"
instance FromText ErrorCode where
parser = errorCode <$> parser
errorCode :: Text -> ErrorCode
errorCode = ErrorCode . strip . unnamespace
where
strip x = fromMaybe x $
Text.stripSuffix "Exception" x <|> Text.stripSuffix "Fault" x
unnamespace x =
case Text.break (== '#') x of
(ns, e) | Text.null e -> ns
| otherwise -> Text.drop 1 e
newtype ErrorMessage = ErrorMessage Text
deriving (Eq, Ord, Show, IsString, FromXML, FromJSON, FromText, ToText, ToLog)
newtype RequestId = RequestId Text
deriving (Eq, Ord, Show, IsString, FromXML, FromJSON, FromText, ToText, ToLog)
data Error
= TransportError HttpException
| SerializeError SerializeError
| ServiceError ServiceError
deriving (Show, Typeable)
instance Exception Error
instance ToLog Error where
build = \case
TransportError e -> build e
SerializeError e -> build e
ServiceError e -> build e
data SerializeError = SerializeError'
{ _serializeAbbrev :: !Abbrev
, _serializeStatus :: !Status
, _serializeMessage :: String
} deriving (Eq, Show, Typeable)
instance ToLog SerializeError where
build SerializeError'{..} = buildLines
[ "[SerializeError] {"
, " service = " <> build _serializeAbbrev
, " status = " <> build _serializeStatus
, " message = " <> build _serializeMessage
, "}"
]
serializeAbbrev :: Lens' SerializeError Abbrev
serializeAbbrev = lens _serializeAbbrev (\s a -> s { _serializeAbbrev = a })
serializeStatus :: Lens' SerializeError Status
serializeStatus = lens _serializeStatus (\s a -> s { _serializeStatus = a })
serializeMessage :: Lens' SerializeError String
serializeMessage = lens _serializeMessage (\s a -> s { _serializeMessage = a })
data ServiceError = ServiceError'
{ _serviceAbbrev :: !Abbrev
, _serviceStatus :: !Status
, _serviceHeaders :: [Header]
, _serviceCode :: !ErrorCode
, _serviceMessage :: Maybe ErrorMessage
, _serviceRequestId :: Maybe RequestId
} deriving (Eq, Show, Typeable)
instance ToLog ServiceError where
build ServiceError'{..} = buildLines
[ "[ServiceError] {"
, " service = " <> build _serviceAbbrev
, " status = " <> build _serviceStatus
, " code = " <> build _serviceCode
, " message = " <> build _serviceMessage
, " request-id = " <> build _serviceRequestId
, "}"
]
serviceAbbrev :: Lens' ServiceError Abbrev
serviceAbbrev = lens _serviceAbbrev (\s a -> s { _serviceAbbrev = a })
serviceStatus :: Lens' ServiceError Status
serviceStatus = lens _serviceStatus (\s a -> s { _serviceStatus = a })
serviceHeaders :: Lens' ServiceError [Header]
serviceHeaders = lens _serviceHeaders (\s a -> s { _serviceHeaders = a })
serviceCode :: Lens' ServiceError ErrorCode
serviceCode = lens _serviceCode (\s a -> s { _serviceCode = a })
serviceMessage :: Lens' ServiceError (Maybe ErrorMessage)
serviceMessage = lens _serviceMessage (\s a -> s { _serviceMessage = a })
serviceRequestId :: Lens' ServiceError (Maybe RequestId)
serviceRequestId = lens _serviceRequestId (\s a -> s { _serviceRequestId = a })
class AsError a where
_Error :: Prism' a Error
_TransportError :: Prism' a HttpException
_SerializeError :: Prism' a SerializeError
_ServiceError :: Prism' a ServiceError
_TransportError = _Error . _TransportError
_SerializeError = _Error . _SerializeError
_ServiceError = _Error . _ServiceError
instance AsError SomeException where
_Error = exception
instance AsError Error where
_Error = id
_TransportError = prism TransportError $ \case
TransportError e -> Right e
x -> Left x
_SerializeError = prism SerializeError $ \case
SerializeError e -> Right e
x -> Left x
_ServiceError = prism ServiceError $ \case
ServiceError e -> Right e
x -> Left x
data Endpoint = Endpoint
{ _endpointHost :: ByteString
, _endpointSecure :: !Bool
, _endpointPort :: !Int
, _endpointScope :: ByteString
} deriving (Eq, Show, Data, Typeable)
endpointHost :: Lens' Endpoint ByteString
endpointHost = lens _endpointHost (\s a -> s { _endpointHost = a })
endpointSecure :: Lens' Endpoint Bool
endpointSecure = lens _endpointSecure (\s a -> s { _endpointSecure = a })
endpointPort :: Lens' Endpoint Int
endpointPort = lens _endpointPort (\s a -> s { _endpointPort = a })
endpointScope :: Lens' Endpoint ByteString
endpointScope = lens _endpointScope (\s a -> s { _endpointScope = a })
data LogLevel
= Info
| Error
| Debug
| Trace
deriving (Eq, Ord, Enum, Show, Data, Typeable)
instance FromText LogLevel where
parser = takeLowerText >>= \case
"info" -> pure Info
"error" -> pure Error
"debug" -> pure Debug
"trace" -> pure Trace
e -> fromTextError $ "Failure parsing LogLevel from " <> e
instance ToText LogLevel where
toText = \case
Info -> "info"
Error -> "error"
Debug -> "debug"
Trace -> "trace"
instance ToByteString LogLevel
type Logger = LogLevel -> Builder -> IO ()
data Retry = Exponential
{ _retryBase :: !Double
, _retryGrowth :: !Int
, _retryAttempts :: !Int
, _retryCheck :: ServiceError -> Maybe Text
}
exponentBase :: Lens' Retry Double
exponentBase = lens _retryBase (\s a -> s { _retryBase = a })
exponentGrowth :: Lens' Retry Int
exponentGrowth = lens _retryGrowth (\s a -> s { _retryGrowth = a })
retryAttempts :: Lens' Retry Int
retryAttempts = lens _retryAttempts (\s a -> s { _retryAttempts = a })
retryCheck :: Lens' Retry (ServiceError -> Maybe Text)
retryCheck = lens _retryCheck (\s a -> s { _retryCheck = a })
data Meta where
Meta :: ToLog a => a -> Meta
instance ToLog Meta where
build (Meta m) = build m
data Signed a = Signed
{ sgMeta :: !Meta
, sgRequest :: !ClientRequest
}
type Algorithm a = Request a -> AuthEnv -> Region -> UTCTime -> Signed a
data Signer = Signer
{ sgSign :: forall a. Algorithm a
, sgPresign :: forall a. Seconds -> Algorithm a
}
data Service = Service
{ _svcAbbrev :: !Abbrev
, _svcSigner :: !Signer
, _svcPrefix :: !ByteString
, _svcVersion :: !ByteString
, _svcEndpoint :: !(Region -> Endpoint)
, _svcTimeout :: !(Maybe Seconds)
, _svcCheck :: !(Status -> Bool)
, _svcError :: !(Abbrev -> Status -> [Header] -> LazyByteString -> Error)
, _svcRetry :: !Retry
}
serviceSigner :: Lens' Service Signer
serviceSigner = lens _svcSigner (\s a -> s { _svcSigner = a })
serviceEndpoint :: Setter' Service Endpoint
serviceEndpoint = sets (\f s -> s { _svcEndpoint = \r -> f (_svcEndpoint s r) })
serviceTimeout :: Lens' Service (Maybe Seconds)
serviceTimeout = lens _svcTimeout (\s a -> s { _svcTimeout = a })
serviceCheck :: Lens' Service (Status -> Bool)
serviceCheck = lens _svcCheck (\s a -> s { _svcCheck = a })
serviceRetry :: Lens' Service Retry
serviceRetry = lens _svcRetry (\s a -> s { _svcRetry = a })
clientRequest :: Endpoint -> Maybe Seconds -> ClientRequest
clientRequest e t = def
{ Client.secure = _endpointSecure e
, Client.host = _endpointHost e
, Client.port = _endpointPort e
, Client.redirectCount = 0
, Client.checkStatus = \_ _ _ -> Nothing
, Client.responseTimeout = microseconds <$> t
}
data Request a = Request
{ _rqService :: !Service
, _rqMethod :: !StdMethod
, _rqPath :: !RawPath
, _rqQuery :: !QueryString
, _rqHeaders :: ![Header]
, _rqBody :: !RqBody
}
rqService :: Lens' (Request a) Service
rqService = lens _rqService (\s a -> s { _rqService = a })
rqBody :: Lens' (Request a) RqBody
rqBody = lens _rqBody (\s a -> s { _rqBody = a })
rqHeaders :: Lens' (Request a) [Header]
rqHeaders = lens _rqHeaders (\s a -> s { _rqHeaders = a })
rqMethod :: Lens' (Request a) StdMethod
rqMethod = lens _rqMethod (\s a -> s { _rqMethod = a })
rqPath :: Lens' (Request a) RawPath
rqPath = lens _rqPath (\s a -> s { _rqPath = a })
rqQuery :: Lens' (Request a) QueryString
rqQuery = lens _rqQuery (\s a -> s { _rqQuery = a })
rqSign :: Algorithm a
rqSign x = sgSign (_svcSigner (_rqService x)) x
rqPresign :: Seconds -> Algorithm a
rqPresign ex x = sgPresign (_svcSigner (_rqService x)) ex x
type Response a = (Status, Rs a)
class AWSRequest a where
type Rs a :: *
request :: a -> Request a
response :: MonadResource m
=> Logger
-> Service
-> Proxy a
-> ClientResponse
-> m (Response a)
newtype AccessKey = AccessKey ByteString
deriving (Eq, Show, IsString, ToText, ToByteString, ToLog)
newtype SecretKey = SecretKey ByteString
deriving (Eq, IsString, ToText, ToByteString)
newtype SessionToken = SessionToken ByteString
deriving (Eq, IsString, ToText, ToByteString)
data AuthEnv = AuthEnv
{ _authAccess :: !AccessKey
, _authSecret :: !SecretKey
, _authToken :: Maybe SessionToken
, _authExpiry :: Maybe UTCTime
}
instance ToLog AuthEnv where
build AuthEnv{..} = buildLines
[ "[Amazonka Auth] {"
, " access key = ****" <> key _authAccess
, " secret key = ****"
, " security token = " <> build (const "****" <$> _authToken :: Maybe Builder)
, " expiry = " <> build _authExpiry
, "}"
]
where
key (AccessKey k) = build . BS.reverse . BS.take 6 $ BS.reverse k
instance FromJSON AuthEnv where
parseJSON = withObject "AuthEnv" $ \o -> AuthEnv
<$> f AccessKey (o .: "AccessKeyId")
<*> f SecretKey (o .: "SecretAccessKey")
<*> fmap (f SessionToken) (o .:? "Token")
<*> o .:? "Expiration"
where
f g = fmap (g . Text.encodeUtf8)
data Auth
= Ref ThreadId (IORef AuthEnv)
| Auth AuthEnv
instance ToLog Auth where
build (Ref t _) = "[Amazonka Auth] { <thread:" <> build (show t) <> "> }"
build (Auth e) = build e
withAuth :: MonadIO m => Auth -> (AuthEnv -> m a) -> m a
withAuth (Ref _ r) f = liftIO (readIORef r) >>= f
withAuth (Auth e) f = f e
data Region
= Ireland
| Frankfurt
| Tokyo
| Singapore
| Sydney
| Beijing
| NorthVirginia
| NorthCalifornia
| Oregon
| GovCloud
| GovCloudFIPS
| SaoPaulo
deriving (Eq, Ord, Read, Show, Data, Typeable, Generic)
instance Hashable Region
instance NFData Region
instance FromText Region where
parser = takeLowerText >>= \case
"eu-west-1" -> pure Ireland
"eu-central-1" -> pure Frankfurt
"ap-northeast-1" -> pure Tokyo
"ap-southeast-1" -> pure Singapore
"ap-southeast-2" -> pure Sydney
"cn-north-1" -> pure Beijing
"us-east-1" -> pure NorthVirginia
"us-west-2" -> pure Oregon
"us-west-1" -> pure NorthCalifornia
"us-gov-west-1" -> pure GovCloud
"fips-us-gov-west-1" -> pure GovCloudFIPS
"sa-east-1" -> pure SaoPaulo
e -> fromTextError $ "Failure parsing Region from " <> e
instance ToText Region where
toText = \case
Ireland -> "eu-west-1"
Frankfurt -> "eu-central-1"
Tokyo -> "ap-northeast-1"
Singapore -> "ap-southeast-1"
Sydney -> "ap-southeast-2"
Beijing -> "cn-north-1"
NorthVirginia -> "us-east-1"
NorthCalifornia -> "us-west-1"
Oregon -> "us-west-2"
GovCloud -> "us-gov-west-1"
GovCloudFIPS -> "fips-us-gov-west-1"
SaoPaulo -> "sa-east-1"
instance ToByteString Region
instance ToLog Region where
build = build . toBS
instance FromXML Region where parseXML = parseXMLText "Region"
instance ToXML Region where toXML = toXMLText
newtype Seconds = Seconds Int
deriving
( Eq
, Ord
, Read
, Show
, Enum
, Num
, Bounded
, Integral
, Real
, Data
, Typeable
, Generic
, ToQuery
, ToByteString
, ToText
)
instance Hashable Seconds
instance NFData Seconds
instance ToLog Seconds where
build s = build (seconds s) <> "s"
seconds :: Seconds -> Int
seconds (Seconds n)
| n < 0 = 0
| otherwise = n
microseconds :: Seconds -> Int
microseconds = (1000000 *) . seconds
_Coerce :: (Coercible a b, Coercible b a) => Iso' a b
_Coerce = iso coerce coerce
_Default :: Monoid a => Iso' (Maybe a) a
_Default = iso f Just
where
f (Just x) = x
f Nothing = mempty