Consolidate icon loading on ChromeOS

chrome/browser/ui/webui/fileicon_source.cc

 // Copyright (c) 2011 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/ui/webui/fileicon_source.h"
 
+#include "base/basictypes.h"
 #include "base/callback.h"
 #include "base/file_path.h"
 #include "base/memory/ref_counted_memory.h"
+#include "base/string_split.h"
 #include "base/utf_string_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/common/time_format.h"
+#include "googleurl/src/gurl.h"
 #include "grit/generated_resources.h"
 #include "net/base/escape.h"
 #include "third_party/skia/include/core/SkBitmap.h"
 #include "ui/gfx/codec/png_codec.h"
 #include "ui/gfx/image/image.h"
 
+namespace {
+
+typedef std::map<std::string, IconLoader::IconSize> QueryIconSizeMap;
+
 // The path used in internal URLs to file icon data.
-static const char kFileIconPath[] = "fileicon";
+const char kFileIconPath[] = "fileicon";
+
+// URL parameter specifying icon size.
+const char kIconSize[] = "iconsize";
+
+void GetPathAndQuery(const std::string& url,
+                     std::string* path,
+                     std::string* query) {
+  // We receive the url with chrome://fileicon/ stripped but GURL expects it.
+  const GURL gurl("chrome://fileicon/" + url);
+  path->assign(net::UnescapeURLComponent(
+      gurl.path().substr(1), (net::UnescapeRule::URL_SPECIAL_CHARS |
+                              net::UnescapeRule::SPACES)));
+  query->assign(gurl.query());
+}
+
+QueryIconSizeMap BuildQueryIconSizeMap() {
+  QueryIconSizeMap::value_type kQueryIconSizeData[] = {
+    std::make_pair("small", IconLoader::SMALL),
+    std::make_pair("normal", IconLoader::NORMAL),
+    std::make_pair("large", IconLoader::LARGE)
+  };
+
+  size_t kQSize = arraysize(kQueryIconSizeData);
+  return QueryIconSizeMap(&kQueryIconSizeData[0],
+                          &kQueryIconSizeData[kQSize]);
+}
+
+// Simple parser for data on the query.
+IconLoader::IconSize QueryToIconSize(const std::string& query) {
+  CR_DEFINE_STATIC_LOCAL(
+      QueryIconSizeMap, kQueryIconSizeMap, (BuildQueryIconSizeMap()));
+  typedef std::pair<std::string, std::string> KVPair;
+  std::vector<KVPair> parameters;
+  if (base::SplitStringIntoKeyValuePairs(query, '=', '&', &parameters)) {
+    for (std::vector<KVPair>::const_iterator itk = parameters.begin();
+         itk != parameters.end(); ++itk) {
+      if (itk->first == kIconSize) {
+        QueryIconSizeMap::const_iterator itq(
+            kQueryIconSizeMap.find(itk->second));
+        if (itq != kQueryIconSizeMap.end())
+          return itq->second;
+      }
+    }
+  }
+  return IconLoader::NORMAL;
+}
+
+}
 
 FileIconSource::FileIconSource()
     : DataSource(kFileIconPath, MessageLoop::current()) {}
 
 FileIconSource::~FileIconSource() {
   cancelable_consumer_.CancelAllRequests();
 }
 
-void FileIconSource::StartDataRequest(const std::string& path,
-                                      bool is_incognito,
-                                      int request_id) {
-  std::string escaped_path = net::UnescapeURLComponent(path,
-      net::UnescapeRule::SPACES);
-#if defined(OS_WIN)
-  // The path we receive has the wrong slashes and escaping for what we need;
-  // this only appears to matter for getting icons from .exe files.
-  std::replace(escaped_path.begin(), escaped_path.end(), '/', '\\');
-  FilePath escaped_filepath(UTF8ToWide(escaped_path));
-#elif defined(OS_POSIX)
-  // The correct encoding on Linux may not actually be UTF8.
-  FilePath escaped_filepath(escaped_path);
-#endif
-
+void FileIconSource::FetchFileIcon(const FilePath& path,
+                                   IconLoader::IconSize icon_size,
+                                   int request_id) {
   IconManager* im = g_browser_process->icon_manager();
-  gfx::Image* icon = im->LookupIcon(escaped_filepath, IconLoader::NORMAL);
+  gfx::Image* icon = im->LookupIcon(path, icon_size);

[ilja, 2012/01/22 02:31:43]

Shouldn't that still be named "escaped_filepath"?

[asanka, 2012/01/23 19:16:47]

The FilePath argument here shouldn't contain any escaping at this point.

 
   if (icon) {
     scoped_refptr<RefCountedBytes> icon_data(new RefCountedBytes);
     gfx::PNGCodec::EncodeBGRASkBitmap(*icon, false, &icon_data->data());
 
     SendResponse(request_id, icon_data);
   } else {
     // Icon was not in cache, go fetch it slowly.
-    IconManager::Handle h = im->LoadIcon(escaped_filepath,
-        IconLoader::NORMAL,
-        &cancelable_consumer_,
+    IconManager::Handle h = im->LoadIcon(
+        path, icon_size, &cancelable_consumer_,
         base::Bind(&FileIconSource::OnFileIconDataAvailable,
                    base::Unretained(this)));
 
     // Attach the ChromeURLDataManager request ID to the history request.
     cancelable_consumer_.SetClientData(im, h, request_id);
   }
 }
 
+void FileIconSource::StartDataRequest(const std::string& path,
+                                      bool is_incognito,
+                                      int request_id) {
+  std::string query;
+  std::string escaped_path;
+  GetPathAndQuery(path, &escaped_path, &query);
+#if defined(OS_WIN)
+  // The path we receive has the wrong slashes and escaping for what we need;
+  // this only appears to matter for getting icons from .exe files.
+  std::replace(escaped_path.begin(), escaped_path.end(), '/', '\\');
+  FilePath escaped_filepath(UTF8ToWide(escaped_path));
+#elif defined(OS_POSIX)
+  // The correct encoding on Linux may not actually be UTF8.
+  FilePath escaped_filepath(escaped_path);
+#endif
+  FetchFileIcon(escaped_filepath, QueryToIconSize(query), request_id);
+}
+
 std::string FileIconSource::GetMimeType(const std::string&) const {
   // Rely on image decoder inferring the correct type.
   return std::string();
 }
 
 void FileIconSource::OnFileIconDataAvailable(IconManager::Handle handle,
                                              gfx::Image* icon) {
   IconManager* im = g_browser_process->icon_manager();
   int request_id = cancelable_consumer_.GetClientData(im, handle);
 
   if (icon) {
     scoped_refptr<RefCountedBytes> icon_data(new RefCountedBytes);
     gfx::PNGCodec::EncodeBGRASkBitmap(*icon, false, &icon_data->data());
 
     SendResponse(request_id, icon_data);
   } else {
     // TODO(glen): send a dummy icon.
     SendResponse(request_id, NULL);
   }
 }