15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#ifndef CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_ 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#define CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_ 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <map> 95821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/callback.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_vector.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/menu_manager.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/models/simple_menu_model.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ExtensionContextMenuBrowserTest; 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class Profile; 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace extensions { 205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// This class contains code that is shared between the various places where 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// context menu items added by the extension or app should be shown. 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class ContextMenuMatcher { 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static const size_t kMaxExtensionItemTitleLength; 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The |filter| will be called on possibly matching menu items, and its 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // result is used to determine which items to actually append to the menu. 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ContextMenuMatcher(Profile* profile, 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::SimpleMenuModel::Delegate* delegate, 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::SimpleMenuModel* menu_model, 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const base::Callback<bool(const MenuItem*)>& filter); 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This is a helper function to append items for one particular extension. 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The |index| parameter is used for assigning id's, and is incremented for 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // each item actually added. 37a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) void AppendExtensionItems(const MenuItem::ExtensionKey& extension_key, 38a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) const base::string16& selection_text, 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int* index); 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void Clear(); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 437d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // This function returns the top level context menu title of an extension 447d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) // based on a printable selection text. 45a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) base::string16 GetTopLevelContextMenuTitle( 46a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const MenuItem::ExtensionKey& extension_key, 47a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) const base::string16& selection_text); 487d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool IsCommandIdChecked(int command_id) const; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool IsCommandIdEnabled(int command_id) const; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void ExecuteCommand(int command_id, 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) content::WebContents* web_contents, 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const content::ContextMenuParams& params); 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) friend class ::ExtensionContextMenuBrowserTest; 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 587d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) bool GetRelevantExtensionTopLevelItems( 59a1401311d1ab56c4ed0a474bd38c108f75cb0cd9Torne (Richard Coles) const MenuItem::ExtensionKey& extension_key, 607d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const Extension** extension, 617d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) bool* can_cross_incognito, 627d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) MenuItem::List& items); 637d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) MenuItem::List GetRelevantExtensionItems( 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const MenuItem::List& items, 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool can_cross_incognito); 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Used for recursively adding submenus of extension items. 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void RecursivelyAppendExtensionItems(const MenuItem::List& items, 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool can_cross_incognito, 71a3f6a49ab37290eeeb8db0f41ec0f1cb74a68be7Torne (Richard Coles) const base::string16& selection_text, 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::SimpleMenuModel* menu_model, 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int* index); 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Attempts to get an MenuItem given the id of a context menu item. 765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) extensions::MenuItem* GetExtensionMenuItem(int id) const; 775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // This will set the icon on the most recently-added item in the menu_model_. 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SetExtensionIcon(const std::string& extension_id); 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Profile* profile_; 825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::SimpleMenuModel* menu_model_; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ui::SimpleMenuModel::Delegate* delegate_; 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Callback<bool(const MenuItem*)> filter_; 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Maps the id from a context menu item to the MenuItem's internal id. 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::map<int, extensions::MenuItem::Id> extension_item_map_; 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Keep track of and clean up menu models for submenus. 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ScopedVector<ui::SimpleMenuModel> extension_menu_models_; 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DISALLOW_COPY_AND_ASSIGN(ContextMenuMatcher); 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace extensions 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#endif // CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_ 99