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 |