fix: make DebugStage optional

Author: EmogurovAnton

plugins/plugin-servers/src/main/kotlin/com/redmadrobot/debug/plugin/servers/ServersPlugin.kt

@@ -35,11 +35,11 @@ public class ServersPlugin(
             return getPlugin<ServersPlugin>().getContainer<ServersPluginContainer>().serversRepository.getDefault()
         }
 
-        public fun getSelectedStage(): DebugStage {
+        public fun getSelectedStage(): DebugStage? {
             return getPlugin<ServersPlugin>().getContainer<ServersPluginContainer>().stagesRepository.getSelectedStage()
         }
 
-        public fun getDefaultStage(): DebugStage {
+        public fun getDefaultStage(): DebugStage? {
             return getPlugin<ServersPlugin>().getContainer<ServersPluginContainer>().stagesRepository.getDefault()
         }
     }

plugins/plugin-servers/src/main/kotlin/com/redmadrobot/debug/plugin/servers/data/DebugStageRepository.kt

@@ -34,7 +34,7 @@ internal class DebugStageRepository(
         }
     }
 
-    fun getSelectedStage(): DebugStage {
+    fun getSelectedStage(): DebugStage? {
         val stageName = sharedPreferences.getString(SELECTED_STAGE_NAME, null)
         val hostsHash = sharedPreferences.getInt(SELECTED_STAGE_HOSTS_HASH, -1)
 
@@ -48,8 +48,8 @@ internal class DebugStageRepository(
         }
     }
 
-    fun getDefault(): DebugStage {
-        return preInstalledStages.first { it.isDefault }
+    fun getDefault(): DebugStage? {
+        return preInstalledStages.firstOrNull { it.isDefault }
     }
 
     suspend fun getStages(): List<DebugStage> {

plugins/plugin-servers/src/main/kotlin/com/redmadrobot/debug/plugin/servers/data/model/DebugStage.kt

@@ -17,7 +17,22 @@ public data class DebugStage(
     }
 
     override fun equals(other: Any?): Boolean {
-        val otherServer = other as DebugStage
-        return this.hosts.size == otherServer.hosts.size && this.hosts == otherServer.hosts
+        if (this === other) return true
+        if (javaClass != other?.javaClass) return false
+
+        other as DebugStage
+
+        if (name != other.name) return false
+        if (hosts != other.hosts) return false
+        if (isDefault != other.isDefault) return false
+
+        return true
+    }
+
+    override fun hashCode(): Int {
+        var result = name.hashCode()
+        result = 31 * result + hosts.hashCode()
+        result = 31 * result + isDefault.hashCode()
+        return result
     }
 }

plugins/plugin-servers/src/main/kotlin/com/redmadrobot/debug/plugin/servers/interceptor/DebugStageInterceptor.kt

@@ -33,14 +33,19 @@ public class DebugStageInterceptor(private val hostName: String) : Interceptor {
         var request = chain.request()
         return if (DebugPanel.isInitialized) {
             val debugStage = stageRepository.getSelectedStage()
-            val host = debugStage.hosts[hostName]
-            if (host != null) {
-                val newUrl = request.getNewUrl(host)
-                request = request.newBuilder()
-                    .url(newUrl)
-                    .build()
-            }
-            chain.proceed(requestModifier?.invoke(request, debugStage) ?: request)
+            val modifiedRequest = debugStage?.let { stage ->
+                val host = stage.hosts[hostName]
+                if (host != null) {
+                    val newUrl = request.getNewUrl(host)
+                    request = request.newBuilder()
+                        .url(newUrl)
+                        .build()
+                }
+
+                requestModifier?.invoke(request, stage)
+            } ?: request
+
+            chain.proceed(modifiedRequest)
         } else {
             chain.proceed(request)
         }