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 #ifndef CHROME_BROWSER_UI_WEBUI_NTP_FAVICON_WEBUI_HANDLER_H_ 6 #define CHROME_BROWSER_UI_WEBUI_NTP_FAVICON_WEBUI_HANDLER_H_ 7 8 #include <map> 9 #include <string> 10 11 #include "base/basictypes.h" 12 #include "base/memory/ref_counted.h" 13 #include "chrome/browser/favicon/favicon_service.h" 14 #include "chrome/common/cancelable_task_tracker.h" 15 #include "content/public/browser/web_ui_message_handler.h" 16 17 class ExtensionIconColorManager; 18 19 namespace base { 20 class ListValue; 21 } 22 23 class FaviconWebUIHandler : public content::WebUIMessageHandler { 24 public: 25 FaviconWebUIHandler(); 26 virtual ~FaviconWebUIHandler(); 27 28 // WebUIMessageHandler 29 virtual void RegisterMessages() OVERRIDE; 30 31 // Called from the JS to get the dominant color of a favicon. The first 32 // argument is a favicon URL, the second is the ID of the DOM node that is 33 // asking for it. 34 void HandleGetFaviconDominantColor(const base::ListValue* args); 35 36 // As above, but for an app tile. The sole argument is the extension ID. 37 void HandleGetAppIconDominantColor(const base::ListValue* args); 38 39 // Callback getting signal that an app icon is loaded. 40 void NotifyAppIconReady(const std::string& extension_id); 41 42 private: 43 // Called when favicon data is available from the history backend. 44 void OnFaviconDataAvailable( 45 FaviconService::Handle request_handle, 46 const chrome::FaviconBitmapResult& bitmap_result); 47 48 CancelableTaskTracker cancelable_task_tracker_; 49 50 // Map from request ID to DOM ID so we can make the appropriate callback when 51 // the favicon request comes back. This map exists because 52 // CancelableRequestConsumerTSimple only takes POD keys. 53 std::map<int, std::string> dom_id_map_; 54 // A counter to track ID numbers as we use them. 55 int id_; 56 57 // Raw PNG representation of the favicon to show when the favicon 58 // database doesn't have a favicon for a webpage. 59 scoped_refptr<base::RefCountedMemory> default_favicon_; 60 61 // Manage retrieval of icons from apps. 62 scoped_ptr<ExtensionIconColorManager> app_icon_color_manager_; 63 64 DISALLOW_COPY_AND_ASSIGN(FaviconWebUIHandler); 65 }; 66 67 #endif // CHROME_BROWSER_UI_WEBUI_NTP_FAVICON_WEBUI_HANDLER_H_ 68