kmath/attributes-kt/src/commonMain/kotlin/space/kscience/attributes/AttributesBuilder.kt

/*
 * Copyright 2018-2023 KMath 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 space.kscience.attributes

/**
 * A safe builder for [Attributes]
 *
 * @param O type marker of an owner object, for which these attributes are made
 */
public class AttributesBuilder<out O> internal constructor(
    private val map: MutableMap<Attribute<*>, Any?>,
) : Attributes {

    override fun toString(): String = "Attributes(value=${content.entries})"
    override fun equals(other: Any?): Boolean = other is Attributes && Attributes.equals(this, other)
    override fun hashCode(): Int = content.hashCode()

    override val content: Map<out Attribute<*>, Any?> get() = map

    public operator fun <T> set(attribute: Attribute<T>, value: T?) {
        if (value == null) {
            map.remove(attribute)
        } else {
            map[attribute] = value
        }
    }

    public operator fun <V> Attribute<V>.invoke(value: V?) {
        set(this, value)
    }

    public infix fun <V> Attribute<V>.put(value: V?) {
        set(this, value)
    }

    /**
     * Put all attributes for given [attributes]
     */
    public fun putAll(attributes: Attributes) {
        map.putAll(attributes.content)
    }

    public infix fun <V> SetAttribute<V>.add(attrValue: V) {
        val currentSet: Set<V> = get(this) ?: emptySet()
        map[this] = currentSet + attrValue
    }

    /**
     * Remove an element from [SetAttribute]
     */
    public infix fun <V> SetAttribute<V>.remove(attrValue: V) {
        val currentSet: Set<V> = get(this) ?: emptySet()
        map[this] = currentSet - attrValue
    }

    public fun build(): Attributes = AttributesImpl(map)
}

/**
 * Create [Attributes] with a given [builder]
 * @param O the type for which attributes are built. The type is used only during compilation phase for static extension dispatch
 */
public fun <O> Attributes(builder: AttributesBuilder<O>.() -> Unit): Attributes =
    AttributesBuilder<O>(mutableMapOf()).apply(builder).build()