1 files changed, 126 insertions, 0 deletions
diff --git a/web-ui/app/js/services/model/mail.js b/web-ui/app/js/services/model/mail.js
new file mode 100644
index 00000000..64a10c1c
--- /dev/null
+++ b/web-ui/app/js/services/model/mail.js
@@ -0,0 +1,126 @@
+/*
+ * Copyright (c) 2014 ThoughtWorks, Inc.
+ *
+ * Pixelated is free software: you can redistribute it and/or modify
+ * it under the terms of the GNU Affero General Public License as published by
+ * the Free Software Foundation, either version 3 of the License, or
+ * (at your option) any later version.
+ *
+ * Pixelated 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 Pixelated. If not, see <http://www.gnu.org/licenses/>.
+ */
+define(['helpers/contenttype'], function (contentType) {
+  'use strict';
+  function isSentMail() {
+    return _.has(this, 'mailbox') && this.mailbox.toUpperCase() === 'SENT';
+  }
+
+  function isDraftMail() {
+    return _.has(this, 'mailbox') && this.mailbox.toUpperCase() === 'DRAFTS';
+  }
+
+  function isInTrash() {
+    return _.has(this, 'mailbox') && this.mailbox.toUpperCase() === 'TRASH';
+  }
+
+  function setDraftReplyFor(ident) {
+    this.draft_reply_for = ident;
+  }
+
+  function replyToAddress() {
+    return {
+      to: [this.replying.single],
+      cc: []
+    };
+  }
+
+  function replyToAllAddress() {
+    return {
+      to: this.replying.all['to-field'],
+      cc: this.replying.all['cc-field']
+    };
+  }
+
+  function getHeadersFromMailPart (rawBody) {
+    var lines, headerLines, endOfHeaders, headers;
+
+    lines = rawBody.split('\n');
+    endOfHeaders = _.indexOf(lines, '');
+    headerLines = lines.slice(0, endOfHeaders);
+
+    headers = _.map(headerLines, function (headerLine) {
+      return _.map(headerLine.split(':'), function(elem){return elem.trim();});
+    });
+
+    return _.object(headers);
+  }
+
+  function getBodyFromMailPart (rawBody) {
+    var lines, endOfHeaders;
+
+    lines = rawBody.split('\n');
+    endOfHeaders = _.indexOf(lines, '');
+
+    return lines.slice(endOfHeaders + 1).join('\n');
+  }
+
+  function parseWithHeaders(rawBody) {
+    return {headers: getHeadersFromMailPart(rawBody), body: getBodyFromMailPart(rawBody)};
+  }
+
+  function getMailMultiParts () {
+    var mediaType = this.getMailMediaType();
+    var boundary = '--' + mediaType.params.boundary + '\n';
+    var finalBoundary = '--' + mediaType.params.boundary + '--';
+
+    var bodyParts = this.body.split(finalBoundary)[0].split(boundary);
+
+    bodyParts = _.reject(bodyParts, function(bodyPart) { return _.isEmpty(bodyPart.trim()); });
+
+    return _.map(bodyParts, parseWithHeaders);
+  }
+
+  function getMailMediaType () {
+    return new contentType.MediaType(this.header.content_type);
+  }
+
+  function isMailMultipartAlternative () {
+    return this.getMailMediaType().type === 'multipart/alternative';
+  }
+
+  function availableBodyPartsContentType () {
+    var bodyParts = this.getMailMultiParts();
+
+    return _.pluck(_.pluck(bodyParts, 'headers'), 'Content-Type');
+  }
+
+  function getMailPartByContentType (contentType) {
+    var bodyParts = this.getMailMultiParts();
+
+    return _.findWhere(bodyParts, {headers: {'Content-Type': contentType}});
+  }
+
+  return {
+    create: function (mail) {
+      if (!mail) { return; }
+
+      mail.isSentMail = isSentMail;
+      mail.isDraftMail = isDraftMail;
+      mail.isInTrash = isInTrash;
+      mail.setDraftReplyFor = setDraftReplyFor;
+      mail.replyToAddress = replyToAddress;
+      mail.replyToAllAddress = replyToAllAddress;
+      mail.getMailMediaType = getMailMediaType;
+      mail.isMailMultipartAlternative = isMailMultipartAlternative;
+      mail.getMailMultiParts = getMailMultiParts;
+      mail.availableBodyPartsContentType = availableBodyPartsContentType;
+      mail.getMailPartByContentType = getMailPartByContentType;
+      return mail;
+    }
+  };
+});