main.dev.schlaubi.lavakord.interop.RoutePlannerUtil.kt Maven / Gradle / Ivy

Go to download

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

Coroutine based client for Lavalink (Kotlin and Java)

There is a newer version: 7.1.0

@file:JvmName("RoutePlannerUtil")

package dev.schlaubi.lavakord.interop

import dev.arbjerg.lavalink.protocol.v4.RoutePlannerStatus
import dev.schlaubi.lavakord.NoRoutePlannerException
import dev.schlaubi.lavakord.audio.Node
import dev.schlaubi.lavakord.rest.addressStatusOrNull
import dev.schlaubi.lavakord.rest.unmarkAddress
import dev.schlaubi.lavakord.rest.unmarkAllAddresses
import java.util.concurrent.CompletableFuture

private fun Node.javaAddressStatusOrNull() =
    lavakord.supply { addressStatusOrNull() }

/**
 * Retrieves the current address status of the route planner api. Can be null if no Route planner is set.
 *
 * @param node the [Node] to retrieve the [RoutePlannerStatus] from
 *
 * @see RoutePlannerStatus
 * @see Node.addressStatusOrNull
 */
public fun addressStatusOrNull(node: Node): CompletableFuture =
    node.javaAddressStatusOrNull()

/**
 * Retrieves the current address status of the route planner api. Can be null if no Route planner is set.
 *
 * @param node the [Node] to retrieve the [RoutePlannerStatus] from
 *
 * @throws NoRoutePlannerException when no route planner is set
 * @see RoutePlannerStatus
 * @see Node.addressStatusOrNull
 */
public fun addressStatus(node: Node): CompletableFuture =
    node.javaAddressStatusOrNull().thenApply { it ?: throw NoRoutePlannerException() }

private fun Node.javaUnmarkAllAddresses() = lavakord.run { unmarkAllAddresses() }

/**
 * Unmarks all failed route planner addresses.
 *
 * @param node the [Node] to unmark all addresses on
 */
public fun unmarkAllAddresses(node: Node): CompletableFuture = node.javaUnmarkAllAddresses()

/**
 * Unmarks all failed route planner addresses.
 *
 * @param link the [JavaLink] to unmark all addresses on
 */
public fun unmarkAllAddresses(link: JavaLink): CompletableFuture =
    link.node.javaUnmarkAllAddresses()

private fun Node.javaUnmarkAddress(address: String) = lavakord.run {
    unmarkAddress(address)
}

/**
 * Unmarks the route planner [address].
 *
 * @param node the [Node] to unmark [address] on
 */
public fun unmarkAddress(node: Node, address: String): CompletableFuture = node.javaUnmarkAddress(address)

/**
 * Unmarks the route planner [address].
 *
 * @param link the [JavaLink] to unmark [address] on
 */
public fun unmarkAddress(link: JavaLink, address: String): CompletableFuture =
    link.node.javaUnmarkAddress(address)