Chromium Code Reviews| OLD | NEW |
|---|---|
| (Empty) | |
| 1 // Copyright 2014 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/app_list/drive/drive_app_converter.h" | |
| 6 | |
| 7 #include <algorithm> | |
| 8 #include <set> | |
| 9 | |
| 10 #include "base/logging.h" | |
| 11 #include "base/memory/ref_counted.h" | |
| 12 #include "base/strings/utf_string_conversions.h" | |
| 13 #include "chrome/browser/extensions/crx_installer.h" | |
| 14 #include "chrome/browser/extensions/install_tracker.h" | |
| 15 #include "chrome/browser/image_decoder.h" | |
| 16 #include "chrome/browser/profiles/profile.h" | |
| 17 #include "content/public/browser/browser_thread.h" | |
| 18 #include "extensions/browser/extension_system.h" | |
| 19 #include "extensions/common/constants.h" | |
| 20 #include "net/base/load_flags.h" | |
| 21 #include "net/url_request/url_fetcher.h" | |
| 22 #include "net/url_request/url_fetcher_delegate.h" | |
| 23 #include "net/url_request/url_request_status.h" | |
| 24 #include "third_party/skia/include/core/SkBitmap.h" | |
| 25 | |
| 26 using content::BrowserThread; | |
| 27 | |
| 28 class DriveAppConverter::IconFetcher : public net::URLFetcherDelegate, | |
|
calamity
2014/05/19 04:01:08
Class comment. It would good to mention somewhere
xiyuan
2014/05/19 06:08:14
Done.
| |
| 29 public ImageDecoder::Delegate { | |
| 30 public: | |
| 31 IconFetcher(DriveAppConverter* converter, | |
| 32 const GURL& icon_url, | |
| 33 int expected_size) | |
| 34 : converter_(converter), | |
| 35 icon_url_(icon_url), | |
| 36 expected_size_(expected_size) {} | |
| 37 virtual ~IconFetcher() { | |
| 38 if (image_decoder_.get()) | |
| 39 image_decoder_->set_delegate(NULL); | |
| 40 } | |
| 41 | |
| 42 void Start() { | |
| 43 fetcher_.reset( | |
| 44 net::URLFetcher::Create(icon_url_, net::URLFetcher::GET, this)); | |
| 45 fetcher_->SetRequestContext(converter_->profile_->GetRequestContext()); | |
| 46 fetcher_->SetLoadFlags(net::LOAD_DO_NOT_SAVE_COOKIES); | |
| 47 fetcher_->Start(); | |
| 48 } | |
| 49 | |
| 50 const GURL& icon_url() const { return icon_url_; } | |
| 51 const SkBitmap& icon() const { return icon_; } | |
| 52 | |
| 53 private: | |
| 54 // net::URLFetcherDelegate overrides: | |
| 55 virtual void OnURLFetchComplete(const net::URLFetcher* source) OVERRIDE { | |
| 56 CHECK_EQ(fetcher_.get(), source); | |
| 57 scoped_ptr<net::URLFetcher> fetcher(fetcher_.Pass()); | |
| 58 | |
| 59 if (!fetcher->GetStatus().is_success() || | |
| 60 fetcher->GetResponseCode() != 200) { | |
|
calamity
2014/05/19 04:01:08
s/200/net::HTTP_OK/
xiyuan
2014/05/19 06:08:14
Done.
| |
| 61 converter_->OnIconFetchComplete(this); | |
| 62 return; | |
| 63 } | |
| 64 | |
| 65 std::string unsafe_icon_data; | |
| 66 fetcher->GetResponseAsString(&unsafe_icon_data); | |
| 67 | |
| 68 image_decoder_ = | |
| 69 new ImageDecoder(this, unsafe_icon_data, ImageDecoder::DEFAULT_CODEC); | |
| 70 image_decoder_->Start( | |
| 71 BrowserThread::GetMessageLoopProxyForThread(BrowserThread::UI)); | |
| 72 } | |
| 73 | |
| 74 // ImageDecoder::Delegate overrides: | |
| 75 virtual void OnImageDecoded(const ImageDecoder* decoder, | |
| 76 const SkBitmap& decoded_image) OVERRIDE { | |
| 77 if (decoded_image.width() == expected_size_) | |
| 78 icon_ = decoded_image; | |
| 79 converter_->OnIconFetchComplete(this); | |
| 80 } | |
| 81 | |
| 82 virtual void OnDecodeImageFailed(const ImageDecoder* decoder) OVERRIDE { | |
| 83 converter_->OnIconFetchComplete(this); | |
| 84 } | |
| 85 | |
| 86 DriveAppConverter* converter_; | |
| 87 const GURL icon_url_; | |
| 88 const int expected_size_; | |
| 89 | |
| 90 scoped_ptr<net::URLFetcher> fetcher_; | |
| 91 scoped_refptr<ImageDecoder> image_decoder_; | |
| 92 SkBitmap icon_; | |
| 93 | |
| 94 DISALLOW_COPY_AND_ASSIGN(IconFetcher); | |
| 95 }; | |
| 96 | |
| 97 DriveAppConverter::DriveAppConverter(Profile* profile, | |
| 98 const drive::DriveAppInfo& app_info, | |
| 99 const FinishedCallback& finished_callback) | |
| 100 : profile_(profile), | |
| 101 app_info_(app_info), | |
| 102 app_(NULL), | |
| 103 finished_callback_(finished_callback) { | |
| 104 DCHECK(profile_); | |
| 105 } | |
| 106 | |
| 107 DriveAppConverter::~DriveAppConverter() { | |
| 108 PostInstallCleanUp(); | |
| 109 } | |
| 110 | |
| 111 void DriveAppConverter::Start() { | |
| 112 if (app_info_.app_name.empty() || | |
| 113 !app_info_.create_url.is_valid()) { | |
| 114 finished_callback_.Run(this, false); | |
| 115 return; | |
| 116 } | |
| 117 | |
| 118 web_app_.title = base::UTF8ToUTF16(app_info_.app_name); | |
| 119 web_app_.app_url = app_info_.create_url; | |
| 120 | |
| 121 const std::set<int> allowed_sizes(extension_misc::kExtensionIconSizes, | |
| 122 extension_misc::kExtensionIconSizes + | |
| 123 extension_misc::kNumExtensionIconSizes); | |
| 124 for (size_t i = 0; i < app_info_.app_icons.size(); ++i) { | |
| 125 const int icon_size = app_info_.app_icons[i].first; | |
| 126 if (allowed_sizes.find(icon_size) == allowed_sizes.end()) | |
| 127 continue; | |
| 128 | |
| 129 const GURL& icon_url = app_info_.app_icons[i].second; | |
| 130 IconFetcher* fetcher = new IconFetcher(this, icon_url, icon_size); | |
| 131 fetchers_.push_back(fetcher); // Pass ownership to |fetchers|. | |
| 132 fetcher->Start(); | |
| 133 } | |
| 134 | |
| 135 if (fetchers_.empty()) | |
| 136 StartInstall(); | |
| 137 } | |
| 138 | |
| 139 void DriveAppConverter::OnIconFetchComplete(IconFetcher* fetcher) { | |
| 140 const SkBitmap& icon = fetcher->icon(); | |
| 141 if (!icon.empty() && icon.width() != 0) { | |
| 142 WebApplicationInfo::IconInfo icon_info; | |
| 143 icon_info.url = fetcher->icon_url(); | |
| 144 icon_info.data = icon; | |
| 145 icon_info.width = icon.width(); | |
| 146 icon_info.height = icon.height(); | |
| 147 web_app_.icons.push_back(icon_info); | |
| 148 } | |
| 149 | |
| 150 fetchers_.erase(std::find(fetchers_.begin(), fetchers_.end(), fetcher)); | |
| 151 | |
| 152 if (fetchers_.empty()) | |
| 153 StartInstall(); | |
| 154 } | |
| 155 | |
| 156 void DriveAppConverter::StartInstall() { | |
| 157 DCHECK(!crx_installer_); | |
| 158 crx_installer_ = extensions::CrxInstaller::CreateSilent( | |
| 159 extensions::ExtensionSystem::Get(profile_)->extension_service()); | |
| 160 extensions::InstallTracker::Get(profile_)->AddObserver(this); | |
| 161 crx_installer_->InstallWebApp(web_app_); | |
| 162 } | |
| 163 | |
| 164 void DriveAppConverter::PostInstallCleanUp() { | |
| 165 if (!crx_installer_) | |
| 166 return; | |
| 167 | |
| 168 extensions::InstallTracker::Get(profile_)->RemoveObserver(this); | |
| 169 crx_installer_ = NULL; | |
| 170 } | |
| 171 | |
| 172 void DriveAppConverter::OnFinishCrxInstall(const std::string& extension_id, | |
| 173 bool success) { | |
| 174 if (!crx_installer_->extension() || | |
| 175 crx_installer_->extension()->id() != extension_id) { | |
| 176 return; | |
| 177 } | |
| 178 | |
| 179 app_ = crx_installer_->extension(); | |
| 180 finished_callback_.Run(this, success); | |
| 181 | |
| 182 PostInstallCleanUp(); | |
| 183 } | |
| OLD | NEW |