com.freya02.botcommands.internal.commands.application.ApplicationCommandDataMap.kt Maven / Gradle / Ivy

Go to download

package com.freya02.botcommands.internal.commands.application

import com.freya02.botcommands.api.commands.CommandPath
import com.freya02.botcommands.internal.enumMapOf
import net.dv8tion.jda.api.interactions.commands.build.CommandData
import java.util.function.Function
import net.dv8tion.jda.api.interactions.commands.Command.Type as CommandType

internal class ApplicationCommandDataMap {
    //The String is CommandPath's base name
    private val typeMap: MutableMap> = enumMapOf()

    val allCommandData: Collection
        get() = typeMap
            .values
            .flatMap { it.values }

    fun computeIfAbsent(type: CommandType, path: CommandPath, mappingFunction: Function): CommandData {
        return this[type].computeIfAbsent(path.name, mappingFunction)
    }

    operator fun set(type: CommandType, name: String, value: CommandData) {
        this[type][name] = value
    }

    private operator fun get(type: CommandType) = typeMap.computeIfAbsent(type) { hashMapOf() }
}