/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
/* vim:set ts=2 sw=2 sts=2 et cindent: */
/* This Source Code Form is subject to the terms of the Mozilla Public
 * License, v. 2.0. If a copy of the MPL was not distributed with this
 * file, You can obtain one at http://mozilla.org/MPL/2.0/. */

#ifndef mozilla_dom_ReadableStreamGenericReader_h
#define mozilla_dom_ReadableStreamGenericReader_h

#include "mozilla/dom/Promise.h"
#include "mozilla/dom/ReadableStreamDefaultReaderBinding.h"
#include "nsISupports.h"
#include "nsCycleCollectionParticipant.h"

namespace mozilla::dom {

class ReadableStream;
class ReadableStreamDefaultReader;
class ReadableStreamBYOBReader;

// Base class for internal slots of readable stream readers
class ReadableStreamGenericReader : public nsISupports {
 public:
  NS_DECL_CYCLE_COLLECTING_ISUPPORTS
  NS_DECL_CYCLE_COLLECTION_SCRIPT_HOLDER_CLASS(ReadableStreamGenericReader)

  explicit ReadableStreamGenericReader(nsCOMPtr<nsIGlobalObject> aGlobal)
      : mGlobal(std::move(aGlobal)) {}

  nsIGlobalObject* GetParentObject() const { return mGlobal; }

  virtual bool IsDefault() = 0;
  virtual bool IsBYOB() = 0;
  virtual ReadableStreamDefaultReader* AsDefault() = 0;
  virtual ReadableStreamBYOBReader* AsBYOB() = 0;

  Promise* ClosedPromise() { return mClosedPromise; }
  void SetClosedPromise(already_AddRefed<Promise>&& aClosedPromise) {
    mClosedPromise = aClosedPromise;
  }

  ReadableStream* GetStream() { return mStream; }
  void SetStream(already_AddRefed<ReadableStream>&& aStream) {
    mStream = aStream;
  }
  void SetStream(ReadableStream* aStream) {
    RefPtr<ReadableStream> stream(aStream);
    SetStream(stream.forget());
  }

  // IDL Methods
  already_AddRefed<Promise> Closed() const;

  MOZ_CAN_RUN_SCRIPT already_AddRefed<Promise> Cancel(
      JSContext* aCx, JS::Handle<JS::Value> aReason, ErrorResult& aRv);

 protected:
  virtual ~ReadableStreamGenericReader() = default;

  nsCOMPtr<nsIGlobalObject> mGlobal;
  RefPtr<Promise> mClosedPromise;
  RefPtr<ReadableStream> mStream;
};

bool ReadableStreamReaderGenericInitialize(ReadableStreamGenericReader* aReader,
                                           ReadableStream* aStream,
                                           ErrorResult& aRv);

void ReadableStreamReaderGenericRelease(ReadableStreamGenericReader* aReader,
                                        ErrorResult& aRv);

}  // namespace mozilla::dom

#endif