Method JSArrayProxy_sort

src/JSArrayProxy.cc:1183–1312

}

PyObject *JSArrayProxyMethodDefinitions::JSArrayProxy_sort(JSArrayProxy *self, PyObject *args, PyObject *kwargs) {
  static const char *const _keywords[] = {"key", "reverse", NULL};

  PyObject *keyfunc = Py_None;
  int reverse = 0;
  if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|$Op:sort", (char **)_keywords, &keyfunc, &reverse)) {
    return NULL;
  }

  if (JSArrayProxy_length(self) > 1) {
    JS::RootedValue jReturnedArray(GLOBAL_CX);
    if (keyfunc != Py_None) {
      if (PyFunction_Check(keyfunc)) {
        // we got a python key function, check if two-argument js style or standard python 1-arg
        PyObject *code = PyFunction_GetCode(keyfunc);
        if (((PyCodeObject *)code)->co_argcount == 1) {
          // adapt to python style, provide js-style comp wrapper that does it the python way, which is < based with calls to keyFunc
          JS::RootedObject funObj(GLOBAL_CX, JS_GetFunctionObject(JS_NewFunction(GLOBAL_CX, sort_compare_key_func, 2, 0, NULL)));

          JS::RootedValue privateValue(GLOBAL_CX, JS::PrivateValue(keyfunc));
          if (!JS_SetProperty(GLOBAL_CX, funObj, "_key_func_param", privateValue)) {  // JS::SetReservedSlot(functionObj, KeyFuncSlot, JS::PrivateValue(keyfunc)); does not work
            PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSArrayProxyType.tp_name);
            return NULL;
          }

          JS::RootedValue reverseValue(GLOBAL_CX);
          reverseValue.setBoolean(reverse);
          if (!JS_SetProperty(GLOBAL_CX, funObj, "_reverse_param", reverseValue)) {
            PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSArrayProxyType.tp_name);
            return NULL;
          }

          JS::Rooted<JS::ValueArray<1>> jArgs(GLOBAL_CX);
          jArgs[0].setObject(*funObj);
          if (!JS_CallFunctionName(GLOBAL_CX, *(self->jsArray), "sort", jArgs, &jReturnedArray)) {
            if (!PyErr_Occurred()) {
              PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSArrayProxyType.tp_name);
            }
            return NULL;
          }

          // cleanup
          if (!JS_DeleteProperty(GLOBAL_CX, funObj, "_key_func_param")) {
            PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSArrayProxyType.tp_name);
            return NULL;
          }

          if (!JS_DeleteProperty(GLOBAL_CX, funObj, "_reverse_param")) {
            PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSArrayProxyType.tp_name);
            return NULL;
          }
        }
        else {
          // two-arg js-style
          JS::Rooted<JS::ValueArray<1>> jArgs(GLOBAL_CX);
          jArgs[0].set(jsTypeFactory(GLOBAL_CX, keyfunc));
          if (!JS_CallFunctionName(GLOBAL_CX, *(self->jsArray), "sort", jArgs, &jReturnedArray)) {
            PyErr_Format(PyExc_SystemError, "%s JSAPI call failed", JSArrayProxyType.tp_name);