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