/*
    This file is part of KAddressBook.

    Copyright (c) 2009 Tobias Koenig <tokoe@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; either version 2 of the License, or
    (at your option) any later version.

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

#include "modelcolumnmanager.h"

#include "settings.h"

#include <kabc/addressee.h>
#include <klocale.h>

#include <QtCore/QEvent>
#include <QtCore/QTimer>
#include <QtGui/QContextMenuEvent>
#include <QtGui/QHeaderView>
#include <QtGui/QMenu>
#include <QtGui/QWidget>

ModelColumnManager::ModelColumnManager( Akonadi::ContactsTreeModel *model, QObject *parent )
  : QObject( parent ), mModel( model ), mWidget( 0 )
{
}

void ModelColumnManager::load()
{
  const QList<int> settingsColumns = Settings::contactModelColumns();
  Akonadi::ContactsTreeModel::Columns columns;

  foreach( int column, settingsColumns )
    columns.append( (Akonadi::ContactsTreeModel::Column)column );

  mModel->setColumns( columns );
}

void ModelColumnManager::store()
{
  const Akonadi::ContactsTreeModel::Columns columns = mModel->columns();
  QList<int> settingsColumns;

  foreach( int column, columns )
    settingsColumns.append( (int)column );

  Settings::setContactModelColumns( settingsColumns );
}

void ModelColumnManager::setWidget( QWidget *widget )
{
  mWidget = widget;
  mWidget->installEventFilter( this );
}

bool ModelColumnManager::eventFilter( QObject *watched, QEvent* event )
{
  if ( watched == mWidget ) {
    if ( event->type() == QEvent::ContextMenu ) {
      QMenu menu;

      Akonadi::ContactsTreeModel::Columns columns = mModel->columns();

      QAction *fullNameAction = menu.addAction( i18n( "Full Name" ) );
      fullNameAction->setCheckable( true );
      fullNameAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::FullName ) );
      fullNameAction->setEnabled( false );

      QAction *birthdayAction = menu.addAction( KABC::Addressee::birthdayLabel() );
      birthdayAction->setCheckable( true );
      birthdayAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::Birthday ) );

      QAction *homeAddressAction = menu.addAction( i18n( "Home Address" ) );
      homeAddressAction->setCheckable( true );
      homeAddressAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::HomeAddress ) );

      QAction *businessAddressAction = menu.addAction( i18n( "Business Address" ) );
      businessAddressAction->setCheckable( true );
      businessAddressAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::BusinessAddress ) );

      QAction *phoneNumbersAction = menu.addAction( i18n( "Phone Numbers" ) );
      phoneNumbersAction->setCheckable( true );
      phoneNumbersAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::PhoneNumbers ) );

      QAction *preferredEmailAction = menu.addAction( i18n( "Preferred EMail" ) );
      preferredEmailAction->setCheckable( true );
      preferredEmailAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::PreferredEmail ) );

      QAction *allEmailsAction = menu.addAction( i18n( "All EMails" ) );
      allEmailsAction->setCheckable( true );
      allEmailsAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::AllEmails ) );

      QAction *organizationAction = menu.addAction( KABC::Addressee::organizationLabel() );
      organizationAction->setCheckable( true );
      organizationAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::Organization ) );

      QAction *homepageAction = menu.addAction( KABC::Addressee::urlLabel() );
      homepageAction->setCheckable( true );
      homepageAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::Homepage ) );

      QAction *noteAction = menu.addAction( KABC::Addressee::noteLabel() );
      noteAction->setCheckable( true );
      noteAction->setChecked( columns.contains( Akonadi::ContactsTreeModel::Note ) );

      if ( menu.exec( ((QContextMenuEvent*)event)->globalPos() ) ) {
        Akonadi::ContactsTreeModel::Columns columns;

        if ( fullNameAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::FullName;
        if ( birthdayAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::Birthday;
        if ( homeAddressAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::HomeAddress;
        if ( businessAddressAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::BusinessAddress;
        if ( phoneNumbersAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::PhoneNumbers;
        if ( preferredEmailAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::PreferredEmail;
        if ( allEmailsAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::AllEmails;
        if ( organizationAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::Organization;
        if ( homepageAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::Homepage;
        if ( noteAction->isChecked() )
          columns << Akonadi::ContactsTreeModel::Note;

        mModel->setColumns( columns );
        QTimer::singleShot( 0, this, SLOT( adaptHeaderView() ) );
      }

      return true;
    } else
      return false;
  }

  return false;
}

void ModelColumnManager::adaptHeaderView()
{
  QHeaderView *view = qobject_cast<QHeaderView*>( mWidget );
  if ( view ) {
    view->resizeSections( QHeaderView::Stretch );

    view->setDefaultAlignment( mModel->columns().count() == 1 ? Qt::AlignCenter : Qt::AlignLeft );
  }
}
