Sending the user agent header with every HTTP request

Author: holasoyender

src/main/kotlin/dev/arbjerg/lavalink/client/LavalinkClient.kt

@@ -19,9 +19,9 @@ import java.util.concurrent.TimeUnit
 
 /**
  * @param userId ID of the bot for authenticating with Discord
- * @param clientName Name of the client, used for connecting to the Lavalink node(s)
+ * @param userAgent Name of the client, used for connecting to the Lavalink node(s)
  */
-class LavalinkClient(val userId: Long, var clientName: String?) : Closeable, Disposable {
+class LavalinkClient(val userId: Long, var userAgent: String?) : Closeable, Disposable {
     private val internalNodes = CopyOnWriteArrayList<LavalinkNode>()
     private val linkMap = ConcurrentHashMap<Long, Link>()
     private var clientOpen = true

src/main/kotlin/dev/arbjerg/lavalink/client/LavalinkNode.kt

@@ -55,8 +55,8 @@ class LavalinkNode(
     val flux: Flux<ClientEvent> = sink.asFlux()
     private val reference: Disposable = flux.subscribe()
 
-    internal val rest = LavalinkRestClient(this)
-    val ws = LavalinkSocket(this, lavalink.clientName)
+    internal val rest = LavalinkRestClient(this, lavalink.userAgent)
+    val ws = LavalinkSocket(this, lavalink.userAgent)
 
     // Stuff for load balancing
     val penalties = Penalties(this)

src/main/kotlin/dev/arbjerg/lavalink/internal/LavalinkRestClient.kt

@@ -15,7 +15,8 @@ import java.io.IOException
 import java.net.URLEncoder
 import java.nio.charset.StandardCharsets
 
-class LavalinkRestClient(val node: LavalinkNode) {
+class LavalinkRestClient(val node: LavalinkNode, private val userAgent: String?) {
+
     fun getPlayers(): Mono<Players> {
         return newRequest {
             path("/v4/sessions/${node.sessionId}/players")
@@ -82,6 +83,10 @@ class LavalinkRestClient(val node: LavalinkNode) {
     internal fun newRequest(configure: HttpBuilder.() -> HttpBuilder): Call {
         val requestBuilder = Request.Builder()
             .addHeader("Authorization", node.password)
+            .apply {
+                if (userAgent != null)
+                    addHeader("User-Agent", userAgent)
+            }
             .get()
         val builder = configure(HttpBuilder(requestBuilder))