chrome/browser/ui/webui/chromeos/image_source.h

   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 #ifndef CHROME_BROWSER_UI_WEBUI_CHROMEOS_IMAGE_SOURCE_H_
   6 #define CHROME_BROWSER_UI_WEBUI_CHROMEOS_IMAGE_SOURCE_H_
   7
   8 #include <string>
   9
  10 #include "base/macros.h"
  11 #include "base/memory/ref_counted.h"
  12 #include "base/memory/weak_ptr.h"
  13 #include "content/public/browser/url_data_source.h"
  14
  15 namespace base {
  16 class SequencedTaskRunner;
  17 }
  18
  19 // TODO(michaelpg): Remove dependency on user_manager.
  20 namespace user_manager {
  21 class UserImage;
  22 }
  23
  24 namespace chromeos {
  25
  26 // TODO(michaelpg): Generalize UserImageLoader for classes like ImageSource.
  27 class UserImageLoader;
  28
  29 // Data source that reads and decodes an image from the RO file system.
  30 class ImageSource : public content::URLDataSource {
  31  public:
  32   ImageSource();
  33   ~ImageSource() override;
  34
  35   // content::URLDataSource implementation.
  36   std::string GetSource() const override;
  37   void StartDataRequest(
  38       const std::string& path,
  39       int render_process_id,
  40       int render_frame_id,
  41       const content::URLDataSource::GotDataCallback& got_data_callback)
  42       override;
  43
  44   std::string GetMimeType(const std::string& path) const override;
  45
  46  private:
  47   // Checks whether we have allowed the image to be loaded.
  48   bool IsWhitelisted(const std::string& path) const;
  49
  50   // The background task runner on which file I/O and image decoding are done.
  51   scoped_refptr<base::SequencedTaskRunner> task_runner_;
  52
  53   scoped_refptr<UserImageLoader> image_loader_;
  54
  55   base::WeakPtrFactory<ImageSource> weak_factory_;
  56
  57   DISALLOW_COPY_AND_ASSIGN(ImageSource);
  58 };
  59
  60 }  // namespace chromeos
  61
  62 #endif  // CHROME_BROWSER_UI_WEBUI_CHROMEOS_IMAGE_SOURCE_H_