chrome/browser/extensions/chrome_extension_function_details.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 "chrome/browser/extensions/chrome_extension_function_details.h"
   6
   7 #include "chrome/browser/extensions/window_controller.h"
   8 #include "chrome/browser/extensions/window_controller_list.h"
   9 #include "chrome/browser/profiles/profile.h"
  10 #include "chrome/browser/ui/browser.h"
  11 #include "chrome/browser/ui/browser_finder.h"
  12 #include "chrome/browser/ui/tabs/tab_strip_model.h"
  13 #include "content/public/browser/render_process_host.h"
  14 #include "content/public/browser/render_view_host.h"
  15 #include "extensions/browser/extension_function.h"
  16 #include "extensions/browser/extension_function_dispatcher.h"
  17
  18 using content::WebContents;
  19 using content::RenderViewHost;
  20 using extensions::WindowController;
  21
  22 ChromeExtensionFunctionDetails::ChromeExtensionFunctionDetails(
  23     UIThreadExtensionFunction* function)
  24     : function_(function) {
  25 }
  26
  27 ChromeExtensionFunctionDetails::~ChromeExtensionFunctionDetails() {
  28 }
  29
  30 Profile* ChromeExtensionFunctionDetails::GetProfile() const {
  31   return Profile::FromBrowserContext(function_->browser_context());
  32 }
  33
  34 bool ChromeExtensionFunctionDetails::CanOperateOnWindow(
  35     const extensions::WindowController* window_controller) const {
  36   // |extension()| is NULL for unit tests only.
  37   if (function_->extension() != NULL &&
  38       !window_controller->IsVisibleToExtension(function_->extension())) {
  39     return false;
  40   }
  41
  42   if (GetProfile() == window_controller->profile())
  43     return true;
  44
  45   if (!function_->include_incognito())
  46     return false;
  47
  48   return GetProfile()->HasOffTheRecordProfile() &&
  49          GetProfile()->GetOffTheRecordProfile() == window_controller->profile();
  50 }
  51
  52 // TODO(stevenjb): Replace this with GetExtensionWindowController().
  53 Browser* ChromeExtensionFunctionDetails::GetCurrentBrowser() const {
  54   // If the delegate has an associated browser, return it.
  55   if (function_->dispatcher()) {
  56     extensions::WindowController* window_controller =
  57         function_->dispatcher()->delegate()->GetExtensionWindowController();
  58     if (window_controller) {
  59       Browser* browser = window_controller->GetBrowser();
  60       if (browser)
  61         return browser;
  62     }
  63   }
  64
  65   // Otherwise, try to default to a reasonable browser. If |include_incognito_|
  66   // is true, we will also search browsers in the incognito version of this
  67   // profile. Note that the profile may already be incognito, in which case
  68   // we will search the incognito version only, regardless of the value of
  69   // |include_incognito|. Look only for browsers on the active desktop as it is
  70   // preferable to pretend no browser is open then to return a browser on
  71   // another desktop.
  72   if (function_->render_view_host()) {
  73     Profile* profile = Profile::FromBrowserContext(
  74         function_->render_view_host()->GetProcess()->GetBrowserContext());
  75     Browser* browser = chrome::FindAnyBrowser(
  76         profile, function_->include_incognito(), chrome::GetActiveDesktop());
  77     if (browser)
  78       return browser;
  79   }
  80
  81   // NOTE(rafaelw): This can return NULL in some circumstances. In particular,
  82   // a background_page onload chrome.tabs api call can make it into here
  83   // before the browser is sufficiently initialized to return here, or
  84   // all of this profile's browser windows may have been closed.
  85   // A similar situation may arise during shutdown.
  86   // TODO(rafaelw): Delay creation of background_page until the browser
  87   // is available. http://code.google.com/p/chromium/issues/detail?id=13284
  88   return NULL;
  89 }
  90
  91 extensions::WindowController*
  92 ChromeExtensionFunctionDetails::GetExtensionWindowController() const {
  93   // If the delegate has an associated window controller, return it.
  94   if (function_->dispatcher()) {
  95     extensions::WindowController* window_controller =
  96         function_->dispatcher()->delegate()->GetExtensionWindowController();
  97     if (window_controller)
  98       return window_controller;
  99   }
 100
 101   return extensions::WindowControllerList::GetInstance()
 102       ->CurrentWindowForFunction(*this);
 103 }
 104
 105 content::WebContents*
 106 ChromeExtensionFunctionDetails::GetAssociatedWebContents() {
 107   content::WebContents* web_contents = function_->GetAssociatedWebContents();
 108   if (web_contents)
 109     return web_contents;
 110
 111   Browser* browser = GetCurrentBrowser();
 112   if (!browser)
 113     return NULL;
 114   return browser->tab_strip_model()->GetActiveWebContents();
 115 }