| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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/fileicon_source_chromeos.h" | |
| 6 | |
| 7 #include <map> | |
| 8 #include <utility> | |
| 9 #include <vector> | |
| 10 | |
| 11 #include "base/file_util.h" | |
| 12 #include "base/memory/singleton.h" | |
| 13 #include "base/string_split.h" | |
| 14 #include "base/string_util.h" | |
| 15 #include "base/threading/thread_restrictions.h" | |
| 16 #include "chrome/browser/icon_loader.h" | |
| 17 #include "chrome/browser/io_thread.h" | |
| 18 #include "chrome/browser/ui/webui/chrome_url_data_manager.h" | |
| 19 #include "chrome/common/url_constants.h" | |
| 20 #include "googleurl/src/gurl.h" | |
| 21 #include "grit/component_extension_resources.h" | |
| 22 #include "grit/component_extension_resources_map.h" | |
| 23 #include "grit/generated_resources.h" | |
| 24 #include "grit/theme_resources.h" | |
| 25 #include "grit/ui_resources.h" | |
| 26 #include "net/base/escape.h" | |
| 27 #include "net/base/mime_util.h" | |
| 28 #include "ui/base/resource/resource_bundle.h" | |
| 29 | |
| 30 namespace { | |
| 31 | |
| 32 struct IdrBySize { | |
| 33 int idr_small_; | |
| 34 int idr_normal_; | |
| 35 int idr_large_; | |
| 36 }; | |
| 37 | |
| 38 typedef std::map<std::string, IconLoader::IconSize> QueryIconSizeMap; | |
| 39 typedef std::map<std::string, IdrBySize> ExtensionIconSizeMap; | |
| 40 | |
| 41 const char *kIconSize = "iconsize"; | |
| 42 | |
| 43 const IdrBySize kAudioIdrs = { | |
| 44 IDR_FILE_MANAGER_IMG_FILETYPE_AUDIO, | |
| 45 IDR_FILE_MANAGER_IMG_FILETYPE_LARGE_AUDIO, | |
| 46 IDR_FILE_MANAGER_IMG_FILETYPE_LARGE_AUDIO | |
| 47 }; | |
| 48 const IdrBySize kDeviceIdrs = { | |
| 49 IDR_FILE_MANAGER_IMG_FILETYPE_DEVICE, | |
| 50 IDR_FILE_MANAGER_IMG_FILETYPE_DEVICE, | |
| 51 IDR_FILE_MANAGER_IMG_FILETYPE_DEVICE | |
| 52 }; | |
| 53 const IdrBySize kDocIdrs = { | |
| 54 IDR_FILE_MANAGER_IMG_FILETYPE_DOC, | |
| 55 IDR_FILE_MANAGER_IMG_FILETYPE_DOC, | |
| 56 IDR_FILE_MANAGER_IMG_FILETYPE_DOC | |
| 57 }; | |
| 58 const IdrBySize kFolderIdrs = { | |
| 59 IDR_FILE_MANAGER_IMG_FILETYPE_FOLDER, | |
| 60 IDR_FILE_MANAGER_IMG_FILETYPE_LARGE_FOLDER, | |
| 61 IDR_FILE_MANAGER_IMG_FILETYPE_LARGE_FOLDER | |
| 62 }; | |
| 63 const IdrBySize kGenericIdrs = { | |
| 64 IDR_FILE_MANAGER_IMG_FILETYPE_GENERIC, | |
| 65 IDR_FILE_MANAGER_IMG_FILETYPE_LARGE_GENERIC, | |
| 66 IDR_FILE_MANAGER_IMG_FILETYPE_LARGE_GENERIC | |
| 67 }; | |
| 68 const IdrBySize kHtmlIdrs = { | |
| 69 IDR_FILE_MANAGER_IMG_FILETYPE_HTML, | |
| 70 IDR_FILE_MANAGER_IMG_FILETYPE_HTML, | |
| 71 IDR_FILE_MANAGER_IMG_FILETYPE_HTML | |
| 72 }; | |
| 73 const IdrBySize kImageIdrs = { | |
| 74 IDR_FILE_MANAGER_IMG_FILETYPE_IMAGE, | |
| 75 IDR_FILE_MANAGER_IMG_FILETYPE_IMAGE, | |
| 76 IDR_FILE_MANAGER_IMG_FILETYPE_IMAGE | |
| 77 }; | |
| 78 const IdrBySize kPdfIdrs = { | |
| 79 IDR_FILE_MANAGER_IMG_FILETYPE_PDF, | |
| 80 IDR_FILE_MANAGER_IMG_FILETYPE_PDF, | |
| 81 IDR_FILE_MANAGER_IMG_FILETYPE_PDF | |
| 82 }; | |
| 83 const IdrBySize kPresentationIdrs = { | |
| 84 IDR_FILE_MANAGER_IMG_FILETYPE_PRESENTATION, | |
| 85 IDR_FILE_MANAGER_IMG_FILETYPE_PRESENTATION, | |
| 86 IDR_FILE_MANAGER_IMG_FILETYPE_PRESENTATION | |
| 87 }; | |
| 88 const IdrBySize kSpreadsheetIdrs = { | |
| 89 IDR_FILE_MANAGER_IMG_FILETYPE_SPREADSHEET, | |
| 90 IDR_FILE_MANAGER_IMG_FILETYPE_SPREADSHEET, | |
| 91 IDR_FILE_MANAGER_IMG_FILETYPE_SPREADSHEET | |
| 92 }; | |
| 93 const IdrBySize kUnreadableDeviceIdrs = { | |
| 94 IDR_FILE_MANAGER_IMG_FILETYPE_UNREADABLE_DEVICE, | |
| 95 IDR_FILE_MANAGER_IMG_FILETYPE_UNREADABLE_DEVICE, | |
| 96 IDR_FILE_MANAGER_IMG_FILETYPE_UNREADABLE_DEVICE | |
| 97 }; | |
| 98 const IdrBySize kTextIdrs = { | |
| 99 IDR_FILE_MANAGER_IMG_FILETYPE_TEXT, | |
| 100 IDR_FILE_MANAGER_IMG_FILETYPE_TEXT, | |
| 101 IDR_FILE_MANAGER_IMG_FILETYPE_TEXT | |
| 102 }; | |
| 103 const IdrBySize kVideoIdrs = { | |
| 104 IDR_FILE_MANAGER_IMG_FILETYPE_VIDEO, | |
| 105 IDR_FILE_MANAGER_IMG_FILETYPE_LARGE_VIDEO, | |
| 106 IDR_FILE_MANAGER_IMG_FILETYPE_LARGE_VIDEO | |
| 107 }; | |
| 108 | |
| 109 QueryIconSizeMap BuildQueryIconSizeMap() { | |
| 110 QueryIconSizeMap::value_type kQueryIconSizeData[] = { | |
| 111 std::make_pair("small", IconLoader::SMALL), | |
| 112 std::make_pair("normal", IconLoader::NORMAL), | |
| 113 std::make_pair("large", IconLoader::LARGE) | |
| 114 }; | |
| 115 | |
| 116 size_t kQSize = arraysize(kQueryIconSizeData); | |
| 117 return QueryIconSizeMap(&kQueryIconSizeData[0], | |
| 118 &kQueryIconSizeData[kQSize]); | |
| 119 } | |
| 120 | |
| 121 // The code below should match translation in | |
| 122 // chrome/browser/resources/file_manager/js/file_manager.js | |
| 123 // chrome/browser/resources/file_manager/css/file_manager.css | |
| 124 // 'audio': /\.(mp3|m4a|oga|ogg|wav)$/i, | |
| 125 // 'html': /\.(html?)$/i, | |
| 126 // 'image': /\.(bmp|gif|jpe?g|ico|png|webp)$/i, | |
| 127 // 'pdf' : /\.(pdf)$/i, | |
| 128 // 'text': /\.(pod|rst|txt|log)$/i, | |
| 129 // 'video': /\.(mov|mp4|m4v|mpe?g4?|ogm|ogv|ogx|webm)$/i | |
| 130 | |
| 131 ExtensionIconSizeMap BuildExtensionIdrSizeMap() { | |
| 132 const ExtensionIconSizeMap::value_type kExtensionIdrBySizeData[] = { | |
| 133 #if defined(GOOGLE_CHROME_BUILD) || defined(USE_PROPRIETARY_CODECS) | |
| 134 std::make_pair(".m4a", kAudioIdrs), | |
| 135 std::make_pair(".mp3", kAudioIdrs), | |
| 136 std::make_pair(".pdf", kPdfIdrs), | |
| 137 std::make_pair(".3gp", kVideoIdrs), | |
| 138 std::make_pair(".avi", kVideoIdrs), | |
| 139 std::make_pair(".m4v", kVideoIdrs), | |
| 140 std::make_pair(".mov", kVideoIdrs), | |
| 141 std::make_pair(".mp4", kVideoIdrs), | |
| 142 std::make_pair(".mpeg", kVideoIdrs), | |
| 143 std::make_pair(".mpg", kVideoIdrs), | |
| 144 std::make_pair(".mpeg4", kVideoIdrs), | |
| 145 std::make_pair(".mpg4", kVideoIdrs), | |
| 146 #endif | |
| 147 std::make_pair(".flac", kAudioIdrs), | |
| 148 std::make_pair(".oga", kAudioIdrs), | |
| 149 std::make_pair(".ogg", kAudioIdrs), | |
| 150 std::make_pair(".wav", kAudioIdrs), | |
| 151 std::make_pair(".htm", kHtmlIdrs), | |
| 152 std::make_pair(".html", kHtmlIdrs), | |
| 153 std::make_pair(".bmp", kImageIdrs), | |
| 154 std::make_pair(".gif", kImageIdrs), | |
| 155 std::make_pair(".ico", kImageIdrs), | |
| 156 std::make_pair(".jpeg", kImageIdrs), | |
| 157 std::make_pair(".jpg", kImageIdrs), | |
| 158 std::make_pair(".png", kImageIdrs), | |
| 159 std::make_pair(".webp", kImageIdrs), | |
| 160 std::make_pair(".log", kTextIdrs), | |
| 161 std::make_pair(".pod", kTextIdrs), | |
| 162 std::make_pair(".rst", kTextIdrs), | |
| 163 std::make_pair(".txt", kTextIdrs), | |
| 164 std::make_pair(".ogm", kVideoIdrs), | |
| 165 std::make_pair(".ogv", kVideoIdrs), | |
| 166 std::make_pair(".ogx", kVideoIdrs), | |
| 167 std::make_pair(".webm", kVideoIdrs), | |
| 168 }; | |
| 169 | |
| 170 const size_t kESize = arraysize(kExtensionIdrBySizeData); | |
| 171 return ExtensionIconSizeMap(&kExtensionIdrBySizeData[0], | |
| 172 &kExtensionIdrBySizeData[kESize]); | |
| 173 } | |
| 174 | |
| 175 // Split on the very first &. The first part is path, the rest query. | |
| 176 void GetExtensionAndQuery(const std::string& url, | |
| 177 std::string* extension, | |
| 178 std::string* query) { | |
| 179 // We receive the url with chrome://fileicon/ stripped but GURL expects it. | |
| 180 const GURL gurl("chrome://fileicon/" + net::EscapePath(url)); | |
| 181 const std::string path = gurl.path(); | |
| 182 *extension = StringToLowerASCII(FilePath(path).Extension()); | |
| 183 *query = gurl.query(); | |
| 184 } | |
| 185 | |
| 186 // Simple parser for data on the query. | |
| 187 IconLoader::IconSize QueryToIconSize(const std::string& query) { | |
| 188 CR_DEFINE_STATIC_LOCAL( | |
| 189 QueryIconSizeMap, kQueryIconSizeMap, (BuildQueryIconSizeMap())); | |
| 190 typedef std::pair<std::string, std::string> KVPair; | |
| 191 std::vector<KVPair> parameters; | |
| 192 if (base::SplitStringIntoKeyValuePairs(query, '=', '&', ¶meters)) { | |
| 193 for (std::vector<KVPair>::const_iterator itk = parameters.begin(); | |
| 194 itk != parameters.end(); ++itk) { | |
| 195 if (itk->first == kIconSize) { | |
| 196 QueryIconSizeMap::const_iterator itq( | |
| 197 kQueryIconSizeMap.find(itk->second)); | |
| 198 if (itq != kQueryIconSizeMap.end()) | |
| 199 return itq->second; | |
| 200 } | |
| 201 } | |
| 202 } | |
| 203 return IconLoader::NORMAL; | |
| 204 } | |
| 205 | |
| 206 // Finds matching resource of proper size. Fallback to generic. | |
| 207 int UrlToIDR(const std::string& url) { | |
| 208 CR_DEFINE_STATIC_LOCAL( | |
| 209 ExtensionIconSizeMap, kExtensionIdrSizeMap, (BuildExtensionIdrSizeMap())); | |
| 210 std::string extension, query; | |
| 211 int idr = -1; | |
| 212 GetExtensionAndQuery(url, &extension, &query); | |
| 213 const IconLoader::IconSize size = QueryToIconSize(query); | |
| 214 ExtensionIconSizeMap::const_iterator it = | |
| 215 kExtensionIdrSizeMap.find(extension); | |
| 216 if (it != kExtensionIdrSizeMap.end()) { | |
| 217 IdrBySize idrbysize = it->second; | |
| 218 if (size == IconLoader::SMALL) { | |
| 219 idr = idrbysize.idr_small_; | |
| 220 } else if (size == IconLoader::NORMAL) { | |
| 221 idr = idrbysize.idr_normal_; | |
| 222 } else if (size == IconLoader::LARGE) { | |
| 223 idr = idrbysize.idr_large_; | |
| 224 } | |
| 225 } | |
| 226 if (idr == -1) { | |
| 227 if (size == IconLoader::SMALL) { | |
| 228 idr = kGenericIdrs.idr_small_; | |
| 229 } else if (size == IconLoader::NORMAL) { | |
| 230 idr = kGenericIdrs.idr_normal_; | |
| 231 } else { | |
| 232 idr = kGenericIdrs.idr_large_; | |
| 233 } | |
| 234 } | |
| 235 return idr; | |
| 236 } | |
| 237 } // namespace | |
| 238 | |
| 239 FileIconSourceCros::FileIconSourceCros() | |
| 240 : DataSource("fileicon", NULL) { | |
| 241 } | |
| 242 | |
| 243 FileIconSourceCros::~FileIconSourceCros() { | |
| 244 } | |
| 245 | |
| 246 void FileIconSourceCros::StartDataRequest(const std::string& url, | |
| 247 bool is_incognito, | |
| 248 int request_id) { | |
| 249 int idr = UrlToIDR(url); | |
| 250 const ResourceBundle& rb = ResourceBundle::GetSharedInstance(); | |
| 251 scoped_refptr<RefCountedStaticMemory> bytes(rb.LoadDataResourceBytes(idr)); | |
| 252 SendResponse(request_id, bytes); | |
| 253 } | |
| 254 | |
| 255 // The mime type refers to the type of the response/icon served. | |
| 256 std::string FileIconSourceCros::GetMimeType( | |
| 257 const std::string& url) const { | |
| 258 return "image/png"; | |
| 259 } | |
| OLD | NEW |