me.jakejmattson.kutils.api.arguments.MemberArg.kt Maven / Gradle / Ivy

Go to download

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

A Discord bot framework for Kotlin.

There is a newer version: 0.18.1

package me.jakejmattson.kutils.api.arguments

import me.jakejmattson.kutils.api.dsl.arguments.*
import me.jakejmattson.kutils.api.dsl.command.CommandEvent
import me.jakejmattson.kutils.api.extensions.stdlib.trimToID
import net.dv8tion.jda.api.entities.Member

open class MemberArg(override val name: String = "Member", private val allowsBot: Boolean = false) : ArgumentType() {
    companion object : MemberArg()

    override fun convert(arg: String, args: List, event: CommandEvent<*>): ArgumentResult {
        val guild = event.guild ?: return ArgumentResult.Error("$name cannot be accessed from outside a guild.")
        val id = arg.trimToID()

        val member = guild.getMemberById(id)
            ?: return ArgumentResult.Error("Couldn't retrieve $name from $arg.")

        if (!allowsBot && member.user.isBot)
            return ArgumentResult.Error("$name cannot be a bot.")

        return ArgumentResult.Success(member)
    }

    override fun generateExamples(event: CommandEvent<*>) = listOf(event.author.id)
}