Convert GetSearchProviderInstallState to Mojo

chrome/browser/search_engines/search_provider_install_state_impl.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/search_engines/search_provider_install_state_message_filter.h"
+#include "chrome/browser/search_engines/search_provider_install_state_impl.h"
 
 #include "base/bind.h"
 #include "base/logging.h"
 #include "chrome/browser/google/google_url_tracker_factory.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/search_engines/template_url_service_factory.h"
 #include "chrome/browser/search_engines/ui_thread_search_terms_data.h"
 #include "chrome/common/render_messages.h"
+#include "chrome/common/search_provider.mojom.h"
 #include "content/public/browser/render_process_host.h"
 #include "url/gurl.h"
 
 using content::BrowserThread;
 
-SearchProviderInstallStateMessageFilter::
-SearchProviderInstallStateMessageFilter(
+SearchProviderInstallStateImpl::SearchProviderInstallStateImpl(
     int render_process_id,
     Profile* profile)
-    : BrowserMessageFilter(ChromeMsgStart),
-      provider_data_(TemplateURLServiceFactory::GetForProfile(profile),
+    : provider_data_(TemplateURLServiceFactory::GetForProfile(profile),
                      UIThreadSearchTermsData(profile).GoogleBaseURLValue(),
                      GoogleURLTrackerFactory::GetForProfile(profile),
                      content::RenderProcessHost::FromID(render_process_id)),
       is_off_the_record_(profile->IsOffTheRecord()),
       weak_factory_(this) {
   // This is initialized by RenderProcessHostImpl. Do not add any non-trivial
   // initialization here. Instead do it lazily when required.
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 }
 
-bool SearchProviderInstallStateMessageFilter::OnMessageReceived(
-    const IPC::Message& message) {
-  DCHECK_CURRENTLY_ON(BrowserThread::IO);
-  bool handled = true;
-  IPC_BEGIN_MESSAGE_MAP(SearchProviderInstallStateMessageFilter, message)
-    IPC_MESSAGE_HANDLER_DELAY_REPLY(
-        ChromeViewHostMsg_GetSearchProviderInstallState,
-        OnGetSearchProviderInstallState)
-    IPC_MESSAGE_UNHANDLED(handled = false)
-  IPC_END_MESSAGE_MAP()
-  return handled;
-}
-
-SearchProviderInstallStateMessageFilter::
-~SearchProviderInstallStateMessageFilter() {
+SearchProviderInstallStateImpl::~SearchProviderInstallStateImpl() {
   DCHECK_CURRENTLY_ON(BrowserThread::IO);
 }
 
+void SearchProviderInstallStateImpl::InstallService(
+    content::RenderProcessHost* host) {
+  int id = host->GetID();
+  Profile* profile = Profile::FromBrowserContext(host->GetBrowserContext());
+  auto* adapter = new SearchProviderInstallStateImplAdapter(
+      std::unique_ptr<SearchProviderInstallStateImpl,
+                      content::BrowserThread::DeleteOnIOThread>(
+          new SearchProviderInstallStateImpl(id, profile)));
+  host->GetServiceRegistry()->AddService<mojom::SearchProviderInstallState>(

[sky, 2016/06/20 15:20:26]

I'm not familiar with the threading requirements of this class. What are they?
What threads do this code need to live on?

[tibell, 2016/06/21 01:31:32]

I'm just preserving the same threading behavior as we had before the switch to
Mojo. Messages arrive and are handled on the IO thread. There are existing
DCHECKs in the code that make sure we're running things on the right threads.

+      base::Bind(&SearchProviderInstallStateImpl::Bind,
+                 base::Unretained(adapter->get())),
+      BrowserThread::GetMessageLoopProxyForThread(BrowserThread::IO));
+  host->SetUserData(SearchProviderInstallStateImpl::kRenderProcessHostKey,
+                    adapter);
+}
+
 search_provider::InstallState
-SearchProviderInstallStateMessageFilter::GetSearchProviderInstallState(
+SearchProviderInstallStateImpl::GetSearchProviderInstallState(
     const GURL& page_location,
     const GURL& requested_host) {
   GURL requested_origin = requested_host.GetOrigin();
 
   // Do the security check before any others to avoid information leaks.
   if (page_location.GetOrigin() != requested_origin)
     return search_provider::DENIED;
 
   // In incognito mode, no search information is exposed. (This check must be
   // done after the security check or else a web site can detect that the
   // user is in incognito mode just by doing a cross origin request.)
   if (is_off_the_record_)
-      return search_provider::NOT_INSTALLED;
+    return search_provider::NOT_INSTALLED;
 
   switch (provider_data_.GetInstallState(requested_origin)) {
     case SearchProviderInstallData::NOT_INSTALLED:
       return search_provider::NOT_INSTALLED;
 
     case SearchProviderInstallData::INSTALLED_BUT_NOT_DEFAULT:
       return search_provider::INSTALLED_BUT_NOT_DEFAULT;
 
     case SearchProviderInstallData::INSTALLED_AS_DEFAULT:
       return search_provider::INSTALLED_AS_DEFAULT;
   }
 
   NOTREACHED();
   return search_provider::NOT_INSTALLED;
 }
 
-void
-SearchProviderInstallStateMessageFilter::OnGetSearchProviderInstallState(
+void SearchProviderInstallStateImpl::Bind(
+    mojom::SearchProviderInstallStateRequest request) {
+  binding_set_.AddBinding(this, std::move(request));
+}
+
+const char SearchProviderInstallStateImpl::kRenderProcessHostKey[] =
+    "kRenderProcessHostKey";
+
+void SearchProviderInstallStateImpl::GetInstallState(
     const GURL& page_location,
     const GURL& requested_host,
-    IPC::Message* reply_msg) {
-  provider_data_.CallWhenLoaded(
-      base::Bind(
-          &SearchProviderInstallStateMessageFilter::
-          ReplyWithProviderInstallState,
-          weak_factory_.GetWeakPtr(),
-          page_location,
-          requested_host,
-          reply_msg));
+    const GetInstallStateCallback& callback) {
+  DCHECK_CURRENTLY_ON(BrowserThread::IO);
+  provider_data_.CallWhenLoaded(base::Bind(
+      &SearchProviderInstallStateImpl::ReplyWithProviderInstallState,
+      weak_factory_.GetWeakPtr(), page_location, requested_host, callback));
 }
 
-void SearchProviderInstallStateMessageFilter::ReplyWithProviderInstallState(
+void SearchProviderInstallStateImpl::ReplyWithProviderInstallState(
     const GURL& page_location,
     const GURL& requested_host,
-    IPC::Message* reply_msg) {
-  DCHECK(reply_msg);
+    const GetInstallStateCallback& callback) {
   search_provider::InstallState install_state =
       GetSearchProviderInstallState(page_location, requested_host);
 
-  ChromeViewHostMsg_GetSearchProviderInstallState::WriteReplyParams(
-      reply_msg,
-      install_state);
-  Send(reply_msg);
+  callback.Run(install_state);
 }
+
+SearchProviderInstallStateImplAdapter::SearchProviderInstallStateImplAdapter(
+    std::unique_ptr<SearchProviderInstallStateImpl,
+                    content::BrowserThread::DeleteOnIOThread> filter)
+    : ptr_(std::move(filter)) {}
+
+SearchProviderInstallStateImplAdapter::
+    ~SearchProviderInstallStateImplAdapter() = default;