main.io.cloudshiftdev.awscdklib.core.Extensions.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-cdk-extensions Show documentation

Kotlin CDK Extensions

There is a newer version: 10.1.0

Show newest version

package io.cloudshiftdev.awscdklib.core

import io.cloudshiftdev.awscdk.ArnComponents
import io.cloudshiftdev.awscdk.CfnResource
import io.cloudshiftdev.awscdk.Duration
import io.cloudshiftdev.awscdk.Names
import io.cloudshiftdev.awscdk.Stack
import io.cloudshiftdev.awscdk.TagProps
import io.cloudshiftdev.awscdk.Tags
import io.cloudshiftdev.awscdk.services.route53.IHostedZone
import io.cloudshiftdev.constructs.Construct
import io.cloudshiftdev.constructs.IConstruct

public fun Construct.tag(key: String, value: String, block: TagProps.Builder.() -> Unit = {}) {
    Tags.of(this).add(key, value, block)
}

public fun Construct.addComment(comment: String) {
    node().children().filterIsInstance().forEach { it.addComment(comment) }
}

public fun Construct.uniqueId(): String = Names.uniqueId(this)

public inline fun  Construct.addPropertyOverride(
    property: String,
    value: String
) {
    val resource = node().children().filterIsInstance().first()
    resource.addPropertyOverride(property, value)
}

public fun CfnResource.addComment(comment: String) {
    addMetadata(mapOf("cloudshift:comment" to comment))
}

public fun CfnResource.addMetadata(newMetadata: Map) {
    if (newMetadata.isEmpty()) {
        return
    }
    var metadata = cfnOptions().metadata()
    metadata = metadata.toMutableMap()
    metadata.putAll(newMetadata)
    cfnOptions().metadata(metadata.toMap())
}

public fun IConstruct.allChildren(): List {
    val list = mutableListOf()
    node().children().forEach {
        list.add(it)
        list.addAll(it.allChildren())
    }
    return list.sortedBy { it.node().path() }
}

public fun resourceArn(scope: Construct, block: (ArnComponents.Builder).() -> Unit): String =
    Stack.of(scope).formatArn(block)

public inline fun  Construct.withSingleton(
    id: String,
    block: (String) -> T
): T = node().tryFindChild(id) as? T ?: block(id)

public inline fun  Construct.withSingleton(
    predicate: (T) -> Boolean = { true },
    block: () -> T
): T {
    return allChildren().filterIsInstance().firstOrNull(predicate) ?: block()
}

public val IConstruct.stack: Stack
    get() = Stack.of(this)

public val Int.seconds: Duration
    get() = Duration.seconds(this)

public fun IHostedZone.dnsName(name: String): String = "${name}.${zoneName()}"

public fun IHostedZone.wildcardDnsName(): String = dnsName("*")

public fun IHostedZone.uniqueDnsName(construct: IConstruct, name: String): String =
    "${name}_${construct.address}.${zoneName()}"

public val IConstruct.address: String
    get() = this.node().addr()