#pragma once
#include "ActiveDOMObject.h"
#include "ExceptionOr.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&, 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; }
ExceptionOr<RefPtr<ReadableStream>> readableStream(JSC::JSGlobalObject&);
bool hasReadableStreamBody() const { return m_body && m_body->hasReadableStream(); }
virtual void consumeBodyAsStream();
virtual void feedStream() { }
virtual void cancel() { }
bool hasLoadingError() const;
ResourceError loadingError() const;
Optional<Exception> loadingException() const;
const String& contentType() const { return m_contentType; }
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&);
ExceptionOr<void> extractBody(FetchBody::Init&&);
void updateContentType();
void consumeOnceLoadingFinished(FetchBodyConsumer::Type, Ref<DeferredPromise>&&);
void setBody(FetchBody&& body) { m_body = WTFMove(body); }
ExceptionOr<void> createReadableStream(JSC::JSGlobalObject&);
void stop() override;
void setDisturbed() { m_isDisturbed = true; }
void setBodyAsOpaque() { m_isBodyOpaque = true; }
bool isBodyOpaque() const { return m_isBodyOpaque; }
void setLoadingError(Exception&&);
void setLoadingError(ResourceError&&);
private:
void blobChunk(const char*, size_t);
void blobLoadingSucceeded();
void blobLoadingFailed();
void finishBlobLoading();
bool virtualHasPendingActivity() const final;
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:
Optional<FetchBody> m_body;
String m_contentType;
bool m_isDisturbed { false };
RefPtr<FetchBodySource> m_readableStreamSource;
Ref<FetchHeaders> m_headers;
private:
Optional<BlobLoader> m_blobLoader;
bool m_isBodyOpaque { false };
Variant<std::nullptr_t, Exception, ResourceError> m_loadingError;
};
}