1 files changed, 131 insertions, 0 deletions
diff --git a/comm/mailnews/imap/src/ImapModuleLoader.jsm b/comm/mailnews/imap/src/ImapModuleLoader.jsm
new file mode 100644
index 0000000000..427fe2a3f2
--- /dev/null
+++ b/comm/mailnews/imap/src/ImapModuleLoader.jsm
@@ -0,0 +1,131 @@
+/* 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/. */
+
+const EXPORTED_SYMBOLS = ["ImapModuleLoader"];
+
+/**
+ * Determine whether to use nsImapService.cpp or ImapService.jsm. When
+ * `mailnews.imap.jsmodule` is `true`, use ImapService.jsm.
+ */
+function ImapModuleLoader() {
+  try {
+    this.loadModule();
+  } catch (e) {
+    console.error(e);
+  }
+}
+
+var imapJSModules = [
+  // moduleName, interfaceId, contractId
+  [
+    "ImapIncomingServer",
+    "{b02a4e1c-0d9e-498c-8b9d-18917ba9f65b}",
+    "@mozilla.org/messenger/server;1?type=imap",
+  ],
+  [
+    "ImapService",
+    "{2ea8fbe6-029b-4bff-ae05-b794cf955afb}",
+    "@mozilla.org/messenger/imapservice;1",
+  ],
+  [
+    "ImapMessageService",
+    "{d63af753-c2f3-4f1d-b650-9d12229de8ad}",
+    "@mozilla.org/messenger/messageservice;1?type=imap",
+    "ImapMessageService",
+  ],
+  [
+    "ImapFolderContentHandler",
+    "{d927a82f-2d15-4972-ab88-6d84601aae68}",
+    "@mozilla.org/uriloader/content-handler;1?type=x-application-imapfolder",
+  ],
+  [
+    "ImapMessageMessageService",
+    "{2532ae4f-a852-4c96-be45-1308ba23d62e}",
+    "@mozilla.org/messenger/messageservice;1?type=imap-message",
+    "ImapMessageService",
+  ],
+  [
+    "ImapProtocolHandler",
+    "{ebb06c58-6ccd-4bde-9087-40663e0388ae}",
+    "@mozilla.org/network/protocol;1?name=imap",
+  ],
+  [
+    "ImapProtocolInfo",
+    "{1d9473bc-423a-4632-ad5d-802154e80f6f}",
+    "@mozilla.org/messenger/protocol/info;1?type=imap",
+  ],
+];
+
+ImapModuleLoader.prototype = {
+  QueryInterface: ChromeUtils.generateQI(["nsIObserver"]),
+
+  observe() {
+    // Nothing to do here, just need the entry so this is instantiated.
+  },
+
+  loadModule() {
+    if (Services.prefs.getBoolPref("mailnews.imap.jsmodule", false)) {
+      let registrar = Components.manager.QueryInterface(
+        Ci.nsIComponentRegistrar
+      );
+
+      for (let [
+        moduleName,
+        interfaceId,
+        contractId,
+        fileName,
+      ] of imapJSModules) {
+        // Register a module.
+        let classId = Components.ID(interfaceId);
+        registrar.registerFactory(
+          classId,
+          "",
+          contractId,
+          lazyFactoryFor(fileName || moduleName, moduleName)
+        );
+      }
+
+      dump("[ImapModuleLoader] Using ImapService.jsm\n");
+
+      const { ImapProtocolHandler } = ChromeUtils.import(
+        `resource:///modules/ImapProtocolHandler.jsm`
+      );
+      let protocolFlags =
+        Ci.nsIProtocolHandler.URI_NORELATIVE |
+        Ci.nsIProtocolHandler.URI_FORBIDS_AUTOMATIC_DOCUMENT_REPLACEMENT |
+        Ci.nsIProtocolHandler.URI_DANGEROUS_TO_LOAD |
+        Ci.nsIProtocolHandler.ALLOWS_PROXY |
+        Ci.nsIProtocolHandler.URI_FORBIDS_COOKIE_ACCESS |
+        Ci.nsIProtocolHandler.ORIGIN_IS_FULL_SPEC;
+
+      Services.io.registerProtocolHandler(
+        "imap",
+        new ImapProtocolHandler(),
+        protocolFlags,
+        Ci.nsIImapUrl.DEFAULT_IMAP_PORT
+      );
+    } else {
+      dump("[ImapModuleLoader] Using nsImapService.cpp\n");
+      // Ensure the imap protocol is actually registered.
+      Cc["@mozilla.org/network/protocol;1?name=imap"].getService(
+        Ci.nsIImapService
+      );
+    }
+  },
+};
+
+function lazyFactoryFor(fileName, constructorName) {
+  let factory = {
+    get scope() {
+      delete this.scope;
+      this.scope = ChromeUtils.import(`resource:///modules/${fileName}.jsm`);
+      return this.scope;
+    },
+    createInstance(interfaceID) {
+      let componentConstructor = this.scope[constructorName];
+      return new componentConstructor().QueryInterface(interfaceID);
+    },
+  };
+  return factory;
+}