Files
bun.sh/src/bun.js/bindings/SQLClient.cpp
2024-11-11 14:40:02 -08:00

322 lines
11 KiB
C++

#include "root.h"
#include "JavaScriptCore/JSGlobalObject.h"
#include <JavaScriptCore/StructureInlines.h>
#include <JavaScriptCore/ObjectPrototype.h>
#include "headers-handwritten.h"
#include <JavaScriptCore/JSCJSValueInlines.h>
#include <JavaScriptCore/ObjectConstructor.h>
#include <JavaScriptCore/JSObjectInlines.h>
#include <JavaScriptCore/JSBigInt.h>
#include <JavaScriptCore/DateInstance.h>
#include "ZigGlobalObject.h"
#include <JavaScriptCore/JSONObject.h>
#include <JavaScriptCore/GCDeferralContext.h>
#include "GCDefferalContext.h"
#include "wtf/Assertions.h"
#include "JavaScriptCore/ArgList.h"
#include "JavaScriptCore/ArrayAllocationProfile.h"
#include "JavaScriptCore/ExceptionScope.h"
#include "JavaScriptCore/JSArrayBufferView.h"
#include "JavaScriptCore/JSCast.h"
#include "JavaScriptCore/JSGlobalObjectInlines.h"
#include "JavaScriptCore/JSType.h"
#include "JavaScriptCore/TypedArrayAdaptersForwardDeclarations.h"
namespace Bun {
using namespace JSC;
typedef struct DataCellArray {
struct DataCell* cells;
unsigned length;
} DataCellArray;
typedef struct TypedArrayDataCell {
void* headPtr;
void* data;
unsigned length;
unsigned byteLength;
JSC::JSType type;
} TypedArrayDataCell;
typedef union DataCellValue {
uint8_t null_value;
WTF::StringImpl* string;
double number;
int32_t integer;
int64_t bigint;
uint8_t boolean;
double date;
double date_with_time_zone;
size_t bytea[2];
WTF::StringImpl* json;
DataCellArray array;
TypedArrayDataCell typed_array;
} DataCellValue;
enum class DataCellTag : uint8_t {
Null = 0,
String = 1,
Double = 2,
Integer = 3,
Bigint = 4,
Boolean = 5,
Date = 6,
DateWithTimeZone = 7,
Bytea = 8,
Json = 9,
Array = 10,
TypedArray = 11,
};
typedef struct DataCell {
DataCellTag tag;
DataCellValue value;
uint8_t freeValue;
} DataCell;
static JSC::JSValue toJS(JSC::VM& vm, JSC::JSGlobalObject* globalObject, DataCell& cell)
{
switch (cell.tag) {
case DataCellTag::Null:
return jsNull();
break;
case DataCellTag::String: {
return jsString(vm, WTF::String(cell.value.string));
break;
}
case DataCellTag::Double:
return jsDoubleNumber(cell.value.number);
break;
case DataCellTag::Integer:
return jsNumber(cell.value.integer);
break;
case DataCellTag::Bigint:
return JSC::JSBigInt::createFrom(globalObject, cell.value.bigint);
break;
case DataCellTag::Boolean:
return jsBoolean(cell.value.boolean);
break;
case DataCellTag::DateWithTimeZone:
case DataCellTag::Date: {
return JSC::DateInstance::create(vm, globalObject->dateStructure(), cell.value.date);
break;
}
case DataCellTag::Bytea: {
Zig::GlobalObject* zigGlobal = jsCast<Zig::GlobalObject*>(globalObject);
auto* subclassStructure = zigGlobal->JSBufferSubclassStructure();
auto* uint8Array = JSC::JSUint8Array::createUninitialized(globalObject, subclassStructure, cell.value.bytea[1]);
if (UNLIKELY(uint8Array == nullptr)) {
return {};
}
if (cell.value.bytea[1] > 0) {
memcpy(uint8Array->vector(), reinterpret_cast<void*>(cell.value.bytea[0]), cell.value.bytea[1]);
}
return uint8Array;
}
case DataCellTag::Json: {
auto str = WTF::String(cell.value.string);
JSC::JSValue json = JSC::JSONParse(globalObject, str);
return json;
break;
}
case DataCellTag::Array: {
MarkedArgumentBuffer args;
unsigned length = cell.value.array.length;
for (unsigned i = 0; i < length; i++) {
JSValue result = toJS(vm, globalObject, cell.value.array.cells[i]);
if (UNLIKELY(result.isEmpty())) {
return {};
}
args.append(result);
}
return JSC::constructArray(globalObject, static_cast<ArrayAllocationProfile*>(nullptr), args);
}
case DataCellTag::TypedArray: {
JSC::JSType type = static_cast<JSC::JSType>(cell.value.typed_array.type);
unsigned length = cell.value.typed_array.length;
switch (type) {
case JSC::JSType::Int32ArrayType: {
JSC::JSInt32Array* array = JSC::JSInt32Array::createUninitialized(globalObject, globalObject->typedArrayStructure(TypedArrayType::TypeInt32, false), length);
if (UNLIKELY(array == nullptr)) {
return {};
}
if (length > 0) {
memcpy(array->vector(), reinterpret_cast<void*>(cell.value.typed_array.data), length * sizeof(int32_t));
}
return array;
}
case JSC::JSType::Uint32ArrayType: {
JSC::JSUint32Array* array = JSC::JSUint32Array::createUninitialized(globalObject, globalObject->typedArrayStructure(TypedArrayType::TypeUint32, false), length);
if (UNLIKELY(array == nullptr)) {
return {};
}
if (length > 0) {
memcpy(array->vector(), reinterpret_cast<void*>(cell.value.typed_array.data), length * sizeof(uint32_t));
}
return array;
}
case JSC::JSType::Int16ArrayType: {
JSC::JSInt16Array* array = JSC::JSInt16Array::createUninitialized(globalObject, globalObject->typedArrayStructure(TypedArrayType::TypeInt16, false), length);
if (UNLIKELY(array == nullptr)) {
return {};
}
if (length > 0) {
memcpy(array->vector(), reinterpret_cast<void*>(cell.value.typed_array.data), length * sizeof(int16_t));
}
return array;
}
case JSC::JSType::Uint16ArrayType: {
JSC::JSUint16Array* array = JSC::JSUint16Array::createUninitialized(globalObject, globalObject->typedArrayStructure(TypedArrayType::TypeUint16, false), length);
if (UNLIKELY(array == nullptr)) {
return {};
}
if (length > 0) {
memcpy(array->vector(), reinterpret_cast<void*>(cell.value.typed_array.data), length * sizeof(uint16_t));
}
return array;
}
case JSC::JSType::Float16ArrayType: {
JSC::JSFloat16Array* array = JSC::JSFloat16Array::createUninitialized(globalObject, globalObject->typedArrayStructure(TypedArrayType::TypeFloat16, false), length);
if (UNLIKELY(array == nullptr)) {
return {};
}
if (length > 0) {
memcpy(array->vector(), reinterpret_cast<void*>(cell.value.typed_array.data), length * 2); // sizeof(float16_t)
}
return array;
}
case JSC::JSType::Float32ArrayType: {
JSC::JSFloat32Array* array = JSC::JSFloat32Array::createUninitialized(globalObject, globalObject->typedArrayStructure(TypedArrayType::TypeFloat32, false), length);
if (UNLIKELY(array == nullptr)) {
return {};
}
if (length > 0) {
memcpy(array->vector(), reinterpret_cast<void*>(cell.value.typed_array.data), length * sizeof(float));
}
return array;
}
case JSC::JSType::Float64ArrayType: {
JSC::JSFloat64Array* array = JSC::JSFloat64Array::createUninitialized(globalObject, globalObject->typedArrayStructure(TypedArrayType::TypeFloat64, false), length);
if (UNLIKELY(array == nullptr)) {
return {};
}
if (length > 0) {
memcpy(array->vector(), reinterpret_cast<void*>(cell.value.typed_array.data), length * sizeof(double));
}
return array;
}
default: {
RELEASE_ASSERT_NOT_REACHED_WITH_MESSAGE("TODO: implement this typed array type");
}
}
}
default: {
RELEASE_ASSERT_NOT_REACHED();
}
}
}
static JSC::JSValue toJS(JSC::Structure* structure, DataCell* cells, unsigned count, JSC::JSGlobalObject* globalObject)
{
auto& vm = globalObject->vm();
auto* object = JSC::constructEmptyObject(vm, structure);
auto scope = DECLARE_THROW_SCOPE(vm);
for (unsigned i = 0; i < count; i++) {
auto& cell = cells[i];
JSValue value = toJS(vm, globalObject, cell);
RETURN_IF_EXCEPTION(scope, {});
object->putDirectOffset(vm, i, value);
}
return object;
}
static JSC::JSValue toJS(JSC::JSArray* array, JSC::Structure* structure, DataCell* cells, unsigned count, JSC::JSGlobalObject* globalObject)
{
JSValue value = toJS(structure, cells, count, globalObject);
if (value.isEmpty())
return {};
if (array) {
array->push(globalObject, value);
return array;
}
auto* newArray = JSC::constructEmptyArray(globalObject, static_cast<ArrayAllocationProfile*>(nullptr), 1);
if (!newArray)
return {};
newArray->putDirectIndex(globalObject, 0, value);
return newArray;
}
extern "C" EncodedJSValue JSC__constructObjectFromDataCell(
JSC::JSGlobalObject* globalObject,
EncodedJSValue encodedArrayValue,
EncodedJSValue encodedStructureValue, DataCell* cells, unsigned count)
{
JSValue arrayValue = JSValue::decode(encodedArrayValue);
JSValue structureValue = JSValue::decode(encodedStructureValue);
auto* array = arrayValue ? jsDynamicCast<JSC::JSArray*>(arrayValue) : nullptr;
auto* structure = jsDynamicCast<JSC::Structure*>(structureValue);
return JSValue::encode(toJS(array, structure, cells, count, globalObject));
}
extern "C" EncodedJSValue JSC__createStructure(JSC::JSGlobalObject* globalObject, JSC::JSCell* owner, unsigned int inlineCapacity, BunString* names)
{
auto& vm = globalObject->vm();
Structure* structure = globalObject->structureCache().emptyObjectStructureForPrototype(globalObject, globalObject->objectPrototype(), inlineCapacity);
if (owner) {
vm.writeBarrier(owner, structure);
} else {
vm.writeBarrier(structure);
}
ensureStillAliveHere(structure);
PropertyNameArray propertyNames(vm, PropertyNameMode::Strings, PrivateSymbolMode::Exclude);
for (unsigned i = 0; i < inlineCapacity; i++) {
propertyNames.add(Identifier::fromString(vm, names[i].toWTFString()));
}
PropertyOffset offset = 0;
for (unsigned i = 0; i < inlineCapacity; i++) {
structure = structure->addPropertyTransition(vm, structure, propertyNames[i], 0, offset);
}
return JSValue::encode(structure);
}
extern "C" EncodedJSValue JSC__createEmptyObjectWithStructure(JSC::JSGlobalObject* globalObject, JSC::Structure* structure)
{
auto& vm = globalObject->vm();
auto* object = JSC::constructEmptyObject(vm, structure);
ensureStillAliveHere(object);
vm.writeBarrier(object);
return JSValue::encode(object);
}
extern "C" void JSC__putDirectOffset(JSC::VM* vm, JSC::EncodedJSValue object, unsigned int offset, JSC::EncodedJSValue value)
{
JSValue::decode(object).getObject()->putDirectOffset(*vm, offset, JSValue::decode(value));
}
}