#pragma once
#include "CachedBytecode.h"
#include "CodeSpecializationKind.h"
#include "SourceOrigin.h"
#include <wtf/RefCounted.h>
#include <wtf/URL.h>
#include <wtf/text/TextPosition.h>
#include <wtf/text/WTFString.h>
namespace JSC {
class SourceCode;
class UnlinkedFunctionExecutable;
class UnlinkedFunctionCodeBlock;
enum class SourceProviderSourceType : uint8_t {
Program,
Module,
WebAssembly,
};
using BytecodeCacheGenerator = Function<RefPtr<CachedBytecode>()>;
class SourceProvider : public RefCounted<SourceProvider> {
public:
static const intptr_t nullID = 1;
JS_EXPORT_PRIVATE SourceProvider(const SourceOrigin&, URL&&, const TextPosition& startPosition, SourceProviderSourceType);
JS_EXPORT_PRIVATE virtual ~SourceProvider();
virtual unsigned hash() const = 0;
virtual StringView source() const = 0;
virtual RefPtr<CachedBytecode> cachedBytecode() const { return nullptr; }
virtual void cacheBytecode(const BytecodeCacheGenerator&) const { }
virtual void updateCache(const UnlinkedFunctionExecutable*, const SourceCode&, CodeSpecializationKind, const UnlinkedFunctionCodeBlock*) const { }
virtual void commitCachedBytecode() const { }
StringView getRange(int start, int end) const
{
return source().substring(start, end - start);
}
const SourceOrigin& sourceOrigin() const { return m_sourceOrigin; }
const URL& url() const { return m_url; }
const String& sourceURLDirective() const { return m_sourceURLDirective; }
const String& sourceMappingURLDirective() const { return m_sourceMappingURLDirective; }
TextPosition startPosition() const { return m_startPosition; }
SourceProviderSourceType sourceType() const { return m_sourceType; }
intptr_t asID()
{
if (!m_id)
getID();
return m_id;
}
void setSourceURLDirective(const String& sourceURLDirective) { m_sourceURLDirective = sourceURLDirective; }
void setSourceMappingURLDirective(const String& sourceMappingURLDirective) { m_sourceMappingURLDirective = sourceMappingURLDirective; }
private:
JS_EXPORT_PRIVATE void getID();
SourceProviderSourceType m_sourceType;
URL m_url;
SourceOrigin m_sourceOrigin;
String m_sourceURLDirective;
String m_sourceMappingURLDirective;
TextPosition m_startPosition;
uintptr_t m_id { 0 };
};
DECLARE_ALLOCATOR_WITH_HEAP_IDENTIFIER(StringSourceProvider);
class StringSourceProvider : public SourceProvider {
WTF_MAKE_FAST_ALLOCATED_WITH_HEAP_IDENTIFIER(StringSourceProvider);
public:
static Ref<StringSourceProvider> create(const String& source, const SourceOrigin& sourceOrigin, URL&& url, const TextPosition& startPosition = TextPosition(), SourceProviderSourceType sourceType = SourceProviderSourceType::Program)
{
return adoptRef(*new StringSourceProvider(source, sourceOrigin, WTFMove(url), startPosition, sourceType));
}
unsigned hash() const override
{
return m_source.get().hash();
}
StringView source() const override
{
return m_source.get();
}
protected:
StringSourceProvider(const String& source, const SourceOrigin& sourceOrigin, URL&& url, const TextPosition& startPosition, SourceProviderSourceType sourceType)
: SourceProvider(sourceOrigin, WTFMove(url), startPosition, sourceType)
, m_source(source.isNull() ? *StringImpl::empty() : *source.impl())
{
}
private:
Ref<StringImpl> m_source;
};
#if ENABLE(WEBASSEMBLY)
class WebAssemblySourceProvider : public SourceProvider {
public:
static Ref<WebAssemblySourceProvider> create(Vector<uint8_t>&& data, const SourceOrigin& sourceOrigin, URL&& url)
{
return adoptRef(*new WebAssemblySourceProvider(WTFMove(data), sourceOrigin, WTFMove(url)));
}
unsigned hash() const override
{
return m_source.impl()->hash();
}
StringView source() const override
{
return m_source;
}
const Vector<uint8_t>& data() const
{
return m_data;
}
private:
WebAssemblySourceProvider(Vector<uint8_t>&& data, const SourceOrigin& sourceOrigin, URL&& url)
: SourceProvider(sourceOrigin, WTFMove(url), TextPosition(), SourceProviderSourceType::WebAssembly)
, m_source("[WebAssembly source]")
, m_data(WTFMove(data))
{
}
String m_source;
Vector<uint8_t> m_data;
};
#endif
}