Files
bun.sh/src/bun.js/scripts/generate-classes.ts

1771 lines
52 KiB
TypeScript

import { unlinkSync } from "fs";
import { readdirSync } from "fs";
import { resolve } from "path";
import type { Field, ClassDefinition } from "./class-definitions";
const directoriesToSearch = [
resolve(`${import.meta.dir}/../api`),
resolve(`${import.meta.dir}/../test`),
resolve(`${import.meta.dir}/../webcore`),
];
function symbolName(typeName, name) {
return `${typeName}__${name}`;
}
function protoSymbolName(typeName, name) {
return `${typeName}Prototype__${name}`;
}
function classSymbolName(typeName, name) {
return `${typeName}Class__${name}`;
}
function subspaceFor(typeName) {
return `m_subspaceFor${typeName}`;
}
function clientSubspaceFor(typeName) {
return `m_clientSubspaceFor${typeName}`;
}
function prototypeName(typeName) {
return `JS${typeName}Prototype`;
}
function className(typeName) {
return `JS${typeName}`;
}
function constructorName(typeName) {
return `JS${typeName}Constructor`;
}
function DOMJITName(fnName) {
return `${fnName}WithoutTypeChecks`;
}
function argTypeName(arg) {
return {
["bool"]: "bool",
["int"]: "int32_t",
["JSUint8Array"]: "JSC::JSUint8Array*",
["JSString"]: "JSC::JSString*",
["JSValue"]: "JSC::JSValue",
}[arg];
}
function DOMJITType(type) {
return {
["bool"]: "JSC::SpecBoolean",
["int"]: "JSC::SpecInt32Only",
["JSUint8Array"]: "JSC::SpecUint8Array",
["JSString"]: "JSC::SpecString",
["JSValue"]: "JSC::SpecHeapTop",
}[type];
}
function ZigDOMJITArgType(type) {
return {
["bool"]: "bool",
["int"]: "i32",
["JSUint8Array"]: "*JSC.JSUint8Array",
["JSString"]: "*JSC.JSString",
["JSValue"]: "JSC.JSValue",
}[type];
}
function ZigDOMJITFunctionType(thisName, { args, returns }) {
return `fn (*${thisName}, *JSC.JSGlobalObject, ${args
.map(ZigDOMJITArgType)
.join(", ")}) callconv(.C) ${ZigDOMJITArgType("JSValue")}`;
}
function DOMJITReturnType(type) {
return {
["bool"]: "bool",
["int"]: "int32_t",
["JSUint8Array"]: "JSC::JSUint8Array*",
["JSString"]: "JSString*",
["JSValue"]: "EncodedJSValue",
}[type];
}
function DOMJITFunctionDeclaration(jsClassName, fnName, { args, returns }) {
const argNames = args.map((arg, i) => `${argTypeName(arg)} arg${i}`);
return `
extern "C" JSC_DECLARE_JIT_OPERATION_WITHOUT_WTF_INTERNAL(${DOMJITName(
fnName
)}Wrapper, EncodedJSValue, (JSC::JSGlobalObject * lexicalGlobalObject, void* thisValue, ${argNames.join(
", "
)}));
extern "C" EncodedJSValue ${DOMJITName(
fnName
)}(void* ptr, JSC::JSGlobalObject * lexicalGlobalObject, ${argNames.join(
", "
)});
static const JSC::DOMJIT::Signature DOMJITSignatureFor${fnName}(${DOMJITName(
fnName
)}Wrapper,
${jsClassName}::info(),
JSC::DOMJIT::Effect::forReadWrite(JSC::DOMJIT::HeapRange::top(), JSC::DOMJIT::HeapRange::top()),
${DOMJITType("JSValue")}, ${args.map(DOMJITType).join(", ")});
`.trim();
}
function DOMJITFunctionDefinition(jsClassName, fnName, { args }) {
const argNames = args.map((arg, i) => `${argTypeName(arg)} arg${i}`);
return `
JSC_DEFINE_JIT_OPERATION(${DOMJITName(
fnName
)}Wrapper, EncodedJSValue, (JSC::JSGlobalObject * lexicalGlobalObject, void* thisValue, ${argNames.join(
", "
)}))
{
VM& vm = JSC::getVM(lexicalGlobalObject);
IGNORE_WARNINGS_BEGIN("frame-address")
CallFrame* callFrame = DECLARE_CALL_FRAME(vm);
IGNORE_WARNINGS_END
JSC::JITOperationPrologueCallFrameTracer tracer(vm, callFrame);
return ${DOMJITName(
fnName
)}(reinterpret_cast<${jsClassName}*>(thisValue)->wrapped(), lexicalGlobalObject, ${args
.map((b, i) => "arg" + i)
.join(", ")});
}
`;
}
function appendSymbols(
to: Map<string, string>,
symbolName: (name: string) => string,
prop
) {
var { defaultValue, getter, setter, accesosr, fn, DOMJIT, cache } = prop;
if (accesosr) {
getter = accesosr.getter;
setter = accesosr.setter;
}
if (getter && !to.get(getter)) {
to.set(getter, symbolName(getter));
}
if (setter && !to.get(setter)) {
to.set(setter, symbolName(setter));
}
if (fn && !to.get(fn)) {
if (DOMJIT) {
to.set(DOMJITName(fn), symbolName(DOMJITName(fn)));
}
to.set(fn, symbolName(fn));
}
}
function propRow(
symbolName: (a: string, b: string) => string,
typeName: string,
name: string,
prop: Field,
isWrapped = true
) {
var {
defaultValue,
getter,
setter,
fn,
accesosr,
fn,
length = 0,
cache,
DOMJIT,
} = prop;
if (accesosr) {
getter = accesosr.getter;
setter = accesosr.setter;
}
var symbol = symbolName(typeName, name);
if (isWrapped) {
if (getter) {
getter = symbol + "GetterWrap";
}
if (setter) {
setter = symbol + "SetterWrap";
}
if (fn) {
fn = symbol + "Callback";
}
} else {
if (getter) {
getter = symbolName(typeName, getter);
}
if (setter) {
setter = symbolName(typeName, setter);
}
if (fn) {
fn = symbolName(typeName, fn);
}
}
if (fn !== undefined) {
if (DOMJIT) {
// { "getElementById"_s, static_cast<unsigned>(JSC::PropertyAttribute::Function | JSC::PropertyAttribute::DOMJITFunction), NoIntrinsic, { HashTableValue::DOMJITFunctionType, jsTestDOMJITPrototypeFunction_getElementById, &DOMJITSignatureForTestDOMJITGetElementById } },
return `
{ "${name}"_s, static_cast<unsigned>(JSC::PropertyAttribute::Function | JSC::PropertyAttribute::DOMJITFunction), NoIntrinsic, { HashTableValue::DOMJITFunctionType, ${fn}, &DOMJITSignatureFor${symbol} } }
`.trim();
}
return `
{ "${name}"_s, static_cast<unsigned>(JSC::PropertyAttribute::Function), NoIntrinsic, { HashTableValue::NativeFunctionType, ${fn}, ${
length || 0
} } }
`.trim();
} else if (getter && setter) {
return `
{ "${name}"_s, static_cast<unsigned>(JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, ${getter}, ${setter} } }
`.trim();
} else if (defaultValue) {
} else if (getter) {
return `{ "${name}"_s, static_cast<unsigned>(JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, ${getter}, 0 } }
`.trim();
} else if (setter) {
return `{ "${name}"_s, static_cast<unsigned>(JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, 0, ${setter} } }
`.trim();
}
throw "Unsupported property";
}
export function generateHashTable(
nameToUse,
symbolName,
typeName,
obj,
props = {},
wrapped
) {
const rows = [];
for (const name in props) {
if ("internal" in props[name]) continue;
rows.push(propRow(symbolName, typeName, name, props[name], wrapped));
}
// static const HashTableValue JSWebSocketPrototypeTableValues[] = {
// { "constructor"_s, static_cast<unsigned>(JSC::PropertyAttribute::DontEnum), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocketConstructor, 0 } },
// { "URL"_s, static_cast<unsigned>(JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_URL, 0 } },
// { "url"_s, static_cast<unsigned>(JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_url, 0 } },
// { "readyState"_s, static_cast<unsigned>(JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_readyState, 0 } },
// { "bufferedAmount"_s, static_cast<unsigned>(JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_bufferedAmount, 0 } },
// { "onopen"_s, static_cast<unsigned>(JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_onopen, setJSWebSocket_onopen } }, },
// { "onmessage"_s, static_cast<unsigned>(JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_onmessage, setJSWebSocket_onmessage } }, },
// { "onerror"_s, static_cast<unsigned>(JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_onerror, setJSWebSocket_onerror } }, },
// { "onclose"_s, static_cast<unsigned>(JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_onclose, setJSWebSocket_onclose } }, },
// { "protocol"_s, static_cast<unsigned>(JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_protocol, 0 } },
// { "extensions"_s, static_cast<unsigned>(JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_extensions, 0 } },
// { "binaryType"_s, static_cast<unsigned>(JSC::PropertyAttribute::CustomAccessor | JSC::PropertyAttribute::DOMAttribute), NoIntrinsic, { HashTableValue::GetterSetterType, jsWebSocket_binaryType, setJSWebSocket_binaryType } }, },
// { "send"_s, static_cast<unsigned>(JSC::PropertyAttribute::Function), NoIntrinsic, { HashTableValue::NativeFunctionType, jsWebSocketPrototypeFunction_send (intptr_t)(1) } },
// { "close"_s, static_cast<unsigned>(JSC::PropertyAttribute::Function), NoIntrinsic, { HashTableValue::NativeFunctionType, jsWebSocketPrototypeFunction_close (intptr_t)(0) } },
// { "CONNECTING"_s, JSC::PropertyAttribute::DontDelete | JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::ConstantInteger, NoIntrinsic, { HashTableValue::ConstantType, 0 } },
// { "OPEN"_s, JSC::PropertyAttribute::DontDelete | JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::ConstantInteger, NoIntrinsic, { HashTableValue::ConstantType, 1 } },
// { "CLOSING"_s, JSC::PropertyAttribute::DontDelete | JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::ConstantInteger, NoIntrinsic, { HashTableValue::ConstantType, 2 } },
// { "CLOSED"_s, JSC::PropertyAttribute::DontDelete | JSC::PropertyAttribute::ReadOnly | JSC::PropertyAttribute::ConstantInteger, NoIntrinsic, { HashTableValue::ConstantType, 3 } },
// };
return `
static const HashTableValue ${nameToUse}TableValues[] = {
${rows.join(" ,\n")}
};
`;
}
function generatePrototype(typeName, obj) {
const proto = prototypeName(typeName);
const { proto: protoFields } = obj;
return `
${
obj.construct
? `extern "C" void* ${classSymbolName(
typeName,
"construct"
)}(JSC::JSGlobalObject*, JSC::CallFrame*);
JSC_DECLARE_CUSTOM_GETTER(js${typeName}Constructor);`
: ""
}
${
"finalize" in obj
? `extern "C" void ${classSymbolName(typeName, "finalize")}(void*);`
: ""
}
${
obj.call
? `extern "C" JSC_DECLARE_HOST_FUNCTION(${classSymbolName(
typeName,
"call"
)});`
: ""
}
${renderDecls(protoSymbolName, typeName, protoFields)}
STATIC_ASSERT_ISO_SUBSPACE_SHARABLE(${proto}, ${proto}::Base);
${generateHashTable(
prototypeName(typeName),
protoSymbolName,
typeName,
obj,
protoFields,
true
)}
const ClassInfo ${proto}::s_info = { "${typeName}"_s, &Base::s_info, nullptr, nullptr, CREATE_METHOD_TABLE(${proto}) };
${renderFieldsImpl(
protoSymbolName,
typeName,
obj,
protoFields,
obj.values || []
)}
void ${proto}::finishCreation(JSC::VM& vm, JSC::JSGlobalObject* globalObject)
{
Base::finishCreation(vm);
reifyStaticProperties(vm, ${className(
typeName
)}::info(), ${proto}TableValues, *this);
JSC_TO_STRING_TAG_WITHOUT_TRANSITION();
}
`;
}
function generateConstructorHeader(typeName) {
const name = constructorName(typeName);
const proto = prototypeName(typeName);
return `
class ${proto} final : public JSC::JSNonFinalObject {
public:
using Base = JSC::JSNonFinalObject;
static ${proto}* create(JSC::VM& vm, JSGlobalObject* globalObject, JSC::Structure* structure)
{
${proto}* ptr = new (NotNull, JSC::allocateCell<${proto}>(vm)) ${proto}(vm, globalObject, structure);
ptr->finishCreation(vm, globalObject);
return ptr;
}
DECLARE_INFO;
template<typename CellType, JSC::SubspaceAccess>
static JSC::GCClient::IsoSubspace* subspaceFor(JSC::VM& vm)
{
return &vm.plainObjectSpace();
}
static JSC::Structure* createStructure(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::JSValue prototype)
{
return JSC::Structure::create(vm, globalObject, prototype, JSC::TypeInfo(JSC::ObjectType, StructureFlags), info());
}
private:
${proto}(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::Structure* structure)
: Base(vm, structure)
{
}
void finishCreation(JSC::VM&, JSC::JSGlobalObject*);
};
class ${name} final : public JSC::InternalFunction {
public:
using Base = JSC::InternalFunction;
static ${name}* create(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::Structure* structure, ${prototypeName(
typeName
)}* prototype);
static constexpr unsigned StructureFlags = Base::StructureFlags;
static constexpr bool needsDestruction = false;
static JSC::Structure* createStructure(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::JSValue prototype)
{
return JSC::Structure::create(vm, globalObject, prototype, JSC::TypeInfo(JSC::InternalFunctionType, StructureFlags), info());
}
template<typename, JSC::SubspaceAccess mode> static JSC::GCClient::IsoSubspace* subspaceFor(JSC::VM& vm)
{
if constexpr (mode == JSC::SubspaceAccess::Concurrently)
return nullptr;
return WebCore::subspaceForImpl<${name}, WebCore::UseCustomHeapCellType::No>(
vm,
[](auto& spaces) { return spaces.${clientSubspaceFor(
typeName
)}Constructor.get(); },
[](auto& spaces, auto&& space) { spaces.${clientSubspaceFor(
typeName
)}Constructor = WTFMove(space); },
[](auto& spaces) { return spaces.${subspaceFor(
typeName
)}Constructor.get(); },
[](auto& spaces, auto&& space) { spaces.${subspaceFor(
typeName
)}Constructor = WTFMove(space); });
}
void initializeProperties(JSC::VM& vm, JSC::JSGlobalObject* globalObject, ${prototypeName(
typeName
)}* prototype);
// Must be defined for each specialization class.
static JSC::EncodedJSValue JSC_HOST_CALL_ATTRIBUTES construct(JSC::JSGlobalObject*, JSC::CallFrame*);
DECLARE_EXPORT_INFO;
private:
${name}(JSC::VM& vm, JSC::Structure* structure);
void finishCreation(JSC::VM&, JSC::JSGlobalObject* globalObject, ${prototypeName(
typeName
)}* prototype);
};
`;
}
function generateConstructorImpl(typeName, obj) {
const name = constructorName(typeName);
const { klass: fields } = obj;
const hashTable =
Object.keys(fields).length > 0
? generateHashTable(name, classSymbolName, typeName, obj, fields, false)
: "";
const hashTableIdentifier = hashTable.length ? `${name}TableValues` : "";
return `
${
obj.estimatedSize
? `extern "C" size_t ${symbolName(typeName, "estimatedSize")}(void* ptr);`
: ""
}
${renderStaticDecls(classSymbolName, typeName, fields)}
${hashTable}
void ${name}::finishCreation(VM& vm, JSC::JSGlobalObject* globalObject, ${prototypeName(
typeName
)}* prototype)
{
Base::finishCreation(vm, 0, "${typeName}"_s, PropertyAdditionMode::WithoutStructureTransition);
${
hashTableIdentifier.length
? `reifyStaticProperties(vm, &${name}::s_info, ${hashTableIdentifier}, *this);`
: ""
}
putDirectWithoutTransition(vm, vm.propertyNames->prototype, prototype, PropertyAttribute::DontEnum | PropertyAttribute::DontDelete | PropertyAttribute::ReadOnly);
ASSERT(inherits(info()));
}
${name}::${name}(JSC::VM& vm, JSC::Structure* structure) : Base(vm, structure, ${
obj.call ? classSymbolName(typeName, "call") : "construct"
}, construct) {
}
${name}* ${name}::create(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::Structure* structure, ${prototypeName(
typeName
)}* prototype) {
${name}* ptr = new (NotNull, JSC::allocateCell<${name}>(vm)) ${name}(vm, structure);
ptr->finishCreation(vm, globalObject, prototype);
return ptr;
}
JSC::EncodedJSValue JSC_HOST_CALL_ATTRIBUTES ${name}::construct(JSC::JSGlobalObject* lexicalGlobalObject, JSC::CallFrame* callFrame)
{
Zig::GlobalObject *globalObject = reinterpret_cast<Zig::GlobalObject*>(lexicalGlobalObject);
JSC::VM &vm = globalObject->vm();
JSObject* newTarget = asObject(callFrame->newTarget());
auto* constructor = globalObject->${className(typeName)}Constructor();
Structure* structure = globalObject->${className(typeName)}Structure();
if (constructor != newTarget) {
auto scope = DECLARE_THROW_SCOPE(vm);
auto* functionGlobalObject = reinterpret_cast<Zig::GlobalObject*>(
// ShadowRealm functions belong to a different global object.
getFunctionRealm(globalObject, newTarget)
);
RETURN_IF_EXCEPTION(scope, {});
structure = InternalFunction::createSubclassStructure(
globalObject,
newTarget,
functionGlobalObject->${className(typeName)}Structure()
);
}
void* ptr = ${classSymbolName(
typeName,
"construct"
)}(globalObject, callFrame);
if (UNLIKELY(!ptr)) {
return JSValue::encode(JSC::jsUndefined());
}
${className(typeName)}* instance = ${className(
typeName
)}::create(vm, globalObject, structure, ptr);
${
obj.estimatedSize
? `vm.heap.reportExtraMemoryAllocated(${symbolName(
obj.name,
"estimatedSize"
)}(instance->wrapped()));`
: ""
}
return JSValue::encode(instance);
}
extern "C" EncodedJSValue ${typeName}__create(Zig::GlobalObject* globalObject, void* ptr) {
auto &vm = globalObject->vm();
JSC::Structure* structure = globalObject->${className(typeName)}Structure();
${className(typeName)}* instance = ${className(
typeName
)}::create(vm, globalObject, structure, ptr);
${
obj.estimatedSize
? `vm.heap.reportExtraMemoryAllocated(${symbolName(
obj.name,
"estimatedSize"
)}(ptr));`
: ""
}
return JSValue::encode(instance);
}
void ${name}::initializeProperties(VM& vm, JSC::JSGlobalObject* globalObject, ${prototypeName(
typeName
)}* prototype)
{
}
const ClassInfo ${name}::s_info = { "Function"_s, &Base::s_info, nullptr, nullptr, CREATE_METHOD_TABLE(${name}) };
${
!obj.noConstructor
? `
extern "C" EncodedJSValue ${typeName}__getConstructor(Zig::GlobalObject* globalObject) {
return JSValue::encode(globalObject->${className(typeName)}Constructor());
}`
: ""
}
`;
}
function renderCachedFieldsHeader(typeName, klass, proto, values) {
const rows: string[] = [];
for (const name in klass) {
if (
("cache" in klass[name] && klass[name].cache === true) ||
klass[name]?.internal
) {
rows.push(`mutable JSC::WriteBarrier<JSC::Unknown> m_${name};`);
}
}
for (const name in proto) {
if (proto[name]?.cache === true || klass[name]?.internal) {
rows.push(`mutable JSC::WriteBarrier<JSC::Unknown> m_${name};`);
}
}
for (const name of values) {
rows.push(`mutable JSC::WriteBarrier<JSC::Unknown> m_${name};`);
}
return rows.join("\n");
}
function renderDecls(symbolName, typeName, proto) {
const rows = [];
for (const name in proto) {
if (
"getter" in proto[name] ||
("accessor" in proto[name] && proto[name].getter)
) {
rows.push(
`extern "C" JSC::EncodedJSValue ${symbolName(
typeName,
proto[name].getter || proto[name].accessor.getter
)}(void* ptr, JSC::JSGlobalObject* lexicalGlobalObject);`,
`
JSC_DECLARE_CUSTOM_GETTER(${symbolName(typeName, name)}GetterWrap);
`.trim(),
"\n"
);
}
if (
"setter" in proto[name] ||
("accessor" in proto[name] && proto[name].setter)
) {
rows.push(
`extern "C" bool ${symbolName(
typeName,
proto[name].setter || proto[name].accessor.setter
)}(void* ptr, JSC::JSGlobalObject* lexicalGlobalObject, JSC::EncodedJSValue value);`,
`
JSC_DECLARE_CUSTOM_SETTER(${symbolName(typeName, name)}SetterWrap);
`.trim(),
"\n"
);
}
if ("fn" in proto[name]) {
rows.push(
`extern "C" EncodedJSValue ${symbolName(
typeName,
proto[name].fn
)}(void* ptr, JSC::JSGlobalObject* lexicalGlobalObject, JSC::CallFrame* callFrame);`,
`
JSC_DECLARE_HOST_FUNCTION(${symbolName(typeName, name)}Callback);
`.trim(),
"\n"
);
if (proto[name].DOMJIT) {
rows.push(
DOMJITFunctionDeclaration(
className(typeName),
symbolName(typeName, name),
proto[name].DOMJIT
),
DOMJITFunctionDefinition(
className(typeName),
symbolName(typeName, name),
proto[name].DOMJIT
)
);
}
}
}
return rows.join("\n");
}
function renderStaticDecls(symbolName, typeName, fields) {
const rows = [];
for (const name in fields) {
if (
"getter" in fields[name] ||
("accessor" in fields[name] && fields[name].getter)
) {
rows.push(
`extern "C" JSC_DECLARE_CUSTOM_GETTER(${symbolName(
typeName,
fields[name].getter || fields[name].accessor.getter
)});`
);
}
if (
"setter" in fields[name] ||
("accessor" in fields[name] && fields[name].setter)
) {
rows.push(
`extern "C" JSC_DECLARE_CUSTOM_SETTER(${symbolName(
typeName,
fields[name].setter || fields[name].accessor.setter
)});`
);
}
if ("fn" in fields[name]) {
rows.push(
`extern "C" JSC_DECLARE_HOST_FUNCTION(${symbolName(
typeName,
fields[name].fn
)});`
);
}
}
return rows.join("\n");
}
function writeBarrier(symbolName, typeName, name, cacheName) {
return `
extern "C" void ${symbolName(
typeName,
name
)}SetCachedValue(JSC::EncodedJSValue thisValue, JSC::JSGlobalObject *globalObject, JSC::EncodedJSValue value)
{
auto& vm = globalObject->vm();
auto* thisObject = jsCast<${className(
typeName
)}*>(JSValue::decode(thisValue));
thisObject->${cacheName}.set(vm, thisObject, JSValue::decode(value));
}
extern "C" EncodedJSValue ${symbolName(
typeName,
name
)}GetCachedValue(JSC::EncodedJSValue thisValue)
{
auto* thisObject = jsCast<${className(
typeName
)}*>(JSValue::decode(thisValue));
return JSValue::encode(thisObject->${cacheName}.get());
}
`;
}
function renderFieldsImpl(
symbolName: (typeName: string, name: string) => string,
typeName: string,
obj: ClassDefinition,
proto: ClassDefinition["proto"],
cachedValues: string[]
) {
const rows: string[] = [];
if (obj.construct) {
rows.push(`
JSC_DEFINE_CUSTOM_GETTER(js${typeName}Constructor, (JSGlobalObject * lexicalGlobalObject, EncodedJSValue thisValue, PropertyName))
{
VM& vm = JSC::getVM(lexicalGlobalObject);
auto throwScope = DECLARE_THROW_SCOPE(vm);
auto* globalObject = reinterpret_cast<Zig::GlobalObject*>(lexicalGlobalObject);
auto* prototype = jsDynamicCast<${prototypeName(
typeName
)}*>(JSValue::decode(thisValue));
if (UNLIKELY(!prototype))
return throwVMTypeError(lexicalGlobalObject, throwScope);
return JSValue::encode(globalObject->${className(typeName)}Constructor());
}
`);
}
for (const name in proto) {
if ("cache" in proto[name] || proto[name]?.internal) {
const cacheName =
typeof proto[name].cache === "string"
? `m_${proto[name].cache}`
: `m_${name}`;
if ("cache" in proto[name]) {
rows.push(`
JSC_DEFINE_CUSTOM_GETTER(${symbolName(
typeName,
name
)}GetterWrap, (JSGlobalObject * lexicalGlobalObject, EncodedJSValue thisValue, PropertyName attributeName))
{
auto& vm = lexicalGlobalObject->vm();
Zig::GlobalObject *globalObject = reinterpret_cast<Zig::GlobalObject*>(lexicalGlobalObject);
auto throwScope = DECLARE_THROW_SCOPE(vm);
${className(typeName)}* thisObject = jsCast<${className(
typeName
)}*>(JSValue::decode(thisValue));
JSC::EnsureStillAliveScope thisArg = JSC::EnsureStillAliveScope(thisObject);
if (JSValue cachedValue = thisObject->${cacheName}.get())
return JSValue::encode(cachedValue);
JSC::JSValue result = JSC::JSValue::decode(
${symbolName(
typeName,
proto[name].getter
)}(thisObject->wrapped(), globalObject)
);
RETURN_IF_EXCEPTION(throwScope, {});
thisObject->${cacheName}.set(vm, thisObject, result);
RELEASE_AND_RETURN(throwScope, JSValue::encode(result));
}`);
}
rows.push(writeBarrier(symbolName, typeName, name, cacheName));
} else if (
"getter" in proto[name] ||
("accessor" in proto[name] && proto[name].getter)
) {
rows.push(`
JSC_DEFINE_CUSTOM_GETTER(${symbolName(
typeName,
name
)}GetterWrap, (JSGlobalObject * lexicalGlobalObject, EncodedJSValue thisValue, PropertyName attributeName))
{
auto& vm = lexicalGlobalObject->vm();
Zig::GlobalObject *globalObject = reinterpret_cast<Zig::GlobalObject*>(lexicalGlobalObject);
auto throwScope = DECLARE_THROW_SCOPE(vm);
${className(typeName)}* thisObject = jsCast<${className(
typeName
)}*>(JSValue::decode(thisValue));
JSC::EnsureStillAliveScope thisArg = JSC::EnsureStillAliveScope(thisObject);
JSC::EncodedJSValue result = ${symbolName(
typeName,
proto[name].getter
)}(thisObject->wrapped(), globalObject);
RETURN_IF_EXCEPTION(throwScope, {});
RELEASE_AND_RETURN(throwScope, result);
}
`);
}
if (
"setter" in proto[name] ||
("accessor" in proto[name] && proto[name].setter)
) {
rows.push(
`
JSC_DEFINE_CUSTOM_SETTER(${symbolName(
typeName,
name
)}SetterWrap, (JSGlobalObject * lexicalGlobalObject, EncodedJSValue thisValue, EncodedJSValue encodedValue, PropertyName attributeName))
{
auto& vm = lexicalGlobalObject->vm();
auto throwScope = DECLARE_THROW_SCOPE(vm);
${className(typeName)}* thisObject = jsCast<${className(
typeName
)}*>(JSValue::decode(thisValue));
JSC::EnsureStillAliveScope thisArg = JSC::EnsureStillAliveScope(thisObject);
auto result = ${symbolName(
typeName,
proto[name].setter || proto[name].accessor.setter
)}(thisObject->wrapped(), lexicalGlobalObject, encodedValue);
RELEASE_AND_RETURN(throwScope, result);
}
`
);
}
if ("fn" in proto[name]) {
rows.push(`
JSC_DEFINE_HOST_FUNCTION(${symbolName(
typeName,
name
)}Callback, (JSGlobalObject * lexicalGlobalObject, CallFrame* callFrame))
{
auto& vm = lexicalGlobalObject->vm();
${className(typeName)}* thisObject = jsDynamicCast<${className(
typeName
)}*>(callFrame->thisValue());
if (UNLIKELY(!thisObject)) {
auto throwScope = DECLARE_THROW_SCOPE(vm);
return throwVMTypeError(lexicalGlobalObject, throwScope);
}
JSC::EnsureStillAliveScope thisArg = JSC::EnsureStillAliveScope(thisObject);
return ${symbolName(
typeName,
proto[name].fn
)}(thisObject->wrapped(), lexicalGlobalObject, callFrame);
}
`);
}
}
if (cachedValues?.length) {
for (const cacheName of cachedValues) {
rows.push(
writeBarrier(symbolName, typeName, cacheName, "m_" + cacheName)
);
}
}
return rows.join("\n");
}
function generateClassHeader(typeName, obj: ClassDefinition) {
var { klass, proto, JSType = "ObjectType", values = [] } = obj;
const name = className(typeName);
const DECLARE_VISIT_CHILDREN =
values.length ||
obj.estimatedSize ||
[...Object.values(klass), ...Object.values(proto)].find((a) => !!a.cache)
? "DECLARE_VISIT_CHILDREN;"
: "";
const sizeEstimator = obj.estimatedSize
? "static size_t estimatedSize(JSCell* cell, VM& vm);"
: "";
var weakOwner = "";
var weakInit = "";
if (obj.hasPendingActivity) {
weakOwner = `
JSC::Weak<${name}> m_weakThis;
bool internalHasPendingActivity();
bool hasPendingActivity() {
if (!m_ctx)
return false;
return this->internalHasPendingActivity();
}
class Owner final : public JSC::WeakHandleOwner {
public:
bool isReachableFromOpaqueRoots(JSC::Handle<JSC::Unknown> handle, void* context, JSC::AbstractSlotVisitor&, const char**) final
{
auto* controller = JSC::jsCast<${name}*>(handle.slot()->asCell());
return controller->hasPendingActivity();
}
void finalize(JSC::Handle<JSC::Unknown>, void* context) final {}
};
static JSC::WeakHandleOwner* getOwner()
{
static NeverDestroyed<Owner> m_owner;
return &m_owner.get();
}
`;
weakInit = `
m_weakThis = JSC::Weak<${name}>(vm, this, getOwner());
`;
}
return `
class ${name} final : public JSC::JSDestructibleObject {
public:
using Base = JSC::JSDestructibleObject;
static ${name}* create(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::Structure* structure, void* ctx);
DECLARE_EXPORT_INFO;
template<typename, JSC::SubspaceAccess mode> static JSC::GCClient::IsoSubspace* subspaceFor(JSC::VM& vm)
{
if constexpr (mode == JSC::SubspaceAccess::Concurrently)
return nullptr;
return WebCore::subspaceForImpl<${name}, WebCore::UseCustomHeapCellType::No>(
vm,
[](auto& spaces) { return spaces.${clientSubspaceFor(
typeName
)}.get(); },
[](auto& spaces, auto&& space) { spaces.${clientSubspaceFor(
typeName
)} = WTFMove(space); },
[](auto& spaces) { return spaces.${subspaceFor(
typeName
)}.get(); },
[](auto& spaces, auto&& space) { spaces.${subspaceFor(
typeName
)} = WTFMove(space); });
}
static void destroy(JSC::JSCell*);
static JSC::Structure* createStructure(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::JSValue prototype)
{
return JSC::Structure::create(vm, globalObject, prototype, JSC::TypeInfo(static_cast<JSC::JSType>(${JSType}), StructureFlags), info());
}
static JSObject* createPrototype(VM& vm, JSDOMGlobalObject* globalObject);
static JSObject* createConstructor(VM& vm, JSGlobalObject* globalObject, JSValue prototype);
~${name}();
void* wrapped() const { return m_ctx; }
void detach()
{
m_ctx = nullptr;
}
static void analyzeHeap(JSCell*, JSC::HeapAnalyzer&);
static ptrdiff_t offsetOfWrapped() { return OBJECT_OFFSETOF(${name}, m_ctx); }
void* m_ctx { nullptr };
${name}(JSC::VM& vm, JSC::Structure* structure, void* sinkPtr)
: Base(vm, structure)
{
m_ctx = sinkPtr;
${weakInit.trim()}
}
void finishCreation(JSC::VM&);
${weakOwner}
${DECLARE_VISIT_CHILDREN}
${renderCachedFieldsHeader(typeName, klass, proto, values)}
};
`;
}
function generateClassImpl(typeName, obj: ClassDefinition) {
const {
klass: fields,
finalize,
proto,
construct,
estimatedSize,
hasPendingActivity = false,
} = obj;
const name = className(typeName);
const DEFINE_VISIT_CHILDREN_LIST = [
...Object.entries(fields),
...Object.entries(proto),
]
.filter(
([name, { cache = false, internal = false }]) =>
(cache || internal) === true
)
.map(([name]) => ` visitor.append(thisObject->m_${name});`)
.join("\n");
const values = (obj.values || [])
.map((val) => {
return `visitor.append(thisObject->m_${val});`;
})
.join("\n");
var DEFINE_VISIT_CHILDREN = "";
if (DEFINE_VISIT_CHILDREN_LIST.length || estimatedSize || values.length) {
DEFINE_VISIT_CHILDREN = `
template<typename Visitor>
void ${name}::visitChildrenImpl(JSCell* cell, Visitor& visitor)
{
${name}* thisObject = jsCast<${name}*>(cell);
ASSERT_GC_OBJECT_INHERITS(thisObject, info());
Base::visitChildren(thisObject, visitor);
${values}
${
estimatedSize
? `if (auto* ptr = thisObject->wrapped()) {
visitor.reportExtraMemoryVisited(${symbolName(obj.name, "estimatedSize")}(ptr));
}`
: ""
}
${DEFINE_VISIT_CHILDREN_LIST}
}
DEFINE_VISIT_CHILDREN(${name});
`.trim();
}
var output = ``;
if (hasPendingActivity) {
output += `
extern "C" bool ${symbolName(typeName, "hasPendingActivity")}(void* ptr);
${name}::internalHasPendingActivity() {
return ${symbolName(typeName, "hasPendingActivity")}(m_ctx);
}
`;
}
if (finalize) {
output += `
${name}::~${name}()
{
if (m_ctx) {
${classSymbolName(typeName, "finalize")}(m_ctx);
}
}
`;
} else {
output += `
${name}::~${name}()
{
}
`;
}
output += `
void ${name}::destroy(JSCell* cell)
{
static_cast<${name}*>(cell)->${name}::~${name}();
}
const ClassInfo ${name}::s_info = { "${typeName}"_s, &Base::s_info, nullptr, nullptr, CREATE_METHOD_TABLE(${name}) };
void ${name}::finishCreation(VM& vm)
{
Base::finishCreation(vm);
ASSERT(inherits(info()));
}
${name}* ${name}::create(JSC::VM& vm, JSC::JSGlobalObject* globalObject, JSC::Structure* structure, void* ctx) {
${name}* ptr = new (NotNull, JSC::allocateCell<${name}>(vm)) ${name}(vm, structure, ctx);
ptr->finishCreation(vm);
return ptr;
}
extern "C" void* ${typeName}__fromJS(JSC::EncodedJSValue value) {
JSC::JSValue decodedValue = JSC::JSValue::decode(value);
if (!decodedValue || decodedValue.isUndefinedOrNull())
return nullptr;
${className(typeName)}* object = JSC::jsDynamicCast<${className(
typeName
)}*>(decodedValue);
if (!object)
return nullptr;
return object->wrapped();
}
extern "C" bool ${typeName}__dangerouslySetPtr(JSC::EncodedJSValue value, void* ptr) {
${className(typeName)}* object = JSC::jsDynamicCast<${className(
typeName
)}*>(JSValue::decode(value));
if (!object)
return false;
object->m_ctx = ptr;
return true;
}
extern "C" const size_t ${typeName}__ptrOffset = ${className(
typeName
)}::offsetOfWrapped();
void ${name}::analyzeHeap(JSCell* cell, HeapAnalyzer& analyzer)
{
auto* thisObject = jsCast<${name}*>(cell);
if (void* wrapped = thisObject->wrapped()) {
// if (thisObject->scriptExecutionContext())
// analyzer.setLabelForCell(cell, "url " + thisObject->scriptExecutionContext()->url().string());
}
Base::analyzeHeap(cell, analyzer);
}
JSObject* ${name}::createConstructor(VM& vm, JSGlobalObject* globalObject, JSValue prototype)
{
return WebCore::${constructorName(
typeName
)}::create(vm, globalObject, WebCore::${constructorName(
typeName
)}::createStructure(vm, globalObject, globalObject->functionPrototype()), jsCast<WebCore::${prototypeName(
typeName
)}*>(prototype));
}
JSObject* ${name}::createPrototype(VM& vm, JSDOMGlobalObject* globalObject)
{
return ${prototypeName(typeName)}::create(vm, globalObject, ${prototypeName(
typeName
)}::createStructure(vm, globalObject, globalObject->objectPrototype()));
}
${DEFINE_VISIT_CHILDREN}
`.trim();
return output;
}
function generateHeader(typeName, obj) {
return generateClassHeader(typeName, obj).trim();
}
function generateImpl(typeName, obj) {
const proto = obj.proto;
return [
generateConstructorHeader(typeName).trim() + "\n",
Object.keys(proto).length > 0 && generatePrototype(typeName, obj).trim(),
!obj.noConstructor ? generateConstructorImpl(typeName, obj).trim() : null,
Object.keys(proto).length > 0 && generateClassImpl(typeName, obj).trim(),
]
.filter(Boolean)
.join("\n\n");
}
function generateZig(
typeName,
{
klass = {},
proto = {},
construct,
finalize,
noConstructor,
estimatedSize,
call = false,
values = [],
hasPendingActivity = false,
} = {} as ClassDefinition
) {
const exports = new Map<string, string>();
if (construct) {
exports.set(`constructor`, classSymbolName(typeName, "construct"));
}
if (call) {
exports.set(`call`, classSymbolName(typeName, "call"));
}
if (finalize) {
exports.set(`finalize`, classSymbolName(typeName, "finalize"));
}
if (estimatedSize) {
exports.set(`estimatedSize`, symbolName(typeName, "estimatedSize"));
}
if (hasPendingActivity) {
exports.set(
"hasPendingActivity",
symbolName(typeName, "hasPendingActivity")
);
}
Object.values(klass).map((a) =>
appendSymbols(exports, (name) => classSymbolName(typeName, name), a)
);
Object.values(proto).map((a) =>
appendSymbols(exports, (name) => protoSymbolName(typeName, name), a)
);
const externs = Object.entries({
...proto,
...Object.fromEntries((values || []).map((a) => [a, { internal: true }])),
})
.filter(
([name, { cache, internal }]) =>
(cache && typeof cache !== "string") || internal
)
.map(
([name]) =>
`extern fn ${protoSymbolName(
typeName,
name
)}SetCachedValue(JSC.JSValue, *JSC.JSGlobalObject, JSC.JSValue) void;
extern fn ${protoSymbolName(
typeName,
name
)}GetCachedValue(JSC.JSValue) JSC.JSValue;
/// \`${typeName}.${name}\` setter
/// This value will be visited by the garbage collector.
pub fn ${name}SetCached(thisValue: JSC.JSValue, globalObject: *JSC.JSGlobalObject, value: JSC.JSValue) void {
JSC.markBinding(@src());
${protoSymbolName(
typeName,
name
)}SetCachedValue(thisValue, globalObject, value);
}
/// \`${typeName}.${name}\` getter
/// This value will be visited by the garbage collector.
pub fn ${name}GetCached(thisValue: JSC.JSValue) ?JSC.JSValue {
JSC.markBinding(@src());
const result = ${protoSymbolName(
typeName,
name
)}GetCachedValue(thisValue);
if (result == .zero)
return null;
return result;
}
`.trim() + "\n"
)
.join("\n");
function typeCheck() {
var output = "";
if (estimatedSize) {
output += `
if (@TypeOf(${typeName}.estimatedSize) != (fn(*${typeName}) callconv(.C) usize)) {
@compileLog("${typeName}.estimatedSize is not a size function");
}
`;
}
if (construct) {
output += `
if (@TypeOf(${typeName}.constructor) != (fn(*JSC.JSGlobalObject, *JSC.CallFrame) callconv(.C) ?*${typeName})) {
@compileLog("${typeName}.constructor is not a constructor");
}
`;
}
if (finalize) {
output += `
if (@TypeOf(${typeName}.finalize) != (fn(*${typeName}) callconv(.C) void)) {
@compileLog("${typeName}.finalize is not a finalizer");
}
`;
}
[...Object.values(proto)].forEach(
({ getter, setter, accessor, fn, cache, DOMJIT }) => {
if (accessor) {
getter = accessor.getter;
setter = accessor.setter;
}
if (getter) {
output += `
if (@TypeOf(${typeName}.${getter}) != GetterType)
@compileLog(
"Expected ${typeName}.${getter} to be a getter"
);
`;
}
if (setter) {
output += `
if (@TypeOf(${typeName}.${setter}) != SetterType)
@compileLog(
"Expected ${typeName}.${setter} to be a setter"
);`;
}
if (fn) {
if (DOMJIT) {
output += `
if (@TypeOf(${typeName}.${DOMJITName(fn)}) != ${ZigDOMJITFunctionType(
typeName,
DOMJIT
)})
@compileLog(
"Expected ${typeName}.${DOMJITName(fn)} to be a DOMJIT function"
);`;
}
output += `
if (@TypeOf(${typeName}.${fn}) != CallbackType)
@compileLog(
"Expected ${typeName}.${fn} to be a callback"
);`;
}
}
);
[...Object.values(klass)].forEach(({ getter, setter, accessor, fn }) => {
if (accessor) {
getter = accessor.getter;
setter = accessor.setter;
}
if (getter) {
output += `
if (@TypeOf(${typeName}.${getter}) != StaticGetterType)
@compileLog(
"Expected ${typeName}.${getter} to be a static getter"
);
`;
}
if (setter) {
output += `
if (@TypeOf(${typeName}.${setter}) != StaticSetterType)
@compileLog(
"Expected ${typeName}.${setter} to be a static setter"
);`;
}
if (fn) {
output += `
if (@TypeOf(${typeName}.${fn}) != StaticCallbackType)
@compileLog(
"Expected ${typeName}.${fn} to be a static callback"
);`;
}
});
if (!!call) {
output += `
if (@TypeOf(${typeName}.call) != StaticCallbackType)
@compileLog(
"Expected ${typeName}.call to be a static callback"
);`;
}
return output;
}
return `
pub const ${className(typeName)} = struct {
const ${typeName} = Classes.${typeName};
const GetterType = fn(*${typeName}, *JSC.JSGlobalObject) callconv(.C) JSC.JSValue;
const SetterType = fn(*${typeName}, *JSC.JSGlobalObject, JSC.JSValue) callconv(.C) bool;
const CallbackType = fn(*${typeName}, *JSC.JSGlobalObject, *JSC.CallFrame) callconv(.C) JSC.JSValue;
/// Return the pointer to the wrapped object.
/// If the object does not match the type, return null.
pub fn fromJS(value: JSC.JSValue) ?*${typeName} {
JSC.markBinding(@src());
return ${symbolName(typeName, "fromJS")}(value);
}
${externs}
${
!noConstructor
? `
/// Get the ${typeName} constructor value.
/// This loads lazily from the global object.
pub fn getConstructor(globalObject: *JSC.JSGlobalObject) JSC.JSValue {
JSC.markBinding(@src());
return ${symbolName(typeName, "getConstructor")}(globalObject);
}
`
: ""
}
/// Create a new instance of ${typeName}
pub fn toJS(this: *${typeName}, globalObject: *JSC.JSGlobalObject) JSC.JSValue {
JSC.markBinding(@src());
if (comptime Environment.allow_assert) {
const value__ = ${symbolName(
typeName,
"create"
)}(globalObject, this);
std.debug.assert(value__.as(${typeName}).? == this); // If this fails, likely a C ABI issue.
return value__;
} else {
return ${symbolName(typeName, "create")}(globalObject, this);
}
}
/// Modify the internal ptr to point to a new instance of ${typeName}.
pub fn dangerouslySetPtr(value: JSC.JSValue, ptr: ?*${typeName}) bool {
JSC.markBinding(@src());
return ${symbolName(typeName, "dangerouslySetPtr")}(value, ptr);
}
/// Detach the ptr from the thisValue
pub fn detachPtr(_: *${typeName}, value: JSC.JSValue) void {
JSC.markBinding(@src());
std.debug.assert(${symbolName(
typeName,
"dangerouslySetPtr"
)}(value, null));
}
extern fn ${symbolName(typeName, "fromJS")}(JSC.JSValue) ?*${typeName};
extern fn ${symbolName(
typeName,
"getConstructor"
)}(*JSC.JSGlobalObject) JSC.JSValue;
extern fn ${symbolName(
typeName,
"create"
)}(globalObject: *JSC.JSGlobalObject, ptr: ?*${typeName}) JSC.JSValue;
extern fn ${typeName}__dangerouslySetPtr(JSC.JSValue, ?*${typeName}) bool;
comptime {
${typeCheck()}
if (!JSC.is_bindgen) {
${[...exports]
.sort(([a], [b]) => a.localeCompare(b))
.map(
([internalName, externalName]) =>
`@export(${typeName}.${internalName}, .{.name = "${externalName}"});`
)
.join("\n ")}
}
}
};
`;
}
function generateLazyClassStructureHeader(
typeName,
{ klass = {}, proto = {} }
) {
return `
JSC::Structure* ${className(
typeName
)}Structure() { return m_${className(
typeName
)}.getInitializedOnMainThread(this); }
JSC::JSObject* ${className(
typeName
)}Constructor() { return m_${className(
typeName
)}.constructorInitializedOnMainThread(this); }
JSC::JSValue ${className(typeName)}Prototype() { return m_${className(
typeName
)}.prototypeInitializedOnMainThread(this); }
JSC::LazyClassStructure m_${className(typeName)};
bool has${className(typeName)}SetterValue { false };
mutable JSC::WriteBarrier<JSC::Unknown> m_${className(typeName)}SetterValue;
`.trim();
}
function generateLazyStructureHeader(typeName, { klass = {}, proto = {} }) {
return `
JSC::Structure* ${className(
typeName
)}Structure() { return m_${className(typeName)}.get(this); }
JSC::LazyProperty<Zig::GlobalObject, Structure> m_${className(typeName)};
bool has${className(typeName)}SetterValue { false };
mutable JSC::WriteBarrier<JSC::Unknown> m_${className(typeName)}SetterValue;
`.trim();
}
function generateLazyStructureImpl(typeName, { klass = {}, proto = {} }) {
return `
m_${className(typeName)}.initLater(
[](const JSC::LazyProperty<JSC::JSGlobalObject, JSC::JSObject>::Initializer& init) {
auto *prototype = WebCore::${className(
typeName
)}::createPrototype(init.vm, reinterpret_cast<Zig::GlobalObject*>(init.owner));
init.set(WebCore::${className(
typeName
)}::createStructure(init.vm, init.owner, prototype));
});
`.trim();
}
function generateLazyClassStructureImpl(typeName, { klass = {}, proto = {} }) {
return `
m_${className(typeName)}.initLater(
[](LazyClassStructure::Initializer& init) {
init.setPrototype(WebCore::${className(
typeName
)}::createPrototype(init.vm, reinterpret_cast<Zig::GlobalObject*>(init.global)));
init.setStructure(WebCore::${className(
typeName
)}::createStructure(init.vm, init.global, init.prototype));
init.setConstructor(WebCore::${className(
typeName
)}::createConstructor(init.vm, init.global, init.prototype));
});
`.trim();
}
const GENERATED_CLASSES_HEADER = `
// GENERATED CODE - DO NOT MODIFY BY HAND
// Generated by make codegen
#pragma once
#include "root.h"
namespace Zig {
}
#include "JSDOMWrapper.h"
#include <wtf/NeverDestroyed.h>
namespace WebCore {
using namespace Zig;
using namespace JSC;
`;
const GENERATED_CLASSES_FOOTER = `
}
`;
const GENERATED_CLASSES_IMPL_HEADER = `
// GENERATED CODE - DO NOT MODIFY BY HAND
// Generated by make codegen
#include "root.h"
#include <JavaScriptCore/JSFunction.h>
#include <JavaScriptCore/InternalFunction.h>
#include <JavaScriptCore/LazyClassStructure.h>
#include <JavaScriptCore/LazyClassStructureInlines.h>
#include <JavaScriptCore/FunctionPrototype.h>
#include "ZigGlobalObject.h"
#include <JavaScriptCore/DOMJITAbstractHeap.h>
#include "DOMJITIDLConvert.h"
#include "DOMJITIDLType.h"
#include "DOMJITIDLTypeFilter.h"
#include "DOMJITHelpers.h"
#include <JavaScriptCore/DFGAbstractHeap.h>
#include "JSDOMConvertBufferSource.h"
#include "ZigGeneratedClasses.h"
namespace WebCore {
using namespace JSC;
using namespace Zig;
`;
const GENERATED_CLASSES_IMPL_FOOTER = `
} // namespace WebCore
`;
function initLazyClasses(initLaterFunctions) {
return `
void GlobalObject::initGeneratedLazyClasses() {
${initLaterFunctions.map((a) => a.trim()).join("\n ")}
}
`.trim();
}
function visitLazyClasses(classes) {
return `
template<typename Visitor>
void GlobalObject::visitGeneratedLazyClasses(GlobalObject *thisObject, Visitor& visitor)
{
${classes
.map(
(a) =>
`thisObject->m_${className(
a.name
)}.visit(visitor); visitor.append(thisObject->m_${className(
a.name
)}SetterValue);`
)
.join("\n ")}
}
`.trim();
}
const ZIG_GENERATED_CLASSES_HEADER = `
const JSC = @import("javascript_core");
const Classes = @import("./generated_classes_list.zig").Classes;
const Environment = @import("../../env.zig");
const std = @import("std");
pub const StaticGetterType = fn(*JSC.JSGlobalObject, JSC.JSValue, JSC.JSValue) callconv(.C) JSC.JSValue;
pub const StaticSetterType = fn(*JSC.JSGlobalObject, JSC.JSValue, JSC.JSValue, JSC.JSValue) callconv(.C) bool;
pub const StaticCallbackType = fn(*JSC.JSGlobalObject, *JSC.CallFrame) callconv(.C) JSC.JSValue;
`;
function findClasses() {
var classes = [];
for (let directory of directoriesToSearch) {
readdirSync(directory).forEach((file) => {
if (file.endsWith(".classes.ts")) {
const result = require(`${directory}/${file}`);
console.log("Generated", result.default.length, "classes from", file);
for (let { name } of result.default) {
console.log(` - ${name}`);
}
classes.push(...result.default);
}
});
}
return classes;
}
const classes = findClasses();
function writeAndUnlink(path, content) {
try {
unlinkSync(path);
} catch (e) {}
return Bun.write(path, content);
}
await writeAndUnlink(`${import.meta.dir}/../bindings/generated_classes.zig`, [
ZIG_GENERATED_CLASSES_HEADER,
...classes.map((a) => generateZig(a.name, a).trim()).join("\n"),
"\n",
`
comptime {
${classes.map((a) => `_ = ${className(a.name)};`).join("\n ")}
}
`,
]);
await writeAndUnlink(`${import.meta.dir}/../bindings/ZigGeneratedClasses.h`, [
GENERATED_CLASSES_HEADER,
...classes.map((a) => generateHeader(a.name, a)),
GENERATED_CLASSES_FOOTER,
]);
await writeAndUnlink(`${import.meta.dir}/../bindings/ZigGeneratedClasses.cpp`, [
GENERATED_CLASSES_IMPL_HEADER,
...classes.map((a) => generateImpl(a.name, a)),
GENERATED_CLASSES_IMPL_FOOTER,
]);
await writeAndUnlink(
`${import.meta.dir}/../bindings/ZigGeneratedClasses+lazyStructureHeader.h`,
classes
.map((a) =>
!a.noConstructor
? generateLazyClassStructureHeader(a.name, a)
: generateLazyStructureHeader(a.name, a)
)
.join("\n")
);
await writeAndUnlink(
`${import.meta.dir}/../bindings/ZigGeneratedClasses+DOMClientIsoSubspaces.h`,
classes.map((a) =>
[
`std::unique_ptr<GCClient::IsoSubspace> ${clientSubspaceFor(a.name)};`,
!a.noConstructor
? `std::unique_ptr<GCClient::IsoSubspace> ${clientSubspaceFor(
a.name
)}Constructor;`
: "",
].join("\n")
)
);
await writeAndUnlink(
`${import.meta.dir}/../bindings/ZigGeneratedClasses+DOMIsoSubspaces.h`,
classes.map((a) =>
[
`std::unique_ptr<IsoSubspace> ${subspaceFor(a.name)};`,
!a.noConstructor
? `std::unique_ptr<IsoSubspace> ${subspaceFor(a.name)}Constructor;`
: ``,
].join("\n")
)
);
await writeAndUnlink(
`${import.meta.dir}/../bindings/ZigGeneratedClasses+lazyStructureImpl.h`,
initLazyClasses(
classes.map((a) =>
!a.noConstructor
? generateLazyClassStructureImpl(a.name, a)
: generateLazyStructureImpl(a.name, a)
)
) +
"\n" +
visitLazyClasses(classes)
);
export {};