src/chrome/browser/ui/webui/extensions/install_extension_handler.cc

   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/install_extension_handler.h"
   6
   7 #include "base/bind.h"
   8 #include "base/strings/string_util.h"
   9 #include "base/strings/utf_string_conversions.h"
  10 #include "chrome/browser/extensions/crx_installer.h"
  11 #include "chrome/browser/extensions/extension_install_prompt.h"
  12 #include "chrome/browser/extensions/extension_service.h"
  13 #include "chrome/browser/extensions/unpacked_installer.h"
  14 #include "chrome/browser/extensions/zipfile_installer.h"
  15 #include "chrome/browser/profiles/profile.h"
  16 #include "content/public/browser/web_contents.h"
  17 #include "content/public/browser/web_ui.h"
  18 #include "content/public/browser/web_ui_data_source.h"
  19 #include "content/public/common/drop_data.h"
  20 #include "extensions/browser/extension_system.h"
  21 #include "extensions/common/feature_switch.h"
  22 #include "grit/generated_resources.h"
  23 #include "net/base/filename_util.h"
  24 #include "ui/base/l10n/l10n_util.h"
  25
  26 namespace extensions {
  27
  28 InstallExtensionHandler::InstallExtensionHandler() {
  29 }
  30
  31 InstallExtensionHandler::~InstallExtensionHandler() {
  32 }
  33
  34 void InstallExtensionHandler::GetLocalizedValues(
  35     content::WebUIDataSource* source) {
  36   source->AddString(
  37       "extensionSettingsInstallDropTarget",
  38       l10n_util::GetStringUTF16(IDS_EXTENSIONS_INSTALL_DROP_TARGET));
  39   source->AddBoolean(
  40       "offStoreInstallEnabled",
  41       FeatureSwitch::easy_off_store_install()->IsEnabled());
  42 }
  43
  44 void InstallExtensionHandler::RegisterMessages() {
  45   web_ui()->RegisterMessageCallback(
  46       "startDrag",
  47       base::Bind(&InstallExtensionHandler::HandleStartDragMessage,
  48                  base::Unretained(this)));
  49   web_ui()->RegisterMessageCallback(
  50       "stopDrag",
  51       base::Bind(&InstallExtensionHandler::HandleStopDragMessage,
  52                  base::Unretained(this)));
  53   web_ui()->RegisterMessageCallback(
  54       "installDroppedFile",
  55       base::Bind(&InstallExtensionHandler::HandleInstallMessage,
  56                  base::Unretained(this)));
  57   web_ui()->RegisterMessageCallback(
  58       "installDroppedDirectory",
  59       base::Bind(&InstallExtensionHandler::HandleInstallDirectoryMessage,
  60                  base::Unretained(this)));
  61 }
  62
  63 void InstallExtensionHandler::HandleStartDragMessage(
  64     const base::ListValue* args) {
  65   content::DropData* drop_data =
  66       web_ui()->GetWebContents()->GetDropData();
  67   if (!drop_data) {
  68     DLOG(ERROR) << "No current drop data.";
  69     return;
  70   }
  71
  72   if (drop_data->filenames.empty()) {
  73     DLOG(ERROR) << "Current drop data contains no files.";
  74     return;
  75   }
  76
  77   const ui::FileInfo& file_info = drop_data->filenames.front();
  78
  79   file_to_install_ = file_info.path;
  80   // Use the display name if provided, for checking file names
  81   // (.path is likely a random hash value in that case).
  82   // TODO(dcheng): It would be nice to make this a FilePath too.
  83   file_display_name_ = file_info.display_name.empty()
  84                            ? file_info.path.AsUTF16Unsafe()
  85                            : file_info.display_name.AsUTF16Unsafe();
  86 }
  87
  88 void InstallExtensionHandler::HandleStopDragMessage(
  89     const base::ListValue* args) {
  90   file_to_install_.clear();
  91   file_display_name_.clear();
  92 }
  93
  94 void InstallExtensionHandler::HandleInstallMessage(
  95     const base::ListValue* args) {
  96   if (file_to_install_.empty()) {
  97     LOG(ERROR) << "No file captured to install.";
  98     return;
  99   }
 100
 101   Profile* profile = Profile::FromBrowserContext(
 102       web_ui()->GetWebContents()->GetBrowserContext());
 103
 104   const bool kCaseSensitive = false;
 105
 106   if (EndsWith(
 107           file_display_name_, base::ASCIIToUTF16(".zip"), kCaseSensitive)) {
 108     ZipFileInstaller::Create(ExtensionSystem::Get(profile)->extension_service())
 109         ->LoadFromZipFile(file_to_install_);
 110   } else {
 111     scoped_ptr<ExtensionInstallPrompt> prompt(
 112         new ExtensionInstallPrompt(web_ui()->GetWebContents()));
 113     scoped_refptr<CrxInstaller> crx_installer(CrxInstaller::Create(
 114         ExtensionSystem::Get(profile)->extension_service(), prompt.Pass()));
 115     crx_installer->set_error_on_unsupported_requirements(true);
 116     crx_installer->set_off_store_install_allow_reason(
 117         CrxInstaller::OffStoreInstallAllowedFromSettingsPage);
 118     crx_installer->set_install_immediately(true);
 119
 120     if (EndsWith(file_display_name_,
 121                  base::ASCIIToUTF16(".user.js"),
 122                  kCaseSensitive)) {
 123       crx_installer->InstallUserScript(
 124           file_to_install_, net::FilePathToFileURL(file_to_install_));
 125     } else if (EndsWith(file_display_name_,
 126                         base::ASCIIToUTF16(".crx"),
 127                         kCaseSensitive)) {
 128       crx_installer->InstallCrx(file_to_install_);
 129     } else {
 130       CHECK(false);
 131     }
 132   }
 133
 134   file_to_install_.clear();
 135   file_display_name_.clear();
 136 }
 137
 138 void InstallExtensionHandler::HandleInstallDirectoryMessage(
 139     const base::ListValue* args) {
 140   Profile* profile = Profile::FromBrowserContext(
 141       web_ui()->GetWebContents()->GetBrowserContext());
 142   UnpackedInstaller::Create(
 143       ExtensionSystem::Get(profile)->
 144           extension_service())->Load(file_to_install_);
 145 }
 146
 147 }  // namespace extensions