OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 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 | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "chrome/browser/automation/automation_provider.h" | 5 #include "chrome/browser/automation/automation_provider.h" |
6 | 6 |
7 #include <set> | 7 #include <set> |
8 | 8 |
9 #include "base/bind.h" | 9 #include "base/bind.h" |
10 #include "base/callback.h" | 10 #include "base/callback.h" |
(...skipping 10 matching lines...) Expand all Loading... |
21 #include "base/stl_util.h" | 21 #include "base/stl_util.h" |
22 #include "base/string_number_conversions.h" | 22 #include "base/string_number_conversions.h" |
23 #include "base/string_util.h" | 23 #include "base/string_util.h" |
24 #include "base/synchronization/waitable_event.h" | 24 #include "base/synchronization/waitable_event.h" |
25 #include "base/threading/thread.h" | 25 #include "base/threading/thread.h" |
26 #include "base/utf_string_conversions.h" | 26 #include "base/utf_string_conversions.h" |
27 #include "base/values.h" | 27 #include "base/values.h" |
28 #include "chrome/app/chrome_command_ids.h" | 28 #include "chrome/app/chrome_command_ids.h" |
29 #include "chrome/browser/autocomplete/autocomplete_edit.h" | 29 #include "chrome/browser/autocomplete/autocomplete_edit.h" |
30 #include "chrome/browser/automation/automation_browser_tracker.h" | 30 #include "chrome/browser/automation/automation_browser_tracker.h" |
31 #include "chrome/browser/automation/automation_extension_tracker.h" | |
32 #include "chrome/browser/automation/automation_provider_list.h" | 31 #include "chrome/browser/automation/automation_provider_list.h" |
33 #include "chrome/browser/automation/automation_provider_observers.h" | 32 #include "chrome/browser/automation/automation_provider_observers.h" |
34 #include "chrome/browser/automation/automation_resource_message_filter.h" | 33 #include "chrome/browser/automation/automation_resource_message_filter.h" |
35 #include "chrome/browser/automation/automation_tab_tracker.h" | 34 #include "chrome/browser/automation/automation_tab_tracker.h" |
36 #include "chrome/browser/automation/automation_window_tracker.h" | 35 #include "chrome/browser/automation/automation_window_tracker.h" |
37 #include "chrome/browser/bookmarks/bookmark_model.h" | 36 #include "chrome/browser/bookmarks/bookmark_model.h" |
38 #include "chrome/browser/bookmarks/bookmark_storage.h" | 37 #include "chrome/browser/bookmarks/bookmark_storage.h" |
39 #include "chrome/browser/browser_process.h" | 38 #include "chrome/browser/browser_process.h" |
40 #include "chrome/browser/browsing_data_remover.h" | 39 #include "chrome/browser/browsing_data_remover.h" |
41 #include "chrome/browser/character_encoding.h" | 40 #include "chrome/browser/character_encoding.h" |
42 #include "chrome/browser/content_settings/host_content_settings_map.h" | 41 #include "chrome/browser/content_settings/host_content_settings_map.h" |
43 #include "chrome/browser/extensions/crx_installer.h" | |
44 #include "chrome/browser/extensions/extension_browser_event_router.h" | |
45 #include "chrome/browser/extensions/extension_host.h" | |
46 #include "chrome/browser/extensions/extension_install_ui.h" | |
47 #include "chrome/browser/extensions/extension_message_service.h" | |
48 #include "chrome/browser/extensions/extension_service.h" | |
49 #include "chrome/browser/extensions/extension_system.h" | |
50 #include "chrome/browser/extensions/extension_tab_util.h" | |
51 #include "chrome/browser/extensions/extension_toolbar_model.h" | |
52 #include "chrome/browser/extensions/unpacked_installer.h" | |
53 #include "chrome/browser/extensions/user_script_master.h" | |
54 #include "chrome/browser/net/url_request_mock_util.h" | 42 #include "chrome/browser/net/url_request_mock_util.h" |
55 #include "chrome/browser/prefs/pref_service.h" | 43 #include "chrome/browser/prefs/pref_service.h" |
56 #include "chrome/browser/printing/print_job.h" | 44 #include "chrome/browser/printing/print_job.h" |
57 #include "chrome/browser/profiles/profile_manager.h" | 45 #include "chrome/browser/profiles/profile_manager.h" |
58 #include "chrome/browser/ssl/ssl_blocking_page.h" | 46 #include "chrome/browser/ssl/ssl_blocking_page.h" |
59 #include "chrome/browser/ui/app_modal_dialogs/app_modal_dialog.h" | 47 #include "chrome/browser/ui/app_modal_dialogs/app_modal_dialog.h" |
60 #include "chrome/browser/ui/app_modal_dialogs/app_modal_dialog_queue.h" | 48 #include "chrome/browser/ui/app_modal_dialogs/app_modal_dialog_queue.h" |
61 #include "chrome/browser/ui/browser_list.h" | 49 #include "chrome/browser/ui/browser_list.h" |
62 #include "chrome/browser/ui/browser_window.h" | 50 #include "chrome/browser/ui/browser_window.h" |
63 #include "chrome/browser/ui/find_bar/find_bar.h" | 51 #include "chrome/browser/ui/find_bar/find_bar.h" |
64 #include "chrome/browser/ui/find_bar/find_bar_controller.h" | 52 #include "chrome/browser/ui/find_bar/find_bar_controller.h" |
65 #include "chrome/browser/ui/find_bar/find_notification_details.h" | 53 #include "chrome/browser/ui/find_bar/find_notification_details.h" |
66 #include "chrome/browser/ui/find_bar/find_tab_helper.h" | 54 #include "chrome/browser/ui/find_bar/find_tab_helper.h" |
67 #include "chrome/browser/ui/login/login_prompt.h" | 55 #include "chrome/browser/ui/login/login_prompt.h" |
68 #include "chrome/browser/ui/omnibox/location_bar.h" | 56 #include "chrome/browser/ui/omnibox/location_bar.h" |
69 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" | 57 #include "chrome/browser/ui/tab_contents/tab_contents_wrapper.h" |
70 #include "chrome/common/automation_constants.h" | 58 #include "chrome/common/automation_constants.h" |
71 #include "chrome/common/automation_messages.h" | 59 #include "chrome/common/automation_messages.h" |
72 #include "chrome/common/chrome_constants.h" | 60 #include "chrome/common/chrome_constants.h" |
73 #include "chrome/common/chrome_paths.h" | 61 #include "chrome/common/chrome_paths.h" |
74 #include "chrome/common/chrome_switches.h" | 62 #include "chrome/common/chrome_switches.h" |
75 #include "chrome/common/chrome_version_info.h" | 63 #include "chrome/common/chrome_version_info.h" |
76 #include "chrome/common/extensions/extension.h" | |
77 #include "chrome/common/pref_names.h" | 64 #include "chrome/common/pref_names.h" |
78 #include "chrome/common/render_messages.h" | 65 #include "chrome/common/render_messages.h" |
79 #include "chrome/common/url_constants.h" | 66 #include "chrome/common/url_constants.h" |
80 #include "content/public/browser/browser_thread.h" | 67 #include "content/public/browser/browser_thread.h" |
81 #include "content/public/browser/download_item.h" | 68 #include "content/public/browser/download_item.h" |
82 #include "content/public/browser/render_view_host.h" | 69 #include "content/public/browser/render_view_host.h" |
83 #include "content/public/browser/trace_controller.h" | 70 #include "content/public/browser/trace_controller.h" |
84 #include "content/public/browser/web_contents.h" | 71 #include "content/public/browser/web_contents.h" |
85 #include "content/public/browser/web_contents_view.h" | 72 #include "content/public/browser/web_contents_view.h" |
86 #include "net/proxy/proxy_config_service_fixed.h" | 73 #include "net/proxy/proxy_config_service_fixed.h" |
(...skipping 88 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
175 use_initial_load_observers_(true), | 162 use_initial_load_observers_(true), |
176 is_connected_(false), | 163 is_connected_(false), |
177 initial_tab_loads_complete_(false), | 164 initial_tab_loads_complete_(false), |
178 network_library_initialized_(true), | 165 network_library_initialized_(true), |
179 login_webui_ready_(true) { | 166 login_webui_ready_(true) { |
180 TRACE_EVENT_BEGIN_ETW("AutomationProvider::AutomationProvider", 0, ""); | 167 TRACE_EVENT_BEGIN_ETW("AutomationProvider::AutomationProvider", 0, ""); |
181 | 168 |
182 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 169 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
183 | 170 |
184 browser_tracker_.reset(new AutomationBrowserTracker(this)); | 171 browser_tracker_.reset(new AutomationBrowserTracker(this)); |
185 extension_tracker_.reset(new AutomationExtensionTracker(this)); | |
186 tab_tracker_.reset(new AutomationTabTracker(this)); | 172 tab_tracker_.reset(new AutomationTabTracker(this)); |
187 window_tracker_.reset(new AutomationWindowTracker(this)); | 173 window_tracker_.reset(new AutomationWindowTracker(this)); |
188 new_tab_ui_load_observer_.reset(new NewTabUILoadObserver(this, profile)); | 174 new_tab_ui_load_observer_.reset(new NewTabUILoadObserver(this, profile)); |
189 metric_event_duration_observer_.reset(new MetricEventDurationObserver()); | 175 metric_event_duration_observer_.reset(new MetricEventDurationObserver()); |
190 extension_test_result_observer_.reset( | |
191 new ExtensionTestResultNotificationObserver(this)); | |
192 | 176 |
193 TRACE_EVENT_END_ETW("AutomationProvider::AutomationProvider", 0, ""); | 177 TRACE_EVENT_END_ETW("AutomationProvider::AutomationProvider", 0, ""); |
194 } | 178 } |
195 | 179 |
196 AutomationProvider::~AutomationProvider() { | 180 AutomationProvider::~AutomationProvider() { |
197 if (channel_.get()) | 181 if (channel_.get()) |
198 channel_->Close(); | 182 channel_->Close(); |
199 } | 183 } |
200 | 184 |
201 bool AutomationProvider::InitializeChannel(const std::string& channel_id) { | 185 bool AutomationProvider::InitializeChannel(const std::string& channel_id) { |
(...skipping 103 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
305 OnNetworkLibraryInit(); | 289 OnNetworkLibraryInit(); |
306 OnLoginWebuiReady(); | 290 OnLoginWebuiReady(); |
307 } | 291 } |
308 | 292 |
309 int AutomationProvider::GetIndexForNavigationController( | 293 int AutomationProvider::GetIndexForNavigationController( |
310 const NavigationController* controller, const Browser* parent) const { | 294 const NavigationController* controller, const Browser* parent) const { |
311 DCHECK(parent); | 295 DCHECK(parent); |
312 return parent->GetIndexOfController(controller); | 296 return parent->GetIndexOfController(controller); |
313 } | 297 } |
314 | 298 |
315 int AutomationProvider::AddExtension(const Extension* extension) { | |
316 DCHECK(extension); | |
317 return extension_tracker_->Add(extension); | |
318 } | |
319 | |
320 // TODO(phajdan.jr): move to TestingAutomationProvider. | 299 // TODO(phajdan.jr): move to TestingAutomationProvider. |
321 DictionaryValue* AutomationProvider::GetDictionaryFromDownloadItem( | 300 DictionaryValue* AutomationProvider::GetDictionaryFromDownloadItem( |
322 const DownloadItem* download) { | 301 const DownloadItem* download) { |
323 std::map<DownloadItem::DownloadState, std::string> state_to_string; | 302 std::map<DownloadItem::DownloadState, std::string> state_to_string; |
324 state_to_string[DownloadItem::IN_PROGRESS] = std::string("IN_PROGRESS"); | 303 state_to_string[DownloadItem::IN_PROGRESS] = std::string("IN_PROGRESS"); |
325 state_to_string[DownloadItem::CANCELLED] = std::string("CANCELLED"); | 304 state_to_string[DownloadItem::CANCELLED] = std::string("CANCELLED"); |
326 state_to_string[DownloadItem::REMOVING] = std::string("REMOVING"); | 305 state_to_string[DownloadItem::REMOVING] = std::string("REMOVING"); |
327 state_to_string[DownloadItem::INTERRUPTED] = std::string("INTERRUPTED"); | 306 state_to_string[DownloadItem::INTERRUPTED] = std::string("INTERRUPTED"); |
328 state_to_string[DownloadItem::COMPLETE] = std::string("COMPLETE"); | 307 state_to_string[DownloadItem::COMPLETE] = std::string("COMPLETE"); |
329 | 308 |
(...skipping 17 matching lines...) Expand all Loading... |
347 dl_item_value->SetBoolean("is_temporary", download->IsTemporary()); | 326 dl_item_value->SetBoolean("is_temporary", download->IsTemporary()); |
348 dl_item_value->SetBoolean("is_otr", download->IsOtr()); // incognito | 327 dl_item_value->SetBoolean("is_otr", download->IsOtr()); // incognito |
349 dl_item_value->SetString("state", state_to_string[download->GetState()]); | 328 dl_item_value->SetString("state", state_to_string[download->GetState()]); |
350 dl_item_value->SetString("safety_state", | 329 dl_item_value->SetString("safety_state", |
351 safety_state_to_string[download->GetSafetyState()]); | 330 safety_state_to_string[download->GetSafetyState()]); |
352 dl_item_value->SetInteger("PercentComplete", download->PercentComplete()); | 331 dl_item_value->SetInteger("PercentComplete", download->PercentComplete()); |
353 | 332 |
354 return dl_item_value; | 333 return dl_item_value; |
355 } | 334 } |
356 | 335 |
357 const Extension* AutomationProvider::GetExtension(int extension_handle) { | |
358 return extension_tracker_->GetResource(extension_handle); | |
359 } | |
360 | |
361 const Extension* AutomationProvider::GetEnabledExtension(int extension_handle) { | |
362 const Extension* extension = | |
363 extension_tracker_->GetResource(extension_handle); | |
364 ExtensionService* service = profile_->GetExtensionService(); | |
365 if (extension && service && | |
366 service->GetExtensionById(extension->id(), false)) | |
367 return extension; | |
368 return NULL; | |
369 } | |
370 | |
371 const Extension* AutomationProvider::GetDisabledExtension( | |
372 int extension_handle) { | |
373 const Extension* extension = | |
374 extension_tracker_->GetResource(extension_handle); | |
375 ExtensionService* service = profile_->GetExtensionService(); | |
376 if (extension && service && | |
377 service->GetExtensionById(extension->id(), true) && | |
378 !service->GetExtensionById(extension->id(), false)) | |
379 return extension; | |
380 return NULL; | |
381 } | |
382 | |
383 void AutomationProvider::OnChannelConnected(int pid) { | 336 void AutomationProvider::OnChannelConnected(int pid) { |
384 is_connected_ = true; | 337 is_connected_ = true; |
385 | 338 |
386 // Send a hello message with our current automation protocol version. | 339 // Send a hello message with our current automation protocol version. |
387 VLOG(2) << "Testing channel connected, sending hello message"; | 340 VLOG(2) << "Testing channel connected, sending hello message"; |
388 channel_->Send(new AutomationMsg_Hello(GetProtocolVersion())); | 341 channel_->Send(new AutomationMsg_Hello(GetProtocolVersion())); |
389 | 342 |
390 SendInitialLoadMessage(); | 343 SendInitialLoadMessage(); |
391 } | 344 } |
392 | 345 |
(...skipping 24 matching lines...) Expand all Loading... |
417 IPC_MESSAGE_HANDLER(AutomationMsg_PrintAsync, PrintAsync) | 370 IPC_MESSAGE_HANDLER(AutomationMsg_PrintAsync, PrintAsync) |
418 IPC_MESSAGE_HANDLER_DELAY_REPLY(AutomationMsg_Find, HandleFindRequest) | 371 IPC_MESSAGE_HANDLER_DELAY_REPLY(AutomationMsg_Find, HandleFindRequest) |
419 IPC_MESSAGE_HANDLER(AutomationMsg_OverrideEncoding, OverrideEncoding) | 372 IPC_MESSAGE_HANDLER(AutomationMsg_OverrideEncoding, OverrideEncoding) |
420 IPC_MESSAGE_HANDLER(AutomationMsg_SelectAll, SelectAll) | 373 IPC_MESSAGE_HANDLER(AutomationMsg_SelectAll, SelectAll) |
421 IPC_MESSAGE_HANDLER(AutomationMsg_Cut, Cut) | 374 IPC_MESSAGE_HANDLER(AutomationMsg_Cut, Cut) |
422 IPC_MESSAGE_HANDLER(AutomationMsg_Copy, Copy) | 375 IPC_MESSAGE_HANDLER(AutomationMsg_Copy, Copy) |
423 IPC_MESSAGE_HANDLER(AutomationMsg_Paste, Paste) | 376 IPC_MESSAGE_HANDLER(AutomationMsg_Paste, Paste) |
424 IPC_MESSAGE_HANDLER(AutomationMsg_ReloadAsync, ReloadAsync) | 377 IPC_MESSAGE_HANDLER(AutomationMsg_ReloadAsync, ReloadAsync) |
425 IPC_MESSAGE_HANDLER(AutomationMsg_StopAsync, StopAsync) | 378 IPC_MESSAGE_HANDLER(AutomationMsg_StopAsync, StopAsync) |
426 IPC_MESSAGE_HANDLER(AutomationMsg_SetPageFontSize, OnSetPageFontSize) | 379 IPC_MESSAGE_HANDLER(AutomationMsg_SetPageFontSize, OnSetPageFontSize) |
427 IPC_MESSAGE_HANDLER_DELAY_REPLY(AutomationMsg_WaitForExtensionTestResult, | |
428 WaitForExtensionTestResult) | |
429 IPC_MESSAGE_HANDLER_DELAY_REPLY(AutomationMsg_InstallExtension, | |
430 InstallExtension) | |
431 IPC_MESSAGE_HANDLER(AutomationMsg_UninstallExtension, | |
432 UninstallExtension) | |
433 IPC_MESSAGE_HANDLER_DELAY_REPLY(AutomationMsg_EnableExtension, | |
434 EnableExtension) | |
435 IPC_MESSAGE_HANDLER(AutomationMsg_DisableExtension, | |
436 DisableExtension) | |
437 IPC_MESSAGE_HANDLER_DELAY_REPLY( | |
438 AutomationMsg_ExecuteExtensionActionInActiveTabAsync, | |
439 ExecuteExtensionActionInActiveTabAsync) | |
440 IPC_MESSAGE_HANDLER(AutomationMsg_MoveExtensionBrowserAction, | |
441 MoveExtensionBrowserAction) | |
442 IPC_MESSAGE_HANDLER(AutomationMsg_GetExtensionProperty, | |
443 GetExtensionProperty) | |
444 IPC_MESSAGE_HANDLER(AutomationMsg_SaveAsAsync, SaveAsAsync) | 380 IPC_MESSAGE_HANDLER(AutomationMsg_SaveAsAsync, SaveAsAsync) |
445 IPC_MESSAGE_HANDLER(AutomationMsg_RemoveBrowsingData, RemoveBrowsingData) | 381 IPC_MESSAGE_HANDLER(AutomationMsg_RemoveBrowsingData, RemoveBrowsingData) |
446 IPC_MESSAGE_HANDLER(AutomationMsg_JavaScriptStressTestControl, | 382 IPC_MESSAGE_HANDLER(AutomationMsg_JavaScriptStressTestControl, |
447 JavaScriptStressTestControl) | 383 JavaScriptStressTestControl) |
448 IPC_MESSAGE_HANDLER(AutomationMsg_BeginTracing, BeginTracing) | 384 IPC_MESSAGE_HANDLER(AutomationMsg_BeginTracing, BeginTracing) |
449 IPC_MESSAGE_HANDLER_DELAY_REPLY(AutomationMsg_EndTracing, EndTracing) | 385 IPC_MESSAGE_HANDLER_DELAY_REPLY(AutomationMsg_EndTracing, EndTracing) |
450 IPC_MESSAGE_HANDLER(AutomationMsg_GetTracingOutput, GetTracingOutput) | 386 IPC_MESSAGE_HANDLER(AutomationMsg_GetTracingOutput, GetTracingOutput) |
451 #if defined(OS_WIN) && !defined(USE_AURA) | 387 #if defined(OS_WIN) && !defined(USE_AURA) |
452 // These are for use with external tabs. | 388 // These are for use with external tabs. |
453 IPC_MESSAGE_HANDLER(AutomationMsg_CreateExternalTab, CreateExternalTab) | 389 IPC_MESSAGE_HANDLER(AutomationMsg_CreateExternalTab, CreateExternalTab) |
(...skipping 357 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
811 return NULL; | 747 return NULL; |
812 } | 748 } |
813 | 749 |
814 RenderViewHost* view_host = web_contents->GetRenderViewHost(); | 750 RenderViewHost* view_host = web_contents->GetRenderViewHost(); |
815 return view_host; | 751 return view_host; |
816 } | 752 } |
817 | 753 |
818 return NULL; | 754 return NULL; |
819 } | 755 } |
820 | 756 |
821 void AutomationProvider::WaitForExtensionTestResult( | |
822 IPC::Message* reply_message) { | |
823 DCHECK(!reply_message_); | |
824 reply_message_ = reply_message; | |
825 // Call MaybeSendResult, because the result might have come in before | |
826 // we were waiting on it. | |
827 extension_test_result_observer_->MaybeSendResult(); | |
828 } | |
829 | |
830 void AutomationProvider::InstallExtension( | |
831 const FilePath& extension_path, bool with_ui, | |
832 IPC::Message* reply_message) { | |
833 ExtensionService* service = profile_->GetExtensionService(); | |
834 ExtensionProcessManager* manager = profile_->GetExtensionProcessManager(); | |
835 if (service && manager) { | |
836 // The observer will delete itself when done. | |
837 new ExtensionReadyNotificationObserver( | |
838 manager, | |
839 service, | |
840 this, | |
841 AutomationMsg_InstallExtension::ID, | |
842 reply_message); | |
843 | |
844 if (extension_path.MatchesExtension(FILE_PATH_LITERAL(".crx"))) { | |
845 ExtensionInstallUI* client = | |
846 (with_ui ? new ExtensionInstallUI(profile_) : NULL); | |
847 scoped_refptr<CrxInstaller> installer( | |
848 CrxInstaller::Create(service, client)); | |
849 if (!with_ui) | |
850 installer->set_allow_silent_install(true); | |
851 installer->set_install_cause(extension_misc::INSTALL_CAUSE_AUTOMATION); | |
852 installer->InstallCrx(extension_path); | |
853 } else { | |
854 scoped_refptr<extensions::UnpackedInstaller> installer( | |
855 extensions::UnpackedInstaller::Create(service)); | |
856 installer->set_prompt_for_plugins(with_ui); | |
857 installer->Load(extension_path); | |
858 } | |
859 } else { | |
860 AutomationMsg_InstallExtension::WriteReplyParams(reply_message, 0); | |
861 Send(reply_message); | |
862 } | |
863 } | |
864 | |
865 void AutomationProvider::UninstallExtension(int extension_handle, | |
866 bool* success) { | |
867 *success = false; | |
868 const Extension* extension = GetExtension(extension_handle); | |
869 ExtensionService* service = profile_->GetExtensionService(); | |
870 if (extension && service) { | |
871 ExtensionUnloadNotificationObserver observer; | |
872 service->UninstallExtension(extension->id(), false, NULL); | |
873 // The extension unload notification should have been sent synchronously | |
874 // with the uninstall. Just to be safe, check that it was received. | |
875 *success = observer.did_receive_unload_notification(); | |
876 } | |
877 } | |
878 | |
879 void AutomationProvider::EnableExtension(int extension_handle, | |
880 IPC::Message* reply_message) { | |
881 const Extension* extension = GetDisabledExtension(extension_handle); | |
882 ExtensionService* service = profile_->GetExtensionService(); | |
883 ExtensionProcessManager* manager = profile_->GetExtensionProcessManager(); | |
884 // Only enable if this extension is disabled. | |
885 if (extension && service && manager) { | |
886 // The observer will delete itself when done. | |
887 new ExtensionReadyNotificationObserver( | |
888 manager, | |
889 service, | |
890 this, | |
891 AutomationMsg_EnableExtension::ID, | |
892 reply_message); | |
893 service->EnableExtension(extension->id()); | |
894 } else { | |
895 AutomationMsg_EnableExtension::WriteReplyParams(reply_message, false); | |
896 Send(reply_message); | |
897 } | |
898 } | |
899 | |
900 void AutomationProvider::DisableExtension(int extension_handle, | |
901 bool* success) { | |
902 *success = false; | |
903 const Extension* extension = GetEnabledExtension(extension_handle); | |
904 ExtensionService* service = profile_->GetExtensionService(); | |
905 if (extension && service) { | |
906 ExtensionUnloadNotificationObserver observer; | |
907 service->DisableExtension(extension->id(), Extension::DISABLE_USER_ACTION); | |
908 // The extension unload notification should have been sent synchronously | |
909 // with the disable. Just to be safe, check that it was received. | |
910 *success = observer.did_receive_unload_notification(); | |
911 } | |
912 } | |
913 | |
914 void AutomationProvider::ExecuteExtensionActionInActiveTabAsync( | |
915 int extension_handle, int browser_handle, | |
916 IPC::Message* reply_message) { | |
917 bool success = false; | |
918 const Extension* extension = GetEnabledExtension(extension_handle); | |
919 ExtensionService* service = profile_->GetExtensionService(); | |
920 ExtensionMessageService* message_service = | |
921 ExtensionSystem::Get(profile_)->message_service(); | |
922 Browser* browser = browser_tracker_->GetResource(browser_handle); | |
923 if (extension && service && message_service && browser) { | |
924 int tab_id = ExtensionTabUtil::GetTabId(browser->GetSelectedWebContents()); | |
925 if (extension->page_action()) { | |
926 service->browser_event_router()->PageActionExecuted( | |
927 browser->profile(), extension->id(), "action", tab_id, "", 1); | |
928 success = true; | |
929 } else if (extension->browser_action()) { | |
930 service->browser_event_router()->BrowserActionExecuted( | |
931 browser->profile(), extension->id(), browser); | |
932 success = true; | |
933 } | |
934 } | |
935 AutomationMsg_ExecuteExtensionActionInActiveTabAsync::WriteReplyParams( | |
936 reply_message, success); | |
937 Send(reply_message); | |
938 } | |
939 | |
940 void AutomationProvider::MoveExtensionBrowserAction( | |
941 int extension_handle, int index, bool* success) { | |
942 *success = false; | |
943 const Extension* extension = GetEnabledExtension(extension_handle); | |
944 ExtensionService* service = profile_->GetExtensionService(); | |
945 if (extension && service) { | |
946 ExtensionToolbarModel* toolbar = service->toolbar_model(); | |
947 if (toolbar) { | |
948 if (index >= 0 && index < static_cast<int>(toolbar->size())) { | |
949 toolbar->MoveBrowserAction(extension, index); | |
950 *success = true; | |
951 } else { | |
952 DLOG(WARNING) << "Attempted to move browser action to invalid index."; | |
953 } | |
954 } | |
955 } | |
956 } | |
957 | |
958 void AutomationProvider::GetExtensionProperty( | |
959 int extension_handle, | |
960 AutomationMsg_ExtensionProperty type, | |
961 bool* success, | |
962 std::string* value) { | |
963 *success = false; | |
964 const Extension* extension = GetExtension(extension_handle); | |
965 ExtensionService* service = profile_->GetExtensionService(); | |
966 if (extension && service) { | |
967 ExtensionToolbarModel* toolbar = service->toolbar_model(); | |
968 int found_index = -1; | |
969 int index = 0; | |
970 switch (type) { | |
971 case AUTOMATION_MSG_EXTENSION_ID: | |
972 *value = extension->id(); | |
973 *success = true; | |
974 break; | |
975 case AUTOMATION_MSG_EXTENSION_NAME: | |
976 *value = extension->name(); | |
977 *success = true; | |
978 break; | |
979 case AUTOMATION_MSG_EXTENSION_VERSION: | |
980 *value = extension->VersionString(); | |
981 *success = true; | |
982 break; | |
983 case AUTOMATION_MSG_EXTENSION_BROWSER_ACTION_INDEX: | |
984 if (toolbar) { | |
985 for (ExtensionList::const_iterator iter = toolbar->begin(); | |
986 iter != toolbar->end(); iter++) { | |
987 // Skip this extension if we are in incognito mode | |
988 // and it is not incognito-enabled. | |
989 if (profile_->IsOffTheRecord() && | |
990 !service->IsIncognitoEnabled((*iter)->id())) | |
991 continue; | |
992 if (*iter == extension) { | |
993 found_index = index; | |
994 break; | |
995 } | |
996 index++; | |
997 } | |
998 *value = base::IntToString(found_index); | |
999 *success = true; | |
1000 } | |
1001 break; | |
1002 default: | |
1003 LOG(WARNING) << "Trying to get undefined extension property"; | |
1004 break; | |
1005 } | |
1006 } | |
1007 } | |
1008 | |
1009 void AutomationProvider::SaveAsAsync(int tab_handle) { | 757 void AutomationProvider::SaveAsAsync(int tab_handle) { |
1010 NavigationController* tab = NULL; | 758 NavigationController* tab = NULL; |
1011 WebContents* web_contents = GetWebContentsForHandle(tab_handle, &tab); | 759 WebContents* web_contents = GetWebContentsForHandle(tab_handle, &tab); |
1012 if (web_contents) | 760 if (web_contents) |
1013 web_contents->OnSavePage(); | 761 web_contents->OnSavePage(); |
1014 } | 762 } |
OLD | NEW |