/*
 * Copyright (c) 2004 Carsten Burghardt <burghardt@kde.org>
 * Copyright (c) 2009 Montel Laurent <montel@kde.org>
 *
 *  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; version 2 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, write to the Free Software
 *  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 *
 *  In addition, as a special exception, the copyright holders give
 *  permission to link the code of this program with any edition of
 *  the Qt library by Trolltech AS, Norway (or with modified versions
 *  of Qt that use the same license as Qt), and distribute linked
 *  combinations including the two.  You must obey the GNU General
 *  Public License in all respects for all of the code used other than
 *  Qt.  If you modify this file, you may extend this exception to
 *  your version of the file, but you are not obligated to do so.  If
 *  you do not wish to do so, delete this exception statement from
 *  your version.
 */

#include "folderrequester.h"
#include "folderselectiondialog.h"
#include "util/mailutil.h"
#include "kernel/mailkernel.h"

#include <messageviewer/utils/autoqpointer.h>

#include <Akonadi/CollectionFetchJob>

#include <KDialog>
#include <KIconLoader>
#include <KLineEdit>
#include <KLocale>

#include <QHBoxLayout>
#include <QKeyEvent>
#include <QToolButton>

namespace MailCommon {

FolderRequester::FolderRequester( QWidget *parent )
    : QWidget( parent ),
      mMustBeReadWrite( true ), mShowOutbox( true ), mNotCreateNewFolder( false )
{
    QHBoxLayout *hlay = new QHBoxLayout( this );
    hlay->setSpacing( KDialog::spacingHint() );
    hlay->setContentsMargins( 0, 0, 0, 0 );

    mEdit = new KLineEdit( this );
    mEdit->setClickMessage( i18n( "Select Folder" ) );
    mEdit->setTrapReturnKey(true);
    mEdit->setReadOnly( true );
    hlay->addWidget( mEdit );

    QToolButton *button = new QToolButton( this );
    button->setIcon( KIcon( QLatin1String("folder") ) );
    button->setIconSize( QSize( KIconLoader::SizeSmall, KIconLoader::SizeSmall ) );
    hlay->addWidget( button );
    connect( button, SIGNAL(clicked()), this, SLOT(slotOpenDialog()) );

    setSizePolicy( QSizePolicy( QSizePolicy::MinimumExpanding,
                                QSizePolicy::Fixed ) );
    setFocusPolicy( Qt::StrongFocus );
}

//-----------------------------------------------------------------------------
void FolderRequester::slotOpenDialog()
{
    FolderSelectionDialog::SelectionFolderOptions options = FolderSelectionDialog::EnableCheck ;
    options |= FolderSelectionDialog::HideVirtualFolder;
    options |= FolderSelectionDialog::NotUseGlobalSettings;
    if ( mNotCreateNewFolder ) {
        options |= FolderSelectionDialog::NotAllowToCreateNewFolder;
    }
    if ( !mShowOutbox ) {
        options |= FolderSelectionDialog::HideOutboxFolder;
    }

    MessageViewer::AutoQPointer<FolderSelectionDialog> dlg(
                new FolderSelectionDialog( this, options ) );

    dlg->setCaption( i18n( "Select Folder" ) );
    dlg->setModal( false );
    dlg->setSelectedCollection( mCollection );

    if ( dlg->exec() && dlg ) {
        setCollection( dlg->selectedCollection(), false );
    }
}

//-----------------------------------------------------------------------------
FolderRequester::~FolderRequester()
{
}

Akonadi::Collection FolderRequester::collection() const
{
    return mCollection;
}

void FolderRequester::setCollectionFullPath( const Akonadi::Collection &col )
{
    if ( KernelIf->collectionModel() ) {
        mEdit->setText( Util::fullCollectionPath( col ) );
    } else {
        mEdit->clear();
    }
}

//-----------------------------------------------------------------------------
void FolderRequester::setCollection( const Akonadi::Collection &collection, bool fetchCollection )
{
    mCollection = collection;
    if ( mCollection.isValid() ) {
        if ( fetchCollection ) {
            Akonadi::CollectionFetchJob *job =
                    new Akonadi::CollectionFetchJob( mCollection, Akonadi::CollectionFetchJob::Base, this );

            connect( job, SIGNAL(result(KJob*)),
                     this, SLOT(slotCollectionsReceived(KJob*)) );
        } else {
            setCollectionFullPath( mCollection );
        }
    } else if ( !mMustBeReadWrite ) { // the Local Folders root node was selected
        mEdit->setText( i18n( "Local Folders" ) );
    }

    emit folderChanged( mCollection );
}

void FolderRequester::slotCollectionsReceived( KJob *job )
{
    if ( job->error() ) {
        mCollection = Akonadi::Collection();
        mEdit->setText( i18n( "Please select a folder" ) );
        return;
    }

    const Akonadi::CollectionFetchJob *fetchJob = qobject_cast<Akonadi::CollectionFetchJob*>( job );
    const Akonadi::Collection::List collections = fetchJob->collections();

    if ( !collections.isEmpty() ) {
        const Akonadi::Collection collection = collections.first();
        // in case this is still the collection we are interested in, update
        if ( collection.id() == mCollection.id() ) {
            mCollection = collection;
            setCollectionFullPath( collection );
        }
    } else {
        // the requested collection doesn't exists anymore
        mCollection = Akonadi::Collection();
        mEdit->setText( i18n( "Please select a folder" ) );
    }
}

bool FolderRequester::hasCollection() const
{
    return mCollection.isValid();
}

//-----------------------------------------------------------------------------
void FolderRequester::keyPressEvent( QKeyEvent *e )
{
    if ( e->key() == Qt::Key_Space ) {
        slotOpenDialog();
    } else {
        e->ignore();
    }
}

void FolderRequester::setMustBeReadWrite( bool readwrite )
{
    mMustBeReadWrite = readwrite;
}

void FolderRequester::setShowOutbox( bool show )
{
    mShowOutbox = show;
}

void FolderRequester::setNotAllowToCreateNewFolder( bool notCreateNewFolder )
{
    mNotCreateNewFolder = notCreateNewFolder;
}


} // namespace MailCommon