chrome/browser/extensions/location_bar_controller.h

// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CHROME_BROWSER_EXTENSIONS_LOCATION_BAR_CONTROLLER_H_
#define CHROME_BROWSER_EXTENSIONS_LOCATION_BAR_CONTROLLER_H_

#include <map>
#include <memory>
#include <vector>

#include "base/macros.h"
#include "base/scoped_observer.h"
#include "extensions/browser/extension_registry_observer.h"

class ExtensionAction;

namespace content {
class WebContents;
class BrowserContext;
}

namespace extensions {
class Extension;
class ExtensionActionManager;
class ExtensionRegistry;

// Provides the UI with the current page actions for extensions. The execution
// of these actions is handled in the ExtensionActionAPI.
class LocationBarController : public ExtensionRegistryObserver {
 public:
  explicit LocationBarController(content::WebContents* web_contents);
  ~LocationBarController() override;

  // Returns the actions which should be displayed in the location bar.
  std::vector<ExtensionAction*> GetCurrentActions();

 private:
  // ExtensionRegistryObserver implementation.
  void OnExtensionLoaded(content::BrowserContext* browser_context,
                         const Extension* extension) override;
  void OnExtensionUnloaded(content::BrowserContext* browser_context,
                           const Extension* extension,
                           UnloadedExtensionInfo::Reason reason) override;

  // The associated WebContents.
  content::WebContents* web_contents_;

  // The associated BrowserContext.
  content::BrowserContext* browser_context_;

  // The ExtensionActionManager to provide page actions.
  ExtensionActionManager* action_manager_;

  // Whether or not to show page actions in the location bar at all. (This is
  // false with the toolbar redesign enabled.)
  bool should_show_page_actions_;

  // Manufactured page actions that have been generated for extensions that want
  // to run a script, but were blocked.
  using ExtensionActionMap =
      std::map<std::string, std::unique_ptr<ExtensionAction>>;
  ExtensionActionMap active_script_actions_;

  ScopedObserver<ExtensionRegistry, ExtensionRegistryObserver>
      extension_registry_observer_;

  DISALLOW_COPY_AND_ASSIGN(LocationBarController);
};

}  // namespace extensions

#endif  // CHROME_BROWSER_EXTENSIONS_LOCATION_BAR_CONTROLLER_H_