fix: make image optional

therajanmaurya · Oct 21, 2022 · 5f812bb · 5f812bb
1 parent 0077156
commit 5f812bb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/Sweet-Error/src/main/java/com/github/therajanmaurya/sweeterror/SweetUIErrorHandler.kt b/Sweet-Error/src/main/java/com/github/therajanmaurya/sweeterror/SweetUIErrorHandler.kt
@@ -153,7 +153,7 @@ class SweetUIErrorHandler(private val context: Context, private val view: View)
         llErrorToLoad.visibility = View.GONE
         llEmptyUI.visibility = View.GONE
         llCustomUI.visibility = View.VISIBLE
-        ivCustomFeatureImage.setImageResource(featureImage!!)
+        featureImage?.let { ivCustomFeatureImage.setImageResource(it) } ?: run { ivCustomFeatureImage.visibility = View.GONE }
         featureName?.let { tvCustomFeatureName.text = it } ?: run { tvCustomFeatureName.visibility = View.GONE }
         subFeatureName?.let { tvCustomSubFeatureName.text = it } ?: run { tvCustomSubFeatureName.visibility = View.GONE }
     }
@@ -172,7 +172,7 @@ class SweetUIErrorHandler(private val context: Context, private val view: View)
         llErrorToLoad.visibility = View.GONE
         llEmptyUI.visibility = View.GONE
         llCustomUI.visibility = View.VISIBLE
-        ivCustomFeatureImage.setImageResource(featureImage!!)
+        featureImage?.let { ivCustomFeatureImage.setImageResource(it) } ?: run { ivCustomFeatureImage.visibility = View.GONE }
         tvCustomFeatureName.text = featureName
         tvCustomSubFeatureName.visibility = View.GONE
     }