/*
  Copyright (c) 2011-2012 - Tőkés Attila

  This file is part of SmtpClient for Qt.

  This library is free software; you can redistribute it and/or
  modify it under the terms of the GNU Lesser General Public
  License as published by the Free Software Foundation; either
  version 2.1 of the License, or (at your option) any later version.

  This library 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
  Lesser General Public License for more details.

  See the LICENSE file for more details.
*/

#include "smtp/mimemessage.h"

#include <QDateTime>
#include <QLocale>
#include <typeinfo>

#include "smtp/quotedprintable.h"

/* [1] Constructors and Destructors */
MimeMessage::MimeMessage(bool createAutoMimeContent)
    : replyTo(nullptr), hEncoding(MimePart::_8Bit) {
  if (createAutoMimeContent) this->content = new MimeMultiPart();

  autoMimeContentCreated = createAutoMimeContent;
}

MimeMessage::~MimeMessage() {
  if (this->autoMimeContentCreated) {
    this->autoMimeContentCreated = false;
    delete (this->content);
  }
}

/* [1] --- */

/* [2] Getters and Setters */
MimePart &MimeMessage::getContent() { return *content; }

void MimeMessage::setContent(MimePart *content) {
  if (this->autoMimeContentCreated) {
    this->autoMimeContentCreated = false;
    delete (this->content);
  }
  this->content = content;
}

void MimeMessage::setReplyTo(EmailAddress *rto) { replyTo = rto; }

void MimeMessage::setSender(EmailAddress *e) {
  this->sender = e;
  e->setParent(this);
}

void MimeMessage::addRecipient(EmailAddress *rcpt, RecipientType type) {
  switch (type) {
    case To:
      recipientsTo << rcpt;
      break;
    case Cc:
      recipientsCc << rcpt;
      break;
    case Bcc:
      recipientsBcc << rcpt;
      break;
  }

  rcpt->setParent(this);
}

void MimeMessage::addTo(EmailAddress *rcpt) { this->recipientsTo << rcpt; }

void MimeMessage::addCc(EmailAddress *rcpt) { this->recipientsCc << rcpt; }

void MimeMessage::addBcc(EmailAddress *rcpt) { this->recipientsBcc << rcpt; }

void MimeMessage::setSubject(const QString &subject) {
  this->subject = subject;
}

void MimeMessage::addPart(MimePart *part) {
  if (typeid(*content) == typeid(MimeMultiPart)) {
    ((MimeMultiPart *)content)->addPart(part);
  };
}

void MimeMessage::setInReplyTo(const QString &inReplyTo) {
  mInReplyTo = inReplyTo;
}

void MimeMessage::setHeaderEncoding(MimePart::Encoding hEnc) {
  this->hEncoding = hEnc;
}

const EmailAddress &MimeMessage::getSender() const { return *sender; }

const QList<EmailAddress *> &MimeMessage::getRecipients(
    RecipientType type) const {
  switch (type) {
    default:
    case To:
      return recipientsTo;
    case Cc:
      return recipientsCc;
    case Bcc:
      return recipientsBcc;
  }
}

const EmailAddress *MimeMessage::getReplyTo() const { return replyTo; }

const QString &MimeMessage::getSubject() const { return subject; }

const QList<MimePart *> &MimeMessage::getParts() const {
  if (typeid(*content) == typeid(MimeMultiPart)) {
    return ((MimeMultiPart *)content)->getParts();
  } else {
    auto *res = new QList<MimePart *>();
    res->append(content);
    return *res;
  }
}

/* [2] --- */

/* [3] Public Methods */

QString MimeMessage::toString() {
  QString mime;

  /* =========== MIME HEADER ============ */

  /* ---------- Sender / From ----------- */
  mime = "From:";
  if (!sender->getName().isEmpty()) {
    switch (hEncoding) {
      case MimePart::Base64:
        mime += " =?utf-8?B?" +
                QByteArray().append(sender->getName().toUtf8()).toBase64() +
                "?=";
        break;
      case MimePart::QuotedPrintable:
        mime += " =?utf-8?Q?" +
                QuotedPrintable::encode(
                    QByteArray().append(sender->getName().toUtf8()))
                    .replace(' ', "_")
                    .replace(':', "=3A") +
                "?=";
        break;
      default:
        mime += " " + sender->getName();
    }
  }
  mime += " <" + sender->getAddress() + ">\r\n";
  /* ---------------------------------- */

  /* ------- Recipients / To ---------- */
  mime += "To:";
  QList<EmailAddress *>::iterator it;
  int i;
  for (i = 0, it = recipientsTo.begin(); it != recipientsTo.end(); ++it, ++i) {
    if (i != 0) {
      mime += ",";
    }

    if (!(*it)->getName().isEmpty()) {
      switch (hEncoding) {
        case MimePart::Base64:
          mime += " =?utf-8?B?" +
                  QByteArray().append((*it)->getName().toUtf8()).toBase64() +
                  "?=";
          break;
        case MimePart::QuotedPrintable:
          mime += " =?utf-8?Q?" +
                  QuotedPrintable::encode(
                      QByteArray().append((*it)->getName().toUtf8()))
                      .replace(' ', "_")
                      .replace(':', "=3A") +
                  "?=";
          break;
        default:
          mime += " " + (*it)->getName();
      }
    }
    mime += " <" + (*it)->getAddress() + ">";
  }
  mime += "\r\n";
  /* ---------------------------------- */

  /* ------- Recipients / Cc ---------- */
  if (!recipientsCc.empty()) {
    mime += "Cc:";
  }
  for (i = 0, it = recipientsCc.begin(); it != recipientsCc.end(); ++it, ++i) {
    if (i != 0) {
      mime += ",";
    }

    if ((*it)->getName() != "") {
      switch (hEncoding) {
        case MimePart::Base64:
          mime += " =?utf-8?B?" +
                  QByteArray().append((*it)->getName().toUtf8()).toBase64() +
                  "?=";
          break;
        case MimePart::QuotedPrintable:
          mime += " =?utf-8?Q?" +
                  QuotedPrintable::encode(
                      QByteArray().append((*it)->getName().toUtf8()))
                      .replace(' ', "_")
                      .replace(':', "=3A") +
                  "?=";
          break;
        default:
          mime += " " + (*it)->getName();
      }
    }
    mime += " <" + (*it)->getAddress() + ">";
  }
  if (!recipientsCc.empty()) {
    mime += "\r\n";
  }
  /* ---------------------------------- */

  /* ------------ Subject ------------- */
  mime += "Subject: ";

  switch (hEncoding) {
    case MimePart::Base64:
      mime += "=?utf-8?B?" + QByteArray().append(subject.toUtf8()).toBase64() +
              "?=";
      break;
    case MimePart::QuotedPrintable:
      mime += "=?utf-8?Q?" +
              QuotedPrintable::encode(QByteArray().append(subject.toUtf8()))
                  .replace(' ', "_")
                  .replace(':', "=3A") +
              "?=";
      break;
    default:
      mime += subject;
  }
  mime += "\r\n";
  /* ---------------------------------- */

  /* ---------- Reply-To -------------- */
  if (replyTo) {
    mime += "Reply-To: ";
    if (!replyTo->getName().isEmpty()) {
      switch (hEncoding) {
        case MimePart::Base64:
          mime += " =?utf-8?B?" +
                  QByteArray().append(replyTo->getName().toUtf8()).toBase64() +
                  "?=";
          break;
        case MimePart::QuotedPrintable:
          mime += " =?utf-8?Q?" +
                  QuotedPrintable::encode(
                      QByteArray().append(replyTo->getName().toUtf8()))
                      .replace(' ', "_")
                      .replace(':', "=3A") +
                  "?=";
          break;
        default:
          mime += " " + replyTo->getName();
      }
    }
    mime += " <" + replyTo->getAddress() + ">\r\n";
  }

  /* ---------------------------------- */

  mime += "MIME-Version: 1.0\r\n";
  if (!mInReplyTo.isEmpty()) {
    mime += "In-Reply-To: <" + mInReplyTo + ">\r\n";
    mime += "References: <" + mInReplyTo + ">\r\n";
  }

  QDateTime now = QDateTime::currentDateTime();
#if QT_VERSION_MAJOR < 5  // Qt4 workaround since RFC2822Date isn't defined
  QString shortDayName =
      QLocale::c().dayName(now.date().dayOfWeek(), QLocale::ShortFormat);
  QString shortMonthName =
      QLocale::c().monthName(now.date().month(), QLocale::ShortFormat);
  int utcOffset = now.secsTo(QDateTime(now.date(), now.time(), Qt::UTC)) / 60;
  char timezoneSign = utcOffset >= 0 ? '+' : '-';
  utcOffset = utcOffset >= 0 ? utcOffset : -utcOffset;
  QString timezone = QString("%1%2%3")
                         .arg(timezoneSign)
                         .arg(utcOffset / 60, 2, 10, QChar('0'))
                         .arg(utcOffset % 60, 2, 10, QChar('0'));
  mime += QString("Date: %1\r\n")
              .arg(now.toString("%1, dd %2 yyyy hh:mm:ss %3")
                       .arg(shortDayName)
                       .arg(shortMonthName)
                       .arg(timezone));
#else   // Qt5 supported
  mime += QString("Date: %1\r\n").arg(now.toString(Qt::RFC2822Date));
#endif  // support RFC2822Date

  mime += content->toString();
  return mime;
}

/* [3] --- */