[Suggestions] Adding a Thumbnail Manager for the Suggestions Service

chrome/browser/search/suggestions/thumbnail_manager.cc

+// Copyright 2014 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/search/suggestions/thumbnail_manager.h"
+
+#include "base/memory/scoped_ptr.h"
+#include "chrome/browser/profiles/profile.h"
+#include "chrome/browser/search/suggestions/proto/suggestions.pb.h"
+#include "content/public/browser/browser_thread.h"
+#include "net/base/load_flags.h"
+
+namespace suggestions {
+
+ThumbnailManager::ThumbnailManager(Profile* profile)
+    : url_request_context_(profile->GetRequestContext()) {}
+
+ThumbnailManager::~ThumbnailManager() {}
+
+ThumbnailManager::ThumbnailRequest::ThumbnailRequest() {}
+
+ThumbnailManager::ThumbnailRequest::~ThumbnailRequest() {}
+
+void ThumbnailManager::InitializeThumbnailMap(
+    const SuggestionsProfile& suggestions) {
+  thumbnail_map_.clear();
+  for (int i = 0; i < suggestions.suggestions_size(); ++i) {
+    const ChromeSuggestion& suggestion = suggestions.suggestions(i);
+    if (suggestion.has_thumbnail()) {
+      thumbnail_map_[GURL(suggestion.url())] = GURL(suggestion.thumbnail());
+    }
+  }
+}
+
+void ThumbnailManager::GetPageThumbnail(
+    const GURL& url,
+    base::Callback<void(const GURL&, const SkBitmap*)> callback) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+  // If |url| is not found in |thumbnail_map_|, then invoke |callback| with NULL
+  // since there is no associated thumbnail.
+  GURL thumbnail_url;
+  if (!GetThumbnailURL(url, &thumbnail_url)) {
+    callback.Run(url, NULL);
+    return;
+  }
+
+  // Look for a request in progress for |thumbnail_url|.
+  ThumbnailRequestMap::iterator it = pending_requests_.find(thumbnail_url);
+  if (it == pending_requests_.end()) {
+    // |thumbnail_url| is not being fetched, so create a request and initiate
+    // the fetch.
+    ThumbnailRequest* request = &pending_requests_[thumbnail_url];
+    request->url = url;
+    request->callbacks.push_back(callback);
+    // This is deallocated in OnFetchComplete().
+    request->fetcher = new chrome::BitmapFetcher(thumbnail_url, this);
+    request->fetcher->Start(
+        url_request_context_, std::string(),
+        net::URLRequest::CLEAR_REFERRER_ON_TRANSITION_FROM_SECURE_TO_INSECURE,
+        net::LOAD_NORMAL);
+  } else {
+    // Request in progress. Register as an interested callback.
+    it->second.callbacks.push_back(callback);
+  }
+}
+
+void ThumbnailManager::OnFetchComplete(const GURL thumbnail_url,
+                                       const SkBitmap* bitmap) {
+  DCHECK(content::BrowserThread::CurrentlyOn(content::BrowserThread::UI));
+
+  ThumbnailRequestMap::iterator it = pending_requests_.find(thumbnail_url);
+  DCHECK(it != pending_requests_.end());
+
+  ThumbnailRequest* request = &it->second;
+
+  // Make sure the request fetcher is deallocated.
+  scoped_ptr<chrome::BitmapFetcher> fetcher_destructor(request->fetcher);
+  // Here |bitmap| could be NULL or a pointer to a bitmap which is owned by the
+  // BitmapFetcher and which ceases to exist after this function. Pass the
+  // un-owned pointer to the registered callbacks.
+  for (CallbackVector::iterator it = request->callbacks.begin();
+       it != request->callbacks.end(); ++it) {
+    it->Run(request->url, bitmap);
+  }
+  pending_requests_.erase(it);
+}
+
+bool ThumbnailManager::GetThumbnailURL(const GURL& url, GURL* thumbnail_url) {
+  std::map<GURL, GURL>::iterator it = thumbnail_map_.find(url);
+  if (it == thumbnail_map_.end())
+    return false;  // Not found.
+  *thumbnail_url = it->second;
+  return true;
+}
+
+}  // namespace suggestions