io.github.freya022.botcommands.internal.parameters.resolvers.localization.LocalizationContextResolvers.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.

There is a newer version: 3.0.0-alpha.18

package io.github.freya022.botcommands.internal.parameters.resolvers.localization

import io.github.freya022.botcommands.api.localization.context.AppLocalizationContext
import io.github.freya022.botcommands.api.localization.context.TextLocalizationContext
import io.github.freya022.botcommands.api.parameters.ClassParameterResolver
import io.github.freya022.botcommands.api.parameters.resolvers.ICustomResolver
import io.github.freya022.botcommands.internal.IExecutableInteractionInfo
import io.github.freya022.botcommands.internal.localization.LocalizationContextImpl
import io.github.freya022.botcommands.internal.utils.throwInternal
import net.dv8tion.jda.api.events.Event
import net.dv8tion.jda.api.events.message.MessageReceivedEvent
import net.dv8tion.jda.api.interactions.Interaction

internal class AppLocalizationContextResolver(private val baseContext: LocalizationContextImpl) :
    ClassParameterResolver(AppLocalizationContext::class),
    ICustomResolver {

    override suspend fun resolveSuspend(info: IExecutableInteractionInfo, event: Event): AppLocalizationContext {
        return when (event) {
            is Interaction -> baseContext.withLocales(event.guildLocale, event.userLocale)
            //MessageReceivedEvent does not provide user locale
            else -> throwInternal("Unsupported event type for localization contexts: ${event.javaClass.name}")
        }
    }
}

internal class TextLocalizationContextResolver(private val baseContext: LocalizationContextImpl) :
    ClassParameterResolver(TextLocalizationContext::class),
    ICustomResolver {

    override suspend fun resolveSuspend(info: IExecutableInteractionInfo, event: Event): TextLocalizationContext {
        return when (event) {
            is Interaction -> baseContext.withLocales(event.guildLocale, event.userLocale)
            is MessageReceivedEvent -> when {
                event.isFromGuild -> baseContext.withGuildLocale(event.guild.locale)
                else -> baseContext
            }
            else -> throwInternal("Unsupported event type for localization contexts: ${event.javaClass.name}")
        }
    }
}