1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
|
// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_
#define CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_
#include <stddef.h>
#include <map>
#include <memory>
#include <string>
#include <vector>
#include "base/functional/callback.h"
#include "base/memory/raw_ptr.h"
#include "chrome/browser/extensions/menu_manager.h"
#include "ui/menus/simple_menu_model.h"
class ExtensionContextMenuBrowserTest;
namespace content {
class BrowserContext;
class RenderFrameHost;
}
namespace extensions {
// This class contains code that is shared between the various places where
// context menu items added by the extension or app should be shown.
class ContextMenuMatcher {
public:
static const size_t kMaxExtensionItemTitleLength;
// Convert a command ID so that it fits within the range for
// extension context menu.
static int ConvertToExtensionsCustomCommandId(int id);
// Returns true if the given id is one generated for extension context menu.
static bool IsExtensionsCustomCommandId(int id);
// The `filter` will be called on possibly matching menu items, and its
// result is used to determine which items to actually append to the menu.
ContextMenuMatcher(content::BrowserContext* context,
ui::SimpleMenuModel::Delegate* delegate,
ui::SimpleMenuModel* menu_model,
base::RepeatingCallback<bool(const MenuItem*)> filter);
ContextMenuMatcher(const ContextMenuMatcher&) = delete;
ContextMenuMatcher& operator=(const ContextMenuMatcher&) = delete;
~ContextMenuMatcher();
// This is a helper function to append items for one particular extension.
// The `index` parameter is used for assigning id's, and is incremented for
// each item actually added. `is_action_menu` is used for browser and page
// action context menus, in which menu items are not placed in submenus
// and the extension's icon is not shown.
void AppendExtensionItems(
const MenuItem::ExtensionKey& extension_key,
const std::u16string& selection_text,
int* index,
bool is_action_menu,
const std::u16string& group_title = std::u16string());
// Returns true if the given menu_model has any visible items.
bool HasVisibleItems(ui::MenuModel* menu_model) const;
void Clear();
// This function returns the top level context menu title of an extension
// based on a printable selection text.
std::u16string GetTopLevelContextMenuTitle(
const MenuItem::ExtensionKey& extension_key,
const std::u16string& selection_text);
void set_smart_text_selection_enabled(bool enabled) {
is_smart_text_selection_enabled_ = enabled;
}
const std::map<int, extensions::MenuItem::Id> extension_item_map() {
return extension_item_map_;
}
bool IsCommandIdChecked(int command_id) const;
bool IsCommandIdVisible(int command_id) const;
bool IsCommandIdEnabled(int command_id) const;
void ExecuteCommand(int command_id,
content::WebContents* web_contents,
content::RenderFrameHost* render_frame_host,
const content::ContextMenuParams& params);
private:
friend class ::ExtensionContextMenuBrowserTest;
bool GetRelevantExtensionTopLevelItems(
const MenuItem::ExtensionKey& extension_key,
const Extension*& extension,
bool& can_cross_incognito,
MenuItem::List* items);
MenuItem::List GetRelevantExtensionItems(const MenuItem::OwnedList& items,
bool can_cross_incognito);
// Used for recursively adding submenus of extension items.
void RecursivelyAppendExtensionItems(const MenuItem::List& items,
bool can_cross_incognito,
const std::u16string& selection_text,
ui::SimpleMenuModel* menu_model,
int* index,
bool is_action_menu_top_level);
// Attempts to get an MenuItem given the id of a context menu item.
extensions::MenuItem* GetExtensionMenuItem(int id) const;
// This will set the icon on the most recently-added item in the menu_model_.
void SetExtensionIcon(const MenuItem::ExtensionKey& extension_key);
raw_ptr<content::BrowserContext, DanglingUntriaged> browser_context_;
raw_ptr<ui::SimpleMenuModel, DanglingUntriaged> menu_model_;
raw_ptr<ui::SimpleMenuModel::Delegate, DanglingUntriaged> delegate_;
base::RepeatingCallback<bool(const MenuItem*)> filter_;
bool is_smart_text_selection_enabled_;
// Maps the id from a context menu item to the MenuItem's internal id.
std::map<int, extensions::MenuItem::Id> extension_item_map_;
// Keep track of and clean up menu models for submenus.
std::vector<std::unique_ptr<ui::SimpleMenuModel>> extension_menu_models_;
};
} // namespace extensions
#endif // CHROME_BROWSER_EXTENSIONS_CONTEXT_MENU_MATCHER_H_
|
