chrome/browser/extensions/chrome_notification_observer.cc

   1 // Copyright 2013 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 "chrome/browser/extensions/chrome_notification_observer.h"
   6
   7 #include "base/logging.h"
   8 #include "chrome/browser/chrome_notification_types.h"
   9 #include "chrome/browser/profiles/profile.h"
  10 #include "chrome/browser/ui/browser.h"
  11 #include "chrome/common/extensions/features/feature_channel.h"
  12 #include "content/public/browser/notification_service.h"
  13 #include "content/public/browser/notification_types.h"
  14 #include "content/public/browser/render_process_host.h"
  15 #include "extensions/browser/extension_system.h"
  16 #include "extensions/browser/process_manager.h"
  17 #include "extensions/common/extension_messages.h"
  18
  19 namespace extensions {
  20
  21 ChromeNotificationObserver::ChromeNotificationObserver() {
  22   registrar_.Add(this,
  23                  chrome::NOTIFICATION_BROWSER_WINDOW_READY,
  24                  content::NotificationService::AllSources());
  25   registrar_.Add(this,
  26                  content::NOTIFICATION_RENDERER_PROCESS_CREATED,
  27                  content::NotificationService::AllBrowserContextsAndSources());
  28 }
  29
  30 ChromeNotificationObserver::~ChromeNotificationObserver() {}
  31
  32 void ChromeNotificationObserver::OnBrowserWindowReady(Browser* browser) {
  33   Profile* profile = browser->profile();
  34   DCHECK(profile);
  35
  36   // Inform the process manager for this profile that the window is ready.
  37   // We continue to observe the notification in case browser windows open for
  38   // a related incognito profile or other regular profiles.
  39   extensions::ProcessManager* manager =
  40       ExtensionSystem::Get(profile)->process_manager();
  41   if (!manager)  // Tests may not have a process manager.
  42     return;
  43   DCHECK_EQ(profile, manager->GetBrowserContext());
  44   manager->OnBrowserWindowReady();
  45
  46   // For incognito profiles also inform the original profile's process manager
  47   // that the window is ready. This will usually be a no-op because the
  48   // original profile's process manager should have been informed when the
  49   // non-incognito window opened.
  50   if (profile->IsOffTheRecord()) {
  51     Profile* original_profile = profile->GetOriginalProfile();
  52     extensions::ProcessManager* original_manager =
  53         ExtensionSystem::Get(original_profile)->process_manager();
  54     DCHECK(original_manager);
  55     DCHECK_EQ(original_profile, original_manager->GetBrowserContext());
  56     original_manager->OnBrowserWindowReady();
  57   }
  58 }
  59
  60 void ChromeNotificationObserver::OnRendererProcessCreated(
  61     content::RenderProcessHost* process) {
  62   // Extensions need to know the channel for API restrictions. Send the channel
  63   // to all renderers, as the non-extension renderers may have content scripts.
  64   process->Send(new ExtensionMsg_SetChannel(GetCurrentChannel()));
  65 }
  66
  67 void ChromeNotificationObserver::Observe(int type,
  68                        const content::NotificationSource& source,
  69                        const content::NotificationDetails& details) {
  70   switch (type) {
  71     case chrome::NOTIFICATION_BROWSER_WINDOW_READY: {
  72       Browser* browser = content::Source<Browser>(source).ptr();
  73       OnBrowserWindowReady(browser);
  74       break;
  75     }
  76
  77     case content::NOTIFICATION_RENDERER_PROCESS_CREATED: {
  78       content::RenderProcessHost* process =
  79           content::Source<content::RenderProcessHost>(source).ptr();
  80       OnRendererProcessCreated(process);
  81       break;
  82     }
  83
  84     default:
  85       NOTREACHED();
  86   }
  87 }
  88
  89 }  // namespace extensions