io.github.freya022.botcommands.internal.commands.application.ApplicationCommandInfoImpl.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.application

import io.github.freya022.botcommands.api.commands.Usability.UnusableReason
import io.github.freya022.botcommands.api.commands.application.ApplicationCommandFilter
import io.github.freya022.botcommands.api.commands.application.ApplicationCommandInfo
import io.github.freya022.botcommands.api.core.Filter
import io.github.freya022.botcommands.api.core.Logging
import io.github.freya022.botcommands.api.core.entities.InputUser
import io.github.freya022.botcommands.api.core.utils.isSubclassOf
import io.github.freya022.botcommands.api.core.utils.loggerOf
import io.github.freya022.botcommands.api.core.utils.simpleNestedName
import io.github.freya022.botcommands.internal.ExecutableMixin
import io.github.freya022.botcommands.internal.commands.AbstractCommandInfoImpl
import io.github.freya022.botcommands.internal.commands.UsabilityImpl
import io.github.freya022.botcommands.internal.commands.application.builder.ApplicationCommandBuilderImpl
import io.github.freya022.botcommands.internal.commands.application.slash.SlashUtils.isFakeSlashFunction
import io.github.freya022.botcommands.internal.core.reflection.MemberParamFunction
import io.github.freya022.botcommands.internal.utils.classRef
import io.github.freya022.botcommands.internal.utils.reference
import io.github.freya022.botcommands.internal.utils.shortSignature
import io.github.freya022.botcommands.internal.utils.throwArgument
import io.github.oshai.kotlinlogging.KotlinLogging
import net.dv8tion.jda.api.entities.channel.middleman.GuildMessageChannel
import net.dv8tion.jda.api.entities.channel.middleman.MessageChannel
import net.dv8tion.jda.api.events.interaction.command.GenericCommandInteractionEvent
import kotlin.reflect.jvm.jvmErasure

private val logger = KotlinLogging.loggerOf()

internal abstract class ApplicationCommandInfoImpl internal constructor(
    builder: ApplicationCommandBuilderImpl<*>
) : AbstractCommandInfoImpl(builder),
    ApplicationCommandInfo,
    ExecutableMixin {

    internal val filters: List> = builder.filters.onEach { filter ->
        require(!filter.global) {
            "Global filter ${filter.javaClass.simpleNestedName} cannot be used explicitly, see ${Filter::global.reference}"
        }
    }

    override fun hasFilters(): Boolean = filters.isNotEmpty()

    // Using the builder to get the scope is required as the info object is still initializing
    // and would NPE when getting the top level instance
    protected inline fun  MemberParamFunction.checkEventScope(
        builder: ApplicationCommandBuilderImpl<*>
    ) {
        if (kFunction.isFakeSlashFunction()) return

        val eventType = firstParameter.type.jvmErasure
        if (builder.topLevelBuilder.scope.isGuildOnly) {
            if (!eventType.isSubclassOf()) {
                // Do not warn about guild-restricted types when everything is forced as a guild command
                if (builder.context.applicationConfig.forceGuildCommands) return

                Logging.getLogger().warn("${kFunction.shortSignature} : First parameter could be a ${classRef()} as to benefit from non-null getters")
            }
        } else if (eventType.isSubclassOf()) {
            throwArgument(kFunction, "Cannot use ${classRef()} on a global application command")
        }
    }

    final override fun getUsability(inputUser: InputUser, channel: MessageChannel): UsabilityImpl = UsabilityImpl.build {
        // Nothing to check outside a guild
        val member = inputUser.member
            ?: return@build logger.trace { "Skipping usability checks for non-members" }
        if (channel !is GuildMessageChannel)
            return@build logger.warn { "Cannot get usability outside of a ${classRef()}" }

        val guild = channel.guild
        if (!guild.selfMember.hasPermission(channel, botPermissions)) add(UnusableReason.BOT_PERMISSIONS)

        val isNotOwner = inputUser !in context.botOwners
        if (isNotOwner && !member.hasPermission(channel, userPermissions)) add(UnusableReason.USER_PERMISSIONS)
    }
}