From 936bb85d6c6ed743b99609a65ffa1a993f53a5e8 Mon Sep 17 00:00:00 2001
From: iorgamgabriel <iorgamgabriel@yahoo.com>
Date: Fri, 23 Feb 2024 17:23:46 +0200
Subject: [PATCH] Bug 1878921 - Translations CouldNotLoadLanguagesError UI "Try
 Again"

---
 .../fenix/translations/TranslationsDialogFragment.kt       | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/app/src/main/java/org/mozilla/fenix/translations/TranslationsDialogFragment.kt b/app/src/main/java/org/mozilla/fenix/translations/TranslationsDialogFragment.kt
index b28fcbb8e..c8c6034c1 100644
--- a/app/src/main/java/org/mozilla/fenix/translations/TranslationsDialogFragment.kt
+++ b/app/src/main/java/org/mozilla/fenix/translations/TranslationsDialogFragment.kt
@@ -29,6 +29,7 @@ import androidx.navigation.fragment.navArgs
 import com.google.android.material.bottomsheet.BottomSheetBehavior
 import com.google.android.material.bottomsheet.BottomSheetDialogFragment
 import mozilla.components.browser.state.store.BrowserStore
+import mozilla.components.concept.engine.translate.TranslationError
 import mozilla.components.lib.state.ext.observeAsComposableState
 import mozilla.components.support.base.feature.ViewBoundFeatureWrapper
 import org.mozilla.fenix.BrowserDirection
@@ -212,7 +213,11 @@ class TranslationsDialogFragment : BottomSheetDialogFragment() {
                 onSettingClicked = onSettingClicked,
                 onLearnMoreClicked = { openBrowserAndLoad(learnMoreUrl) },
                 onPositiveButtonClicked = {
-                    translationsDialogStore.dispatch(TranslationsDialogAction.TranslateAction)
+                    if (state.error is TranslationError.CouldNotLoadLanguagesError) {
+                        translationsDialogStore.dispatch(TranslationsDialogAction.FetchSupportedLanguages)
+                    } else {
+                        translationsDialogStore.dispatch(TranslationsDialogAction.TranslateAction)
+                    }
                 },
                 onNegativeButtonClicked = {
                     if (state.isTranslated) {