kaddressbook/utils.cpp

   1 /*
   2   Copyright (c) 2013-2015 Montel Laurent <montel@kde.org>
   3
   4   This program is free software; you can redistribute it and/or modify it
   5   under the terms of the GNU General Public License, version 2, as
   6   published by the Free Software Foundation.
   7
   8   This program is distributed in the hope that it will be useful, but
   9   WITHOUT ANY WARRANTY; without even the implied warranty of
  10   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11   General Public License for more details.
  12
  13   You should have received a copy of the GNU General Public License along
  14   with this program; if not, write to the Free Software Foundation, Inc.,
  15   51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
  16 */
  17
  18 #include "utils.h"
  19
  20 #include <QItemSelectionModel>
  21
  22 #include <KContacts/Addressee>
  23 #include <AkonadiCore/EntityTreeModel>
  24 #include <KContacts/ContactGroup>
  25
  26 Akonadi::Item::List Utils::collectSelectedContactsItem(QItemSelectionModel *model)
  27 {
  28     Akonadi::Item::List lst;
  29
  30     const QModelIndexList indexes = model->selectedRows(0);
  31     for (int i = 0; i < indexes.count(); ++i) {
  32         const QModelIndex index = indexes.at(i);
  33         if (index.isValid()) {
  34             const Akonadi::Item item =
  35                 index.data(Akonadi::EntityTreeModel::ItemRole).value<Akonadi::Item>();
  36             if (item.isValid() && item.hasPayload<KContacts::Addressee>()) {
  37                 lst.append(item);
  38             }
  39         }
  40     }
  41     return lst;
  42 }
  43
  44 Akonadi::Item::List Utils::collectSelectedAllContactsItem(QItemSelectionModel *model)
  45 {
  46     Akonadi::Item::List lst;
  47
  48     const QModelIndexList indexes = model->selectedRows(0);
  49     for (int i = 0; i < indexes.count(); ++i) {
  50         const QModelIndex index = indexes.at(i);
  51         if (index.isValid()) {
  52             const Akonadi::Item item =
  53                 index.data(Akonadi::EntityTreeModel::ItemRole).value<Akonadi::Item>();
  54             if (item.isValid()) {
  55                 if (item.hasPayload<KContacts::Addressee>()  || item.hasPayload<KContacts::ContactGroup>()) {
  56                     lst.append(item);
  57                 }
  58             }
  59         }
  60     }
  61     return lst;
  62 }