Implement outgoing Jukebox messaging

Author: FluxCapacitor2

common/src/main/kotlin/com/bluedragonmc/server/api/OutgoingRPCHandler.kt

@@ -2,6 +2,7 @@ package com.bluedragonmc.server.api
 
 import com.bluedragonmc.api.grpc.CommonTypes.GameState
 import com.bluedragonmc.api.grpc.CommonTypes.GameType
+import com.bluedragonmc.api.grpc.JukeboxOuterClass.PlayerSongQueue
 import com.bluedragonmc.api.grpc.PartySvc.PartyListResponse
 import com.bluedragonmc.api.grpc.PlayerTrackerOuterClass.QueryPlayerResponse
 import com.bluedragonmc.server.Game
@@ -50,4 +51,11 @@ interface OutgoingRPCHandler {
     suspend fun transferParty(partyOwner: Player, newOwner: UUID)
     suspend fun listPartyMembers(member: UUID): PartyListResponse
 
+    // Jukebox controls
+    suspend fun getSongInfo(player: Player): PlayerSongQueue
+    suspend fun playSong(player: Player, songName: String, queuePosition: Int, startTimeInTicks: Int, tags: List<String>): Boolean
+    suspend fun removeSongByName(player: Player, songName: String)
+    suspend fun removeSongByTag(player: Player, matchTags: List<String>)
+    suspend fun stopSongAndClearQueue(player: Player)
+
 }

common/src/main/kotlin/com/bluedragonmc/server/api/OutgoingRPCHandlerStub.kt

@@ -1,8 +1,6 @@
 package com.bluedragonmc.server.api
 
-import com.bluedragonmc.api.grpc.CommonTypes
-import com.bluedragonmc.api.grpc.PartySvc
-import com.bluedragonmc.api.grpc.PlayerTrackerOuterClass
+import com.bluedragonmc.api.grpc.*
 import com.bluedragonmc.server.Game
 import net.kyori.adventure.text.Component
 import net.minestom.server.command.CommandSender
@@ -100,4 +98,31 @@ class OutgoingRPCHandlerStub : OutgoingRPCHandler {
         return PartySvc.PartyListResponse.getDefaultInstance()
     }
 
+    override suspend fun getSongInfo(player: Player): JukeboxOuterClass.PlayerSongQueue {
+        return playerSongQueue {
+            isPlaying = false
+        }
+    }
+
+    override suspend fun playSong(
+        player: Player,
+        songName: String,
+        queuePosition: Int,
+        startTimeInTicks: Int,
+        tags: List<String>,
+    ): Boolean {
+        return false
+    }
+
+    override suspend fun removeSongByName(player: Player, songName: String) {
+
+    }
+
+    override suspend fun removeSongByTag(player: Player, matchTags: List<String>) {
+
+    }
+
+    override suspend fun stopSongAndClearQueue(player: Player) {
+
+    }
 }

gradle/libs.versions.toml

@@ -11,7 +11,7 @@ okhttp = "4.10.0"
 serialization = "1.5.0-RC"
 tinylog = "2.6.2"
 # Auto-generated GRPC/Protobuf messaging code
-rpc = "0953adbab3"
+rpc = "b7071251fb"
 # Agones SDK and its necessary runtime dependencies
 agones-kt = "0.1.2"
 grpc = "1.50.2"

src/main/kotlin/com/bluedragonmc/server/impl/OutgoingRPCHandlerImpl.kt

@@ -15,6 +15,8 @@ import com.bluedragonmc.server.service.Messaging
 import com.bluedragonmc.server.utils.listen
 import com.bluedragonmc.server.utils.listenSuspend
 import com.bluedragonmc.server.utils.miniMessage
+import com.github.benmanes.caffeine.cache.Caffeine
+import io.grpc.ManagedChannel
 import io.grpc.ManagedChannelBuilder
 import kotlinx.coroutines.launch
 import kotlinx.coroutines.runBlocking
@@ -28,6 +30,7 @@ import net.minestom.server.event.instance.AddEntityToInstanceEvent
 import net.minestom.server.event.player.PlayerDisconnectEvent
 import net.minestom.server.event.player.PlayerSpawnEvent
 import net.minestom.server.instance.Instance
+import java.time.Duration
 import java.util.*
 import java.util.concurrent.TimeUnit
 
@@ -111,6 +114,27 @@ class OutgoingRPCHandlerImpl(serverAddress: String) : OutgoingRPCHandler {
     private val queueStub = QueueServiceGrpcKt.QueueServiceCoroutineStub(channel)
     private val partyStub = PartyServiceGrpcKt.PartyServiceCoroutineStub(channel)
 
+    private val proxyChannelCache = Caffeine
+        .newBuilder()
+        .expireAfterAccess(Duration.ofMinutes(10))
+        .evictionListener<String, ManagedChannel> { _, value, _ ->
+            value?.shutdown()?.awaitTermination(10, TimeUnit.SECONDS)
+        }
+        .build<String, ManagedChannel>()
+
+    private fun getChannelToProxyOf(player: Player): ManagedChannel {
+        val address = player.playerConnection.remoteAddress.toString()
+        return proxyChannelCache.get(address) { _ ->
+            ManagedChannelBuilder.forAddress(address, 50051)
+                .defaultLoadBalancingPolicy("round_robin")
+                .usePlaintext()
+                .enableRetry()
+                .build()
+        }
+    }
+
+    private fun getJukeboxStubOf(player: Player) = JukeboxGrpcKt.JukeboxCoroutineStub(getChannelToProxyOf(player))
+
     override fun isConnected(): Boolean {
         return !channel.isShutdown && !channel.isTerminated && ::serverName.isInitialized
     }
@@ -299,4 +323,46 @@ class OutgoingRPCHandlerImpl(serverAddress: String) : OutgoingRPCHandler {
                 .build()
         )
     }
-}
+
+    override suspend fun getSongInfo(player: Player): JukeboxOuterClass.PlayerSongQueue {
+        return getJukeboxStubOf(player).getSongInfo(songInfoRequest {
+            playerUuid = player.uuid.toString()
+        })
+    }
+
+    override suspend fun playSong(
+        player: Player,
+        songName: String,
+        queuePosition: Int,
+        startTimeInTicks: Int,
+        tags: List<String>,
+    ): Boolean {
+        return getJukeboxStubOf(player).playSong(playSongRequest {
+            this.playerUuid = player.uuid.toString()
+            this.songName = songName
+            this.queuePosition = queuePosition
+            this.startTimeTicks = startTimeInTicks
+            tags.forEach { this.tags.add(it) }
+        }).startedPlaying
+    }
+
+    override suspend fun removeSongByName(player: Player, songName: String) {
+        getJukeboxStubOf(player).removeSong(songRemoveRequest {
+            this.playerUuid = player.uuid.toString()
+            this.songName = songName
+        })
+    }
+
+    override suspend fun removeSongByTag(player: Player, matchTags: List<String>) {
+        getJukeboxStubOf(player).removeSongs(batchSongRemoveRequest {
+            this.playerUuid = player.uuid.toString()
+            matchTags.forEach { this.matchTags.add(it) }
+        })
+    }
+
+    override suspend fun stopSongAndClearQueue(player: Player) {
+        getJukeboxStubOf(player).stopSong(stopSongRequest {
+            this.playerUuid = player.uuid.toString()
+        })
+    }
+}