fix: skip network requests for inactive server sources

- GetStableDiffusionModelsUseCaseImpl: only fetch when A1111 is active
- FetchAndGetSwarmUiModelsUseCaseImpl: only fetch when SwarmUI is active
- FetchAndGetHuggingFaceModelsUseCaseImpl: only fetch when HuggingFace is active
- FetchAndGetStabilityAiEnginesUseCaseImpl: only fetch when StabilityAI is active

This prevents unnecessary network requests to servers that are not currently
configured as the active generation source.

Author: crim50n

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/huggingface/FetchAndGetHuggingFaceModelsUseCaseImpl.kt

@@ -1,10 +1,20 @@
 package com.shifthackz.aisdv1.domain.usecase.huggingface
 
+import com.shifthackz.aisdv1.domain.entity.HuggingFaceModel
+import com.shifthackz.aisdv1.domain.entity.ServerSource
+import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.HuggingFaceModelsRepository
+import io.reactivex.rxjava3.core.Single
 
 internal class FetchAndGetHuggingFaceModelsUseCaseImpl(
+    private val preferenceManager: PreferenceManager,
     private val huggingFaceModelsRepository: HuggingFaceModelsRepository,
 ) : FetchAndGetHuggingFaceModelsUseCase {
 
-    override fun invoke() = huggingFaceModelsRepository.fetchAndGetHuggingFaceModels()
+    override fun invoke(): Single<List<HuggingFaceModel>> {
+        if (preferenceManager.source != ServerSource.HUGGING_FACE) {
+            return Single.just(emptyList())
+        }
+        return huggingFaceModelsRepository.fetchAndGetHuggingFaceModels()
+    }
 }

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/sdmodel/GetStableDiffusionModelsUseCaseImpl.kt

@@ -1,17 +1,24 @@
 package com.shifthackz.aisdv1.domain.usecase.sdmodel
 
+import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.entity.StableDiffusionModel
+import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.ServerConfigurationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionModelsRepository
 import io.reactivex.rxjava3.core.Single
 
 internal class GetStableDiffusionModelsUseCaseImpl(
+    private val preferenceManager: PreferenceManager,
     private val serverConfigurationRepository: ServerConfigurationRepository,
     private val sdModelsRepository: StableDiffusionModelsRepository,
 ) : GetStableDiffusionModelsUseCase {
 
-    override operator fun invoke(): Single<List<Pair<StableDiffusionModel, Boolean>>> =
-        serverConfigurationRepository
+    override operator fun invoke(): Single<List<Pair<StableDiffusionModel, Boolean>>> {
+        // Only fetch models if A1111/Forge is the active source
+        if (preferenceManager.source != ServerSource.AUTOMATIC1111) {
+            return Single.just(emptyList())
+        }
+        return serverConfigurationRepository
             .fetchAndGetConfiguration()
             .flatMap { config ->
                 sdModelsRepository
@@ -23,4 +30,5 @@ internal class GetStableDiffusionModelsUseCaseImpl(
                     model to (config.sdModelCheckpoint == model.title)
                 }
             }
+    }
 }

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/stabilityai/FetchAndGetStabilityAiEnginesUseCaseImpl.kt

@@ -1,5 +1,6 @@
 package com.shifthackz.aisdv1.domain.usecase.stabilityai
 
+import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.entity.StabilityAiEngine
 import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.StabilityAiEnginesRepository
@@ -10,12 +11,17 @@ internal class FetchAndGetStabilityAiEnginesUseCaseImpl(
     private val preferenceManager: PreferenceManager,
 ) : FetchAndGetStabilityAiEnginesUseCase {
 
-    override fun invoke() = repository
-        .fetchAndGet()
-        .flatMap { engines ->
-            if (!engines.map(StabilityAiEngine::id).contains(preferenceManager.stabilityAiEngineId)) {
-                preferenceManager.stabilityAiEngineId = engines.firstOrNull()?.id ?: ""
-            }
-            Single.just(engines)
+    override fun invoke(): Single<List<StabilityAiEngine>> {
+        if (preferenceManager.source != ServerSource.STABILITY_AI) {
+            return Single.just(emptyList())
         }
+        return repository
+            .fetchAndGet()
+            .flatMap { engines ->
+                if (!engines.map(StabilityAiEngine::id).contains(preferenceManager.stabilityAiEngineId)) {
+                    preferenceManager.stabilityAiEngineId = engines.firstOrNull()?.id ?: ""
+                }
+                Single.just(engines)
+            }
+    }
 }

domain/src/main/java/com/shifthackz/aisdv1/domain/usecase/swarmmodel/FetchAndGetSwarmUiModelsUseCaseImpl.kt

@@ -1,5 +1,6 @@
 package com.shifthackz.aisdv1.domain.usecase.swarmmodel
 
+import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.entity.SwarmUiModel
 import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.SwarmUiModelsRepository
@@ -10,12 +11,17 @@ internal class FetchAndGetSwarmUiModelsUseCaseImpl(
     private val repository: SwarmUiModelsRepository,
 ) : FetchAndGetSwarmUiModelsUseCase {
 
-    override fun invoke(): Single<List<SwarmUiModel>> = repository
-        .fetchAndGetModels()
-        .map { models ->
-            if (!models.map(SwarmUiModel::name).contains(preferenceManager.swarmUiModel)) {
-                preferenceManager.swarmUiModel = models.firstOrNull()?.name ?: ""
-            }
-            models
+    override fun invoke(): Single<List<SwarmUiModel>> {
+        if (preferenceManager.source != ServerSource.SWARM_UI) {
+            return Single.just(emptyList())
         }
+        return repository
+            .fetchAndGetModels()
+            .map { models ->
+                if (!models.map(SwarmUiModel::name).contains(preferenceManager.swarmUiModel)) {
+                    preferenceManager.swarmUiModel = models.firstOrNull()?.name ?: ""
+                }
+                models
+            }
+    }
 }

domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/huggingface/FetchAndGetHuggingFaceModelsUseCaseImplTest.kt

@@ -2,16 +2,29 @@ package com.shifthackz.aisdv1.domain.usecase.huggingface
 
 import com.nhaarman.mockitokotlin2.mock
 import com.nhaarman.mockitokotlin2.whenever
+import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.mocks.mockHuggingFaceModels
+import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.HuggingFaceModelsRepository
 import io.reactivex.rxjava3.core.Single
+import org.junit.Before
 import org.junit.Test
 
 class FetchAndGetHuggingFaceModelsUseCaseImplTest {
 
+    private val stubPreferenceManager = mock<PreferenceManager>()
     private val stubRepository = mock<HuggingFaceModelsRepository>()
 
-    private val useCase = FetchAndGetHuggingFaceModelsUseCaseImpl(stubRepository)
+    private val useCase = FetchAndGetHuggingFaceModelsUseCaseImpl(
+        preferenceManager = stubPreferenceManager,
+        huggingFaceModelsRepository = stubRepository,
+    )
+
+    @Before
+    fun initialize() {
+        whenever(stubPreferenceManager.source)
+            .thenReturn(ServerSource.HUGGING_FACE)
+    }
 
     @Test
     fun `given repository provided models list, expected valid list value`() {

domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/sdmodel/GetStableDiffusionModelsUseCaseImplTest.kt

@@ -3,24 +3,35 @@ package com.shifthackz.aisdv1.domain.usecase.sdmodel
 import com.nhaarman.mockitokotlin2.mock
 import com.nhaarman.mockitokotlin2.whenever
 import com.shifthackz.aisdv1.domain.entity.ServerConfiguration
+import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.mocks.mockServerConfiguration
 import com.shifthackz.aisdv1.domain.mocks.mockStableDiffusionModels
+import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.ServerConfigurationRepository
 import com.shifthackz.aisdv1.domain.repository.StableDiffusionModelsRepository
 import io.reactivex.rxjava3.core.Single
 import org.junit.Assert
+import org.junit.Before
 import org.junit.Test
 
 class GetStableDiffusionModelsUseCaseImplTest {
 
+    private val stubPreferenceManager = mock<PreferenceManager>()
     private val stubServerConfigurationRepository = mock<ServerConfigurationRepository>()
     private val stubSdModelsRepository = mock<StableDiffusionModelsRepository>()
 
     private val useCase = GetStableDiffusionModelsUseCaseImpl(
+        preferenceManager = stubPreferenceManager,
         serverConfigurationRepository = stubServerConfigurationRepository,
         sdModelsRepository = stubSdModelsRepository,
     )
 
+    @Before
+    fun initialize() {
+        whenever(stubPreferenceManager.source)
+            .thenReturn(ServerSource.AUTOMATIC1111)
+    }
+
     @Test
     fun `given repository returns list with value present in configuration, expected list with selected value`() {
         whenever(stubServerConfigurationRepository.fetchAndGetConfiguration())
@@ -105,4 +116,17 @@ class GetStableDiffusionModelsUseCaseImplTest {
             .await()
             .assertNotComplete()
     }
+
+    @Test
+    fun `given source is not AUTOMATIC1111, expected empty list without network requests`() {
+        whenever(stubPreferenceManager.source)
+            .thenReturn(ServerSource.FAL_AI)
+
+        useCase()
+            .test()
+            .assertNoErrors()
+            .assertValue(emptyList())
+            .await()
+            .assertComplete()
+    }
 }

domain/src/test/java/com/shifthackz/aisdv1/domain/usecase/stabilityai/FetchAndGetStabilityAiEnginesUseCaseImplTest.kt

@@ -2,10 +2,12 @@ package com.shifthackz.aisdv1.domain.usecase.stabilityai
 
 import com.nhaarman.mockitokotlin2.mock
 import com.nhaarman.mockitokotlin2.whenever
+import com.shifthackz.aisdv1.domain.entity.ServerSource
 import com.shifthackz.aisdv1.domain.mocks.mockStabilityAiEngines
 import com.shifthackz.aisdv1.domain.preference.PreferenceManager
 import com.shifthackz.aisdv1.domain.repository.StabilityAiEnginesRepository
 import io.reactivex.rxjava3.core.Single
+import org.junit.Before
 import org.junit.Test
 
 class FetchAndGetStabilityAiEnginesUseCaseImplTest {
@@ -19,6 +21,12 @@ class FetchAndGetStabilityAiEnginesUseCaseImplTest {
         preferenceManager = stubPreferenceManager,
     )
 
+    @Before
+    fun initialize() {
+        whenever(stubPreferenceManager.source)
+            .thenReturn(ServerSource.STABILITY_AI)
+    }
+
     @Test
     fun `given repository returned engines list, id present in preference, expected the same engines list, id not changed`() {
         whenever(stubRepository.fetchAndGet())