Feed back to the ServiceTabLauncher when the tab has been created.

chrome/browser/android/service_tab_launcher.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/service_tab_launcher.h"
 
 #include "base/android/jni_string.h"
 #include "base/callback.h"
 #include "content/public/browser/browser_context.h"
 #include "content/public/browser/page_navigator.h"
+#include "content/public/browser/web_contents.h"
 #include "jni/ServiceTabLauncher_jni.h"
 
 using base::android::AttachCurrentThread;
 using base::android::ConvertUTF8ToJavaString;
 using base::android::GetApplicationContext;
 
+// Called by Java when the WebContents instance for a request Id is available.
+void OnWebContentsForRequestAvailable(
+    JNIEnv* env, jclass clazz, jint request_id, jobject android_web_contents) {
+  chrome::android::ServiceTabLauncher::GetInstance()->OnTabLaunched(
+      request_id,
+      content::WebContents::FromJavaWebContents(android_web_contents));
+}
+
 namespace chrome {
 namespace android {
 
 // static
 ServiceTabLauncher* ServiceTabLauncher::GetInstance() {
   return Singleton<ServiceTabLauncher>::get();
 }
 
-ServiceTabLauncher::ServiceTabLauncher() {
+ServiceTabLauncher::ServiceTabLauncher()
+    : last_request_id_(0) {
   java_object_.Reset(
       Java_ServiceTabLauncher_getInstance(AttachCurrentThread(),
                                           GetApplicationContext()));
 }
 
 ServiceTabLauncher::~ServiceTabLauncher() {}
 
 void ServiceTabLauncher::LaunchTab(
     content::BrowserContext* browser_context,
     const content::OpenURLParams& params,
@@ skipping 15 matching lines @@
   JNIEnv* env = AttachCurrentThread();
   ScopedJavaLocalRef<jstring> url = ConvertUTF8ToJavaString(
       env, params.url.spec());
   ScopedJavaLocalRef<jstring> referrer_url =
       ConvertUTF8ToJavaString(env, params.referrer.url.spec());
   ScopedJavaLocalRef<jstring> headers = ConvertUTF8ToJavaString(
       env, params.extra_headers);
 
   ScopedJavaLocalRef<jbyteArray> post_data;
 
+  tab_launched_callbacks_[++last_request_id_] = callback;
   Java_ServiceTabLauncher_launchTab(env,
                                     java_object_.obj(),
                                     GetApplicationContext(),
-                                    0 /* request_id */,
+                                    last_request_id_,
                                     browser_context->IsOffTheRecord(),
                                     url.obj(),
                                     disposition,
                                     referrer_url.obj(),
                                     params.referrer.policy,
                                     headers.obj(),
                                     post_data.obj());
+}
 
-  // TODO(peter): We need to wait for the Android Activity to reply to the
-  // launch intent with the ID of the launched Web Contents, so that the Java
-  // side can invoke a method on the native side with the request id and the
-  // WebContents enabling us to invoke |callback|. See https://crbug.com/454809.
+void ServiceTabLauncher::OnTabLaunched(int request_id,
+                                       content::WebContents* web_contents) {
+  DCHECK_EQ(tab_launched_callbacks_.count(request_id), 1u);
+  DCHECK(web_contents);
+
+  tab_launched_callbacks_[request_id].Run(web_contents);
+  tab_launched_callbacks_.erase(request_id);
 }
 
 bool ServiceTabLauncher::RegisterServiceTabLauncher(JNIEnv* env) {
   return RegisterNativesImpl(env);
 }
 
 }  // namespace android
 }  // namespace chrome