1 // Copyright (c) 2012 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 #ifndef CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_ 6 #define CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_ 7 8 #include <map> 9 10 #include "base/callback.h" 11 #include "base/memory/scoped_ptr.h" 12 #include "base/memory/scoped_vector.h" 13 #include "chrome/browser/extensions/menu_manager.h" 14 #include "ui/base/models/simple_menu_model.h" 15 16 class ExtensionContextMenuBrowserTest; 17 class Profile; 18 19 namespace extensions { 20 21 // This class contains code that is shared between the various places where 22 // context menu items added by the extension or app should be shown. 23 class ContextMenuMatcher { 24 public: 25 static const size_t kMaxExtensionItemTitleLength; 26 27 // The |filter| will be called on possibly matching menu items, and its 28 // result is used to determine which items to actually append to the menu. 29 ContextMenuMatcher(Profile* profile, 30 ui::SimpleMenuModel::Delegate* delegate, 31 ui::SimpleMenuModel* menu_model, 32 const base::Callback<bool(const MenuItem*)>& filter); 33 34 // This is a helper function to append items for one particular extension. 35 // The |index| parameter is used for assigning id's, and is incremented for 36 // each item actually added. 37 void AppendExtensionItems(const MenuItem::ExtensionKey& extension_key, 38 const base::string16& selection_text, 39 int* index); 40 41 void Clear(); 42 43 // This function returns the top level context menu title of an extension 44 // based on a printable selection text. 45 base::string16 GetTopLevelContextMenuTitle( 46 const MenuItem::ExtensionKey& extension_key, 47 const base::string16& selection_text); 48 49 bool IsCommandIdChecked(int command_id) const; 50 bool IsCommandIdEnabled(int command_id) const; 51 void ExecuteCommand(int command_id, 52 content::WebContents* web_contents, 53 const content::ContextMenuParams& params); 54 55 private: 56 friend class ::ExtensionContextMenuBrowserTest; 57 58 bool GetRelevantExtensionTopLevelItems( 59 const MenuItem::ExtensionKey& extension_key, 60 const Extension** extension, 61 bool* can_cross_incognito, 62 MenuItem::List& items); 63 64 MenuItem::List GetRelevantExtensionItems( 65 const MenuItem::List& items, 66 bool can_cross_incognito); 67 68 // Used for recursively adding submenus of extension items. 69 void RecursivelyAppendExtensionItems(const MenuItem::List& items, 70 bool can_cross_incognito, 71 const base::string16& selection_text, 72 ui::SimpleMenuModel* menu_model, 73 int* index); 74 75 // Attempts to get an MenuItem given the id of a context menu item. 76 extensions::MenuItem* GetExtensionMenuItem(int id) const; 77 78 // This will set the icon on the most recently-added item in the menu_model_. 79 void SetExtensionIcon(const std::string& extension_id); 80 81 Profile* profile_; 82 ui::SimpleMenuModel* menu_model_; 83 ui::SimpleMenuModel::Delegate* delegate_; 84 85 base::Callback<bool(const MenuItem*)> filter_; 86 87 // Maps the id from a context menu item to the MenuItem's internal id. 88 std::map<int, extensions::MenuItem::Id> extension_item_map_; 89 90 // Keep track of and clean up menu models for submenus. 91 ScopedVector<ui::SimpleMenuModel> extension_menu_models_; 92 93 DISALLOW_COPY_AND_ASSIGN(ContextMenuMatcher); 94 }; 95 96 } // namespace extensions 97 98 #endif // CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_ 99