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

#ifndef CONTENT_BROWSER_DOWNLOAD_BYTE_STREAM_INPUT_STREAM_H_
#define CONTENT_BROWSER_DOWNLOAD_BYTE_STREAM_INPUT_STREAM_H_

#include "components/download/public/common/input_stream.h"
#include "content/common/content_export.h"

namespace content {

class ByteStreamReader;

// Download input stream backed by a ByteStreamReader.
class CONTENT_EXPORT ByteStreamInputStream : public download::InputStream {
 public:
  explicit ByteStreamInputStream(
      std::unique_ptr<ByteStreamReader> stream_reader);
  ~ByteStreamInputStream() override;

  // download::InputStream
  bool IsEmpty() override;
  void RegisterDataReadyCallback(
      const mojo::SimpleWatcher::ReadyCallback& callback) override;
  void ClearDataReadyCallback() override;
  download::InputStream::StreamState Read(scoped_refptr<net::IOBuffer>* data,
                                          size_t* length) override;
  download::DownloadInterruptReason GetCompletionStatus() override;

 private:
  // ByteStreamReader to read from.
  std::unique_ptr<ByteStreamReader, base::OnTaskRunnerDeleter> stream_reader_;

  // Status when the response completes.
  download::DownloadInterruptReason completion_status_;

  DISALLOW_COPY_AND_ASSIGN(ByteStreamInputStream);
};

}  // namespace content

#endif  // CONTENT_BROWSER_DOWNLOAD_BYTE_STREAM_INPUT_STREAM_H_