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)#include "chrome/browser/ui/webui/extensions/install_extension_handler.h" 65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 75821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/bind.h" 8868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h" 9868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/crx_installer.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/extension_install_prompt.h" 125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/extension_service.h" 135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/extensions/extension_system.h" 147dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch#include "chrome/browser/extensions/unpacked_installer.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/profiles/profile.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/web_contents.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/web_contents_view.h" 185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "content/public/browser/web_ui.h" 192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "content/public/browser/web_ui_data_source.h" 20eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "content/public/common/drop_data.h" 21f2477e01787aa58f445919b809d89e252beef54fTorne (Richard Coles)#include "extensions/common/feature_switch.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "grit/generated_resources.h" 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_util.h" 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "ui/base/l10n/l10n_util.h" 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 26a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles)namespace extensions { 27a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)InstallExtensionHandler::InstallExtensionHandler() { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)InstallExtensionHandler::~InstallExtensionHandler() { 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void InstallExtensionHandler::GetLocalizedValues( 352a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) content::WebUIDataSource* source) { 362a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source->AddString( 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "extensionSettingsInstallDropTarget", 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) l10n_util::GetStringUTF16(IDS_EXTENSIONS_INSTALL_DROP_TARGET)); 392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) source->AddBoolean( 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "offStoreInstallEnabled", 41a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) FeatureSwitch::easy_off_store_install()->IsEnabled()); 425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void InstallExtensionHandler::RegisterMessages() { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->RegisterMessageCallback( 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "startDrag", 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&InstallExtensionHandler::HandleStartDragMessage, 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->RegisterMessageCallback( 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "stopDrag", 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&InstallExtensionHandler::HandleStopDragMessage, 525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->RegisterMessageCallback( 545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) "installDroppedFile", 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Bind(&InstallExtensionHandler::HandleInstallMessage, 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Unretained(this))); 577dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch web_ui()->RegisterMessageCallback( 587dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch "installDroppedDirectory", 597dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::Bind(&InstallExtensionHandler::HandleInstallDirectoryMessage, 607dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::Unretained(this))); 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void InstallExtensionHandler::HandleStartDragMessage(const ListValue* args) { 64eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch content::DropData* drop_data = 65eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch web_ui()->GetWebContents()->GetView()->GetDropData(); 665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!drop_data) { 675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DLOG(ERROR) << "No current drop data."; 685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (drop_data->filenames.empty()) { 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DLOG(ERROR) << "Current drop data contains no files."; 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 76eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch const content::DropData::FileInfo& file_info = drop_data->filenames.front(); 7790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) 78d0247b1b59f9c528cb6df88b4f2b9afaf80d181eTorne (Richard Coles) file_to_install_ = base::FilePath::FromUTF16Unsafe(file_info.path); 7990dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // Use the display name if provided, for checking file names 8090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) // (.path is likely a random hash value in that case). 8190dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) file_display_name_ = 8290dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) file_info.display_name.empty() ? file_info.path : file_info.display_name; 835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void InstallExtensionHandler::HandleStopDragMessage(const ListValue* args) { 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file_to_install_.clear(); 8790dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) file_display_name_.clear(); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void InstallExtensionHandler::HandleInstallMessage(const ListValue* args) { 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (file_to_install_.empty()) { 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) LOG(ERROR) << "No file captured to install."; 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) Profile* profile = Profile::FromBrowserContext( 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) web_ui()->GetWebContents()->GetBrowserContext()); 98ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch scoped_ptr<ExtensionInstallPrompt> prompt( 99ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch new ExtensionInstallPrompt(web_ui()->GetWebContents())); 100a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) scoped_refptr<CrxInstaller> crx_installer(CrxInstaller::Create( 101a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) ExtensionSystem::Get(profile)->extension_service(), 102ba5b9a6411cb1792fd21f0a078d7a25cd1ceec16Ben Murdoch prompt.Pass())); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crx_installer->set_error_on_unsupported_requirements(true); 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crx_installer->set_off_store_install_allow_reason( 105a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) CrxInstaller::OffStoreInstallAllowedFromSettingsPage); 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crx_installer->set_install_wait_for_idle(false); 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const bool kCaseSensitive = false; 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 11090dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) if (EndsWith(file_display_name_, ASCIIToUTF16(".user.js"), kCaseSensitive)) { 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crx_installer->InstallUserScript( 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file_to_install_, 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::FilePathToFileURL(file_to_install_)); 11490dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) } else if (EndsWith(file_display_name_, 11590dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) ASCIIToUTF16(".crx"), 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kCaseSensitive)) { 1175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crx_installer->InstallCrx(file_to_install_); 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } else { 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CHECK(false); 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file_to_install_.clear(); 12390dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles) file_display_name_.clear(); 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1257dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch 1267dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdochvoid InstallExtensionHandler::HandleInstallDirectoryMessage( 1277dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch const ListValue* args) { 1287dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch Profile* profile = Profile::FromBrowserContext( 1297dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch web_ui()->GetWebContents()->GetBrowserContext()); 130a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) UnpackedInstaller::Create( 131a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) ExtensionSystem::Get(profile)-> 1327dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch extension_service())->Load(file_to_install_); 1337dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch} 134a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles) 135a36e5920737c6adbddd3e43b760e5de8431db6e0Torne (Richard Coles)} // namespace extensions 136