1 files changed, 138 insertions, 0 deletions
diff --git a/comm/mailnews/jsaccount/src/JaUrl.h b/comm/mailnews/jsaccount/src/JaUrl.h
new file mode 100644
index 0000000000..6de4feba60
--- /dev/null
+++ b/comm/mailnews/jsaccount/src/JaUrl.h
@@ -0,0 +1,138 @@
+/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* vim: set ts=8 sts=2 et sw=2 tw=80: */
+/* 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 _JaUrl_H_
+#define _JaUrl_H_
+
+#include "DelegateList.h"
+#include "msgCore.h"
+#include "msgIOverride.h"
+#include "nsCycleCollectionParticipant.h"
+#include "nsTHashMap.h"
+#include "nsIFile.h"
+#include "nsIInterfaceRequestor.h"
+#include "nsIMsgFolder.h"
+#include "nsISupports.h"
+#include "nsIURI.h"
+#include "nsIURL.h"
+#include "nsMsgMailNewsUrl.h"
+#include "nsWeakReference.h"
+#include "msgIJaUrl.h"
+#include "nsProxyRelease.h"
+
+namespace mozilla {
+namespace mailnews {
+
+/* Header file */
+
+// This class is an XPCOM component, usable in JS, that calls the methods
+// in the C++ base class (bypassing any JS override).
+class JaBaseCppUrl : public nsMsgMailNewsUrl,
+                     public nsIMsgMessageUrl,
+                     public msgIJaUrl,
+                     public nsIInterfaceRequestor,
+                     public nsSupportsWeakReference {
+ public:
+  NS_DECL_ISUPPORTS_INHERITED
+  NS_DECL_NSIMSGMESSAGEURL
+  NS_DECL_MSGIJAURL
+  NS_DECL_NSIINTERFACEREQUESTOR
+  JaBaseCppUrl() {}
+
+  // nsIMsgMailNewsUrl overrides
+  NS_IMETHOD GetFolder(nsIMsgFolder** aFolder) override;
+  NS_IMETHOD SetFolder(nsIMsgFolder* aFolder) override;
+  NS_IMETHOD IsUrlType(uint32_t type, bool* isType) override;
+  NS_IMETHOD GetServer(nsIMsgIncomingServer** aIncomingServer) override;
+
+ protected:
+  virtual ~JaBaseCppUrl() {}
+
+  // nsIMsgMailUrl variables.
+
+  nsCOMPtr<nsIMsgFolder> mFolder;
+
+  // nsIMsgMessageUrl variables.
+
+  // the uri for the original message, like ews-message://server/folder#123
+  nsCString mUri;
+  nsCOMPtr<nsIFile> mMessageFile;
+  bool mCanonicalLineEnding = false;
+  nsCString mOriginalSpec;
+
+  // msgIJaUrl variables
+  unsigned int m_urlType{0};
+};
+
+class JaCppUrlDelegator : public JaBaseCppUrl, public msgIOverride {
+ public:
+  NS_DECL_ISUPPORTS_INHERITED
+  NS_DECL_MSGIOVERRIDE
+
+  NS_FORWARD_NSIMSGMESSAGEURL(
+      DELEGATE_JS(mJsIMsgMessageUrl, mMethods,
+                  (nsCOMPtr<nsIMsgMessageUrl>(do_QueryInterface(mCppBase))))
+          ->)
+  NS_FORWARD_NSIINTERFACEREQUESTOR(
+      DELEGATE_JS(
+          mJsIInterfaceRequestor, mMethods,
+          (nsCOMPtr<nsIInterfaceRequestor>(do_QueryInterface(mCppBase))))
+          ->)
+
+  JaCppUrlDelegator();
+
+  class Super : public nsIMsgMailNewsUrl,
+                public nsIURIWithSpecialOrigin,
+                public nsIMsgMessageUrl,
+                public msgIJaUrl,
+                public nsIInterfaceRequestor,
+                public nsISupportsWeakReference {
+   public:
+    explicit Super(JaCppUrlDelegator* aFakeThis) { mFakeThis = aFakeThis; }
+    NS_DECL_ISUPPORTS
+    NS_FORWARD_NSIMSGMAILNEWSURL(mFakeThis->JaBaseCppUrl::)
+    NS_FORWARD_NSIURI(mFakeThis->JaBaseCppUrl::)
+    NS_FORWARD_NSIURL(mFakeThis->JaBaseCppUrl::)
+    NS_FORWARD_NSIURIWITHSPECIALORIGIN(mFakeThis->JaBaseCppUrl::)
+    NS_FORWARD_NSIMSGMESSAGEURL(mFakeThis->JaBaseCppUrl::)
+    NS_FORWARD_MSGIJAURL(mFakeThis->JaBaseCppUrl::)
+    NS_FORWARD_NSIINTERFACEREQUESTOR(mFakeThis->JaBaseCppUrl::)
+    NS_FORWARD_NSISUPPORTSWEAKREFERENCE(mFakeThis->JaBaseCppUrl::)
+   private:
+    virtual ~Super() {}
+    JaCppUrlDelegator* mFakeThis;
+  };
+
+ private:
+  virtual ~JaCppUrlDelegator() {
+    NS_ReleaseOnMainThread("JaCppUrlDelegator::mJsIMsgMessageUrl",
+                           mJsIMsgMessageUrl.forget());
+    NS_ReleaseOnMainThread("JaCppUrlDelegator::mJsIInterfaceRequestor",
+                           mJsIInterfaceRequestor.forget());
+    NS_ReleaseOnMainThread("JaCppUrlDelegator::mJsISupports",
+                           mJsISupports.forget());
+    NS_ReleaseOnMainThread("JaCppUrlDelegator::mDelegateList",
+                           mDelegateList.forget());
+  }
+
+  // Interfaces that may be overridden by JS.
+  nsCOMPtr<nsIMsgMessageUrl> mJsIMsgMessageUrl;
+  nsCOMPtr<nsIInterfaceRequestor> mJsIInterfaceRequestor;
+
+  // Owning reference to the JS override.
+  nsCOMPtr<nsISupports> mJsISupports;
+
+  // Class to bypass JS delegates. nsCOMPtr for when we do cycle collection.
+  nsCOMPtr<nsIMsgMailNewsUrl> mCppBase;
+
+  RefPtr<DelegateList> mDelegateList;
+  nsTHashMap<nsCStringHashKey, bool>* mMethods;
+};
+
+}  // namespace mailnews
+}  // namespace mozilla
+
+#endif