io.github.freya022.botcommands.internal.commands.builder.CommandBuilderImpl.kt Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of BotCommands Show documentation

A Kotlin-first (and Java) framework that makes creating Discord bots a piece of cake, using the JDA library.

The newest version!

package io.github.freya022.botcommands.internal.commands.builder

import io.github.freya022.botcommands.api.commands.CommandPath
import io.github.freya022.botcommands.api.commands.CommandType
import io.github.freya022.botcommands.api.commands.builder.CommandBuilder
import io.github.freya022.botcommands.api.commands.builder.RateLimitBuilder
import io.github.freya022.botcommands.api.commands.ratelimit.RateLimitInfo
import io.github.freya022.botcommands.api.commands.ratelimit.RateLimiter
import io.github.freya022.botcommands.api.core.BContext
import io.github.freya022.botcommands.api.core.DeclarationSite
import io.github.freya022.botcommands.api.core.service.getService
import io.github.freya022.botcommands.api.core.setCallerAsDeclarationSite
import io.github.freya022.botcommands.api.core.utils.enumSetOf
import io.github.freya022.botcommands.internal.commands.ratelimit.RateLimitContainer
import io.github.freya022.botcommands.internal.utils.lazyPath
import net.dv8tion.jda.api.Permission
import java.util.*

internal abstract class CommandBuilderImpl internal constructor(
    override val context: BContext,
    override val name: String
) : CommandBuilder {
    internal abstract val type: CommandType
    final override lateinit var declarationSite: DeclarationSite

    final override var userPermissions: EnumSet = enumSetOf()

    final override var botPermissions: EnumSet = enumSetOf()

    final override val path: CommandPath by lazyPath()

    internal var rateLimitInfo: RateLimitInfo? = null
        private set

    final override fun rateLimit(
        rateLimiter: RateLimiter,
        block: RateLimitBuilder.() -> Unit
    ) {
        rateLimitInfo = RateLimitBuilderImpl("$type: ${path.fullPath}", rateLimiter)
            .setCallerAsDeclarationSite()
            .apply(block)
            .build()
    }

    final override fun rateLimitReference(group: String) {
        rateLimitInfo = context.getService()[group]
            ?: throw NoSuchElementException("Could not find a rate limiter for '$group'")
    }
}