Index: extensions/browser/api/capture_web_contents_function.cc |
diff --git a/extensions/browser/api/capture_web_contents_function.cc b/extensions/browser/api/capture_web_contents_function.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..e10842d8aee0aa103031261e8dec0ca9045f5479 |
--- /dev/null |
+++ b/extensions/browser/api/capture_web_contents_function.cc |
@@ -0,0 +1,150 @@ |
+// Copyright 2013 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 "extensions/browser/api/capture_web_contents_function.h" |
+ |
+#include "base/base64.h" |
+#include "base/strings/stringprintf.h" |
+#include "content/public/browser/render_widget_host.h" |
+#include "content/public/browser/render_widget_host_view.h" |
+#include "content/public/browser/web_contents.h" |
+#include "extensions/browser/extension_function.h" |
+#include "extensions/common/constants.h" |
+#include "ui/gfx/codec/jpeg_codec.h" |
+#include "ui/gfx/codec/png_codec.h" |
+#include "ui/gfx/display.h" |
+#include "ui/gfx/geometry/size_conversions.h" |
+#include "ui/gfx/screen.h" |
+ |
+using content::RenderWidgetHost; |
+using content::RenderWidgetHostView; |
+using content::WebContents; |
+ |
+namespace extensions { |
+ |
+using api::extension_types::ImageDetails; |
+ |
+bool CaptureWebContentsFunction::HasPermission() { |
+ return true; |
+} |
+ |
+bool CaptureWebContentsFunction::RunAsync() { |
+ EXTENSION_FUNCTION_VALIDATE(args_); |
+ |
+ context_id_ = extension_misc::kCurrentWindowId; |
+ args_->GetInteger(0, &context_id_); |
+ |
+ scoped_ptr<ImageDetails> image_details; |
+ if (args_->GetSize() > 1) { |
+ base::Value* spec = NULL; |
+ EXTENSION_FUNCTION_VALIDATE(args_->Get(1, &spec) && spec); |
+ image_details = ImageDetails::FromValue(*spec); |
+ } |
+ |
+ if (!IsScreenshotEnabled()) |
+ return false; |
+ |
+ WebContents* contents = GetWebContentsForID(context_id_); |
+ if (!contents) |
+ return false; |
+ |
+ // The default format and quality setting used when encoding jpegs. |
+ const api::extension_types::ImageFormat kDefaultFormat = |
+ api::extension_types::IMAGE_FORMAT_JPEG; |
+ const int kDefaultQuality = 90; |
+ |
+ image_format_ = kDefaultFormat; |
+ image_quality_ = kDefaultQuality; |
+ |
+ if (image_details) { |
+ if (image_details->format != api::extension_types::IMAGE_FORMAT_NONE) |
+ image_format_ = image_details->format; |
+ if (image_details->quality.get()) |
+ image_quality_ = *image_details->quality; |
+ } |
+ |
+ // TODO(miu): Account for fullscreen render widget? http://crbug.com/419878 |
+ RenderWidgetHostView* const view = contents->GetRenderWidgetHostView(); |
+ RenderWidgetHost* const host = view ? view->GetRenderWidgetHost() : nullptr; |
+ if (!view || !host) { |
+ OnCaptureFailure(FAILURE_REASON_VIEW_INVISIBLE); |
+ return false; |
+ } |
+ |
+ // By default, the requested bitmap size is the view size in screen |
+ // coordinates. However, if there's more pixel detail available on the |
+ // current system, increase the requested bitmap size to capture it all. |
+ const gfx::Size view_size = view->GetViewBounds().size(); |
+ gfx::Size bitmap_size = view_size; |
+ const gfx::NativeView native_view = view->GetNativeView(); |
+ gfx::Screen* const screen = gfx::Screen::GetScreenFor(native_view); |
+ const float scale = |
+ screen->GetDisplayNearestWindow(native_view).device_scale_factor(); |
+ if (scale > 1.0f) |
+ bitmap_size = gfx::ScaleToCeiledSize(view_size, scale); |
+ |
+ host->CopyFromBackingStore( |
+ gfx::Rect(view_size), |
+ bitmap_size, |
+ base::Bind(&CaptureWebContentsFunction::CopyFromBackingStoreComplete, |
+ this), |
+ kN32_SkColorType); |
+ return true; |
+} |
+ |
+void CaptureWebContentsFunction::CopyFromBackingStoreComplete( |
+ const SkBitmap& bitmap, |
+ content::ReadbackResponse response) { |
+ if (response == content::READBACK_SUCCESS) { |
+ OnCaptureSuccess(bitmap); |
+ return; |
+ } |
+ OnCaptureFailure(FAILURE_REASON_UNKNOWN); |
+} |
+ |
+void CaptureWebContentsFunction::OnCaptureSuccess(const SkBitmap& bitmap) { |
+ std::vector<unsigned char> data; |
+ SkAutoLockPixels screen_capture_lock(bitmap); |
+ bool encoded = false; |
+ std::string mime_type; |
+ switch (image_format_) { |
+ case api::extension_types::IMAGE_FORMAT_JPEG: |
+ encoded = gfx::JPEGCodec::Encode( |
+ reinterpret_cast<unsigned char*>(bitmap.getAddr32(0, 0)), |
+ gfx::JPEGCodec::FORMAT_SkBitmap, |
+ bitmap.width(), |
+ bitmap.height(), |
+ static_cast<int>(bitmap.rowBytes()), |
+ image_quality_, |
+ &data); |
+ mime_type = kMimeTypeJpeg; |
+ break; |
+ case api::extension_types::IMAGE_FORMAT_PNG: |
+ encoded = |
+ gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, |
+ true, // Discard transparency. |
+ &data); |
+ mime_type = kMimeTypePng; |
+ break; |
+ default: |
+ NOTREACHED() << "Invalid image format."; |
+ } |
+ |
+ if (!encoded) { |
+ OnCaptureFailure(FAILURE_REASON_ENCODING_FAILED); |
+ return; |
+ } |
+ |
+ std::string base64_result; |
+ base::StringPiece stream_as_string(reinterpret_cast<const char*>(data.data()), |
+ data.size()); |
+ |
+ base::Base64Encode(stream_as_string, &base64_result); |
+ base64_result.insert( |
+ 0, base::StringPrintf("data:%s;base64,", mime_type.c_str())); |
+ SetResult(new base::StringValue(base64_result)); |
+ SendResponse(true); |
+} |
+ |
+} // namespace extensions |