Changing tab closure handling logic in speech recognition code and cleaning bubble controller. (Spe…

chrome/browser/speech/tab_watcher.cc

+// Copyright (c) 2012 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/speech/tab_watcher.h"
+
+#include "base/bind.h"
+#include "chrome/browser/tab_contents/tab_util.h"
+#include "content/public/browser/notification_registrar.h"
+#include "content/public/browser/notification_source.h"
+#include "content/public/browser/notification_types.h"
+#include "content/public/browser/render_process_host.h"
+#include "content/public/browser/render_view_host.h"
+#include "content/public/browser/web_contents.h"
+
+using content::BrowserThread;
+using content::WebContents;
+
+namespace {
+const int kNotificationType = content::NOTIFICATION_WEB_CONTENTS_DISCONNECTED;
+}
+
+TabWatcher::TabWatcher(
+    TabClosedCallback tab_closed_callback, BrowserThread::ID callback_thread)
+  : tab_closed_callback_(tab_closed_callback),
+    callback_thread_(callback_thread) {
+}
+
+TabWatcher::~TabWatcher() {
+  // Must be destroyed on the UI thread due to |registrar_| non thread-safety.
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+}
+
+void TabWatcher::Watch(
+    int render_process_id, int render_view_id) {
+  if (!BrowserThread::CurrentlyOn(BrowserThread::UI)) {
+    BrowserThread::PostTask(BrowserThread::UI, FROM_HERE, base::Bind(
+        &TabWatcher::Watch, this, render_process_id, render_view_id));
+    return;
+  }
+
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  WebContents* web_contents = tab_util::GetWebContentsByID(render_process_id,
+                                                           render_view_id);
+
+  // Sessions initiated by extensions might not have a corresponding WebContent.
+  if (!web_contents)
+    return;
+
+  // Avoid multiple registrations on |registrar_| for the same |web_contents|.
+  if (registered_contents_.find(web_contents) != registered_contents_.end())
+    return;
+  registered_contents_.insert(web_contents);
+
+  // Lazy initialize the registrar.
+  if (!registrar_.get())
+    registrar_.reset(new content::NotificationRegistrar());
+
+  registrar_->Add(this,
+                  kNotificationType,
+                  content::Source<WebContents>(web_contents));
+}
+
+void TabWatcher::Observe(
+    int type,
+    const content::NotificationSource& source,
+    const content::NotificationDetails& details) {
+  DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
+  DCHECK_EQ(kNotificationType, type);
+
+  WebContents* web_contents = content::Source<WebContents>(source).ptr();
+  const int render_process_id = web_contents->GetRenderProcessHost()->GetID();
+  const int render_view_id = web_contents->GetRenderViewHost()->GetRoutingID();
+
+  registrar_->Remove(this,
+                     kNotificationType,
+                     content::Source<WebContents>(web_contents));
+  registered_contents_.erase(web_contents);
+
+  BrowserThread::PostTask(callback_thread_, FROM_HERE, base::Bind(
+      tab_closed_callback_, render_process_id, render_view_id));
+}