Index: chrome/browser/ui/snapshot_tab_helper.cc |
diff --git a/chrome/browser/ui/snapshot_tab_helper.cc b/chrome/browser/ui/snapshot_tab_helper.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..76ed5ca7201cda393a0c1773eb2247a8e32ea914 |
--- /dev/null |
+++ b/chrome/browser/ui/snapshot_tab_helper.cc |
@@ -0,0 +1,47 @@ |
+// Copyright (c) 2012 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/ui/snapshot_tab_helper.h" |
+ |
+#include "chrome/common/chrome_notification_types.h" |
+#include "chrome/common/render_messages.h" |
+#include "content/public/browser/notification_service.h" |
+#include "content/public/browser/web_contents.h" |
+ |
+using content::WebContents; |
+ |
+DEFINE_WEB_CONTENTS_USER_DATA_KEY(SnapshotTabHelper); |
+ |
+SnapshotTabHelper::SnapshotTabHelper(WebContents* web_contents) |
+ : content::WebContentsObserver(web_contents) { |
+} |
+ |
+SnapshotTabHelper::~SnapshotTabHelper() { |
+} |
+ |
+void SnapshotTabHelper::CaptureSnapshot() { |
+ Send(new ChromeViewMsg_CaptureSnapshot(routing_id())); |
+} |
+ |
+//////////////////////////////////////////////////////////////////////////////// |
+// WebContentsObserver overrides |
+ |
+bool SnapshotTabHelper::OnMessageReceived(const IPC::Message& message) { |
+ bool handled = true; |
+ IPC_BEGIN_MESSAGE_MAP(SnapshotTabHelper, message) |
+ IPC_MESSAGE_HANDLER(ChromeViewHostMsg_Snapshot, OnSnapshot) |
+ IPC_MESSAGE_UNHANDLED(handled = false) |
+ IPC_END_MESSAGE_MAP() |
+ return handled; |
+} |
+ |
+//////////////////////////////////////////////////////////////////////////////// |
+// Internal helpers |
+ |
+void SnapshotTabHelper::OnSnapshot(const SkBitmap& bitmap) { |
+ content::NotificationService::current()->Notify( |
+ chrome::NOTIFICATION_TAB_SNAPSHOT_TAKEN, |
+ content::Source<WebContents>(web_contents()), |
+ content::Details<const SkBitmap>(&bitmap)); |
+} |