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
130
|
/****************************************************************************
* Copyright (C) 2011 Instituto Nokia de Tecnologia (INdT) *
* *
* This file may be used under the terms of the GNU Lesser *
* General Public License version 2.1 as published by the Free Software *
* Foundation and appearing in the file LICENSE.LGPL included in the *
* packaging of this file. Please review the following information to *
* ensure the GNU Lesser General Public License version 2.1 requirements *
* will be met: http://www.gnu.org/licenses/old-licenses/lgpl-2.1.html. *
* *
* 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 Lesser General Public License for more details. *
****************************************************************************/
#include "BookmarkModel.h"
#include <QtSql/QSqlError>
#include <QtSql/QSqlQuery>
#include <QtSql/QSqlRecord>
BookmarkModel::BookmarkModel(QSqlDatabase database, QObject *parent)
: QSqlTableModel(parent, database)
{
connect(this, SIGNAL(rowsInserted(const QModelIndex&, int, int)), SIGNAL(countChanged()));
connect(this, SIGNAL(rowsRemoved(const QModelIndex&, int, int)), SIGNAL(countChanged()));
setEditStrategy(OnManualSubmit);
}
void BookmarkModel::generateRoleNames()
{
for (int i = 0; i < this->columnCount(); i++)
m_roles[Qt::UserRole + i] = this->headerData(i, Qt::Horizontal).toByteArray();
setRoleNames(m_roles);
}
QString BookmarkModel::tableCreateQuery() const
{
const QLatin1String bookmarkQuery("CREATE TABLE IF NOT EXISTS bookmarks (id INTEGER PRIMARY KEY AUTOINCREMENT,"
"name VARCHAR, url VARCHAR, dateAdded DATE);");
return bookmarkQuery;
}
void BookmarkModel::setFilter(const QString& filter)
{
QSqlTableModel::setFilter(filter);
}
bool BookmarkModel::select()
{
return QSqlTableModel::select();
}
void BookmarkModel::insert(const QString& name, const QString& url)
{
if (contains(url))
return;
QSqlRecord record = this->record();
record.setValue(QLatin1String("name"), name);
record.setValue(QLatin1String("url"), url);
insertRecord(-1, record);
submitAll();
}
void BookmarkModel::remove(const QString& url)
{
if (!contains(url))
return;
QSqlQuery sqlQuery(database());
sqlQuery.prepare(QLatin1String("SELECT id FROM bookmarks WHERE url = ?"));
sqlQuery.addBindValue(url);
sqlQuery.exec();
sqlQuery.first();
int indexToDelete = -1;
for (int row = 0; row < rowCount(); ++row) {
if (index(row, 0).data(Qt::DisplayRole).toInt() == sqlQuery.value(0).toInt()) {
indexToDelete = row;
break;
}
}
if (indexToDelete >= 0) {
beginRemoveRows(QModelIndex(), indexToDelete, indexToDelete);
removeRow(indexToDelete);
submitAll();
endRemoveRows();
}
}
void BookmarkModel::togglePin(const QString& url)
{
if (contains(url))
remove(url);
else
insert(url, url);
}
void BookmarkModel::update(int index, const QString& name, const QString& url)
{
setData(this->index(index, 1), name);
setData(this->index(index, 2), url);
}
bool BookmarkModel::contains(const QString& url)
{
QSqlQuery sqlQuery(database());
sqlQuery.prepare(QLatin1String("SELECT id FROM bookmarks WHERE url = ?"));
sqlQuery.addBindValue(url);
sqlQuery.exec();
return sqlQuery.first();
}
QVariant BookmarkModel::data(const QModelIndex& index, int role) const
{
QVariant value;
if (role < Qt::UserRole)
value = QSqlQueryModel::data(index, role);
else {
const int columnId = role - Qt::UserRole;
const QModelIndex modelIndex = this->index(index.row(), columnId);
value = QSqlTableModel::data(modelIndex, Qt::DisplayRole);
}
return value;
}
|
