1 files changed, 53 insertions, 0 deletions
diff --git a/js/xpconnect/src/qsObjectHelper.h b/js/xpconnect/src/qsObjectHelper.h
new file mode 100644
index 000000000..a3c753800
--- /dev/null
+++ b/js/xpconnect/src/qsObjectHelper.h
@@ -0,0 +1,53 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
+/* vim: set ts=8 sts=4 et sw=4 tw=99: */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this
+ * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#ifndef qsObjectHelper_h
+#define qsObjectHelper_h
+
+#include "xpcObjectHelper.h"
+
+#include "nsCOMPtr.h"
+#include "nsWrapperCache.h"
+#include "mozilla/TypeTraits.h"
+
+class qsObjectHelper : public xpcObjectHelper
+{
+public:
+    template <class T>
+    inline
+    qsObjectHelper(T* aObject, nsWrapperCache* aCache)
+        : xpcObjectHelper(ToSupports(aObject), ToCanonicalSupports(aObject),
+                          aCache)
+    {}
+
+    template <class T>
+    inline
+    qsObjectHelper(nsCOMPtr<T>& aObject, nsWrapperCache* aCache)
+        : xpcObjectHelper(ToSupports(aObject.get()),
+                          ToCanonicalSupports(aObject.get()), aCache)
+    {
+        if (mCanonical) {
+            // Transfer the strong reference.
+            mCanonicalStrong = dont_AddRef(mCanonical);
+            aObject.forget();
+        }
+    }
+
+    template <class T>
+    inline
+    qsObjectHelper(RefPtr<T>& aObject, nsWrapperCache* aCache)
+        : xpcObjectHelper(ToSupports(aObject.get()),
+                          ToCanonicalSupports(aObject.get()), aCache)
+    {
+        if (mCanonical) {
+            // Transfer the strong reference.
+            mCanonicalStrong = dont_AddRef(mCanonical);
+            aObject.forget();
+        }
+    }
+};
+
+#endif