/*
    Copyright (c) 2010 Klarälvdalens Datakonsult AB,
                       a KDAB Group company <info@kdab.com>
    Author: Kevin Ottens <kevin@kdab.com>

    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.
*/

#ifndef RETRIEVEITEMSTASK_H
#define RETRIEVEITEMSTASK_H

#include <kimap/fetchjob.h>

#include "resourcetask.h"

namespace Akonadi {
  class Session;
}

class RetrieveItemsTask : public ResourceTask
{
  Q_OBJECT

public:
  explicit RetrieveItemsTask( ResourceStateInterface::Ptr resource, Akonadi::Session *session = 0, QObject *parent = 0 );
  virtual ~RetrieveItemsTask();

  void setFastSyncEnabled( bool fastSync );
  bool isFastSyncEnabled() const;

private slots:
  void onFetchForBodyCheckDone( KJob *job );
  void onPreExpungeSelectDone( KJob *job );
  void onExpungeDone( KJob *job );

  void onFinalSelectDone( KJob *job );

  void onHeadersReceived( const QString &mailBox, const QMap<qint64, qint64> &uids,
                          const QMap<qint64, qint64> &sizes,
                          const QMap<qint64, KIMAP::MessageFlags> &flags,
                          const QMap<qint64, KIMAP::MessagePtr> &messages );
  void onHeadersFetchDone( KJob *job );

  void onFlagsReceived( const QString &mailBox, const QMap<qint64, qint64> &uids,
                        const QMap<qint64, qint64> &sizes,
                        const QMap<qint64, KIMAP::MessageFlags> &flags,
                        const QMap<qint64, KIMAP::MessagePtr> &messages );
  void onFlagsFetchDone( KJob *job );

protected:
  virtual void doStart( KIMAP::Session *session );

private:
  void checkForMissingBodies();
  void startRetrievalTasks();
  void triggerPreExpungeSelect( const QString &mailBox );
  void triggerExpunge( const QString &mailBox );
  void triggerFinalSelect( const QString &mailBox );

  void listFlagsForImapSet( const KIMAP::ImapSet& set );

  KIMAP::Session *m_session;
  Akonadi::Session *m_akonadiSession;
  bool m_fastSync;
  QList<qint64> m_messageUidsMissingBody;
  int m_fetchedMissingBodies;
};

#endif
