1 files changed, 143 insertions, 0 deletions
diff --git a/src/mongo/scripting/mozjs/dbquery.cpp b/src/mongo/scripting/mozjs/dbquery.cpp
new file mode 100644
index 00000000000..e9bddf4b30e
--- /dev/null
+++ b/src/mongo/scripting/mozjs/dbquery.cpp
@@ -0,0 +1,143 @@
+/**
+ * Copyright (C) 2015 MongoDB Inc.
+ *
+ * This program is free software: you can redistribute it and/or  modify
+ * it under the terms of the GNU Affero General Public License, version 3,
+ * as published by the Free Software Foundation.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU Affero General Public License for more details.
+ *
+ * You should have received a copy of the GNU Affero General Public License
+ * along with this program.  If not, see <http://www.gnu.org/licenses/>.
+ *
+ * As a special exception, the copyright holders give permission to link the
+ * code of portions of this program with the OpenSSL library under certain
+ * conditions as described in each individual source file and distribute
+ * linked combinations including the program with the OpenSSL library. You
+ * must comply with the GNU Affero General Public License in all respects
+ * for all of the code used other than as permitted herein. If you modify
+ * file(s) with this exception, you may extend this exception to your
+ * version of the file(s), but you are not obligated to do so. If you do not
+ * wish to do so, delete this exception statement from your version. If you
+ * delete this exception statement from all source files in the program,
+ * then also delete it in the license file.
+ */
+
+#include "mongo/platform/basic.h"
+
+#include "mongo/scripting/mozjs/dbquery.h"
+
+#include "mongo/scripting/mozjs/idwrapper.h"
+#include "mongo/scripting/mozjs/implscope.h"
+#include "mongo/scripting/mozjs/objectwrapper.h"
+
+namespace mongo {
+namespace mozjs {
+
+const char* const DBQueryInfo::className = "DBQuery";
+
+void DBQueryInfo::construct(JSContext* cx, JS::CallArgs args) {
+    auto scope = getScope(cx);
+
+    if (args.length() < 4)
+        uasserted(ErrorCodes::BadValue, "dbQuery constructor requires at least 4 arguments");
+
+    JS::RootedObject thisv(cx);
+    scope->getDbQueryProto().newObject(&thisv);
+    ObjectWrapper o(cx, thisv);
+
+    o.setValue("_mongo", args.get(0));
+    o.setValue("_db", args.get(1));
+    o.setValue("_collection", args.get(2));
+    o.setValue("_ns", args.get(3));
+
+    JS::RootedObject emptyObj(cx);
+    JS::RootedValue emptyObjVal(cx);
+    emptyObjVal.setObjectOrNull(emptyObj);
+
+    JS::RootedValue nullVal(cx);
+    nullVal.setNull();
+
+    if (args.length() > 4 && args.get(4).isObject()) {
+        o.setValue("_query", args.get(4));
+    } else {
+        o.setValue("_query", emptyObjVal);
+    }
+
+    if (args.length() > 5 && args.get(5).isObject()) {
+        o.setValue("_fields", args.get(5));
+    } else {
+        o.setValue("_fields", nullVal);
+    }
+
+    if (args.length() > 6 && args.get(6).isNumber()) {
+        o.setValue("_limit", args.get(6));
+    } else {
+        o.setNumber("_limit", 0);
+    }
+
+    if (args.length() > 7 && args.get(7).isNumber()) {
+        o.setValue("_skip", args.get(7));
+    } else {
+        o.setNumber("_skip", 0);
+    }
+
+    if (args.length() > 8 && args.get(8).isNumber()) {
+        o.setValue("_batchSize", args.get(8));
+    } else {
+        o.setNumber("_batchSize", 0);
+    }
+
+    if (args.length() > 9 && args.get(9).isNumber()) {
+        o.setValue("_options", args.get(9));
+    } else {
+        o.setNumber("_options", 0);
+    }
+
+    o.setValue("_cursor", nullVal);
+    o.setNumber("_numReturned", 0);
+    o.setBoolean("_special", false);
+
+    args.rval().setObjectOrNull(thisv);
+}
+
+void DBQueryInfo::getProperty(JSContext* cx,
+                              JS::HandleObject obj,
+                              JS::HandleId id,
+                              JS::MutableHandleValue vp) {
+    if (!vp.isUndefined()) {
+        return;
+    }
+
+    IdWrapper wid(cx, id);
+
+    // We only use this for index access
+    if (!wid.isInt()) {
+        return;
+    }
+
+    JS::RootedObject parent(cx);
+    if (!JS_GetPrototype(cx, obj, &parent))
+        uasserted(ErrorCodes::InternalError, "Couldn't get prototype");
+
+    ObjectWrapper parentWrapper(cx, parent);
+
+    JS::RootedValue arrayAccess(cx);
+    parentWrapper.getValue("arrayAccess", &arrayAccess);
+
+    if (arrayAccess.isObject() && JS_ObjectIsFunction(cx, arrayAccess.toObjectOrNull())) {
+        JS::AutoValueArray<1> args(cx);
+
+        args[0].setInt32(wid.toInt32());
+
+        ObjectWrapper(cx, obj).callMethod(arrayAccess, args, vp);
+    } else {
+        uasserted(ErrorCodes::BadValue, "arrayAccess is not a function");
+    }
+}
+
+}  // namespace mozjs
+}  // namespace mongo