/* Copyright (C) 2006 - 2014 Jan Kundrát <jkt@flaska.net>

   This file is part of the Trojita Qt IMAP e-mail client,
   http://trojita.flaska.net/

   This program is free software; you can redistribute it and/or
   modify it under the terms of the GNU General Public License as
   published by the Free Software Foundation; either version 2 of
   the License or (at your option) version 3 or any later version
   accepted by the membership of KDE e.V. (or its successor approved
   by the membership of KDE e.V.), which shall act as a proxy
   defined in Section 14 of version 3 of the license.

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

   You should have received a copy of the GNU General Public License
   along with this program.  If not, see <http://www.gnu.org/licenses/>.
*/
#include "Sendmail.h"

namespace MSA
{

Sendmail::Sendmail(QObject *parent, const QString &command, const QStringList &args):
    AbstractMSA(parent), command(command), args(args)
{
    proc = new QProcess(this);
    connect(proc, &QProcess::started, this, &Sendmail::handleStarted);
    connect(proc, static_cast<void (QProcess::*)(int)>(&QProcess::finished), this, &Sendmail::handleFinished);
    connect(proc, static_cast<void (QProcess::*)(QProcess::ProcessError)>(&QProcess::error), this, &Sendmail::handleError);
    connect(proc, &QIODevice::bytesWritten, this, &Sendmail::handleBytesWritten);
}

Sendmail::~Sendmail()
{
    proc->kill();
    proc->waitForFinished();
}

void Sendmail::sendMail(const QByteArray &from, const QList<QByteArray> &to, const QByteArray &data)
{
    // first +1 for the process startup
    // second +1 for waiting for the result
    emit progressMax(data.size() + 2);
    emit progress(0);
    QStringList myArgs = args;
    myArgs << QStringLiteral("-f") << QString::fromUtf8(from);
    for (QList<QByteArray>::const_iterator it = to.begin(); it != to.end(); ++it) {
        // On posix systems, process args are bytearrays, not strings--- but QProcess
        // takes strings.
        myArgs << QString::fromUtf8(*it);
    }
    writtenSoFar = 0;
    emit connecting();
    proc->start(command, myArgs);
    dataToSend = data;
}

void Sendmail::cancel()
{
    proc->terminate();
}

void Sendmail::handleStarted()
{
    // The process has started already -> +1
    emit progress(1);

    emit sending();
    proc->write(dataToSend);
    proc->closeWriteChannel();
}

void Sendmail::handleError(QProcess::ProcessError e)
{
    Q_UNUSED(e);
    emit error(tr("The sendmail process has failed: %1").arg(proc->errorString()));
}

void Sendmail::handleBytesWritten(qint64 bytes)
{
    writtenSoFar += bytes;
    // +1 due to starting at one
    emit progress(writtenSoFar + 1);
}

void Sendmail::handleFinished(const int exitCode)
{
    // that's the last one
    emit progressMax(dataToSend.size() + 2);

    if (exitCode == 0) {
        emit sent();
        return;
    }

    QByteArray allStdout = proc->readAllStandardOutput();
    QByteArray allStderr = proc->readAllStandardError();
    emit error(tr("The sendmail process has failed (%1):\n%2\n%3").arg(QString::number(exitCode), QString::fromUtf8(allStdout),
                                                                       QString::fromUtf8(allStderr)));
}

SendmailFactory::SendmailFactory(const QString &command, const QStringList &args):
    m_command(command), m_args(args)
{
}

SendmailFactory::~SendmailFactory()
{
}

AbstractMSA *SendmailFactory::create(QObject *parent) const
{
    return new Sendmail(parent, m_command, m_args);
}

}
