• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• aws4s-master
  • src
    • main
      • scala
        • org
          • aws4s
            • sqs
              • ReceiptHandle.scala
              • SqsParam.scala
              • SendMessage.scala
              • ReceiveRequestAttemptId.scala
              • Sqs.scala
              • Message.scala
              • ReceiveMessage.scala
              • MessageDeduplicationId.scala
              • TextParamValue.scala
              • Queue.scala
              • SqsCommand.scala
              • DeleteMessage.scala
              • MaxNumberOfMessages.scala
              • SequenceNumber.scala
              • MessageId.scala
              • VisibilityTimeout.scala
              • DelaySeconds.scala
              • WaitTimeSeconds.scala
              • MessageBody.scala
            • Arn.scala
            • s3
              • ListBuckets.scala
              • ObjectPath.scala
              • S3.scala
              • DeleteObject.scala
              • PutObject.scala
              • GetObject.scala
              • S3ObjectCommand.scala
              • BucketName.scala
              • S3ServiceCommand.scala
            • utils
              • Sized.scala
            • kms
              • Decrypt.scala
              • KeyDescription.scala
              • CreateKey.scala
              • GrantToken.scala
              • Kms.scala
              • Ciphertext.scala
              • Encrypt.scala
              • KmsParam.scala
              • GrantTokens.scala
              • Blob.scala
              • ScheduleKeyDeletion.scala
              • KeyMetadata.scala
              • KeyId.scala
              • PendingWindowInDays.scala
              • KmsAggregateParam.scala
              • EncryptionContext.scala
              • Plaintext.scala
              • KmsCommand.scala
            • Failure.scala
            • Region.scala
            • Credentials.scala
            • PayloadSigning.scala
            • dynamodb
              • AttributeType.scala
              • KeySchemaElement.scala
              • TableDescription.scala
              • CreateTable.scala
              • DynamoDbAggregateParam.scala
              • TableName.scala
              • ProvisionedThroughput.scala
              • Index.scala
              • CapacityUnit.scala
              • KeySchema.scala
              • KeyType.scala
              • DynamoDbCommand.scala
              • AttributeDefinition.scala
              • DeleteTable.scala
              • AttributeDefinitions.scala
              • DynamoDbParam.scala
              • DeleteTableSuccess.scala
              • DynamoDb.scala
              • AttributeName.scala
            • core
              • Param.scala
              • RequestSigning.scala
              • ResponseContent.scala
              • CommandPayload.scala
              • Command.scala
              • XmlParsing.scala
              • ServiceName.scala
              • AggregateParamValidator.scala
              • RenderedParam.scala
              • ParamValidator.scala
              • ExtraCirceDecoders.scala
              • Service.scala
              • AggregateParamRenderer.scala
              • Clock.scala
              • ExtraEntityDecoderInstances.scala
              • ParamRenderer.scala
    • test
      • scala
        • org
          • aws4s
            • sqs
              • SqsSmokeTest.scala
            • s3
              • S3SmokeTest.scala
            • kms
              • KmsSmokeTest.scala
            • dynamodb
              • DynamoDbSmokeTest.scala
            • core
              • SmokeTest.scala
              • RequestSigningSpec.scala
  • _config.yml
  • LICENSE
  • project
    • build.properties
    • plugins.sbt
  • .travis.yml
  • README.md
  • .scalafmt.conf
  • build.sbt
  • .gitignore
  • version.sbt

package org.aws4s.core

import cats.data.NonEmptyList
import cats.effect.{Effect, Sync}
import io.circe.Json
import org.http4s.{EntityDecoder, MediaRange, Message}
import org.http4s.scalaxml._
import org.http4s.circe._

private[aws4s] sealed trait ResponseContent {
  final def tryParse[A](pf: PartialFunction[ResponseContent, Option[A]]): Option[A] =
    pf.orElse[ResponseContent, Option[A]]({ case _ => None })(this)
}

private[aws4s] case class XmlContent(elem:    scala.xml.Elem) extends ResponseContent
private[aws4s] case class JsonContent(json:   Json) extends ResponseContent
private[aws4s] case class StringContent(text: String) extends ResponseContent
private[aws4s] case object NoContent extends ResponseContent

private[aws4s] object ResponseContent {

  implicit def entityDecoder[F[_]: Effect]: EntityDecoder[F, ResponseContent] =
    EntityDecoder[F, scala.xml.Elem].map(elem => XmlContent(elem)).widen[ResponseContent] orElse
      EntityDecoder[F, Json].map(json         => JsonContent(json)).widen[ResponseContent] orElse
      inclusiveJsonEntityDecoder.map(json     => JsonContent(json)).widen[ResponseContent] orElse
      EntityDecoder[F, String].map(text       => StringContent(text)).widen[ResponseContent] orElse
      EntityDecoder[F, Unit].map(_            => NoContent).widen[ResponseContent]

  private def inclusiveJsonEntityDecoder[F[_]: Sync]: EntityDecoder[F, Json] = {
    val json = jsonDecoder[F]
    val extraMediaRanges = NonEmptyList.of(
      "application/x-amz-json-1.0"
    ) map (mr => MediaRange.parse(mr).getOrElse(throw new RuntimeException(s"Invalid Media Range: $mr")))
    val allMediaRanges = extraMediaRanges concat json.consumes.toList
    EntityDecoder.decodeBy[F, Json](allMediaRanges.head, allMediaRanges.tail: _*)((msg: Message[F]) => json.decode(msg, strict = false))
  }
}