src/chrome/browser/ui/webui/theme_source.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/theme_source.h"
   6
   7 #include "base/memory/ref_counted_memory.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "base/strings/string_number_conversions.h"
  10 #include "chrome/browser/profiles/profile.h"
  11 #include "chrome/browser/resources_util.h"
  12 #include "chrome/browser/search/instant_io_context.h"
  13 #include "chrome/browser/themes/theme_properties.h"
  14 #include "chrome/browser/themes/theme_service.h"
  15 #include "chrome/browser/themes/theme_service_factory.h"
  16 #include "chrome/browser/ui/webui/ntp/ntp_resource_cache.h"
  17 #include "chrome/browser/ui/webui/ntp/ntp_resource_cache_factory.h"
  18 #include "chrome/common/url_constants.h"
  19 #include "content/public/browser/browser_thread.h"
  20 #include "net/url_request/url_request.h"
  21 #include "ui/base/layout.h"
  22 #include "ui/base/resource/resource_bundle.h"
  23 #include "ui/base/webui/web_ui_util.h"
  24 #include "url/gurl.h"
  25
  26 using content::BrowserThread;
  27
  28 namespace {
  29
  30 std::string GetThemePath() {
  31   return std::string(content::kChromeUIScheme) + "://" +
  32          std::string(chrome::kChromeUIThemePath) + "/";
  33 }
  34
  35 // use a resource map rather than hard-coded strings.
  36 static const char* kNewTabCSSPath = "css/new_tab_theme.css";
  37 static const char* kNewIncognitoTabCSSPath = "css/incognito_new_tab_theme.css";
  38
  39 }  // namespace
  40
  41 ////////////////////////////////////////////////////////////////////////////////
  42 // ThemeSource, public:
  43
  44 ThemeSource::ThemeSource(Profile* profile)
  45     : profile_(profile->GetOriginalProfile()) {
  46   NTPResourceCache::WindowType win_type = NTPResourceCache::GetWindowType(
  47       profile_, NULL);
  48   css_bytes_ =
  49       NTPResourceCacheFactory::GetForProfile(profile)->GetNewTabCSS(win_type);
  50 }
  51
  52 ThemeSource::~ThemeSource() {
  53 }
  54
  55 std::string ThemeSource::GetSource() const {
  56   return chrome::kChromeUIThemePath;
  57 }
  58
  59 void ThemeSource::StartDataRequest(
  60     const std::string& path,
  61     int render_process_id,
  62     int render_frame_id,
  63     const content::URLDataSource::GotDataCallback& callback) {
  64   // Default scale factor if not specified.
  65   ui::ScaleFactor scale_factor;
  66   std::string uncached_path;
  67   webui::ParsePathAndScale(GURL(GetThemePath() + path),
  68                            &uncached_path,
  69                            &scale_factor);
  70
  71   if (uncached_path == kNewTabCSSPath ||
  72       uncached_path == kNewIncognitoTabCSSPath) {
  73     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
  74
  75     callback.Run(css_bytes_.get());
  76     return;
  77   }
  78
  79
  80   int resource_id = ResourcesUtil::GetThemeResourceId(uncached_path);
  81   if (resource_id != -1) {
  82     SendThemeBitmap(callback, resource_id, scale_factor);
  83     return;
  84   }
  85
  86   // We don't have any data to send back.
  87   callback.Run(NULL);
  88 }
  89
  90 std::string ThemeSource::GetMimeType(const std::string& path) const {
  91   std::string uncached_path;
  92   webui::ParsePathAndScale(GURL(GetThemePath() + path), &uncached_path, NULL);
  93
  94   if (uncached_path == kNewTabCSSPath ||
  95       uncached_path == kNewIncognitoTabCSSPath) {
  96     return "text/css";
  97   }
  98
  99   return "image/png";
 100 }
 101
 102 base::MessageLoop* ThemeSource::MessageLoopForRequestPath(
 103     const std::string& path) const {
 104   std::string uncached_path;
 105   webui::ParsePathAndScale(GURL(GetThemePath() + path), &uncached_path, NULL);
 106
 107   if (uncached_path == kNewTabCSSPath ||
 108       uncached_path == kNewIncognitoTabCSSPath) {
 109     // We generated and cached this when we initialized the object.  We don't
 110     // have to go back to the UI thread to send the data.
 111     return NULL;
 112   }
 113
 114   // If it's not a themeable image, we don't need to go to the UI thread.
 115   int resource_id = ResourcesUtil::GetThemeResourceId(uncached_path);
 116   if (!ThemeProperties::IsThemeableImage(resource_id))
 117     return NULL;
 118
 119   return content::URLDataSource::MessageLoopForRequestPath(path);
 120 }
 121
 122 bool ThemeSource::ShouldReplaceExistingSource() const {
 123   // We currently get the css_bytes_ in the ThemeSource constructor, so we need
 124   // to recreate the source itself when a theme changes.
 125   return true;
 126 }
 127
 128 bool ThemeSource::ShouldServiceRequest(const net::URLRequest* request) const {
 129   if (request->url().SchemeIs(chrome::kChromeSearchScheme))
 130     return InstantIOContext::ShouldServiceRequest(request);
 131   return URLDataSource::ShouldServiceRequest(request);
 132 }
 133
 134 ////////////////////////////////////////////////////////////////////////////////
 135 // ThemeSource, private:
 136
 137 void ThemeSource::SendThemeBitmap(
 138     const content::URLDataSource::GotDataCallback& callback,
 139     int resource_id,
 140     ui::ScaleFactor scale_factor) {
 141   if (ThemeProperties::IsThemeableImage(resource_id)) {
 142     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
 143     ui::ThemeProvider* tp = ThemeServiceFactory::GetForProfile(profile_);
 144     DCHECK(tp);
 145
 146     scoped_refptr<base::RefCountedMemory> image_data(tp->GetRawData(
 147         resource_id, scale_factor));
 148     callback.Run(image_data.get());
 149   } else {
 150     DCHECK(BrowserThread::CurrentlyOn(BrowserThread::IO));
 151     const ResourceBundle& rb = ResourceBundle::GetSharedInstance();
 152     callback.Run(rb.LoadDataResourceBytesForScale(resource_id, scale_factor));
 153   }
 154 }