| OLD | NEW |
| (Empty) |
| 1 // Copyright 2014 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 #ifndef COMPONENTS_ENHANCED_BOOKMARKS_BOOKMARK_IMAGE_SERVICE_H_ | |
| 5 #define COMPONENTS_ENHANCED_BOOKMARKS_BOOKMARK_IMAGE_SERVICE_H_ | |
| 6 | |
| 7 #include "base/files/file_path.h" | |
| 8 #include "base/memory/scoped_ptr.h" | |
| 9 #include "base/memory/singleton.h" | |
| 10 #include "components/bookmarks/browser/bookmark_model_observer.h" | |
| 11 #include "components/enhanced_bookmarks/image_record.h" | |
| 12 #include "components/enhanced_bookmarks/image_store.h" | |
| 13 #include "components/keyed_service/core/keyed_service.h" | |
| 14 #include "net/url_request/url_request.h" | |
| 15 #include "url/gurl.h" | |
| 16 | |
| 17 namespace base { | |
| 18 class SingleThreadTaskRunner; | |
| 19 } | |
| 20 | |
| 21 namespace bookmarks { | |
| 22 class BookmarkNode; | |
| 23 } | |
| 24 | |
| 25 namespace enhanced_bookmarks { | |
| 26 | |
| 27 class EnhancedBookmarkModel; | |
| 28 | |
| 29 // The BookmarkImageService stores salient images for bookmarks. | |
| 30 class BookmarkImageService : public KeyedService, | |
| 31 public bookmarks::BookmarkModelObserver, | |
| 32 public base::NonThreadSafe { | |
| 33 public: | |
| 34 BookmarkImageService(const base::FilePath& path, | |
| 35 EnhancedBookmarkModel* enhanced_bookmark_model, | |
| 36 scoped_refptr<base::SequencedWorkerPool> pool); | |
| 37 BookmarkImageService(scoped_ptr<ImageStore> store, | |
| 38 EnhancedBookmarkModel* enhanced_bookmark_model, | |
| 39 scoped_refptr<base::SequencedWorkerPool> pool); | |
| 40 | |
| 41 ~BookmarkImageService() override; | |
| 42 | |
| 43 typedef base::Callback<void(scoped_refptr<ImageRecord>)> ImageCallback; | |
| 44 | |
| 45 // KeyedService: | |
| 46 void Shutdown() override; | |
| 47 | |
| 48 // Returns a salient image for a URL. This may trigger a network request for | |
| 49 // the image if the image was not retrieved before and if a bookmark node has | |
| 50 // a URL for this salient image available. The image (which may be empty) is | |
| 51 // sent via the callback. The callback may be called synchronously if it is | |
| 52 // possible. The callback is always triggered on the main thread. | |
| 53 void SalientImageForUrl(const GURL& page_url, ImageCallback callback); | |
| 54 | |
| 55 // bookmarks::BookmarkModelObserver: | |
| 56 void BookmarkNodeRemoved(bookmarks::BookmarkModel* model, | |
| 57 const bookmarks::BookmarkNode* parent, | |
| 58 int old_index, | |
| 59 const bookmarks::BookmarkNode* node, | |
| 60 const std::set<GURL>& removed_urls) override; | |
| 61 void BookmarkModelLoaded(bookmarks::BookmarkModel* model, | |
| 62 bool ids_reassigned) override; | |
| 63 void BookmarkNodeMoved(bookmarks::BookmarkModel* model, | |
| 64 const bookmarks::BookmarkNode* old_parent, | |
| 65 int old_index, | |
| 66 const bookmarks::BookmarkNode* new_parent, | |
| 67 int new_index) override; | |
| 68 void BookmarkNodeAdded(bookmarks::BookmarkModel* model, | |
| 69 const bookmarks::BookmarkNode* parent, | |
| 70 int index) override; | |
| 71 void OnWillChangeBookmarkNode(bookmarks::BookmarkModel* model, | |
| 72 const bookmarks::BookmarkNode* node) override; | |
| 73 void BookmarkNodeChanged(bookmarks::BookmarkModel* model, | |
| 74 const bookmarks::BookmarkNode* node) override; | |
| 75 void BookmarkNodeFaviconChanged(bookmarks::BookmarkModel* model, | |
| 76 const bookmarks::BookmarkNode* node) override; | |
| 77 void BookmarkNodeChildrenReordered( | |
| 78 bookmarks::BookmarkModel* model, | |
| 79 const bookmarks::BookmarkNode* node) override; | |
| 80 void BookmarkAllUserNodesRemoved(bookmarks::BookmarkModel* model, | |
| 81 const std::set<GURL>& removed_urls) override; | |
| 82 | |
| 83 protected: | |
| 84 // Returns true if the image for the page_url is currently being fetched. | |
| 85 bool IsPageUrlInProgress(const GURL& page_url); | |
| 86 | |
| 87 // Stores the new image to local storage. If update_bookmarks is true, relates | |
| 88 // the corresponding bookmark to image_url. | |
| 89 void ProcessNewImage(const GURL& page_url, | |
| 90 bool update_bookmarks, | |
| 91 const GURL& image_url, | |
| 92 scoped_ptr<gfx::Image> image); | |
| 93 | |
| 94 // Resizes large images to proper size that fits device display. This method | |
| 95 // should _not_ run on the UI thread. | |
| 96 virtual scoped_ptr<gfx::Image> ResizeImage(const gfx::Image& image) = 0; | |
| 97 | |
| 98 // Sets a new image for a bookmark. If the given page_url is bookmarked and | |
| 99 // the image is retrieved from the image_url, then the image is locally | |
| 100 // stored. If update_bookmark is true the URL is also added to the bookmark. | |
| 101 virtual void RetrieveSalientImage( | |
| 102 const GURL& page_url, | |
| 103 const GURL& image_url, | |
| 104 const std::string& referrer, | |
| 105 net::URLRequest::ReferrerPolicy referrer_policy, | |
| 106 bool update_bookmark) = 0; | |
| 107 | |
| 108 // Retrieves a salient image for a given page_url by downloading the image in | |
| 109 // one of the bookmark. | |
| 110 virtual void RetrieveSalientImageForPageUrl(const GURL& page_url); | |
| 111 | |
| 112 // PageUrls currently in the progress of being retrieved. | |
| 113 std::set<GURL> in_progress_page_urls_; | |
| 114 | |
| 115 // Cached pointer to the bookmark model. | |
| 116 EnhancedBookmarkModel* enhanced_bookmark_model_; | |
| 117 | |
| 118 private: | |
| 119 // If fetch_from_web is true, retrieves the salient image via a network | |
| 120 // request; else only gets the image from local storage. | |
| 121 void SalientImageForUrl(const GURL& page_url, | |
| 122 bool fetch_from_web, | |
| 123 ImageCallback stack_callback); | |
| 124 | |
| 125 // Processes the requests that have been waiting on an image. | |
| 126 void ProcessRequests(const GURL& page_url, scoped_refptr<ImageRecord> image); | |
| 127 | |
| 128 // Once an image is retrieved this method calls ResizeImage() and updates the | |
| 129 // store with the smaller image, then returns the newly formed ImageRecord. | |
| 130 // This is typically called on |pool_|, the background sequenced worker pool | |
| 131 // for this object. | |
| 132 scoped_refptr<ImageRecord> ResizeAndStoreImage( | |
| 133 scoped_refptr<ImageRecord> image_info, | |
| 134 const GURL& page_url); | |
| 135 | |
| 136 // Calls |StoreImage| in the background. This should only be called from the | |
| 137 // main thread. | |
| 138 void PostTaskToStoreImage(scoped_ptr<gfx::Image> image, | |
| 139 const GURL& image_url, | |
| 140 const GURL& page_url); | |
| 141 | |
| 142 // Called when |StoreImage| as been posted. This should only be called from | |
| 143 // the main thread. | |
| 144 void OnStoreImagePosted(const GURL& page_url, | |
| 145 scoped_refptr<ImageRecord> image); | |
| 146 | |
| 147 // Called when retrieving an image from the image store fails, to trigger | |
| 148 // retrieving the image from the url stored in the bookmark (if any). | |
| 149 void FetchCallback(const GURL& page_url, | |
| 150 ImageCallback original_callback, | |
| 151 scoped_refptr<ImageRecord> record); | |
| 152 | |
| 153 // Remove the image stored for this bookmark (if it exists). Called when a | |
| 154 // bookmark is deleted. | |
| 155 void RemoveImageForUrl(const GURL& url); | |
| 156 | |
| 157 // Moves an image from one url to another. | |
| 158 void ChangeImageURL(const GURL& from, const GURL& to); | |
| 159 | |
| 160 // Removes all the entries in the image service. | |
| 161 void ClearAll(); | |
| 162 | |
| 163 // The image store can only be accessed from the blocking pool. | |
| 164 // RetrieveImageFromStore starts a request to retrieve the image and returns | |
| 165 // the result via a callback. RetrieveImageFromStore must be called on the | |
| 166 // main thread and the callback will be called on the main thread as well. The | |
| 167 // callback will always be called. The returned image is nil if the image is | |
| 168 // not present in the store. | |
| 169 void RetrieveImageFromStore(const GURL& page_url, ImageCallback callback); | |
| 170 | |
| 171 // Maps a pageUrl to an image. | |
| 172 scoped_ptr<ImageStore> store_; | |
| 173 | |
| 174 // All the callbacks waiting for a particular image. | |
| 175 std::map<const GURL, std::vector<ImageCallback>> callbacks_; | |
| 176 | |
| 177 // When a bookmark is changed, two messages are received on the | |
| 178 // bookmarkModelObserver, one with the old state, one with the new. The url | |
| 179 // before the change is saved in this instance variable. | |
| 180 GURL previous_url_; | |
| 181 | |
| 182 // The worker pool to enqueue the requests onto. | |
| 183 scoped_refptr<base::SequencedWorkerPool> pool_; | |
| 184 DISALLOW_COPY_AND_ASSIGN(BookmarkImageService); | |
| 185 }; | |
| 186 | |
| 187 } // namespace enhanced_bookmarks | |
| 188 | |
| 189 #endif // COMPONENTS_ENHANCED_BOOKMARKS_BOOKMARK_IMAGE_SERVICE_H_ | |
| 190 | |
| OLD | NEW |