io.github.freya022.botcommands.internal.parameters.ServiceMethodOption.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

import io.github.freya022.botcommands.api.core.service.ServiceContainer
import io.github.freya022.botcommands.api.parameters.AggregatedParameter
import io.github.freya022.botcommands.internal.core.options.OptionImpl
import io.github.freya022.botcommands.internal.core.options.OptionType
import io.github.freya022.botcommands.internal.core.service.tryGetWrappedService

internal class ServiceMethodOption internal constructor(
    override val parent: AggregatedParameter,
    optionParameter: OptionParameter,
    private val serviceContainer: ServiceContainer,
) : OptionImpl(optionParameter, OptionType.SERVICE) {

    override val executable get() = parent.executable

    private lateinit var cachedService: Any

    // Caches the service if:
    // 1. Is non-null
    // 2. Is not Lazy/List
    internal fun getService(): Any? {
        if (::cachedService.isInitialized) return cachedService

        val result = serviceContainer.tryGetWrappedService(kParameter)
        val service = if (isOptionalOrNullable) {
            result.getOrNull() ?: return null
        } else {
            result.getOrThrow()
        }

        // Always retry getting lazy services and lists as they may be updated later
        if (service is List<*>)
            return service
        if (service is Lazy<*>) {
            if (service.isInitialized() && service.value != null)
                cachedService = service
            return service
        } else {
            cachedService = service
            return service
        }
    }
}