third_party/blink/renderer/modules/compression/inflate_transformer.h

#ifndef THIRD_PARTY_BLINK_RENDERER_MODULES_COMPRESSION_INFLATE_TRANSFORMER_H_
#define THIRD_PARTY_BLINK_RENDERER_MODULES_COMPRESSION_INFLATE_TRANSFORMER_H_

#include "base/util/type_safety/strong_alias.h"

#include "third_party/blink/renderer/core/streams/transform_stream_transformer.h"
#include "third_party/blink/renderer/core/typed_arrays/dom_typed_array.h"
#include "third_party/blink/renderer/platform/wtf/vector.h"
#include "third_party/zlib/zlib.h"

namespace blink {

class InflateTransformer final : public TransformStreamTransformer {
 public:
  enum class Algorithm { kDeflate, kGzip };

  InflateTransformer(ScriptState*, Algorithm algorithm);
  ~InflateTransformer() override;

  void Transform(v8::Local<v8::Value> chunk,
                 TransformStreamDefaultControllerInterface*,
                 ExceptionState&) override;

  void Flush(TransformStreamDefaultControllerInterface*,
             ExceptionState&) override;

  ScriptState* GetScriptState() override { return script_state_; }

  void Trace(Visitor*) override;

 private:
  using IsFinished = util::StrongAlias<class IsFinishedTag, bool>;

  void Inflate(const Bytef*,
               wtf_size_t,
               IsFinished,
               TransformStreamDefaultControllerInterface*,
               ExceptionState&);

  Member<ScriptState> script_state_;

  z_stream stream_;

  Vector<Bytef> out_buffer_;

  bool was_flush_called_ = false;

  static constexpr wtf_size_t kBufferSize = 65536;

  DISALLOW_COPY_AND_ASSIGN(InflateTransformer);
};

}  // namespace blink

#endif  // THIRD_PARTY_BLINK_RENDERER_MODULES_COMPRESSION_INFLATE_TRANSFORMER_H_