Function FlattenIntoArrayWithCallBack

src/PyListProxyHandler.cc:1317–1413

private

// private
static uint32_t FlattenIntoArrayWithCallBack(JSContext *cx,
  JSObject *retArray, PyObject *source,
  Py_ssize_t sourceLen, uint32_t start, uint32_t depth,
  JS::HandleValue callBack, JS::HandleObject thisArg) {

  uint32_t targetIndex = start;

  JS::RootedValue sourceValue(cx, jsTypeFactory(cx, source));
  JS::Rooted<JS::ValueArray<3>> jArgs(cx);
  JS::RootedValue elementVal(cx);
  JS::RootedValue retVal(cx);

  for (uint32_t sourceIndex = 0; sourceIndex < sourceLen; sourceIndex++) {
    if (PyObject_TypeCheck(source, &JSArrayProxyType)) {
      JS_GetElement(cx, *(((JSArrayProxy *)source)->jsArray), sourceIndex, &elementVal);
    }
    else if (PyObject_TypeCheck(source, &PyList_Type)) {
      elementVal.set(jsTypeFactory(cx, PyList_GetItem(source, sourceIndex)));
    }

    jArgs[0].set(elementVal);
    jArgs[1].setInt32(sourceIndex);
    jArgs[2].set(sourceValue);
    if (!JS_CallFunctionValue(cx, thisArg, callBack, jArgs, &retVal)) {
      return false;
    }

    PyObject *element = pyTypeFactory(cx, retVal);

    bool shouldFlatten;
    if (depth > 0) {
      shouldFlatten = PyObject_TypeCheck(element, &JSArrayProxyType) || PyObject_TypeCheck(element, &PyList_Type);
    } else {
      shouldFlatten = false;
    }

    Py_ssize_t elementLen;
    if (PyObject_TypeCheck(element, &JSArrayProxyType)) {
      elementLen = JSArrayProxyMethodDefinitions::JSArrayProxy_length((JSArrayProxy *)element);
    }
    else if (PyObject_TypeCheck(element, &PyList_Type)) {
      elementLen = PyList_GET_SIZE(element);
    }

    if (shouldFlatten) {
      targetIndex = FlattenIntoArrayWithCallBack(cx,
        retArray,
        element,
        elementLen,
        targetIndex,
        depth - 1,
        callBack,
        thisArg
      );
    }
    else {
      JS::RootedObject rootedRetArray(cx, retArray);