pcimcioch.gitlabci.dsl.job.InheritDsl.kt Maven / Gradle / Ivy

Go to download
package pcimcioch.gitlabci.dsl.job

import kotlinx.serialization.PrimitiveDescriptor
import kotlinx.serialization.PrimitiveKind
import kotlinx.serialization.Serializable
import kotlinx.serialization.Transient
import kotlinx.serialization.builtins.serializer
import kotlinx.serialization.builtins.set
import pcimcioch.gitlabci.dsl.DslBase
import pcimcioch.gitlabci.dsl.StringRepresentation
import pcimcioch.gitlabci.dsl.serializer.MultiTypeSerializer
import pcimcioch.gitlabci.dsl.serializer.StringRepresentationSerializer

@Serializable
class InheritDsl : DslBase() {
    @Transient
    private var defaultBoolean: Boolean? = null

    @Transient
    private var variablesBoolean: Boolean? = null

    @Transient
    private var defaultSet: MutableSet? = null

    @Transient
    private var variablesSet: MutableSet? = null

    @Serializable(with = DefaultSerializer::class)
    var default: Any? = null
        get() = defaultBoolean ?: defaultSet
        private set

    @Serializable(with = VariablesSerializer::class)
    var variables: Any? = null
        get() = variablesBoolean ?: variablesSet
        private set

    fun default(vararg elements: InheritDefaultType) = default(elements.toList())
    fun default(elements: Iterable) {
        defaultBoolean = null
        ensureDefaultSet().addAll(elements)
    }

    fun default(value: Boolean) {
        defaultSet = null
        defaultBoolean = value
    }

    fun variables(vararg elements: String) = variables(elements.toList())
    fun variables(elements: Iterable) {
        variablesBoolean = null
        ensureVariablesSet().addAll(elements)
    }

    fun variables(value: Boolean) {
        variablesSet = null
        variablesBoolean = value
    }

    private fun ensureDefaultSet() = defaultSet ?: mutableSetOf().also { defaultSet = it }
    private fun ensureVariablesSet() = variablesSet ?: mutableSetOf().also { variablesSet = it }

    object VariablesSerializer : MultiTypeSerializer(
            PrimitiveDescriptor("Variables", PrimitiveKind.BOOLEAN),
            mapOf(
                    Boolean::class to Boolean.serializer(),
                    Set::class to String.serializer().set))

    object DefaultSerializer : MultiTypeSerializer(
            PrimitiveDescriptor("Default", PrimitiveKind.BOOLEAN),
            mapOf(
                    Boolean::class to Boolean.serializer(),
                    Set::class to InheritDefaultType.InheritDefaultTypeSerializer.set))

    companion object {
        init {
            addSerializer(InheritDsl::class, serializer())
        }
    }
}

fun createInherit(block: InheritDsl.() -> Unit = {}) = InheritDsl().apply(block)

@Serializable(with = InheritDefaultType.InheritDefaultTypeSerializer::class)
enum class InheritDefaultType(
        override val stringRepresentation: String
) : StringRepresentation {
    IMAGE("image"),
    SERVICES("services"),
    BEFORE_SCRIPT("before_script"),
    AFTER_SCRIPT("after_script"),
    TAGS("tags"),
    CACHE("cache"),
    ARTIFACTS("artifacts"),
    RETRY("retry"),
    TIMEOUT("timeout"),
    INTERRUPTIBLE("interruptible");

    object InheritDefaultTypeSerializer : StringRepresentationSerializer("InheritDefaultType")
}