#include "include/JSObjectProxy.hh"

#include "include/JSObjectIterProxy.hh"

#include "include/JSObjectKeysProxy.hh"

#include "include/JSObjectValuesProxy.hh"

#include "include/JSObjectItemsProxy.hh"

#include "include/modules/pythonmonkey/pythonmonkey.hh"

#include "include/jsTypeFactory.hh"

#include "include/pyTypeFactory.hh"

#include "include/PyBaseProxyHandler.hh"

#include "include/JSFunctionProxy.hh"

#include <jsapi.h>

#include <jsfriendapi.h>

#include <Python.h>

#include <object.h>

JSContext *GLOBAL_CX; /**< pointer to PythonMonkey's JSContext */

bool keyToId(PyObject *key, JS::MutableHandleId idp) {

}

void JSObjectProxyMethodDefinitions::JSObjectProxy_dealloc(JSObjectProxy *self)

}

int JSObjectProxyMethodDefinitions::JSObjectProxy_traverse(JSObjectProxy *self, visitproc visit, void *arg)

}

int JSObjectProxyMethodDefinitions::JSObjectProxy_clear(JSObjectProxy *self)

}

Py_ssize_t JSObjectProxyMethodDefinitions::JSObjectProxy_length(JSObjectProxy *self)

}

static inline PyObject *getKey(JSObjectProxy *self, PyObject *key, JS::HandleId id, bool checkPropertyShadowsMethod) {

// look through the methods for dispatch

for (size_t index = 0;; index++) {

const char *methodName = JSObjectProxyType.tp_methods[index].ml_name;

if (methodName == NULL || !PyUnicode_Check(key)) {

JS::RootedValue value(GLOBAL_CX);

JS_GetPropertyById(GLOBAL_CX, *(self->jsObject), id, &value);

// if value is a JSFunction, bind `this` to self

/* (Caleb Aikens) its potentially problematic to bind it like this since if the function

if (value.isObject()) {

JS::RootedObject valueObject(GLOBAL_CX);

JS_ValueToObject(GLOBAL_CX, value, &valueObject);

js::ESClass cls;

JS::GetBuiltinClass(GLOBAL_CX, valueObject, &cls);

if (cls == js::ESClass::Function) {

JS::Rooted<JS::ValueArray<1>> args(GLOBAL_CX);

args[0].setObject(*((*(self->jsObject)).get()));

JS::RootedValue boundFunction(GLOBAL_CX);

if (!JS_CallFunctionName(GLOBAL_CX, valueObject, "bind", args, &boundFunction)) {

PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSObjectProxyType.tp_name);

return NULL;

}

value.set(boundFunction);

}

}

else if (value.isUndefined() && PyUnicode_Check(key)) {

if (strcmp("__class__", PyUnicode_AsUTF8(key)) == 0) {

return PyObject_GenericGetAttr((PyObject *)self, key);

}

}

return pyTypeFactory(GLOBAL_CX, value);

}

else {

if (strcmp(methodName, PyUnicode_AsUTF8(key)) == 0) {

if (checkPropertyShadowsMethod) {

// just make sure no property is shadowing a method by name

JS::RootedValue value(GLOBAL_CX);

JS_GetPropertyById(GLOBAL_CX, *(self->jsObject), id, &value);

if (!value.isUndefined()) {

return pyTypeFactory(GLOBAL_CX, value);

}

}

return PyObject_GenericGetAttr((PyObject *)self, key);

}

}

}

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_get(JSObjectProxy *self, PyObject *key)

{

JS::RootedId id(GLOBAL_CX);

if (!keyToId(key, &id)) {

PyErr_SetString(PyExc_AttributeError, "JSObjectProxy property name must be of type str or int");

return NULL;

}

return getKey(self, key, id, false);

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_get_subscript(JSObjectProxy *self, PyObject *key)

{

JS::RootedId id(GLOBAL_CX);

if (!keyToId(key, &id)) {

PyErr_SetString(PyExc_AttributeError, "JSObjectProxy property name must be of type str or int");

return NULL;

}

return getKey(self, key, id, true);

}

int JSObjectProxyMethodDefinitions::JSObjectProxy_contains(JSObjectProxy *self, PyObject *key)

}

static inline void assignKeyValue(JSObjectProxy *self, PyObject *key, JS::HandleId id, PyObject *value) {

}

int JSObjectProxyMethodDefinitions::JSObjectProxy_assign(JSObjectProxy *self, PyObject *key, PyObject *value)

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_richcompare(JSObjectProxy *self, PyObject *other, int op)

{

if (op != Py_EQ && op != Py_NE) {

Py_RETURN_NOTIMPLEMENTED;

}

std::unordered_map<PyObject *, PyObject *> visited;

bool isEqual = JSObjectProxy_richcompare_helper(self, other, visited);

switch (op)

{

case Py_EQ:

return PyBool_FromLong(isEqual);

case Py_NE:

return PyBool_FromLong(!isEqual);

default:

return NULL;

}

}

bool JSObjectProxyMethodDefinitions::JSObjectProxy_richcompare_helper(JSObjectProxy *self, PyObject *other, std::unordered_map<PyObject *, PyObject *> &visited)

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_iter(JSObjectProxy *self) {

// key iteration

JSObjectIterProxy *iterator = PyObject_GC_New(JSObjectIterProxy, &JSObjectIterProxyType);

if (iterator == NULL) {

return NULL;

}

iterator->it.it_index = 0;

iterator->it.reversed = false;

iterator->it.kind = KIND_KEYS;

Py_INCREF(self);

iterator->it.di_dict = (PyDictObject *)self;

iterator->it.props = new JS::PersistentRootedIdVector(GLOBAL_CX);

// Get **enumerable** own properties

if (!js::GetPropertyKeys(GLOBAL_CX, *(self->jsObject), JSITER_OWNONLY, iterator->it.props)) {

return NULL;

}

PyObject_GC_Track(iterator);

return (PyObject *)iterator;

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_iter_next(JSObjectProxy *self) {

PyObject *key = PyUnicode_FromString("next");

JS::RootedId id(GLOBAL_CX);

if (!keyToId(key, &id)) {

PyErr_SetString(PyExc_SystemError, "JSObjectProxy failed type conversion");

return NULL;

}

PyObject *nextFunction = getKey(self, key, id, false);

Py_DECREF(key);

if (nextFunction == NULL) {

PyErr_SetString(PyExc_SystemError, "JSObjectProxy could not retrieve key");

return NULL;

}

PyObject *retVal = JSFunctionProxyMethodDefinitions::JSFunctionProxy_call(nextFunction, PyTuple_New(0), NULL);

Py_DECREF(nextFunction);

// check if end of iteration

key = PyUnicode_FromString("done");

PyObject *done = JSObjectProxy_get((JSObjectProxy *)retVal, key);

Py_DECREF(key);

if (done == Py_True) {

PyErr_SetNone(PyExc_StopIteration);

return NULL;

}

key = PyUnicode_FromString("value");

PyObject *value = JSObjectProxy_get((JSObjectProxy *)retVal, key);

Py_DECREF(key);

return value;

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_repr(JSObjectProxy *self) {

Py_ssize_t i = Py_ReprEnter((PyObject *)self);

if (i != 0) {

return i > 0 ? PyUnicode_FromString("{...}") : NULL;

}

Py_ssize_t selfLength = JSObjectProxy_length(self);

if (selfLength == 0) {

Py_ReprLeave((PyObject *)self);

return PyUnicode_FromString("{}");

}

_PyUnicodeWriter writer;

_PyUnicodeWriter_Init(&writer);

writer.overallocate = 1;

/* "{" + "1: 2" + ", 3: 4" * (len - 1) + "}" */

writer.min_length = 1 + 4 + (2 + 4) * (selfLength - 1) + 1;

PyObject *key = NULL, *value = NULL;

JS::RootedIdVector props(GLOBAL_CX);

if (_PyUnicodeWriter_WriteChar(&writer, '{') < 0) {

goto error;

}

/* Do repr() on each key+value pair, and insert ": " between them. Note that repr may mutate the dict. */

// Get **enumerable** own properties

if (!js::GetPropertyKeys(GLOBAL_CX, *(self->jsObject), JSITER_OWNONLY, &props)) {

return NULL;

}

for (Py_ssize_t index = 0; index < selfLength; index++) {

if (index > 0) {

if (_PyUnicodeWriter_WriteASCIIString(&writer, ", ", 2) < 0) {

goto error;

}

}

JS::HandleId id = props[index];

key = idToKey(GLOBAL_CX, id);

// escape infinite recur on superclass reference

if (strcmp(PyUnicode_AsUTF8(key), "$super") == 0) {

continue;

}

// Prevent repr from deleting key or value during key format.

Py_INCREF(key);

PyObject *s = PyObject_Repr(key);

if (s == NULL) {

goto error;

}

int res = _PyUnicodeWriter_WriteStr(&writer, s);

Py_DECREF(s);

if (res < 0) {

goto error;

}

if (_PyUnicodeWriter_WriteASCIIString(&writer, ": ", 2) < 0) {

goto error;

}

JS::RootedValue elementVal(GLOBAL_CX);

JS_GetPropertyById(GLOBAL_CX, *(self->jsObject), id, &elementVal);

if (&elementVal.toObject() == (*(self->jsObject)).get()) {

value = (PyObject *)self;

Py_INCREF(value);

} else {

value = pyTypeFactory(GLOBAL_CX, elementVal);

}

s = PyObject_Repr(value);

if (s == NULL) {

goto error;

}

res = _PyUnicodeWriter_WriteStr(&writer, s);

Py_DECREF(s);

if (res < 0) {

goto error;

}

Py_CLEAR(key);

Py_CLEAR(value);

}

writer.overallocate = 0;

if (_PyUnicodeWriter_WriteChar(&writer, '}') < 0) {

goto error;

}

Py_ReprLeave((PyObject *)self);

return _PyUnicodeWriter_Finish(&writer);

error:

Py_ReprLeave((PyObject *)self);

_PyUnicodeWriter_Dealloc(&writer);

Py_XDECREF(key);

Py_XDECREF(value);

return NULL;

}

static int mergeFromSeq2(JSObjectProxy *self, PyObject *seq2) {

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_or(JSObjectProxy *self, PyObject *other) {

#if PY_VERSION_HEX < 0x03090000

// | is not supported on dicts in python3.8 or less, so only allow if both

// operands are JSObjectProxy

if (!PyObject_TypeCheck(self, &JSObjectProxyType) || !PyObject_TypeCheck(other, &JSObjectProxyType)) {

Py_RETURN_NOTIMPLEMENTED;

}

#endif

if (!PyDict_Check(self) || !PyDict_Check(other)) {

Py_RETURN_NOTIMPLEMENTED;

}

if (!PyObject_TypeCheck(self, &JSObjectProxyType) && PyObject_TypeCheck(other, &JSObjectProxyType)) {

return PyDict_Type.tp_as_number->nb_or((PyObject *)&(self->dict), other);

}

else {

JS::Rooted<JS::ValueArray<3>> args(GLOBAL_CX);

args[0].setObjectOrNull(JS_NewPlainObject(GLOBAL_CX));

args[1].setObjectOrNull(*(self->jsObject)); // this is null is left operand is real dict

JS::RootedValue jValueOther(GLOBAL_CX, jsTypeFactory(GLOBAL_CX, other));

args[2].setObject(jValueOther.toObject());

JS::RootedObject global(GLOBAL_CX, JS::GetNonCCWObjectGlobal(*(self->jsObject)));

// call Object.assign

JS::RootedValue Object(GLOBAL_CX);

if (!JS_GetProperty(GLOBAL_CX, global, "Object", &Object)) {

PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSObjectProxyType.tp_name);

return NULL;

}

JS::RootedObject rootedObject(GLOBAL_CX, Object.toObjectOrNull());

JS::RootedValue ret(GLOBAL_CX);

if (!JS_CallFunctionName(GLOBAL_CX, rootedObject, "assign", args, &ret)) {

PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSObjectProxyType.tp_name);

return NULL;

}

return pyTypeFactory(GLOBAL_CX, ret);

}

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_ior(JSObjectProxy *self, PyObject *other) {

if (PyDict_Check(other)) {

JS::Rooted<JS::ValueArray<2>> args(GLOBAL_CX);

args[0].setObjectOrNull(*(self->jsObject));

JS::RootedValue jValueOther(GLOBAL_CX, jsTypeFactory(GLOBAL_CX, other));

args[1].setObject(jValueOther.toObject());

JS::RootedObject global(GLOBAL_CX, JS::GetNonCCWObjectGlobal(*(self->jsObject)));

// call Object.assign

JS::RootedValue Object(GLOBAL_CX);

if (!JS_GetProperty(GLOBAL_CX, global, "Object", &Object)) {

PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSObjectProxyType.tp_name);

return NULL;

}

JS::RootedObject rootedObject(GLOBAL_CX, Object.toObjectOrNull());

JS::RootedValue ret(GLOBAL_CX);

if (!JS_CallFunctionName(GLOBAL_CX, rootedObject, "assign", args, &ret)) {

PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSObjectProxyType.tp_name);

return NULL;

}

}

else {

if (mergeFromSeq2(self, other) < 0) {

return NULL;

}

}

Py_INCREF(self);

return (PyObject *)self;

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_get_method(JSObjectProxy *self, PyObject *const *args, Py_ssize_t nargs) {

PyObject *key;

PyObject *default_value = Py_None;

if (!_PyArg_CheckPositional("get", nargs, 1, 2)) {

return NULL;

}

key = args[0];

if (nargs < 2) {

goto skip_optional;

}

default_value = args[1];

skip_optional:

PyObject *value = JSObjectProxy_get(self, key);

if (value == Py_None) {

value = default_value;

}

return value;

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_setdefault_method(JSObjectProxy *self, PyObject *const *args, Py_ssize_t nargs) {

PyObject *key;

PyObject *default_value = Py_None;

if (!_PyArg_CheckPositional("setdefault", nargs, 1, 2)) {

return NULL;

}

key = args[0];

if (nargs < 2) {

goto skip_optional;

}

default_value = args[1];

skip_optional:

JS::RootedId id(GLOBAL_CX);

if (!keyToId(key, &id)) { // invalid key

// TODO (Caleb Aikens): raise exception here

return NULL;

}

PyObject *value = getKey(self, key, id, true);

if (value == Py_None) {

assignKeyValue(self, key, id, default_value);

Py_XINCREF(default_value);

return default_value;

}

return value;

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_pop_method(JSObjectProxy *self, PyObject *const *args, Py_ssize_t nargs) {

PyObject *key;

PyObject *default_value = NULL;

if (!_PyArg_CheckPositional("pop", nargs, 1, 2)) {

return NULL;

}

key = args[0];

if (nargs < 2) {

goto skip_optional;

}

default_value = args[1];

skip_optional:

JS::RootedId id(GLOBAL_CX);

if (!keyToId(key, &id)) {

PyErr_SetString(PyExc_AttributeError, "JSObjectProxy property name must be of type str or int");

return NULL;

}

JS::RootedValue value(GLOBAL_CX);

JS_GetPropertyById(GLOBAL_CX, *(self->jsObject), id, &value);

if (value.isUndefined()) {

if (default_value != NULL) {

Py_INCREF(default_value);

return default_value;

}

_PyErr_SetKeyError(key);

return NULL;

}

else {

JS::ObjectOpResult ignoredResult;

JS_DeletePropertyById(GLOBAL_CX, *(self->jsObject), id, ignoredResult);

return pyTypeFactory(GLOBAL_CX, value);

}

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_clear_method(JSObjectProxy *self) {

JS::RootedIdVector props(GLOBAL_CX);

if (!js::GetPropertyKeys(GLOBAL_CX, *(self->jsObject), JSITER_OWNONLY, &props))

{

PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSObjectProxyType.tp_name);

return NULL;

}

JS::ObjectOpResult ignoredResult;

size_t length = props.length();

for (size_t index = 0; index < length; index++) {

JS_DeletePropertyById(GLOBAL_CX, *(self->jsObject), props[index], ignoredResult);

}

Py_RETURN_NONE;

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_copy_method(JSObjectProxy *self) {

JS::Rooted<JS::ValueArray<2>> args(GLOBAL_CX);

args[0].setObjectOrNull(JS_NewPlainObject(GLOBAL_CX));

args[1].setObjectOrNull(*(self->jsObject));

JS::RootedObject global(GLOBAL_CX, JS::GetNonCCWObjectGlobal(*(self->jsObject)));

// call Object.assign

JS::RootedValue Object(GLOBAL_CX);

if (!JS_GetProperty(GLOBAL_CX, global, "Object", &Object)) {

PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSObjectProxyType.tp_name);

return NULL;

}

JS::RootedObject rootedObject(GLOBAL_CX, Object.toObjectOrNull());

JS::RootedValue ret(GLOBAL_CX);

if (!JS_CallFunctionName(GLOBAL_CX, rootedObject, "assign", args, &ret)) {

PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSObjectProxyType.tp_name);

return NULL;

}

return pyTypeFactory(GLOBAL_CX, ret);

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_update_method(JSObjectProxy *self, PyObject *args, PyObject *kwds) {

PyObject *arg = NULL;

int result = 0;

if (!PyArg_UnpackTuple(args, "update", 0, 1, &arg)) {

return NULL;

}

else if (arg != NULL) {

if (PyDict_CheckExact(arg) || PyObject_TypeCheck(arg, &JSObjectProxyType)) {

JSObjectProxyMethodDefinitions::JSObjectProxy_ior((JSObjectProxy *)self, arg);

result = 0;

} else { // iterable

result = mergeFromSeq2((JSObjectProxy *)self, arg);

if (result < 0) {

return NULL;

}

}

}

if (result == 0 && kwds != NULL) {

if (PyArg_ValidateKeywordArguments(kwds)) {

JSObjectProxyMethodDefinitions::JSObjectProxy_ior((JSObjectProxy *)self, kwds);

}

}

Py_RETURN_NONE;

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_keys_method(JSObjectProxy *self) {

return _PyDictView_New((PyObject *)self, &JSObjectKeysProxyType);

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_values_method(JSObjectProxy *self) {

return _PyDictView_New((PyObject *)self, &JSObjectValuesProxyType);

}

PyObject *JSObjectProxyMethodDefinitions::JSObjectProxy_items_method(JSObjectProxy *self) {

return _PyDictView_New((PyObject *)self, &JSObjectItemsProxyType);

}