org.jetbrains.kotlin.gradle.utils.gradleAttributesContainerUtils.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of kotlin-gradle-plugin Show documentation

Kotlin Gradle Plugin

There is a newer version: 2.1.0-RC

/*
 * Copyright 2010-2023 JetBrains s.r.o. and Kotlin Programming Language contributors.
 * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
 */

package org.jetbrains.kotlin.gradle.utils

import org.gradle.api.attributes.Attribute
import org.gradle.api.attributes.AttributeContainer
import org.gradle.api.attributes.HasAttributes
import org.gradle.api.provider.ProviderFactory

/**
 * KGP's internal analog of [org.gradle.api.internal.attributes.AttributeContainerInternal.asMap]
 * Can be used to compare attributes
 */
internal fun AttributeContainer.toMap(): Map, Any?> {
    val result = mutableMapOf, Any?>()
    for (key in keySet()) {
        result[key] = getAttribute(key)
    }

    return result
}

internal fun  HasAttributes.setAttributeProvider(
    providerFactory: ProviderFactory,
    key: Attribute,
    value: () -> T
) {
    attributes.attributeProvider(
        key,
        providerFactory.provider { value() }
    )
}

/**
 * Should only be used to configure simple attributes values!
 *
 * When in doubt, prefer lazy method overload.
 */
internal fun  HasAttributes.setAttribute(
    key: Attribute,
    value: T
) {
    attributes.attribute(key, value)
}

internal fun  HasAttributes.copyAttributeTo(
    providerFactory: ProviderFactory,
    dest: HasAttributes,
    key: Attribute,
) {
    dest.setAttributeProvider(providerFactory, key) {
        attributes.getAttribute(key)
            ?: throw IllegalStateException("Failed to copy attribute. Source container is missing $key (named ${key.name}).")
    }
}

internal fun HasAttributes.copyAttributesTo(
    providerFactory: ProviderFactory,
    dest: HasAttributes,
    keys: Iterable> = attributes.keySet(),
) {
    for (key in keys) {
        copyAttributeTo(providerFactory, dest, key)
    }
}

internal inline fun  attributeOf(
    name: String
): Attribute = Attribute.of(name, T::class.java)