/*
  Copyright (C) 2009 Klaralvdalens Datakonsult AB, a KDAB Group company, info@kdab.net
  Copyright (c) 2009 Leo Franchi <lfranchi@kde.org>

  This library is free software; you can redistribute it and/or modify it
  under the terms of the GNU Library General Public License as published by
  the Free Software Foundation; either version 2 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 Library General Public
  License for more details.

  You should have received a copy of the GNU Library General Public License
  along with this library; see the file COPYING.LIB.  If not, write to the
  Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
  02110-1301, USA.
*/

#include "signencryptjob.h"

#include "contentjobbase_p.h"
#include "kleo/cryptobackendfactory.h"
#include "kleo/cryptobackend.h"
#include "kleo/enum.h"
#include "kleo/signencryptjob.h"
#include "util.h"

#include <kdebug.h>
#include <kmime/kmime_message.h>
#include <kmime/kmime_content.h>
#include <kmime/kmime_util.h>
#include <QBuffer>

#include <gpgme++/global.h>
#include <gpgme++/signingresult.h>
#include <gpgme++/encryptionresult.h>
#include <sstream>

using namespace Message;

class Message::SignEncryptJobPrivate : public ContentJobBasePrivate
{
  public:
    SignEncryptJobPrivate( SignEncryptJob *qq )
      : ContentJobBasePrivate( qq )
      , content( 0 )
    {
    }

    KMime::Content* content;
    std::vector<GpgME::Key> signers;
    Kleo::CryptoMessageFormat format;

    std::vector<GpgME::Key> encKeys;
    QStringList recipients;
    
    // copied from messagecomposer.cpp
    bool binaryHint( Kleo::CryptoMessageFormat f )
    {
      switch ( f ) {
      case Kleo::SMIMEFormat:
      case Kleo::SMIMEOpaqueFormat:
        return true;
      default:
      case Kleo::OpenPGPMIMEFormat:
      case Kleo::InlineOpenPGPFormat:
        return false;
      }
    }

    Q_DECLARE_PUBLIC( SignEncryptJob )
};

SignEncryptJob::SignEncryptJob( QObject *parent )
  : ContentJobBase( *new SignEncryptJobPrivate( this ), parent )
{
}

SignEncryptJob::~SignEncryptJob()
{
}

void SignEncryptJob::setContent( KMime::Content* content )
{
  Q_D( SignEncryptJob );

  Q_ASSERT( content );

  d->content = content;
}

void SignEncryptJob::setCryptoMessageFormat( Kleo::CryptoMessageFormat format)
{
  Q_D( SignEncryptJob );

  // There *must* be a concrete format set at this point.
  Q_ASSERT( format == Kleo::OpenPGPMIMEFormat
            || format == Kleo::InlineOpenPGPFormat
            || format == Kleo::SMIMEFormat
            || format == Kleo::SMIMEOpaqueFormat );
  d->format = format;
}

void SignEncryptJob::setSigningKeys( std::vector<GpgME::Key>& signers )
{
  Q_D( SignEncryptJob );

  d->signers = signers;
}

KMime::Content* SignEncryptJob::origContent()
{
  Q_D( SignEncryptJob );

  return d->content;

}


void SignEncryptJob::setEncryptionKeys( std::vector<GpgME::Key> keys )
{
  Q_D( SignEncryptJob );

  d->encKeys = keys;
}

void SignEncryptJob::setRecipients( QStringList recipients ) {
  Q_D( SignEncryptJob );

  d->recipients = recipients;
}

QStringList SignEncryptJob::recipients() {
  Q_D( SignEncryptJob );

  return d->recipients;
}

std::vector<GpgME::Key> SignEncryptJob::encryptionKeys() {
  Q_D( SignEncryptJob );

  return d->encKeys;
}


void SignEncryptJob::process()
{
  Q_D( SignEncryptJob );
  Q_ASSERT( d->resultContent == 0 ); // Not processed before.

  // if setContent hasn't been called, we assume that a subjob was added
  // and we want to use that
  if( !d->content || !d->content->hasContent() ) {
    Q_ASSERT( d->subjobContents.size() == 1 );
    d->content = d->subjobContents.first();
  }

  d->resultContent = new KMime::Content;

  const Kleo::CryptoBackend::Protocol *proto = 0;
  if( d->format & Kleo::AnyOpenPGP ) {
    proto = Kleo::CryptoBackendFactory::instance()->openpgp();
  } else if( d->format & Kleo::AnySMIME ) {
    proto = Kleo::CryptoBackendFactory::instance()->smime();
  }

  Q_ASSERT( proto );


  kDebug() << "creating signencrypt from:" << proto->name() << proto->displayName();
  std::auto_ptr<Kleo::SignEncryptJob> job( proto->signEncryptJob( !d->binaryHint( d->format ), d->format == Kleo::InlineOpenPGPFormat ) );
  QByteArray encBody;

  // replace simple LFs by CRLFs for all MIME supporting CryptPlugs
  // according to RfC 2633, 3.1.1 Canonicalization
  d->content->assemble();
  QByteArray content;
  if( d->format & Kleo::InlineOpenPGPFormat ) {
    content = KMime::LFtoCRLF( d->content->body() );
  } else {
    content = KMime::LFtoCRLF( d->content->encodedContent() );
  }

  kDebug() << "signing and encrypting content:" << content;
  const std::pair<GpgME::SigningResult,GpgME::EncryptionResult> res = job->exec( d->signers, d->encKeys,
                                        content,
                                        false,
                                        encBody );

  if ( res.first.error() ) {
    kDebug() << "signing failed:" << res.first.error().asString();
    setError( res.first.error().code() );
    setErrorText( QString::fromLocal8Bit( res.first.error().asString() ) );
  }
  if ( res.second.error() ) {
    kDebug() << "encrypyting failed:" << res.second.error().asString();
    setError( res.second.error().code() );
    setErrorText( QString::fromLocal8Bit( res.second.error().asString() ) );
  }

  // exec'ed jobs don't delete themselves
  job->deleteLater();
  QByteArray signatureHashAlgo =  res.first.createdSignature( 0 ).hashAlgorithmAsString();

  d->resultContent = Message::Util::composeHeadersAndBody( d->content, encBody, d->format, true, signatureHashAlgo );
//  d->resultContent->setBody( signature );
  emitResult();
}

#include "signencryptjob.moc"
