From 3e9a764d73e7feaaaf49a6121c33043e79a779bd Mon Sep 17 00:00:00 2001
From: Andris Reinman <andris@reinman.eu>
Date: Thu, 25 May 2023 11:28:30 +0300
Subject: [PATCH] Fixed tests

---
 imap-core/lib/indexer/parse-mime-tree.js | 87 ++++++++++++++++++++----
 imap-core/test/parse-mime-tree-test.js   | 19 +++++-
 lib/mailbox-handler.js                   |  1 -
 3 files changed, 91 insertions(+), 16 deletions(-)

diff --git a/imap-core/lib/indexer/parse-mime-tree.js b/imap-core/lib/indexer/parse-mime-tree.js
index 1fe767eb..baf35ce6 100644
--- a/imap-core/lib/indexer/parse-mime-tree.js
+++ b/imap-core/lib/indexer/parse-mime-tree.js
@@ -1,7 +1,6 @@
 'use strict';
 
 const addressparser = require('nodemailer/lib/addressparser');
-const libmime = require('libmime');
 
 /**
  * Parses a RFC822 message into a structured object (JSON compatible)
@@ -255,19 +254,81 @@ class MIMEParser {
      * @return {Object} Parsed value
      */
     parseValueParams(headerValue) {
-        let parsed = libmime.parseHeaderValue(headerValue) || {};
-
-        let subtype = (parsed.value || '').split('/');
-        let type = (subtype.shift() || '').toLowerCase();
-        subtype = subtype.join('/');
-
-        return {
-            value: parsed.value || '',
-            type,
-            subtype,
-            params: parsed.params || {},
-            hasParams: !!Object.keys(parsed.params || {}).length
+        let data = {
+            value: '',
+            type: '',
+            subtype: '',
+            params: {}
         };
+        let match;
+        let processEncodedWords = {};
+
+        let charsetRequired = new WeakSet();
+
+        (headerValue || '').split(';').forEach((part, i) => {
+            let key, value;
+            if (!i) {
+                data.value = part.trim();
+                data.subtype = data.value.split('/');
+                data.type = (data.subtype.shift() || '').toLowerCase();
+                data.subtype = data.subtype.join('/');
+                return;
+            }
+            value = part.split('=');
+            key = (value.shift() || '').trim().toLowerCase();
+            value = value.join('=').replace(/^['"\s]*|['"\s]*$/g, '');
+
+            // Do not touch headers that have strange looking keys, keep these
+            // only in the unparsed array
+            if (/[^a-zA-Z0-9\-*]/.test(key) || key.length >= 100) {
+                return;
+            }
+
+            // This regex allows for an optional trailing asterisk, for headers
+            // which are encoded with lang/charset info as well as a continuation.
+            // See https://tools.ietf.org/html/rfc2231 section 4.1.
+
+            if ((match = key.match(/^([^*]+)\*(\d)?\*?$/))) {
+                if (!processEncodedWords[match[1]]) {
+                    processEncodedWords[match[1]] = [];
+
+                    if (key.match(/^([^*]+)\*(\d)?\*$/)) {
+                        // must have charset
+                        charsetRequired.add(processEncodedWords[match[1]]);
+                    }
+                }
+                processEncodedWords[match[1]][Number(match[2]) || 0] = value;
+            } else {
+                data.params[key] = value;
+            }
+            data.hasParams = true;
+        });
+
+        // convert extended mime word into a regular one
+        Object.keys(processEncodedWords).forEach(key => {
+            let charset = '';
+            let value = '';
+
+            let isCharsetRequired = charsetRequired.has(processEncodedWords[key]);
+            if (!isCharsetRequired) {
+                charset = 'utf-8';
+                value = processEncodedWords[key].join('').replace(/%/g, '=');
+            } else {
+                processEncodedWords[key].forEach((val, i) => {
+                    if (!i) {
+                        let parts = val.split("'"); // eslint-disable-line quotes
+                        charset = parts.shift();
+                        parts.shift(); // lang argument, ignored
+                        val = parts.join("'"); // eslint-disable-line quotes
+                    }
+                    value += val.replace(/%/g, '=');
+                });
+            }
+
+            data.params[key] = '=?' + (charset || 'ISO-8859-1').toUpperCase() + '?Q?' + value + '?=';
+        });
+
+        return data;
     }
 
     /**
diff --git a/imap-core/test/parse-mime-tree-test.js b/imap-core/test/parse-mime-tree-test.js
index 1316a334..f4821a28 100644
--- a/imap-core/test/parse-mime-tree-test.js
+++ b/imap-core/test/parse-mime-tree-test.js
@@ -9,7 +9,7 @@ const expect = chai.expect;
 chai.config.includeStack = true;
 
 describe('#parseValueParams', function () {
-    it('should return as is', function () {
+    it('should return continuation value as mime-word', function () {
         let parser = new MIMEParser();
         const parsed = parser.parseValueParams(
             'text/plain;\n' +
@@ -24,7 +24,22 @@ describe('#parseValueParams', function () {
             subtype: 'plain',
             params: {
                 'x-apple-part-url': '99AFDE83-8953-43B4-BE59-F59D6160AFAB',
-                name: 'emailengine_uuendamise_kasud_ja_muud_asjad_ja_veelgi_pikem_pealkiri.txt'
+                name: '=?UTF-8?Q?emailengine_uuendamise_kasud_ja_muud_asjad_ja_veelgi_pikem_pealkiri.txt?='
+            },
+            hasParams: true
+        });
+    });
+
+    it('should return continuation value as mime-word', function () {
+        let parser = new MIMEParser();
+        const parsed = parser.parseValueParams('image/jpeg; name="=?UTF-8?Q?sw=C3=A4n=2Ejpg?="');
+
+        expect(parsed).to.deep.equal({
+            value: 'image/jpeg',
+            type: 'image',
+            subtype: 'jpeg',
+            params: {
+                name: '=?UTF-8?Q?sw=C3=A4n=2Ejpg?='
             },
             hasParams: true
         });
diff --git a/lib/mailbox-handler.js b/lib/mailbox-handler.js
index 28ed413d..998b0b54 100644
--- a/lib/mailbox-handler.js
+++ b/lib/mailbox-handler.js
@@ -79,7 +79,6 @@ class MailboxHandler {
                                 return callback(err);
                             }
 
-                            console.log('MAILBOX CREATED');
                             publish(this.redis, {
                                 ev: MAILBOX_CREATED,
                                 user,