Use ShellClientFactory interface to load mojo:profile

content/browser/browser_context.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 "content/public/browser/browser_context.h"
 
 #include <stddef.h>
 #include <stdint.h>
 #include <algorithm>
 #include <limits>
+#include <memory>
 #include <utility>
+#include <vector>
 
+#include "base/command_line.h"
 #include "base/guid.h"
 #include "base/lazy_instance.h"
+#include "base/macros.h"
 #include "base/rand_util.h"
 #include "build/build_config.h"
+#include "components/profile_service/profile_app.h"
+#include "components/profile_service/public/cpp/constants.h"
 #include "components/profile_service/user_id_map.h"
 #include "content/browser/download/download_manager_impl.h"
 #include "content/browser/fileapi/chrome_blob_storage_context.h"
 #include "content/browser/indexed_db/indexed_db_context_impl.h"
 #include "content/browser/loader/resource_dispatcher_host_impl.h"
+#include "content/browser/mojo/browser_shell_connection.h"
+#include "content/browser/mojo/constants.h"
 #include "content/browser/push_messaging/push_messaging_router.h"
 #include "content/browser/storage_partition_impl_map.h"
 #include "content/common/child_process_host_impl.h"
 #include "content/public/browser/blob_handle.h"
 #include "content/public/browser/browser_thread.h"
 #include "content/public/browser/content_browser_client.h"
 #include "content/public/browser/site_instance.h"
+#include "content/public/common/content_switches.h"
+#include "content/public/common/mojo_shell_connection.h"
 #include "net/cookies/cookie_store.h"
 #include "net/ssl/channel_id_service.h"
 #include "net/ssl/channel_id_store.h"
 #include "net/url_request/url_request_context.h"
 #include "net/url_request/url_request_context_getter.h"
+#include "services/shell/public/cpp/connection.h"
+#include "services/shell/public/cpp/connector.h"
+#include "services/shell/public/interfaces/shell_client.mojom.h"
 #include "storage/browser/database/database_tracker.h"
 #include "storage/browser/fileapi/external_mount_points.h"
 
 using base::UserDataAdapter;
 
 namespace content {
 
 namespace {
 
 base::LazyInstance<std::set<std::string>> g_used_user_ids =
     LAZY_INSTANCE_INITIALIZER;
 base::LazyInstance<std::vector<std::pair<BrowserContext*, std::string>>>
 g_context_to_user_id = LAZY_INSTANCE_INITIALIZER;
 
 // Key names on BrowserContext.
 const char kDownloadManagerKeyName[] = "download_manager";
+const char kMojoShellConnection[] = "mojo-shell-connection";
+const char kMojoWasInitialized[] = "mojo-was-initialized";
 const char kStoragePartitionMapKeyName[] = "content_storage_partition_map";
 
-const char kMojoWasInitialized[] = "mojo-was-initialized";
-
 #if defined(OS_CHROMEOS)
 const char kMountPointsKey[] = "mount_points";
 #endif  // defined(OS_CHROMEOS)
 
 StoragePartitionImplMap* GetStoragePartitionMap(
     BrowserContext* browser_context) {
   StoragePartitionImplMap* partition_map =
       static_cast<StoragePartitionImplMap*>(
           browser_context->GetUserData(kStoragePartitionMapKeyName));
   if (!partition_map) {
@@ skipping 39 matching lines @@
   wrapper->process_manager()->Shutdown();
 }
 
 void SetDownloadManager(BrowserContext* context,
                         content::DownloadManager* download_manager) {
   DCHECK_CURRENTLY_ON(BrowserThread::UI);
   DCHECK(download_manager);
   context->SetUserData(kDownloadManagerKeyName, download_manager);
 }
 
+class BrowserContextShellConnectionHolder
+    : public base::SupportsUserData::Data {
+ public:
+  BrowserContextShellConnectionHolder(
+      std::unique_ptr<mojo::Connection> connection,
+      mojo::shell::mojom::ShellClientRequest request)
+      : root_connection_(std::move(connection)),
+        shell_connection_(new BrowserShellConnection(std::move(request))) {}
+  ~BrowserContextShellConnectionHolder() override {}
+
+  BrowserShellConnection* shell_connection() { return shell_connection_.get(); }
+
+ private:
+  std::unique_ptr<mojo::Connection> root_connection_;
+  std::unique_ptr<BrowserShellConnection> shell_connection_;
+
+  DISALLOW_COPY_AND_ASSIGN(BrowserContextShellConnectionHolder);
+};
+
 }  // namespace
 
 // static
 void BrowserContext::AsyncObliterateStoragePartition(
     BrowserContext* browser_context,
     const GURL& site,
     const base::Closure& on_gc_required) {
   GetStoragePartitionMap(browser_context)->AsyncObliterate(site,
                                                            on_gc_required);
 }
@@ skipping 200 matching lines @@
                    make_scoped_refptr(indexed_db_context_impl)));
   }
 }
 
 void BrowserContext::SetDownloadManagerForTesting(
     BrowserContext* browser_context,
     DownloadManager* download_manager) {
   SetDownloadManager(browser_context, download_manager);
 }
 
+// static
 void BrowserContext::Initialize(
     BrowserContext* browser_context,
     const base::FilePath& path) {
   // Generate a GUID for |browser_context| to use as the Mojo user id.
   std::string new_id = base::GenerateGUID();
   while (g_used_user_ids.Get().find(new_id) != g_used_user_ids.Get().end())
     new_id = base::GenerateGUID();
 
   g_used_user_ids.Get().insert(new_id);
   g_context_to_user_id.Get().push_back(std::make_pair(browser_context, new_id));
 
   profile::AssociateMojoUserIDWithProfileDir(new_id, path);
   browser_context->SetUserData(kMojoWasInitialized,
                                new base::SupportsUserData::Data);
+
+  MojoShellConnection* shell = MojoShellConnection::Get();
+  if (shell) {
+    // NOTE: Many unit tests create a TestBrowserContext without initializing
+    // Mojo or the global Mojo shell connection.
+
+    mojo::shell::mojom::ShellClientPtr shell_client;
+    mojo::shell::mojom::ShellClientRequest shell_client_request =
+        mojo::GetProxy(&shell_client);
+
+    mojo::shell::mojom::PIDReceiverPtr pid_receiver;
+    mojo::Connector::ConnectParams params(
+        mojo::Identity(kBrowserMojoApplicationName, new_id));
+    params.set_client_process_connection(std::move(shell_client),
+                                         mojo::GetProxy(&pid_receiver));
+    pid_receiver->SetPID(base::GetCurrentProcId());
+
+    BrowserContextShellConnectionHolder* connection_holder =
+        new BrowserContextShellConnectionHolder(
+          shell->GetConnector()->Connect(&params),
+          std::move(shell_client_request));
+    browser_context->SetUserData(kMojoShellConnection, connection_holder);
+
+    BrowserShellConnection* connection = connection_holder->shell_connection();
+
+    // New embedded application factories should be added to |connection| here.
+
+    if (base::CommandLine::ForCurrentProcess()->HasSwitch(
+            switches::kMojoLocalStorage)) {
+      connection->AddEmbeddedApplication(
+          profile::kProfileMojoApplicationName,
+          base::Bind(
+              &profile::CreateProfileApp,
+              BrowserThread::GetMessageLoopProxyForThread(BrowserThread::FILE),
+              BrowserThread::GetMessageLoopProxyForThread(BrowserThread::DB)),
+          nullptr);
+    }
+  }
 }
 
+// static
 const std::string& BrowserContext::GetMojoUserIdFor(
     BrowserContext* browser_context) {
   CHECK(browser_context->GetUserData(kMojoWasInitialized))
       << "Attempting to get the mojo user id for a BrowserContext that was "
       << "never Initialize()ed.";
 
   auto it = std::find_if(
       g_context_to_user_id.Get().begin(),
       g_context_to_user_id.Get().end(),
       [&browser_context](const std::pair<BrowserContext*, std::string>& p) {
         return p.first == browser_context; });
   CHECK(it != g_context_to_user_id.Get().end());
   return it->second;
 }
 
+// static
+mojo::Connector* BrowserContext::GetMojoConnectorFor(
+    BrowserContext* browser_context) {
+  BrowserContextShellConnectionHolder* connection_holder =
+      static_cast<BrowserContextShellConnectionHolder*>(
+          browser_context->GetUserData(kMojoShellConnection));
+  if (!connection_holder)
+    return nullptr;
+  return connection_holder->shell_connection()->GetConnector();
+}
+
 BrowserContext::~BrowserContext() {
   CHECK(GetUserData(kMojoWasInitialized))
       << "Attempting to destroy a BrowserContext that never called "
       << "Initialize()";
 
   if (GetUserData(kDownloadManagerKeyName))
     GetDownloadManager(this)->Shutdown();
 }
 
 }  // namespace content