1 files changed, 186 insertions, 0 deletions
diff --git a/comm/mail/extensions/openpgp/content/modules/filtersWrapper.jsm b/comm/mail/extensions/openpgp/content/modules/filtersWrapper.jsm
new file mode 100644
index 0000000000..a43fb29e87
--- /dev/null
+++ b/comm/mail/extensions/openpgp/content/modules/filtersWrapper.jsm
@@ -0,0 +1,186 @@
+/*
+ * 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 https://mozilla.org/MPL/2.0/.
+ */
+
+"use strict";
+
+var EXPORTED_SYMBOLS = ["EnigmailFiltersWrapper"];
+
+const { EnigmailConstants } = ChromeUtils.import(
+  "chrome://openpgp/content/modules/constants.jsm"
+);
+
+var gEnigmailFilters = null;
+
+var l10n = new Localization(["messenger/openpgp/openpgp.ftl"], true);
+
+/**
+ * filter action for creating a decrypted version of the mail and
+ * deleting the original mail at the same time
+ */
+const filterActionMoveDecrypt = {
+  id: EnigmailConstants.FILTER_MOVE_DECRYPT,
+  name: l10n.formatValueSync("filter-decrypt-move-label"),
+  value: "movemessage",
+  applyAction(aMsgHdrs, aActionValue, aListener, aType, aMsgWindow) {
+    if (gEnigmailFilters) {
+      gEnigmailFilters.moveDecrypt.applyAction(
+        aMsgHdrs,
+        aActionValue,
+        aListener,
+        aType,
+        aMsgWindow
+      );
+    } else {
+      aListener.OnStartCopy();
+      aListener.OnStopCopy(0);
+    }
+  },
+
+  isValidForType(type, scope) {
+    return gEnigmailFilters
+      ? gEnigmailFilters.moveDecrypt.isValidForType(type, scope)
+      : false;
+  },
+
+  validateActionValue(value, folder, type) {
+    if (gEnigmailFilters) {
+      return gEnigmailFilters.moveDecrypt.validateActionValue(
+        value,
+        folder,
+        type
+      );
+    }
+    return null;
+  },
+
+  allowDuplicates: false,
+  isAsync: true,
+  needsBody: true,
+};
+
+/**
+ * filter action for creating a decrypted copy of the mail, leaving the original
+ * message untouched
+ */
+const filterActionCopyDecrypt = {
+  id: EnigmailConstants.FILTER_COPY_DECRYPT,
+  name: l10n.formatValueSync("filter-decrypt-copy-label"),
+  value: "copymessage",
+  applyAction(aMsgHdrs, aActionValue, aListener, aType, aMsgWindow) {
+    if (gEnigmailFilters) {
+      gEnigmailFilters.copyDecrypt.applyAction(
+        aMsgHdrs,
+        aActionValue,
+        aListener,
+        aType,
+        aMsgWindow
+      );
+    } else {
+      aListener.OnStartCopy();
+      aListener.OnStopCopy(0);
+    }
+  },
+
+  isValidForType(type, scope) {
+    return gEnigmailFilters
+      ? gEnigmailFilters.copyDecrypt.isValidForType(type, scope)
+      : false;
+  },
+
+  validateActionValue(value, folder, type) {
+    if (gEnigmailFilters) {
+      return gEnigmailFilters.copyDecrypt.validateActionValue(
+        value,
+        folder,
+        type
+      );
+    }
+    return null;
+  },
+
+  allowDuplicates: false,
+  isAsync: true,
+  needsBody: true,
+};
+
+/**
+ * filter action for to encrypt a mail to a specific key
+ */
+const filterActionEncrypt = {
+  id: EnigmailConstants.FILTER_ENCRYPT,
+  name: l10n.formatValueSync("filter-encrypt-label"),
+  value: "encryptto",
+  applyAction(aMsgHdrs, aActionValue, aListener, aType, aMsgWindow) {
+    if (gEnigmailFilters) {
+      gEnigmailFilters.encrypt.applyAction(
+        aMsgHdrs,
+        aActionValue,
+        aListener,
+        aType,
+        aMsgWindow
+      );
+    } else {
+      aListener.OnStartCopy();
+      aListener.OnStopCopy(0);
+    }
+  },
+
+  isValidForType(type, scope) {
+    return gEnigmailFilters ? gEnigmailFilters.encrypt.isValidForType() : false;
+  },
+
+  validateActionValue(value, folder, type) {
+    if (gEnigmailFilters) {
+      return gEnigmailFilters.encrypt.validateActionValue(value, folder, type);
+    }
+    return null;
+  },
+
+  allowDuplicates: false,
+  isAsync: true,
+  needsBody: true,
+};
+
+/**
+ * Add a custom filter action. If the filter already exists, do nothing
+ * (for example, if addon is disabled and re-enabled)
+ *
+ * @param filterObj - nsIMsgFilterCustomAction
+ */
+function addFilterIfNotExists(filterObj) {
+  let filterService = Cc[
+    "@mozilla.org/messenger/services/filters;1"
+  ].getService(Ci.nsIMsgFilterService);
+
+  let foundFilter = null;
+  try {
+    foundFilter = filterService.getCustomAction(filterObj.id);
+  } catch (ex) {}
+
+  if (!foundFilter) {
+    filterService.addCustomAction(filterObj);
+  }
+}
+
+var EnigmailFiltersWrapper = {
+  onStartup() {
+    let { EnigmailFilters } = ChromeUtils.import(
+      "chrome://openpgp/content/modules/filters.jsm"
+    );
+    gEnigmailFilters = EnigmailFilters;
+
+    addFilterIfNotExists(filterActionMoveDecrypt);
+    addFilterIfNotExists(filterActionCopyDecrypt);
+    addFilterIfNotExists(filterActionEncrypt);
+
+    gEnigmailFilters.onStartup();
+  },
+
+  onShutdown() {
+    gEnigmailFilters.onShutdown();
+    gEnigmailFilters = null;
+  },
+};