1 files changed, 77 insertions, 0 deletions
diff --git a/dom/streams/ReadableStreamGenericReader.h b/dom/streams/ReadableStreamGenericReader.h
new file mode 100644
index 0000000000..ccc966bdc4
--- /dev/null
+++ b/dom/streams/ReadableStreamGenericReader.h
@@ -0,0 +1,77 @@
+/* -*- 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;
+};
+
+namespace streams_abstract {
+
+bool ReadableStreamReaderGenericInitialize(ReadableStreamGenericReader* aReader,
+                                           ReadableStream* aStream);
+
+void ReadableStreamReaderGenericRelease(ReadableStreamGenericReader* aReader,
+                                        ErrorResult& aRv);
+
+}  // namespace streams_abstract
+
+}  // namespace mozilla::dom
+
+#endif