SWF Filtering implementation (#32)

Author: ShiftHackZ

build.gradle

@@ -1,6 +1,6 @@
 buildscript {
     ext {
-        appVersion = "0.3.0"
+        appVersion = "0.3.1"
         minSdk = 26
         targetSdk = 33
     }

data/src/main/java/com/shifthackz/aisdv1/data/remote/StableDiffusionGenerationRemoteDataSource.kt

@@ -9,6 +9,7 @@ import com.shifthackz.aisdv1.domain.entity.TextToImagePayload
 import com.shifthackz.aisdv1.network.api.StableDiffusionWebUiAutomaticRestApi
 import com.shifthackz.aisdv1.network.api.StableDiffusionWebUiAutomaticRestApi.Companion.PATH_IMG_TO_IMG
 import com.shifthackz.aisdv1.network.api.StableDiffusionWebUiAutomaticRestApi.Companion.PATH_TXT_TO_IMG
+import com.shifthackz.aisdv1.network.extensions.withExceptionMapper
 import com.shifthackz.aisdv1.network.response.SdGenerationResponse
 
 internal class StableDiffusionGenerationRemoteDataSource(
@@ -25,9 +26,11 @@ internal class StableDiffusionGenerationRemoteDataSource(
         .flatMap { url -> api.textToImage(url, payload.mapToRequest()) }
         .map { response -> payload to response }
         .map(Pair<TextToImagePayload, SdGenerationResponse>::mapToAiGenResult)
+        .withExceptionMapper()
 
     override fun imageToImage(payload: ImageToImagePayload) = serverUrlProvider(PATH_IMG_TO_IMG)
         .flatMap { url -> api.imageToImage(url, payload.mapToRequest()) }
         .map { response -> payload to response }
         .map(Pair<ImageToImagePayload, SdGenerationResponse>::mapToAiGenResult)
+        .withExceptionMapper()
 }

network/src/main/java/com/shifthackz/aisdv1/network/exception/BadKeywordException.kt

@@ -0,0 +1,3 @@
+package com.shifthackz.aisdv1.network.exception
+
+class BadKeywordException(val keyword: String) : Throwable("Keyword '$keyword' is not allowed")

network/src/main/java/com/shifthackz/aisdv1/network/extensions/ErrorParserExtensions.kt

@@ -0,0 +1,16 @@
+package com.shifthackz.aisdv1.network.extensions
+
+import com.shifthackz.aisdv1.network.exception.BadKeywordException
+import io.reactivex.rxjava3.core.Single
+import retrofit2.HttpException
+
+fun <T: Any> Single<T>.withExceptionMapper(): Single<T> =
+    onErrorResumeNext { t -> Single.error(t.mapException()) }
+
+private fun Throwable.mapException(): Throwable {
+    if (this is HttpException && code() == 400) {
+        val keyword = response()?.errorBody()?.string()
+        return keyword?.let(::BadKeywordException) ?: this
+    }
+    return this
+}