Implement API for retrieve ICEServer config

fioan89 · fioan89 · commit 13e090bfa3bd · 2022-05-27T00:38:32.000+03:00
- gson type adapter for the RTCICeServer
- retrofit calls to retrieve ICE configurations
diff --git a/src/main/kotlin/com/coder/gateway/sdk/CoderClientService.kt b/src/main/kotlin/com/coder/gateway/sdk/CoderClientService.kt
@@ -1,14 +1,16 @@
 package com.coder.gateway.sdk
 
+import com.coder.gateway.models.SSHKeys
 import com.coder.gateway.models.UriScheme
 import com.coder.gateway.models.User
-import com.coder.gateway.models.SSHKeys
 import com.coder.gateway.models.Workspace
 import com.coder.gateway.sdk.ex.AuthenticationException
 import com.google.gson.Gson
 import com.google.gson.GsonBuilder
 import com.intellij.openapi.components.Service
 import com.jetbrains.gateway.sdk.convertors.InstantConverter
+import com.jetbrains.gateway.sdk.convertors.RTCIceServerAdapter
+import dev.onvoid.webrtc.RTCIceServer
 import okhttp3.OkHttpClient
 import okhttp3.logging.HttpLoggingInterceptor
 import retrofit2.Retrofit
@@ -31,6 +33,7 @@ class CoderClientService {
 
         val gson: Gson = GsonBuilder()
             .registerTypeAdapter(Instant::class.java, InstantConverter())
+            .registerTypeAdapter(RTCIceServer::class.java, RTCIceServerAdapter())
             .setPrettyPrinting()
             .create()
 
@@ -77,4 +80,12 @@ class CoderClientService {
 
         return sshKeysResponse.body()!!
     }
-}
+
+    fun iceServers(): List<RTCIceServer> {
+        val iceServersResponse = retroRestClient.iceServers(sessionToken).execute()
+        if (!iceServersResponse.isSuccessful) {
+            throw IllegalStateException("Could not retrieve retrieve ICE servers:${iceServersResponse.code()}, reason: ${iceServersResponse.message()}")
+        }
+        return iceServersResponse.body()!!.iceServers
+    }
+}
diff --git a/src/main/kotlin/com/coder/gateway/sdk/CoderRestService.kt b/src/main/kotlin/com/coder/gateway/sdk/CoderRestService.kt
@@ -1,7 +1,7 @@
 package com.coder.gateway.sdk
 
-import com.coder.gateway.models.User
 import com.coder.gateway.models.SSHKeys
+import com.coder.gateway.models.User
 import com.coder.gateway.models.Workspace
 import retrofit2.Call
 import retrofit2.http.Body
@@ -25,4 +25,7 @@ interface CoderRestService {
 
     @GET("api/v0/users/{userId}/sshkey")
     fun sshKeys(@Header("Session-Token") sessionToken: String, @Path("userId") userID: String): Call<SSHKeys>
+
+    @GET("api/private/webrtc/ice")
+    fun iceServers(@Header("Session-Token") sessionToken: String): Call<IceServersWrapper>
 }
diff --git a/src/main/kotlin/com/coder/gateway/sdk/IceServersWrapper.kt b/src/main/kotlin/com/coder/gateway/sdk/IceServersWrapper.kt
@@ -0,0 +1,6 @@
+package com.coder.gateway.sdk
+
+import com.google.gson.annotations.SerializedName
+import dev.onvoid.webrtc.RTCIceServer
+
+data class IceServersWrapper(@SerializedName("data") val iceServers: List<RTCIceServer>)
diff --git a/src/main/kotlin/com/jetbrains/gateway/sdk/convertors/RTCIceServerAdapter.kt b/src/main/kotlin/com/jetbrains/gateway/sdk/convertors/RTCIceServerAdapter.kt
@@ -0,0 +1,52 @@
+package com.jetbrains.gateway.sdk.convertors
+
+import com.google.gson.TypeAdapter
+import com.google.gson.stream.JsonReader
+import com.google.gson.stream.JsonWriter
+import dev.onvoid.webrtc.RTCIceServer
+
+private const val URLS = "urls"
+
+private const val USERNAME = "username"
+
+private const val PASSWORD = "password"
+
+/**
+ * GSON adapter for converting [RTCIceServer] objects.
+ */
+class RTCIceServerAdapter : TypeAdapter<RTCIceServer>() {
+    override fun write(writer: JsonWriter, iceServer: RTCIceServer) {
+        writer.beginObject()
+        writer.name(URLS).beginArray()
+        iceServer.urls.forEach {
+            writer.value(it)
+        }
+        writer.endArray()
+        writer.name(USERNAME).value(iceServer.username)
+        writer.name(PASSWORD).value(iceServer.password)
+        writer.endObject()
+    }
+
+    override fun read(reader: JsonReader): RTCIceServer {
+        val iceServer = RTCIceServer()
+        reader.beginObject()
+        while (reader.hasNext()) {
+            when (reader.nextName()) {
+                URLS -> {
+                    val urls = mutableListOf<String>()
+                    reader.beginArray()
+                    while (reader.hasNext()) {
+                        urls.add(reader.nextString())
+                    }
+                    reader.endArray()
+                    iceServer.urls = urls
+                }
+                USERNAME -> iceServer.username = reader.nextString()
+                "credential" -> iceServer.password = reader.nextString()
+            }
+        }
+        reader.endObject()
+        return iceServer
+    }
+
+}
