extensions/browser/mojo/service_registration_manager.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "extensions/browser/mojo/service_registration_manager.h"
   6
   7 #include "base/command_line.h"
   8 #include "base/lazy_instance.h"
   9 #include "content/public/browser/browser_thread.h"
  10 #include "content/public/browser/render_frame_host.h"
  11 #include "content/public/browser/render_process_host.h"
  12 #include "content/public/browser/site_instance.h"
  13 #include "device/serial/serial_service_impl.h"
  14 #include "extensions/browser/extension_registry.h"
  15 #include "extensions/browser/process_map.h"
  16 #include "extensions/common/constants.h"
  17 #include "extensions/common/extension_api.h"
  18 #include "extensions/common/switches.h"
  19
  20 namespace extensions {
  21 namespace {
  22
  23 base::LazyInstance<ServiceRegistrationManager> g_lazy_instance =
  24     LAZY_INSTANCE_INITIALIZER;
  25
  26 ServiceRegistrationManager* g_test_instance = nullptr;
  27
  28 }  // namespace
  29
  30 ServiceRegistrationManager::ServiceRegistrationManager() {
  31   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
  32           switches::kEnableMojoSerialService)) {
  33     AddServiceFactory(
  34         "serial",
  35         base::Bind(device::SerialServiceImpl::CreateOnMessageLoop,
  36                    content::BrowserThread::GetMessageLoopProxyForThread(
  37                        content::BrowserThread::FILE),
  38                    content::BrowserThread::GetMessageLoopProxyForThread(
  39                        content::BrowserThread::IO),
  40                    content::BrowserThread::GetMessageLoopProxyForThread(
  41                        content::BrowserThread::UI)));
  42   }
  43 }
  44
  45 ServiceRegistrationManager::~ServiceRegistrationManager() {
  46 }
  47
  48 ServiceRegistrationManager* ServiceRegistrationManager::GetSharedInstance() {
  49   if (g_test_instance)
  50     return g_test_instance;
  51   return g_lazy_instance.Pointer();
  52 }
  53
  54 void ServiceRegistrationManager::AddServicesToRenderFrame(
  55     content::RenderFrameHost* render_frame_host) {
  56   content::BrowserContext* context =
  57       render_frame_host->GetProcess()->GetBrowserContext();
  58   content::SiteInstance* site_instance = render_frame_host->GetSiteInstance();
  59   GURL extension_url = site_instance->GetSiteURL();
  60   ExtensionRegistry* registry = ExtensionRegistry::Get(context);
  61
  62   // TODO(sammc): Handle content scripts and web pages that have access to some
  63   // extension APIs.
  64   if (!extension_url.SchemeIs(kExtensionScheme)) {
  65     return;
  66   }
  67
  68   const Extension* extension =
  69       registry->enabled_extensions().GetByID(extension_url.host());
  70   if (!extension)
  71     return;
  72
  73   Feature::Context context_type =
  74       ProcessMap::Get(context)->GetMostLikelyContextType(
  75           extension, render_frame_host->GetProcess()->GetID());
  76   for (const auto& factory : factories_) {
  77     auto availability = ExtensionAPI::GetSharedInstance()->IsAvailable(
  78         factory.first, extension, context_type, extension_url);
  79     if (availability.is_available()) {
  80       AddServiceToServiceRegistry(render_frame_host->GetServiceRegistry(),
  81                                   factory.second.get());
  82     }
  83   }
  84 }
  85
  86 void ServiceRegistrationManager::AddServiceToServiceRegistry(
  87     content::ServiceRegistry* service_registry,
  88     internal::ServiceFactoryBase* factory) {
  89   factory->Register(service_registry);
  90 }
  91
  92 // static
  93 void ServiceRegistrationManager::SetServiceRegistrationManagerForTest(
  94     ServiceRegistrationManager* service_registration_manager) {
  95   g_test_instance = service_registration_manager;
  96 }
  97
  98 }  // namespace extensions