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