Miscellaneous fixes to DataUse UI

chrome/browser/android/data_usage/data_use_tab_ui_manager_android.cc

 // Copyright 2015 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/android/data_usage/data_use_tab_ui_manager_android.h"
 
 #include <string>
 
 #include "base/android/jni_android.h"
 #include "base/android/jni_string.h"
@@ skipping 30 matching lines @@
       chrome::android::DataUseUITabModelFactory::GetForBrowserContext(profile);
   DCHECK_LE(0, static_cast<SessionID::id_type>(tab_id));
   if (data_use_ui_tab_model) {
     return data_use_ui_tab_model->HasDataUseTrackingEnded(
         static_cast<SessionID::id_type>(tab_id));
   }
   return false;
 }
 
 // static
+void UserClickedContinueOnDialogBox(JNIEnv* env,
+                                    const JavaParamRef<jclass>& clazz,
+                                    jint tab_id,
+                                    const JavaParamRef<jobject>& jprofile) {
+  Profile* profile = ProfileAndroid::FromProfileAndroid(jprofile);
+  chrome::android::DataUseUITabModel* data_use_ui_tab_model =
+      chrome::android::DataUseUITabModelFactory::GetForBrowserContext(profile);
+  DCHECK_LE(0, static_cast<SessionID::id_type>(tab_id));
+  if (data_use_ui_tab_model) {
+    data_use_ui_tab_model->UserClickedContinueOnDialogBox(
+        static_cast<SessionID::id_type>(tab_id));
+  }
+}
+
+// static
+jboolean WouldDataUseTrackingEnd(JNIEnv* env,
+                                 const JavaParamRef<jclass>& clazz,
+                                 jint tab_id,
+                                 const JavaParamRef<jstring>& url,
+                                 jint transition_type,
+                                 const JavaParamRef<jobject>& jprofile) {
+  Profile* profile = ProfileAndroid::FromProfileAndroid(jprofile);
+  chrome::android::DataUseUITabModel* data_use_ui_tab_model =
+      chrome::android::DataUseUITabModelFactory::GetForBrowserContext(profile);
+  DCHECK_LE(0, static_cast<SessionID::id_type>(tab_id));
+  if (data_use_ui_tab_model) {
+    return data_use_ui_tab_model->WouldDataUseTrackingEnd(
+        ConvertJavaStringToUTF8(env, url), transition_type,
+        static_cast<SessionID::id_type>(tab_id));
+  }
+  return false;
+}
+
+// static
 void OnCustomTabInitialNavigation(JNIEnv* env,
                                   const JavaParamRef<jclass>& clazz,
                                   jint tab_id,
+                                  const JavaParamRef<jstring>& package_name,
                                   const JavaParamRef<jstring>& url,
-                                  const JavaParamRef<jstring>& package_name,
                                   const JavaParamRef<jobject>& jprofile) {
   Profile* profile = ProfileAndroid::FromProfileAndroid(jprofile);
   chrome::android::DataUseUITabModel* data_use_ui_tab_model =
       chrome::android::DataUseUITabModelFactory::GetForBrowserContext(profile);
   DCHECK_LE(0, static_cast<SessionID::id_type>(tab_id));
   if (data_use_ui_tab_model) {
     data_use_ui_tab_model->ReportCustomTabInitialNavigation(
         static_cast<SessionID::id_type>(tab_id),
-        ConvertJavaStringToUTF8(env, url),
-        ConvertJavaStringToUTF8(env, package_name));
+        ConvertJavaStringToUTF8(env, package_name),
+        ConvertJavaStringToUTF8(env, url));
   }
 }
 
 bool RegisterDataUseTabUIManager(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }