| 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 // Download code which handles CRX files (extensions, themes, apps, ...). | 5 // Download code which handles CRX files (extensions, themes, apps, ...). |
| 6 | 6 |
| 7 #include "chrome/browser/download/download_util.h" | 7 #include "chrome/browser/download/download_util.h" |
| 8 #include "chrome/browser/extensions/crx_installer.h" | 8 #include "chrome/browser/extensions/crx_installer.h" |
| 9 #include "chrome/browser/extensions/extension_install_ui.h" | 9 #include "chrome/browser/extensions/extension_install_prompt.h" |
| 10 #include "chrome/browser/extensions/extension_service.h" | 10 #include "chrome/browser/extensions/extension_service.h" |
| 11 #include "chrome/browser/extensions/webstore_installer.h" | 11 #include "chrome/browser/extensions/webstore_installer.h" |
| 12 #include "chrome/browser/profiles/profile.h" | 12 #include "chrome/browser/profiles/profile.h" |
| 13 #include "chrome/common/chrome_notification_types.h" | 13 #include "chrome/common/chrome_notification_types.h" |
| 14 #include "chrome/common/extensions/extension_switch_utils.h" | 14 #include "chrome/common/extensions/extension_switch_utils.h" |
| 15 #include "content/public/browser/download_item.h" | 15 #include "content/public/browser/download_item.h" |
| 16 #include "content/public/browser/notification_service.h" | 16 #include "content/public/browser/notification_service.h" |
| 17 | 17 |
| 18 using content::BrowserThread; | 18 using content::BrowserThread; |
| 19 using content::DownloadItem; | 19 using content::DownloadItem; |
| 20 | 20 |
| 21 namespace download_crx_util { | 21 namespace download_crx_util { |
| 22 | 22 |
| 23 namespace { | 23 namespace { |
| 24 | 24 |
| 25 // Hold a mock ExtensionInstallUI object that will be used when the | 25 // Hold a mock ExtensionInstallPrompt object that will be used when the |
| 26 // download system opens a CRX. | 26 // download system opens a CRX. |
| 27 ExtensionInstallUI* mock_install_ui_for_testing = NULL; | 27 ExtensionInstallPrompt* mock_install_prompt_for_testing = NULL; |
| 28 | 28 |
| 29 // Called to get an extension install UI object. In tests, will return | 29 // Called to get an extension install UI object. In tests, will return |
| 30 // a mock if the test calls download_util::SetMockInstallUIForTesting() | 30 // a mock if the test calls download_util::SetMockInstallUIForTesting() |
| 31 // to set one. | 31 // to set one. |
| 32 ExtensionInstallUI* CreateExtensionInstallUI(Profile* profile) { | 32 ExtensionInstallPrompt* CreateExtensionInstallPrompt(Profile* profile) { |
| 33 // Use a mock if one is present. Otherwise, create a real extensions | 33 // Use a mock if one is present. Otherwise, create a real extensions |
| 34 // install UI. | 34 // install UI. |
| 35 ExtensionInstallUI* result = NULL; | 35 ExtensionInstallPrompt* result = NULL; |
| 36 if (mock_install_ui_for_testing) { | 36 if (mock_install_prompt_for_testing) { |
| 37 result = mock_install_ui_for_testing; | 37 result = mock_install_prompt_for_testing; |
| 38 mock_install_ui_for_testing = NULL; | 38 mock_install_prompt_for_testing = NULL; |
| 39 } else { | 39 } else { |
| 40 result = new ExtensionInstallUI(profile); | 40 result = new ExtensionInstallPrompt(profile); |
| 41 } | 41 } |
| 42 | 42 |
| 43 return result; | 43 return result; |
| 44 } | 44 } |
| 45 | 45 |
| 46 } // namespace | 46 } // namespace |
| 47 | 47 |
| 48 // Tests can call this method to inject a mock ExtensionInstallUI | 48 // Tests can call this method to inject a mock ExtensionInstallPrompt |
| 49 // to be used to confirm permissions on a downloaded CRX. | 49 // to be used to confirm permissions on a downloaded CRX. |
| 50 void SetMockInstallUIForTesting(ExtensionInstallUI* mock_ui) { | 50 void SetMockInstallPromptForTesting(ExtensionInstallPrompt* mock_prompt) { |
| 51 mock_install_ui_for_testing = mock_ui; | 51 mock_install_prompt_for_testing = mock_prompt; |
| 52 } | 52 } |
| 53 | 53 |
| 54 scoped_refptr<CrxInstaller> OpenChromeExtension( | 54 scoped_refptr<CrxInstaller> OpenChromeExtension( |
| 55 Profile* profile, | 55 Profile* profile, |
| 56 const DownloadItem& download_item) { | 56 const DownloadItem& download_item) { |
| 57 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); | 57 DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI)); |
| 58 | 58 |
| 59 ExtensionService* service = profile->GetExtensionService(); | 59 ExtensionService* service = profile->GetExtensionService(); |
| 60 CHECK(service); | 60 CHECK(service); |
| 61 | 61 |
| 62 scoped_refptr<CrxInstaller> installer( | 62 scoped_refptr<CrxInstaller> installer( |
| 63 CrxInstaller::Create( | 63 CrxInstaller::Create( |
| 64 service, | 64 service, |
| 65 CreateExtensionInstallUI(profile), | 65 CreateExtensionInstallPrompt(profile), |
| 66 WebstoreInstaller::GetAssociatedApproval(download_item))); | 66 WebstoreInstaller::GetAssociatedApproval(download_item))); |
| 67 | 67 |
| 68 installer->set_delete_source(true); | 68 installer->set_delete_source(true); |
| 69 installer->set_install_cause(extension_misc::INSTALL_CAUSE_USER_DOWNLOAD); | 69 installer->set_install_cause(extension_misc::INSTALL_CAUSE_USER_DOWNLOAD); |
| 70 | 70 |
| 71 if (UserScript::IsURLUserScript(download_item.GetURL(), | 71 if (UserScript::IsURLUserScript(download_item.GetURL(), |
| 72 download_item.GetMimeType())) { | 72 download_item.GetMimeType())) { |
| 73 installer->InstallUserScript(download_item.GetFullPath(), | 73 installer->InstallUserScript(download_item.GetFullPath(), |
| 74 download_item.GetURL()); | 74 download_item.GetURL()); |
| 75 } else { | 75 } else { |
| (...skipping 20 matching lines...) Expand all Loading... |
| 96 if (download_item.GetMimeType() == extensions::Extension::kMimeType || | 96 if (download_item.GetMimeType() == extensions::Extension::kMimeType || |
| 97 UserScript::IsURLUserScript(download_item.GetURL(), | 97 UserScript::IsURLUserScript(download_item.GetURL(), |
| 98 download_item.GetMimeType())) { | 98 download_item.GetMimeType())) { |
| 99 return true; | 99 return true; |
| 100 } else { | 100 } else { |
| 101 return false; | 101 return false; |
| 102 } | 102 } |
| 103 } | 103 } |
| 104 | 104 |
| 105 } // namespace download_crx_util | 105 } // namespace download_crx_util |
| OLD | NEW |