org.http4k.connect.amazon.s3.model.S3BucketPreSigner.kt Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http4k-connect-amazon-s3 Show documentation
http4k-connect-amazon-s3
There is a newer version: 5.41.0.0
Show newest version
package org.http4k.connect.amazon.s3.model

import org.http4k.aws.AwsCredentialScope
import org.http4k.aws.AwsCredentials
import org.http4k.aws.AwsRequestPreSigner
import org.http4k.cloudnative.env.Environment
import org.http4k.connect.amazon.CredentialsProvider
import org.http4k.connect.amazon.Environment
import org.http4k.connect.amazon.core.model.Region
import org.http4k.connect.amazon.s3.S3
import org.http4k.core.Headers
import org.http4k.core.Method.DELETE
import org.http4k.core.Method.GET
import org.http4k.core.Method.PUT
import org.http4k.core.Request
import org.http4k.core.Uri
import org.http4k.core.appendToPath
import java.time.Clock
import java.time.Duration

class S3BucketPreSigner(
    bucketName: BucketName,
    region: Region,
    credentialsProvider: CredentialsProvider,
    clock: Clock = Clock.systemUTC(),
    forcePathStyle: Boolean = false
) {
    constructor(
        bucketName: BucketName,
        region: Region,
        credentials: AwsCredentials,
        clock: Clock = Clock.systemUTC(),
        forcePathStyle: Boolean = false
    ) : this(
        bucketName = bucketName,
        region = region,
        credentialsProvider = { credentials },
        clock = clock,
        forcePathStyle = forcePathStyle
    )

    /**
     * Convenience constructor to create an S3BucketPreSigner from an http4k Environment
     */
    constructor(
        bucketName: BucketName,
        region: Region,
        env: Environment = Environment.ENV,
        clock: Clock = Clock.systemUTC(),
        forcePathStyle: Boolean = false,
        credentialsProvider: CredentialsProvider = CredentialsProvider.Environment(env)
    ): this(
        bucketName = bucketName,
        region = region,
        credentialsProvider = credentialsProvider,
        clock = clock,
        forcePathStyle = forcePathStyle
    )

    private val preSigner = AwsRequestPreSigner(
        credentialsProvider = credentialsProvider,
        scope = AwsCredentialScope(region.value, S3.awsService.value),
        clock = clock
    )

    private val bucketUri = let {
        val usePathStyleApi = forcePathStyle || bucketName.requiresPathStyleApi()
        val pathPrefix = if (usePathStyleApi) "/$bucketName" else ""
        val subdomain = if (usePathStyleApi) "" else "$bucketName."
        Uri.of("https://$subdomain${S3.awsService}.$region.amazonaws.com").path(pathPrefix)
    }

    fun get(key: BucketKey, duration: Duration, headers: Headers = emptyList()) = preSigner(
        Request(GET, bucketUri.appendToPath(key.value)).headers(headers),
        duration
    )

    fun put(key: BucketKey, duration: Duration, headers: Headers = emptyList()) = preSigner(
        Request(PUT, bucketUri.appendToPath(key.value)).headers(headers),
        duration
    )

    fun delete(key: BucketKey, duration: Duration, headers: Headers = emptyList()) = preSigner(
        Request(DELETE, bucketUri.appendToPath(key.value)).headers(headers),
        duration
    )
}