Translate: New Bubble UX (for the view toolkit)

chrome/browser/translate/translate_ui_delegate.cc

Index: chrome/browser/translate/translate_ui_delegate.cc
diff --git a/chrome/browser/translate/translate_ui_delegate.cc b/chrome/browser/translate/translate_ui_delegate.cc
index f28941d4ddd623717576eef779bdfd49284cd63b..ef711783daad861a43c8accf26c0524348b3bb39 100644
--- a/chrome/browser/translate/translate_ui_delegate.cc
+++ b/chrome/browser/translate/translate_ui_delegate.cc
@@ -91,7 +91,6 @@ size_t TranslateUIDelegate::GetOriginalLanguageIndex() const {
 }
 
 void TranslateUIDelegate::SetOriginalLanguageIndex(size_t language_index) {
-  DCHECK_LT(language_index, GetNumberOfLanguages());
   original_language_index_ = language_index;
 }
 
@@ -167,10 +166,15 @@ bool TranslateUIDelegate::IsLanguageBlocked() {
 }
 
 void TranslateUIDelegate::SetLanguageBlocked(bool value) {
-  if (value)
+  if (value) {
     prefs_->BlockLanguage(GetOriginalLanguageCode());
-  else
+    TranslateTabHelper* translate_tab_helper =
+        TranslateTabHelper::FromWebContents(web_contents());
+    DCHECK(translate_tab_helper);
+    translate_tab_helper->language_state().SetTranslateEnabled(false);
+  } else {
     prefs_->UnblockLanguage(GetOriginalLanguageCode());
+  }
 
   UMA_HISTOGRAM_BOOLEAN(kNeverTranslateLang, true);
 }
@@ -185,10 +189,15 @@ void TranslateUIDelegate::SetSiteBlacklist(bool value) {
   if (host.empty())
     return;
 
-  if (value)
+  if (value) {
     prefs_->BlacklistSite(host);
-  else
+    TranslateTabHelper* translate_tab_helper =
+        TranslateTabHelper::FromWebContents(web_contents());
+    DCHECK(translate_tab_helper);
+    translate_tab_helper->language_state().SetTranslateEnabled(false);
+  } else {
     prefs_->RemoveSiteFromBlacklist(host);
+  }
 
   UMA_HISTOGRAM_BOOLEAN(kNeverTranslateSite, true);
 }