extensions/browser/api/capture_web_contents_function.h

   1 // Copyright 2013 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 #ifndef EXTENSIONS_BROWSER_API_CAPTURE_WEB_CONTENTS_FUNCTION_H_
   6 #define EXTENSIONS_BROWSER_API_CAPTURE_WEB_CONTENTS_FUNCTION_H_
   7
   8 #include "content/public/browser/readback_types.h"
   9 #include "extensions/browser/extension_function.h"
  10 #include "extensions/common/api/extension_types.h"
  11
  12 class SkBitmap;
  13
  14 namespace content {
  15 class WebContents;
  16 }
  17
  18 namespace extensions {
  19
  20 // Base class for capturing visibile area of a WebContents.
  21 // This is used by both webview.captureVisibleRegion and tabs.captureVisibleTab.
  22 class CaptureWebContentsFunction : public AsyncExtensionFunction {
  23  public:
  24   CaptureWebContentsFunction() {}
  25
  26  protected:
  27   ~CaptureWebContentsFunction() override {}
  28
  29   // ExtensionFunction implementation.
  30   bool HasPermission() override;
  31   bool RunAsync() override;
  32
  33   virtual bool IsScreenshotEnabled() = 0;
  34   virtual content::WebContents* GetWebContentsForID(int context_id) = 0;
  35
  36   enum FailureReason {
  37     FAILURE_REASON_UNKNOWN,
  38     FAILURE_REASON_ENCODING_FAILED,
  39     FAILURE_REASON_VIEW_INVISIBLE
  40   };
  41   virtual void OnCaptureFailure(FailureReason reason) = 0;
  42
  43  private:
  44   typedef core_api::extension_types::ImageDetails ImageDetails;
  45
  46   void CopyFromBackingStoreComplete(const SkBitmap& bitmap,
  47                                     content::ReadbackResponse response);
  48   void OnCaptureSuccess(const SkBitmap& bitmap);
  49
  50   // |context_id_| is the ID used to find the relevant WebContents. In the
  51   // |tabs.captureVisibleTab()| api, this represents the window-id, and in the
  52   // |webview.captureVisibleRegion()| api, this represents the instance-id of
  53   // the guest.
  54   int context_id_;
  55
  56   // The format (JPEG vs PNG) of the resulting image.  Set in RunAsync().
  57   ImageDetails::Format image_format_;
  58
  59   // Quality setting to use when encoding jpegs.  Set in RunAsync().
  60   int image_quality_;
  61
  62   DISALLOW_COPY_AND_ASSIGN(CaptureWebContentsFunction);
  63 };
  64
  65 }  // namespace extensions
  66
  67 #endif  // EXTENSIONS_BROWSER_API_CAPTURE_WEB_CONTENTS_FUNCTION_H_