diff options
author | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
---|---|---|
committer | Daniel Baumann <daniel.baumann@progress-linux.org> | 2024-04-07 19:33:14 +0000 |
commit | 36d22d82aa202bb199967e9512281e9a53db42c9 (patch) | |
tree | 105e8c98ddea1c1e4784a60a5a6410fa416be2de /dom/streams/ReadableStreamTee.h | |
parent | Initial commit. (diff) | |
download | firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.tar.xz firefox-esr-36d22d82aa202bb199967e9512281e9a53db42c9.zip |
Adding upstream version 115.7.0esr.upstream/115.7.0esr
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'dom/streams/ReadableStreamTee.h')
-rw-r--r-- | dom/streams/ReadableStreamTee.h | 91 |
1 files changed, 91 insertions, 0 deletions
diff --git a/dom/streams/ReadableStreamTee.h b/dom/streams/ReadableStreamTee.h new file mode 100644 index 0000000000..855fcf4286 --- /dev/null +++ b/dom/streams/ReadableStreamTee.h @@ -0,0 +1,91 @@ +/* -*- 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_ReadableStreamTee_h +#define mozilla_dom_ReadableStreamTee_h + +#include "mozilla/dom/ReadRequest.h" +#include "mozilla/dom/ReadableStreamController.h" +#include "mozilla/dom/UnderlyingSourceCallbackHelpers.h" +#include "nsCycleCollectionParticipant.h" +#include "nsISupportsImpl.h" + +namespace mozilla::dom { +struct TeeState; +enum class TeeBranch : bool; +class ReadableStream; + +// Implementation of the Pull algorithm steps for ReadableStreamDefaultTee, +// from +// https://streams.spec.whatwg.org/#abstract-opdef-readablestreamdefaulttee +// Step 12. +class ReadableStreamDefaultTeeSourceAlgorithms final + : public UnderlyingSourceAlgorithmsBase { + public: + NS_DECL_ISUPPORTS_INHERITED + NS_DECL_CYCLE_COLLECTION_CLASS_INHERITED( + ReadableStreamDefaultTeeSourceAlgorithms, UnderlyingSourceAlgorithmsBase) + + ReadableStreamDefaultTeeSourceAlgorithms(TeeState* aTeeState, + TeeBranch aBranch) + : mTeeState(aTeeState), mBranch(aBranch) {} + + MOZ_CAN_RUN_SCRIPT void StartCallback(JSContext* aCx, + ReadableStreamController& aController, + JS::MutableHandle<JS::Value> aRetVal, + ErrorResult& aRv) override { + aRetVal.setUndefined(); + } + + MOZ_CAN_RUN_SCRIPT already_AddRefed<Promise> PullCallback( + JSContext* aCx, ReadableStreamController& aController, + ErrorResult& aRv) override; + + MOZ_CAN_RUN_SCRIPT already_AddRefed<Promise> CancelCallback( + JSContext* aCx, const Optional<JS::Handle<JS::Value>>& aReason, + ErrorResult& aRv) override; + + protected: + ~ReadableStreamDefaultTeeSourceAlgorithms() override = default; + + private: + // Virtually const, but is cycle collected + MOZ_KNOWN_LIVE RefPtr<TeeState> mTeeState; + TeeBranch mBranch; +}; + +// https://streams.spec.whatwg.org/#abstract-opdef-readablestreamdefaulttee +// Step 12.3 +struct ReadableStreamDefaultTeeReadRequest final : public ReadRequest { + NS_DECL_ISUPPORTS_INHERITED + NS_DECL_CYCLE_COLLECTION_CLASS_INHERITED(ReadableStreamDefaultTeeReadRequest, + ReadRequest) + + RefPtr<TeeState> mTeeState; + + explicit ReadableStreamDefaultTeeReadRequest(TeeState* aTeeState) + : mTeeState(aTeeState) {} + + void ChunkSteps(JSContext* aCx, JS::Handle<JS::Value> aChunk, + ErrorResult& aRv) override; + + MOZ_CAN_RUN_SCRIPT void CloseSteps(JSContext* aCx, ErrorResult& aRv) override; + + void ErrorSteps(JSContext* aCx, JS::Handle<JS::Value> aError, + ErrorResult& aRv) override; + + protected: + ~ReadableStreamDefaultTeeReadRequest() override = default; +}; + +namespace streams_abstract { +MOZ_CAN_RUN_SCRIPT void ReadableByteStreamTee( + JSContext* aCx, ReadableStream* aStream, + nsTArray<RefPtr<ReadableStream>>& aResult, ErrorResult& aRv); +} + +} // namespace mozilla::dom +#endif |