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 #include "chrome/browser/ui/webui/extensions/command_handler.h"
6
7 #include "base/bind.h"
8 #include "base/values.h"
9 #include "chrome/browser/chrome_notification_types.h"
10 #include "chrome/browser/extensions/api/commands/command_service.h"
11 #include "chrome/browser/extensions/extension_keybinding_registry.h"
12 #include "chrome/browser/extensions/extension_service.h"
13 #include "chrome/browser/extensions/extension_system.h"
14 #include "chrome/browser/profiles/profile.h"
15 #include "chrome/common/extensions/extension_set.h"
16 #include "content/public/browser/web_ui.h"
17 #include "content/public/browser/web_ui_data_source.h"
18 #include "grit/generated_resources.h"
19 #include "ui/base/l10n/l10n_util.h"
20
21 namespace extensions {
22
CommandHandler(Profile * profile)23 CommandHandler::CommandHandler(Profile* profile) : profile_(profile) {
24 }
25
~CommandHandler()26 CommandHandler::~CommandHandler() {
27 }
28
GetLocalizedValues(content::WebUIDataSource * source)29 void CommandHandler::GetLocalizedValues(content::WebUIDataSource* source) {
30 source->AddString("extensionCommandsOverlay",
31 l10n_util::GetStringUTF16(IDS_EXTENSION_COMMANDS_DIALOG_TITLE));
32 source->AddString("extensionCommandsEmpty",
33 l10n_util::GetStringUTF16(IDS_EXTENSION_COMMANDS_EMPTY));
34 source->AddString("extensionCommandsInactive",
35 l10n_util::GetStringUTF16(IDS_EXTENSION_COMMANDS_INACTIVE));
36 source->AddString("extensionCommandsStartTyping",
37 l10n_util::GetStringUTF16(IDS_EXTENSION_TYPE_SHORTCUT));
38 source->AddString("extensionCommandsDelete",
39 l10n_util::GetStringUTF16(IDS_EXTENSION_DELETE_SHORTCUT));
40 source->AddString("extensionCommandsGlobal",
41 l10n_util::GetStringUTF16(IDS_EXTENSION_COMMANDS_GLOBAL));
42 source->AddString("extensionCommandsRegular",
43 l10n_util::GetStringUTF16(IDS_EXTENSION_COMMANDS_NOT_GLOBAL));
44 source->AddString("ok", l10n_util::GetStringUTF16(IDS_OK));
45 }
46
RegisterMessages()47 void CommandHandler::RegisterMessages() {
48 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_LOADED,
49 content::Source<Profile>(profile_));
50 registrar_.Add(this, chrome::NOTIFICATION_EXTENSION_UNLOADED,
51 content::Source<Profile>(profile_));
52
53 web_ui()->RegisterMessageCallback("extensionCommandsRequestExtensionsData",
54 base::Bind(&CommandHandler::HandleRequestExtensionsData,
55 base::Unretained(this)));
56 web_ui()->RegisterMessageCallback("setShortcutHandlingSuspended",
57 base::Bind(&CommandHandler::HandleSetShortcutHandlingSuspended,
58 base::Unretained(this)));
59 web_ui()->RegisterMessageCallback("setExtensionCommandShortcut",
60 base::Bind(&CommandHandler::HandleSetExtensionCommandShortcut,
61 base::Unretained(this)));
62 web_ui()->RegisterMessageCallback("setCommandScope",
63 base::Bind(&CommandHandler::HandleSetCommandScope,
64 base::Unretained(this)));
65 }
66
Observe(int type,const content::NotificationSource & source,const content::NotificationDetails & details)67 void CommandHandler::Observe(
68 int type,
69 const content::NotificationSource& source,
70 const content::NotificationDetails& details) {
71 DCHECK(type == chrome::NOTIFICATION_EXTENSION_LOADED ||
72 type == chrome::NOTIFICATION_EXTENSION_UNLOADED);
73 UpdateCommandDataOnPage();
74 }
75
UpdateCommandDataOnPage()76 void CommandHandler::UpdateCommandDataOnPage() {
77 DictionaryValue results;
78 GetAllCommands(&results);
79 web_ui()->CallJavascriptFunction(
80 "extensions.ExtensionCommandsOverlay.returnExtensionsData", results);
81 }
82
HandleRequestExtensionsData(const ListValue * args)83 void CommandHandler::HandleRequestExtensionsData(const ListValue* args) {
84 UpdateCommandDataOnPage();
85 }
86
HandleSetExtensionCommandShortcut(const base::ListValue * args)87 void CommandHandler::HandleSetExtensionCommandShortcut(
88 const base::ListValue* args) {
89 std::string extension_id;
90 std::string command_name;
91 std::string keystroke;
92 if (!args->GetString(0, &extension_id) ||
93 !args->GetString(1, &command_name) ||
94 !args->GetString(2, &keystroke)) {
95 NOTREACHED();
96 return;
97 }
98
99 Profile* profile = Profile::FromWebUI(web_ui());
100 CommandService* command_service = CommandService::Get(profile);
101 command_service->UpdateKeybindingPrefs(extension_id, command_name, keystroke);
102
103 UpdateCommandDataOnPage();
104 }
105
HandleSetCommandScope(const base::ListValue * args)106 void CommandHandler::HandleSetCommandScope(
107 const base::ListValue* args) {
108 std::string extension_id;
109 std::string command_name;
110 bool global;
111 if (!args->GetString(0, &extension_id) ||
112 !args->GetString(1, &command_name) ||
113 !args->GetBoolean(2, &global)) {
114 NOTREACHED();
115 return;
116 }
117
118 Profile* profile = Profile::FromWebUI(web_ui());
119 CommandService* command_service = CommandService::Get(profile);
120 if (command_service->SetScope(extension_id, command_name, global))
121 UpdateCommandDataOnPage();
122 }
123
HandleSetShortcutHandlingSuspended(const ListValue * args)124 void CommandHandler::HandleSetShortcutHandlingSuspended(const ListValue* args) {
125 bool suspended;
126 if (args->GetBoolean(0, &suspended))
127 ExtensionKeybindingRegistry::SetShortcutHandlingSuspended(suspended);
128 }
129
GetAllCommands(base::DictionaryValue * commands)130 void CommandHandler::GetAllCommands(base::DictionaryValue* commands) {
131 ListValue* results = new ListValue;
132
133 Profile* profile = Profile::FromWebUI(web_ui());
134 CommandService* command_service = CommandService::Get(profile);
135
136 const ExtensionSet* extensions = extensions::ExtensionSystem::Get(profile)->
137 extension_service()->extensions();
138 for (ExtensionSet::const_iterator extension = extensions->begin();
139 extension != extensions->end(); ++extension) {
140 scoped_ptr<DictionaryValue> extension_dict(new DictionaryValue);
141 extension_dict->SetString("name", (*extension)->name());
142 extension_dict->SetString("id", (*extension)->id());
143
144 // Add the keybindings to a list structure.
145 scoped_ptr<ListValue> extensions_list(new ListValue());
146
147 bool active = false;
148
149 extensions::Command browser_action;
150 if (command_service->GetBrowserActionCommand((*extension)->id(),
151 CommandService::ALL,
152 &browser_action,
153 &active)) {
154 extensions_list->Append(
155 browser_action.ToValue((extension->get()), active));
156 }
157
158 extensions::Command page_action;
159 if (command_service->GetPageActionCommand((*extension)->id(),
160 CommandService::ALL,
161 &page_action,
162 &active)) {
163 extensions_list->Append(page_action.ToValue((extension->get()), active));
164 }
165
166 extensions::Command script_badge;
167 if (command_service->GetScriptBadgeCommand((*extension)->id(),
168 CommandService::ALL,
169 &script_badge,
170 &active)) {
171 extensions_list->Append(script_badge.ToValue((extension->get()), active));
172 }
173
174 extensions::CommandMap named_commands;
175 if (command_service->GetNamedCommands((*extension)->id(),
176 CommandService::ALL,
177 extensions::CommandService::ANY_SCOPE,
178 &named_commands)) {
179 for (extensions::CommandMap::const_iterator iter = named_commands.begin();
180 iter != named_commands.end(); ++iter) {
181 extensions::Command command = command_service->FindCommandByName(
182 (*extension)->id(), iter->second.command_name());
183 ui::Accelerator shortcut_assigned = command.accelerator();
184
185 active = (shortcut_assigned.key_code() != ui::VKEY_UNKNOWN);
186
187 extensions_list->Append(
188 iter->second.ToValue((extension->get()), active));
189 }
190 }
191
192 if (!extensions_list->empty()) {
193 extension_dict->Set("commands", extensions_list.release());
194 results->Append(extension_dict.release());
195 }
196 }
197
198 commands->Set("commands", results);
199 }
200
201 } // namespace extensions
202