org.http4k.connect.amazon.s3.action.PutObject.kt Maven / Gradle / Ivy

Go to download
package org.http4k.connect.amazon.s3.action

import dev.forkhandles.result4k.Failure
import dev.forkhandles.result4k.Success
import org.http4k.connect.Http4kConnectAction
import org.http4k.connect.amazon.core.model.Tag
import org.http4k.connect.amazon.s3.S3BucketAction
import org.http4k.connect.amazon.s3.model.BucketKey
import org.http4k.connect.amazon.s3.model.StorageClass
import org.http4k.connect.asRemoteFailure
import org.http4k.core.Headers
import org.http4k.core.Method.PUT
import org.http4k.core.Request
import org.http4k.core.Response
import org.http4k.core.Uri
import java.io.InputStream

@Http4kConnectAction
data class PutObject(
    val key: BucketKey,
    val content: InputStream,
    val headers: Headers = emptyList(),
    val tags: List = emptyList(),
    val storageClass: StorageClass? = null,
) : S3BucketAction {

    override fun toRequest() = Request(PUT, Uri.of("/$key"))
        .headers(headers + headersFor(tags) + headersFor(storageClass))
        .body(content)

    override fun toResult(response: Response) = with(response) {
        when {
            status.successful || status.redirection -> Success(Unit)
            else -> Failure(asRemoteFailure(this))
        }
    }
}

internal fun headersFor(tags: Collection) = if (tags.isEmpty()) {
    emptyList()
} else {
    listOf("x-amz-tagging" to tags.joinToString("&") { (key, value) -> "$key=$value" })
}

private fun headersFor(storageClass: StorageClass?) = if (storageClass == null) {
    emptyList()
} else {
    listOf("x-amz-storage-class" to storageClass.toString())
}