chrome/browser/extensions/extension_action_test_util.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/extension_action_test_util.h"
   6
   7 #include "chrome/browser/extensions/extension_action.h"
   8 #include "chrome/browser/extensions/extension_action_manager.h"
   9 #include "chrome/browser/extensions/extension_toolbar_model.h"
  10 #include "chrome/browser/extensions/location_bar_controller.h"
  11 #include "chrome/browser/extensions/tab_helper.h"
  12 #include "chrome/browser/profiles/profile.h"
  13 #include "chrome/browser/sessions/session_tab_helper.h"
  14 #include "content/public/browser/web_contents.h"
  15 #include "extensions/common/extension.h"
  16 #include "extensions/common/feature_switch.h"
  17
  18 namespace extensions {
  19 namespace extension_action_test_util {
  20
  21 namespace {
  22
  23 size_t GetPageActionCount(content::WebContents* web_contents,
  24                           bool only_count_visible) {
  25   DCHECK(web_contents);
  26   size_t count = 0u;
  27   int tab_id = SessionTabHelper::IdForTab(web_contents);
  28   // Page actions are either stored in the location bar (and provided by the
  29   // LocationBarController), or in the main toolbar (and provided by the
  30   // ExtensionToolbarModel), depending on whether or not the extension action
  31   // redesign is enabled.
  32   if (!FeatureSwitch::extension_action_redesign()->IsEnabled()) {
  33     std::vector<ExtensionAction*> page_actions =
  34         TabHelper::FromWebContents(web_contents)->
  35             location_bar_controller()->GetCurrentActions();
  36     count = page_actions.size();
  37     // Trim any invisible page actions, if necessary.
  38     if (only_count_visible) {
  39       for (std::vector<ExtensionAction*>::iterator iter = page_actions.begin();
  40            iter != page_actions.end(); ++iter) {
  41         if (!(*iter)->GetIsVisible(tab_id))
  42           --count;
  43       }
  44     }
  45   } else {
  46     ExtensionToolbarModel* toolbar_model =
  47         ExtensionToolbarModel::Get(
  48             Profile::FromBrowserContext(web_contents->GetBrowserContext()));
  49     const ExtensionList& toolbar_extensions = toolbar_model->toolbar_items();
  50     ExtensionActionManager* action_manager =
  51         ExtensionActionManager::Get(web_contents->GetBrowserContext());
  52     for (ExtensionList::const_iterator iter = toolbar_extensions.begin();
  53          iter != toolbar_extensions.end(); ++iter) {
  54       ExtensionAction* extension_action = action_manager->GetPageAction(**iter);
  55       if (extension_action &&
  56           (!only_count_visible || extension_action->GetIsVisible(tab_id)))
  57         ++count;
  58     }
  59   }
  60
  61   return count;
  62 }
  63
  64 }  // namespace
  65
  66 size_t GetVisiblePageActionCount(content::WebContents* web_contents) {
  67   return GetPageActionCount(web_contents, true);
  68 }
  69
  70 size_t GetTotalPageActionCount(content::WebContents* web_contents) {
  71   return GetPageActionCount(web_contents, false);
  72 }
  73
  74 }  // namespace extension_action_test_util
  75 }  // namespace extensions