module Network.AWS.Types
(
AccessKey (..)
, SecretKey (..)
, SessionToken (..)
, AuthEnv (..)
, Auth (..)
, withAuth
, LogLevel (..)
, Logger
, Abbrev
, AWSService (..)
, Service (..)
, serviceOf
, Retry (..)
, AWSSigner (..)
, AWSPresigner (..)
, Meta
, Signed (..)
, sgMeta
, sgRequest
, AWSRequest (..)
, Request (..)
, rqMethod
, rqHeaders
, rqPath
, rqQuery
, rqBody
, Response
, AsError (..)
, Error (..)
, HttpException
, SerializeError (..)
, serializeAbbrev
, serializeStatus
, serializeMessage
, ServiceError (..)
, serviceAbbrev
, serviceStatus
, serviceHeaders
, serviceCode
, serviceMessage
, serviceRequestId
, ErrorCode (..)
, ErrorMessage (..)
, RequestId (..)
, Endpoint (..)
, Region (..)
, ClientRequest
, ClientResponse
, ResponseBody
, clientRequest
, Seconds (..)
, _Seconds
, seconds
, microseconds
, _Coerce
, _Default
) where
import Control.Exception
import Control.Exception.Lens (exception)
import Control.Applicative
import Control.Concurrent (ThreadId)
import Control.Lens hiding (coerce)
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 qualified Data.ByteString.Builder as Build
import qualified Data.ByteString.Lazy.Char8 as LBS8
import Data.Coerce
import Data.Conduit
import Data.Data (Data, Typeable)
import Data.Hashable
import Data.IORef
import Data.Monoid
import Data.Proxy
import Data.String
import qualified Data.Text.Encoding as Text
import Data.Time
import GHC.Generics (Generic)
import Network.AWS.Data.Body
import Network.AWS.Data.Crypto
import Network.AWS.Data.Log
import Network.AWS.Data.ByteString
import Network.AWS.Data.Path
import Network.AWS.Data.Query
import Network.AWS.Data.Text
import Network.AWS.Data.XML
import Network.HTTP.Client hiding (Request, Response, Proxy)
import qualified Network.HTTP.Client as Client
import Network.HTTP.Types.Header
import Network.HTTP.Types.Method
import Network.HTTP.Types.Status (Status)
import Text.XML (def)
import Prelude
type ClientRequest = Client.Request
type ClientResponse = Client.Response ResponseBody
type ResponseBody = ResumableSource (ResourceT IO) ByteString
clientRequest :: ClientRequest
clientRequest = def
{ Client.secure = True
, Client.port = 443
, Client.redirectCount = 0
, Client.checkStatus = \_ _ _ -> Nothing
}
newtype Abbrev = Abbrev Text
deriving (Eq, Ord, Show, IsString, FromXML, FromJSON, FromText, ToText, ToLog)
newtype ErrorCode = ErrorCode Text
deriving (Eq, Ord, Show, IsString, FromXML, FromJSON, FromText, ToText, ToLog)
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
, _endpointScope :: ByteString
} deriving (Eq, Show, Data, Typeable)
data LogLevel
= Error
| Info
| Debug
| Trace
deriving (Eq, Ord, Enum, Show, Data, Typeable)
type Logger = LogLevel -> Builder -> IO ()
data Retry = Exponential
{ _retryBase :: !Double
, _retryGrowth :: !Int
, _retryAttempts :: !Int
, _retryCheck :: ServiceError -> Maybe Text
}
data Service s = Service
{ _svcAbbrev :: !Abbrev
, _svcPrefix :: ByteString
, _svcVersion :: ByteString
, _svcEndpoint :: Region -> Endpoint
, _svcTimeout :: Maybe Seconds
, _svcStatus :: Status -> Bool
, _svcError :: Abbrev -> Status -> [Header] -> LazyByteString -> Error
, _svcRetry :: Retry
}
data Request a = Request
{ _rqMethod :: !StdMethod
, _rqPath :: !RawPath
, _rqQuery :: !QueryString
, _rqHeaders :: ![Header]
, _rqBody :: !RqBody
}
instance Show (Request a) where
show = LBS8.unpack . Build.toLazyByteString . build
instance ToLog (Request a) where
build Request{..} = buildLines
[ "[Raw Request] {"
, " method = " <> build _rqMethod
, " path = " <> build (escapePath _rqPath)
, " query = " <> build _rqQuery
, " headers = " <> build _rqHeaders
, " body = {"
, " hash = " <> build (digestToBase Base16 (bodySHA256 _rqBody))
, " payload =\n" <> build (bodyRequest _rqBody)
, " }"
, "}"
]
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 })
class AWSSigner v where
signed :: v ~ Sg s
=> AuthEnv
-> Region
-> UTCTime
-> Service s
-> Request a
-> Signed v a
class AWSPresigner v where
presigned :: v ~ Sg s
=> AuthEnv
-> Region
-> UTCTime
-> Seconds
-> Service s
-> Request a
-> Signed v a
data family Meta v :: *
data Signed v a where
Signed :: ToLog (Meta v)
=> { _sgMeta :: Meta v
, _sgRequest :: ClientRequest
}
-> Signed v a
sgMeta :: ToLog (Meta v) => Lens' (Signed v a) (Meta v)
sgMeta f (Signed m rq) = f m <&> \y -> Signed y rq
sgRequest :: Lens' (Signed v a) ClientRequest
sgRequest f (Signed m rq) = f rq <&> \y -> Signed m y
class AWSSigner (Sg a) => AWSService a where
type Sg a :: *
service :: Sv p ~ a => Proxy p -> Service a
serviceOf :: forall a. AWSService (Sv a) => a -> Service (Sv a)
serviceOf = const $ service (Proxy :: Proxy a)
type Response a = (Status, Rs a)
class AWSService (Sv a) => AWSRequest a where
type Rs a :: *
type Sv a :: *
request :: a -> Request a
response :: MonadResource m
=> Logger
-> Service s
-> Request 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 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 -> fail $
"Failure parsing Region from " ++ show 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
)
_Seconds :: Iso' Seconds Int
_Seconds = iso seconds Seconds
instance ToLog Seconds where
build (Seconds n) = build n <> "s"
seconds :: Seconds -> Int
seconds (Seconds n) = n
microseconds :: Seconds -> Int
microseconds (Seconds n) = n * 1000000
_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