Implement webview.captureVisibleRegion()

chrome/browser/extensions/api/tabs/tabs_api.cc

 // 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/extensions/api/tabs/tabs_api.h"
 
 #include <stddef.h>
 #include <algorithm>
 #include <limits>
 #include <utility>
@@ skipping 1633 matching lines @@
   // under such circumstances, whereas |TabStripModel::CloseWebContentsAt()|
   // does not.
   contents->Close();
   return true;
 }
 
 TabsCaptureVisibleTabFunction::TabsCaptureVisibleTabFunction()
     : chrome_details_(this) {
 }
 
+bool TabsCaptureVisibleTabFunction::HasPermission() {
+  return true;
+}
+
 bool TabsCaptureVisibleTabFunction::IsScreenshotEnabled() {
   PrefService* service = chrome_details_.GetProfile()->GetPrefs();
   if (service->GetBoolean(prefs::kDisableScreenshots)) {
     error_ = keys::kScreenshotsDisabled;
     return false;
   }
   return true;
 }
 
 WebContents* TabsCaptureVisibleTabFunction::GetWebContentsForID(int window_id) {
   Browser* browser = NULL;
   if (!GetBrowserFromWindowID(&chrome_details_, window_id, &browser))
     return NULL;
 
   WebContents* contents = browser->tab_strip_model()->GetActiveWebContents();
   if (!contents) {
     error_ = "No active web contents to capture";
     return NULL;
   }
 
   if (!extension()->permissions_data()->CanCaptureVisiblePage(
           SessionTabHelper::IdForTab(contents), &error_)) {
     return NULL;
   }
   return contents;
 }
 
+bool TabsCaptureVisibleTabFunction::RunAsync() {
+  using api::extension_types::ImageDetails;
+
+  EXTENSION_FUNCTION_VALIDATE(args_);
+
+  int 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);
+  }
+
+  WebContents* contents = GetWebContentsForID(context_id);
+
+  return CaptureAsync(
+      contents, image_details.get(),
+      base::Bind(&TabsCaptureVisibleTabFunction::CopyFromBackingStoreComplete,
+                 this));
+}
+
+void TabsCaptureVisibleTabFunction::OnCaptureSuccess(const SkBitmap& bitmap) {
+  std::string base64_result;
+  if (!EncodeBitmap(bitmap, &base64_result)) {
+    OnCaptureFailure(FAILURE_REASON_ENCODING_FAILED);
+    return;
+  }
+
+  SetResult(new base::StringValue(base64_result));
+  SendResponse(true);
+}
+
 void TabsCaptureVisibleTabFunction::OnCaptureFailure(FailureReason reason) {
   const char* reason_description = "internal error";
   switch (reason) {
     case FAILURE_REASON_UNKNOWN:
       reason_description = "unknown error";
       break;
     case FAILURE_REASON_ENCODING_FAILED:
       reason_description = "encoding failed";
       break;
     case FAILURE_REASON_VIEW_INVISIBLE:
@@ skipping 374 matching lines @@
   ZoomModeToZoomSettings(zoom_mode, &zoom_settings);
   zoom_settings.default_zoom_factor.reset(new double(
       content::ZoomLevelToZoomFactor(zoom_controller->GetDefaultZoomLevel())));
 
   results_ = api::tabs::GetZoomSettings::Results::Create(zoom_settings);
   SendResponse(true);
   return true;
 }
 
 }  // namespace extensions