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
18namespace extensions {
19namespace extension_action_test_util {
20
21namespace {
22
23size_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
66size_t GetVisiblePageActionCount(content::WebContents* web_contents) {
67  return GetPageActionCount(web_contents, true);
68}
69
70size_t GetTotalPageActionCount(content::WebContents* web_contents) {
71  return GetPageActionCount(web_contents, false);
72}
73
74}  // namespace extension_action_test_util
75}  // namespace extensions
76