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
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56

content / browser / download / mhtml_extra_parts_impl.h [blame]

// Copyright 2017 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_DOWNLOAD_MHTML_EXTRA_PARTS_IMPL_H_
#define CONTENT_BROWSER_DOWNLOAD_MHTML_EXTRA_PARTS_IMPL_H_

#include "content/public/browser/mhtml_extra_parts.h"

namespace content {

// Data fields used to build an additional MHTML part in the output file.
struct MHTMLExtraDataPart {
  std::string content_type;
  std::string content_location;
  std::string extra_headers;
  std::string body;

  MHTMLExtraDataPart();
  MHTMLExtraDataPart(const MHTMLExtraDataPart& other);
  MHTMLExtraDataPart(MHTMLExtraDataPart&& other);
  MHTMLExtraDataPart& operator=(const MHTMLExtraDataPart& other);
  MHTMLExtraDataPart& operator=(MHTMLExtraDataPart&& other);
  ~MHTMLExtraDataPart();
};

// Class used as a data object for WebContents UserData to represent an MHTML
// part that we plan to write into the output MHTML file.  Each MHTMLExtraPart
// object in the contained vector lets us hold enough information to generate
// one MHTML part.  This allows arbitrary extra MHTML parts to be added into the
// complete file.  For instance, this might be used for gathering load time
// signals in debug code for analysis.
class MHTMLExtraPartsImpl : public content::MHTMLExtraParts {
 public:
  MHTMLExtraPartsImpl();
  ~MHTMLExtraPartsImpl() override;

  // Return the vector of parts to be serialized.
  const std::vector<MHTMLExtraDataPart>& parts() const { return parts_; }

  // Return the number of extra parts added.
  int64_t size() override;

  // Creates a MHTMLExtraDataPart and adds it to our vector of parts.
  void AddExtraMHTMLPart(const std::string& content_type,
                         const std::string& content_location,
                         const std::string& extra_headers,
                         const std::string& body) override;

 private:
  std::vector<MHTMLExtraDataPart> parts_;
};

}  // namespace content

#endif  // CONTENT_BROWSER_DOWNLOAD_MHTML_EXTRA_PARTS_IMPL_H_