#pragma once
#include "ActiveDOMObject.h"
#include "FetchBody.h"
#include "FetchBodySource.h"
#include "FetchHeaders.h"
#include "FetchLoader.h"
#include "FetchLoaderClient.h"
#include "ResourceError.h"
namespace WebCore {
class FetchBodyOwner : public RefCounted<FetchBodyOwner>, public ActiveDOMObject {
public:
FetchBodyOwner(ScriptExecutionContext&, std::optional<FetchBody>&&, Ref<FetchHeaders>&&);
~FetchBodyOwner();
bool bodyUsed() const { return isDisturbed(); }
void arrayBuffer(Ref<DeferredPromise>&&);
void blob(Ref<DeferredPromise>&&);
void formData(Ref<DeferredPromise>&&);
void json(Ref<DeferredPromise>&&);
void text(Ref<DeferredPromise>&&);
bool isDisturbed() const;
bool isDisturbedOrLocked() const;
void loadBlob(const Blob&, FetchBodyConsumer*);
bool isActive() const { return !!m_blobLoader; }
RefPtr<ReadableStream> readableStream(JSC::ExecState&);
bool hasReadableStreamBody() const { return m_body && m_body->hasReadableStream(); }
#if ENABLE(STREAMS_API)
virtual void consumeBodyAsStream();
virtual void feedStream() { }
virtual void cancel() { }
#endif
protected:
const FetchBody& body() const { return *m_body; }
FetchBody& body() { return *m_body; }
bool isBodyNull() const { return !m_body; }
bool isBodyNullOrOpaque() const { return !m_body || m_isBodyOpaque; }
void cloneBody(FetchBodyOwner&);
void extractBody(ScriptExecutionContext&, FetchBody::Init&&);
void updateContentType();
void consumeOnceLoadingFinished(FetchBodyConsumer::Type, Ref<DeferredPromise>&&);
void setBody(FetchBody&& body) { m_body = WTFMove(body); }
void createReadableStream(JSC::ExecState&);
void stop() override;
void setDisturbed() { m_isDisturbed = true; }
void setBodyAsOpaque() { m_isBodyOpaque = true; }
bool isBodyOpaque() const { return m_isBodyOpaque; }
private:
void blobChunk(const char*, size_t);
void blobLoadingSucceeded();
void blobLoadingFailed();
void finishBlobLoading();
struct BlobLoader final : FetchLoaderClient {
BlobLoader(FetchBodyOwner&);
void didReceiveResponse(const ResourceResponse&) final;
void didReceiveData(const char* data, size_t size) final { owner.blobChunk(data, size); }
void didFail(const ResourceError&) final;
void didSucceed() final { owner.blobLoadingSucceeded(); }
FetchBodyOwner& owner;
std::unique_ptr<FetchLoader> loader;
};
protected:
std::optional<FetchBody> m_body;
String m_contentType;
bool m_isDisturbed { false };
#if ENABLE(STREAMS_API)
RefPtr<FetchBodySource> m_readableStreamSource;
#endif
Ref<FetchHeaders> m_headers;
std::optional<ResourceError> m_loadingError;
private:
std::optional<BlobLoader> m_blobLoader;
bool m_isBodyOpaque { false };
};
}