File: entity-filter-model.cpp

package info (click to toggle)
ktp-text-ui 0.8.1-1
  • links: PTS, VCS
  • area: main
  • in suites: jessie, jessie-kfreebsd
  • size: 3,680 kB
  • ctags: 1,142
  • sloc: cpp: 9,451; sh: 15; makefile: 9
file content (129 lines) | stat: -rw-r--r-- 4,784 bytes parent folder | download
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
/***************************************************************************
 *   Copyright (C) 2012 by Dan Vratil <dan@progdan.cz>                     *
 *                                                                         *
 *   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 "entity-filter-model.h"
#include "entity-model.h"
#include "person-entity-merge-model.h"

#include <TelepathyQt/Types>

#include <KDE/KDebug>

EntityFilterModel::EntityFilterModel(QObject *parent):
    QSortFilterProxyModel(parent)
{
}

EntityFilterModel::~EntityFilterModel()
{
}

bool EntityFilterModel::filterAcceptsRow(int source_row, const QModelIndex &source_parent) const
{
    /* Fast path for groups */
    if (source_parent == QModelIndex()) {
        return true;
    }

    const QModelIndex index = source_parent.child(source_row, 0);

    const PersonEntityMergeModel::ItemType itemType =
        static_cast<PersonEntityMergeModel::ItemType>(index.data(PersonEntityMergeModel::ItemTypeRole).toUInt());
    if (itemType == PersonEntityMergeModel::Entity) {
        const Tp::AccountPtr account = index.data(PersonEntityMergeModel::AccountRole).value< Tp::AccountPtr >();
        const KTp::LogEntity entity = index.data(PersonEntityMergeModel::EntityRole).value< KTp::LogEntity >();
        Q_ASSERT(entity.isValid());

        bool matches_filter = false;

        if (!m_searchHits.isEmpty() && !account.isNull() && entity.isValid()) {
            Q_FOREACH(const KTp::LogSearchHit &searchHit, m_searchHits) {
                const Tp::AccountPtr searchHitAccount = searchHit.account();
                const KTp::LogEntity searchHitTarget = searchHit.entity();

                /* Don't display search hits with empty account or target */
                if (searchHitAccount.isNull() || !searchHitTarget.isValid()) {
                    continue;
                }

                if ((searchHitAccount->uniqueIdentifier() == account->uniqueIdentifier()) &&
                    (searchHitTarget.id() == entity.id()))
                {
                    matches_filter = true;
                    break;
                }
            }
        } else {
            matches_filter = true;
        }

        const QString term = filterRegExp().pattern();
        if (term.isEmpty()) {
            return matches_filter;
        }

        const KTp::ContactPtr contact = index.data(PersonEntityMergeModel::ContactRole).value< KTp::ContactPtr >();

        /* Check if contact's account name matches */
        if (entity.alias().contains(term, Qt::CaseInsensitive) && matches_filter) {
            kDebug() << entity.alias() << "matches" << term;
            return matches_filter;
        }

        /* If there's information about contact's real name try to match it too */
        if (!contact.isNull()) {
            if (contact->alias().contains(term, Qt::CaseInsensitive) && matches_filter) {
                kDebug() << contact->alias() << "matches" << term;
                return matches_filter;
            }
        }

        return false;
    } else if (itemType == PersonEntityMergeModel::Persona) {
        if (sourceModel()->rowCount(index) == 0) {
            return true;
        }

        for (int i = 0; i < sourceModel()->rowCount(index); ++i) {
            if (filterAcceptsRow(i, index)) {
                return true;
            }
        }
        return false;
    }

    Q_ASSERT(false);
    return false;
}

void EntityFilterModel::setSearchHits(const QList<KTp::LogSearchHit> &searchHits)
{
    m_searchHits = searchHits;

    invalidate();
}


void EntityFilterModel::clearSearchHits()
{
    m_searchHits.clear();

    invalidate();
}