Automatically trigger Always-Translate in new translation infobar

chrome/browser/ui/android/infobars/translate_compact_infobar.cc

 // Copyright 2017 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/ui/android/infobars/translate_compact_infobar.h"
 
 #include <stddef.h>
 
 #include "base/android/jni_android.h"
 #include "base/android/jni_array.h"
@@ skipping 45 matching lines @@
       target_language_code, delegate->ShouldAlwaysTranslate(),
       delegate->triggered_from_menu(), java_languages, java_codes);
 }
 
 void TranslateCompactInfoBar::ProcessButton(int action) {
   if (!owner())
     return;  // We're closing; don't call anything, it might access the owner.
 
   // TODO(ramyasharma): Handle other button clicks.
   translate::TranslateInfoBarDelegate* delegate = GetDelegate();
-  if (action == InfoBarAndroid::ACTION_TRANSLATE)
+  if (action == InfoBarAndroid::ACTION_TRANSLATE) {
     delegate->Translate();
-  else if (action == InfoBarAndroid::ACTION_TRANSLATE_SHOW_ORIGINAL)
+    if (!delegate->ShouldAlwaysTranslate() &&
+        delegate->ShouldAutoAlwaysTranslate()) {
+      JNIEnv* env = base::android::AttachCurrentThread();
+      Java_TranslateCompactInfoBar_setAutoAlwaysTranslate(env,
+                                                          GetJavaInfoBar());
+    }
+  } else if (action == InfoBarAndroid::ACTION_TRANSLATE_SHOW_ORIGINAL) {
     delegate->RevertTranslation();
-  else
+  } else if (action == InfoBarAndroid::ACTION_CANCEL) {
+    delegate->TranslationDeclined();
+  } else {
     DCHECK_EQ(InfoBarAndroid::ACTION_NONE, action);
+  }
 }
 
 void TranslateCompactInfoBar::SetJavaInfoBar(
     const base::android::JavaRef<jobject>& java_info_bar) {
   InfoBarAndroid::SetJavaInfoBar(java_info_bar);
   JNIEnv* env = base::android::AttachCurrentThread();
   Java_TranslateCompactInfoBar_setNativePtr(env, java_info_bar,
                                             reinterpret_cast<intptr_t>(this));
 }
 
@@ skipping 47 matching lines @@
     translate::TranslateErrors::Type error_type) {
   if (!owner())
     return;  // We're closing; don't call anything, it might access the owner.
 
   DCHECK(translate_driver_);
   JNIEnv* env = base::android::AttachCurrentThread();
   Java_TranslateCompactInfoBar_onPageTranslated(env, GetJavaInfoBar(),
                                                 error_type);
 }
 
+bool TranslateCompactInfoBar::ShouldAutoAlwaysTranslate() {
+  translate::TranslateInfoBarDelegate* delegate = GetDelegate();
+  return delegate->ShouldAutoAlwaysTranslate();
+}
+
 translate::TranslateInfoBarDelegate* TranslateCompactInfoBar::GetDelegate() {
   return delegate()->AsTranslateInfoBarDelegate();
 }
 
 // Native JNI methods ---------------------------------------------------------
 
 // static
 bool RegisterTranslateCompactInfoBar(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }