chrome/browser/chromeos/extensions/wallpaper_function_base.cc

   1 // Copyright 2013 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/chromeos/extensions/wallpaper_function_base.h"
   6
   7 #include "base/synchronization/cancellation_flag.h"
   8 #include "chrome/browser/image_decoder.h"
   9 #include "chrome/grit/generated_resources.h"
  10 #include "chromeos/login/login_state.h"
  11 #include "components/wallpaper/wallpaper_layout.h"
  12 #include "ui/base/l10n/l10n_util.h"
  13
  14 using content::BrowserThread;
  15
  16 namespace wallpaper_api_util {
  17 namespace {
  18
  19 // Keeps in sync (same order) with WallpaperLayout enum in header file.
  20 const char* const kWallpaperLayoutArrays[] = {
  21   "CENTER",
  22   "CENTER_CROPPED",
  23   "STRETCH",
  24   "TILE"
  25 };
  26
  27 const int kWallpaperLayoutCount = arraysize(kWallpaperLayoutArrays);
  28
  29 }  // namespace
  30
  31 const char kCancelWallpaperMessage[] = "Set wallpaper was canceled.";
  32
  33 wallpaper::WallpaperLayout GetLayoutEnum(const std::string& layout) {
  34   for (int i = 0; i < kWallpaperLayoutCount; i++) {
  35     if (layout.compare(kWallpaperLayoutArrays[i]) == 0)
  36       return static_cast<wallpaper::WallpaperLayout>(i);
  37   }
  38   // Default to use CENTER layout.
  39   return wallpaper::WALLPAPER_LAYOUT_CENTER;
  40 }
  41
  42 }  // namespace wallpaper_api_util
  43
  44 class WallpaperFunctionBase::UnsafeWallpaperDecoder
  45     : public ImageDecoder::ImageRequest {
  46  public:
  47   explicit UnsafeWallpaperDecoder(scoped_refptr<WallpaperFunctionBase> function)
  48       : ImageRequest(
  49             BrowserThread::GetMessageLoopProxyForThread(BrowserThread::UI)),
  50         function_(function) {}
  51
  52   void Start(const std::vector<char>& image_data) {
  53     DCHECK_CURRENTLY_ON(BrowserThread::UI);
  54
  55     // This function can only be called after user login. It is fine to use
  56     // unsafe image decoder here. Before user login, a robust jpeg decoder will
  57     // be used.
  58     CHECK(chromeos::LoginState::Get()->IsUserLoggedIn());
  59     std::string image_data_str(image_data.begin(), image_data.end());
  60     ImageDecoder::StartWithOptions(this, image_data_str,
  61                                    ImageDecoder::DEFAULT_CODEC, true);
  62   }
  63
  64   void Cancel() {
  65     cancel_flag_.Set();
  66   }
  67
  68   void OnImageDecoded(const SkBitmap& decoded_image) override {
  69     // Make the SkBitmap immutable as we won't modify it. This is important
  70     // because otherwise it gets duplicated during painting, wasting memory.
  71     SkBitmap immutable(decoded_image);
  72     immutable.setImmutable();
  73     gfx::ImageSkia final_image = gfx::ImageSkia::CreateFrom1xBitmap(immutable);
  74     final_image.MakeThreadSafe();
  75     if (cancel_flag_.IsSet()) {
  76       function_->OnCancel();
  77       delete this;
  78       return;
  79     }
  80     function_->OnWallpaperDecoded(final_image);
  81     delete this;
  82   }
  83
  84   void OnDecodeImageFailed() override {
  85     function_->OnFailure(
  86         l10n_util::GetStringUTF8(IDS_WALLPAPER_MANAGER_INVALID_WALLPAPER));
  87     delete this;
  88   }
  89
  90  private:
  91   scoped_refptr<WallpaperFunctionBase> function_;
  92   base::CancellationFlag cancel_flag_;
  93
  94   DISALLOW_COPY_AND_ASSIGN(UnsafeWallpaperDecoder);
  95 };
  96
  97 WallpaperFunctionBase::UnsafeWallpaperDecoder*
  98     WallpaperFunctionBase::unsafe_wallpaper_decoder_;
  99
 100 WallpaperFunctionBase::WallpaperFunctionBase() {
 101 }
 102
 103 WallpaperFunctionBase::~WallpaperFunctionBase() {
 104 }
 105
 106 void WallpaperFunctionBase::StartDecode(const std::vector<char>& data) {
 107   DCHECK_CURRENTLY_ON(BrowserThread::UI);
 108   if (unsafe_wallpaper_decoder_)
 109     unsafe_wallpaper_decoder_->Cancel();
 110   unsafe_wallpaper_decoder_ = new UnsafeWallpaperDecoder(this);
 111   unsafe_wallpaper_decoder_->Start(data);
 112 }
 113
 114 void WallpaperFunctionBase::OnCancel() {
 115   unsafe_wallpaper_decoder_ = NULL;
 116   SetError(wallpaper_api_util::kCancelWallpaperMessage);
 117   SendResponse(false);
 118 }
 119
 120 void WallpaperFunctionBase::OnFailure(const std::string& error) {
 121   unsafe_wallpaper_decoder_ = NULL;
 122   SetError(error);
 123   SendResponse(false);
 124 }