1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46

content / browser / devtools / protocol / devtools_mhtml_helper.h [blame]

// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_MHTML_HELPER_H_
#define CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_MHTML_HELPER_H_

#include "content/browser/devtools/protocol/page_handler.h"
#include "storage/browser/blob/shareable_file_reference.h"

namespace content {
namespace protocol {

class DevToolsMHTMLHelper
    : public base::RefCountedThreadSafe<DevToolsMHTMLHelper> {
 public:
  static void Capture(
      const WebContents::Getter& web_contents_getter,
      std::unique_ptr<PageHandler::CaptureSnapshotCallback> callback);

 private:
  DevToolsMHTMLHelper(
      const WebContents::Getter& web_contents_getter,
      std::unique_ptr<PageHandler::CaptureSnapshotCallback> callback);
  ~DevToolsMHTMLHelper();

  void CreateTemporaryFile();
  void TemporaryFileCreatedOnIO();
  void TemporaryFileCreatedOnUI();
  void MHTMLGeneratedOnUI(int64_t mhtml_file_size);
  void ReadMHTML();
  void ReportFailure(const std::string& message);
  void ReportSuccess(std::unique_ptr<std::string> mhtml_snapshot);

  WebContents::Getter web_contents_getter_;
  std::unique_ptr<PageHandler::CaptureSnapshotCallback> callback_;
  scoped_refptr<storage::ShareableFileReference> mhtml_file_;
  base::FilePath mhtml_snapshot_path_;

  friend class base::RefCountedThreadSafe<DevToolsMHTMLHelper>;
};

}  // namespace protocol
}  // namespace content

#endif  // CONTENT_BROWSER_DEVTOOLS_PROTOCOL_DEVTOOLS_MHTML_HELPER_H_