| OLD | NEW |
| (Empty) |
| 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 #include "chrome/browser/search/thumbnail_source.h" | |
| 6 | |
| 7 #include "base/callback.h" | |
| 8 #include "base/memory/ref_counted_memory.h" | |
| 9 #include "base/message_loop/message_loop.h" | |
| 10 #include "chrome/browser/search/instant_io_context.h" | |
| 11 #include "chrome/browser/thumbnails/thumbnail_service.h" | |
| 12 #include "chrome/browser/thumbnails/thumbnail_service_factory.h" | |
| 13 #include "chrome/common/url_constants.h" | |
| 14 #include "net/url_request/url_request.h" | |
| 15 #include "url/gurl.h" | |
| 16 | |
| 17 // Set ThumbnailService now as Profile isn't thread safe. | |
| 18 ThumbnailSource::ThumbnailSource(Profile* profile, bool capture_thumbnails) | |
| 19 : thumbnail_service_(ThumbnailServiceFactory::GetForProfile(profile)), | |
| 20 capture_thumbnails_(capture_thumbnails) { | |
| 21 } | |
| 22 | |
| 23 ThumbnailSource::~ThumbnailSource() { | |
| 24 } | |
| 25 | |
| 26 std::string ThumbnailSource::GetSource() const { | |
| 27 return capture_thumbnails_ ? | |
| 28 chrome::kChromeUIThumbnailHost2 : chrome::kChromeUIThumbnailHost; | |
| 29 } | |
| 30 | |
| 31 void ThumbnailSource::StartDataRequest( | |
| 32 const std::string& path, | |
| 33 int render_process_id, | |
| 34 int render_frame_id, | |
| 35 const content::URLDataSource::GotDataCallback& callback) { | |
| 36 scoped_refptr<base::RefCountedMemory> data; | |
| 37 if (thumbnail_service_->GetPageThumbnail(GURL(path), capture_thumbnails_, | |
| 38 &data)) { | |
| 39 // We have the thumbnail. | |
| 40 callback.Run(data.get()); | |
| 41 } else { | |
| 42 callback.Run(default_thumbnail_.get()); | |
| 43 } | |
| 44 if (capture_thumbnails_) | |
| 45 thumbnail_service_->AddForcedURL(GURL(path)); | |
| 46 } | |
| 47 | |
| 48 std::string ThumbnailSource::GetMimeType(const std::string&) const { | |
| 49 // We need to explicitly return a mime type, otherwise if the user tries to | |
| 50 // drag the image they get no extension. | |
| 51 return "image/png"; | |
| 52 } | |
| 53 | |
| 54 base::MessageLoop* ThumbnailSource::MessageLoopForRequestPath( | |
| 55 const std::string& path) const { | |
| 56 // TopSites can be accessed from the IO thread. | |
| 57 return thumbnail_service_.get() ? | |
| 58 NULL : content::URLDataSource::MessageLoopForRequestPath(path); | |
| 59 } | |
| 60 | |
| 61 bool ThumbnailSource::ShouldServiceRequest( | |
| 62 const net::URLRequest* request) const { | |
| 63 if (request->url().SchemeIs(chrome::kChromeSearchScheme)) | |
| 64 return InstantIOContext::ShouldServiceRequest(request); | |
| 65 return URLDataSource::ShouldServiceRequest(request); | |
| 66 } | |
| OLD | NEW |