chrome/browser/extensions/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 CHROME_BROWSER_EXTENSIONS_API_CAPTURE_WEB_CONTENTS_FUNCTION_H_
   6 #define CHROME_BROWSER_EXTENSIONS_API_CAPTURE_WEB_CONTENTS_FUNCTION_H_
   7
   8 #include "chrome/browser/extensions/chrome_extension_function.h"
   9 #include "chrome/common/extensions/api/types.h"
  10
  11 class SkBitmap;
  12
  13 namespace content {
  14 class WebContents;
  15 }
  16
  17 namespace extensions {
  18
  19 // Base class for capturing visibile area of a WebContents.
  20 // This is used by both webview.captureVisibleRegion and tabs.captureVisibleTab.
  21 class CaptureWebContentsFunction : public ChromeAsyncExtensionFunction {
  22  public:
  23   CaptureWebContentsFunction();
  24
  25  protected:
  26   virtual ~CaptureWebContentsFunction();
  27
  28   // ExtensionFunction implementation.
  29   virtual bool HasPermission() OVERRIDE;
  30   virtual bool RunImpl() OVERRIDE;
  31
  32   virtual bool IsScreenshotEnabled() = 0;
  33   virtual content::WebContents* GetWebContentsForID(int context_id) = 0;
  34
  35   enum FailureReason {
  36     FAILURE_REASON_UNKNOWN,
  37     FAILURE_REASON_CONTENT_NOT_FOUND,
  38     FAILURE_REASON_ENCODING_FAILED,
  39     FAILURE_REASON_VIEW_INVISIBLE
  40   };
  41   virtual void OnCaptureFailure(FailureReason reason) = 0;
  42
  43  private:
  44   typedef api::types::ImageDetails ImageDetails;
  45
  46   void CopyFromBackingStoreComplete(bool succeed, const SkBitmap& bitmap);
  47   void GetSnapshotFromRendererComplete(bool succeeded, const SkBitmap& bitmap);
  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 RunImpl().
  57   ImageDetails::Format image_format_;
  58
  59   // Quality setting to use when encoding jpegs.  Set in RunImpl().
  60   int image_quality_;
  61
  62   DISALLOW_COPY_AND_ASSIGN(CaptureWebContentsFunction);
  63 };
  64
  65 }  // namespace extensions
  66
  67 #endif  // CHROME_BROWSER_EXTENSIONS_API_CAPTURE_WEB_CONTENTS_FUNCTION_H_