modelcolumnmanager.cpp

Go to the documentation of this file.

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

#include <KLocale>

#include <QtCore/QEvent>
#include <QtCore/QTimer>
#include <QContextMenuEvent>
#include <QHeaderView>
#include <QMenu>
#include <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 *familyNameAction = menu.addAction( i18n( "Family Name" ) );
      familyNameAction->setCheckable( true );
      familyNameAction->setChecked(
        columns.contains( Akonadi::ContactsTreeModel::FamilyName ) );

      QAction *givenNameAction = menu.addAction( i18n( "Given Name" ) );
      givenNameAction->setCheckable( true );
      givenNameAction->setChecked(
        columns.contains( Akonadi::ContactsTreeModel::GivenName ) );

      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 *roleAction = menu.addAction( KABC::Addressee::roleLabel() );
      roleAction->setCheckable( true );
      roleAction->setChecked(
        columns.contains( Akonadi::ContactsTreeModel::Role ) );

      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 ( familyNameAction->isChecked() ) {
          columns << Akonadi::ContactsTreeModel::FamilyName;
        }
        if ( givenNameAction->isChecked() ) {
          columns << Akonadi::ContactsTreeModel::GivenName;
        }
        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 ( roleAction->isChecked() ) {
          columns << Akonadi::ContactsTreeModel::Role;
        }
        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 );
  }
}

#include "modelcolumnmanager.moc"