forked from blue/squawk
142 lines
3.7 KiB
C++
142 lines
3.7 KiB
C++
// Squawk messenger.
|
|
// Copyright (C) 2019 Yury Gubich <blue@macaw.me>
|
|
//
|
|
// 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 3 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, see <http://www.gnu.org/licenses/>.
|
|
|
|
#include "keysmodel.h"
|
|
|
|
const QHash<int, QByteArray> UI::KeysModel::roles = {
|
|
{Label, "label"},
|
|
{FingerPrint, "fingerPrint"},
|
|
{TrustLevel, "trustLevel"}
|
|
};
|
|
|
|
UI::KeysModel::KeysModel(QObject* parent):
|
|
QAbstractListModel(parent),
|
|
keys(),
|
|
modified()
|
|
{
|
|
|
|
}
|
|
|
|
UI::KeysModel::~KeysModel() {
|
|
for (Shared::KeyInfo* key : keys) {
|
|
delete key;
|
|
}
|
|
|
|
for (std::pair<const int, Shared::KeyInfo*>& pair: modified) {
|
|
delete pair.second;
|
|
}
|
|
}
|
|
|
|
std::deque<Shared::KeyInfo> UI::KeysModel::modifiedKeys() const {
|
|
std::deque<Shared::KeyInfo> response(modified.size());
|
|
|
|
int i = 0;
|
|
for (const std::pair<const int, Shared::KeyInfo*>& pair: modified) {
|
|
response[i] = *(pair.second);
|
|
++i;
|
|
}
|
|
|
|
return response;
|
|
}
|
|
|
|
void UI::KeysModel::addKey(const Shared::KeyInfo& info) {
|
|
beginInsertRows(QModelIndex(), keys.size(), keys.size());
|
|
keys.push_back(new Shared::KeyInfo(info));
|
|
endInsertRows();
|
|
}
|
|
|
|
QVariant UI::KeysModel::data(const QModelIndex& index, int role) const {
|
|
int i = index.row();
|
|
const Shared::KeyInfo* info;
|
|
bool dirty;
|
|
std::map<int, Shared::KeyInfo*>::const_iterator itr = modified.find(i);
|
|
if (itr != modified.end()) {
|
|
info = itr->second;
|
|
dirty = true;
|
|
} else {
|
|
dirty = false;
|
|
info = keys[i];
|
|
}
|
|
QVariant answer;
|
|
|
|
switch (role) {
|
|
case Qt::DisplayRole:
|
|
case Label:
|
|
answer = info->label;
|
|
break;
|
|
case FingerPrint:
|
|
answer = info->fingerPrint;
|
|
break;
|
|
case TrustLevel:
|
|
answer = static_cast<uint8_t>(info->trustLevel);
|
|
break;
|
|
case LastInteraction:
|
|
answer = info->lastInteraction;
|
|
break;
|
|
case Dirty:
|
|
answer = dirty;
|
|
break;
|
|
}
|
|
|
|
return answer;
|
|
}
|
|
|
|
int UI::KeysModel::rowCount(const QModelIndex& parent) const {
|
|
return keys.size();
|
|
}
|
|
|
|
QHash<int, QByteArray> UI::KeysModel::roleNames() const {return roles;}
|
|
|
|
QModelIndex UI::KeysModel::index(int row, int column, const QModelIndex& parent) const {
|
|
if (!hasIndex(row, column, parent)) {
|
|
return QModelIndex();
|
|
}
|
|
|
|
return createIndex(row, column, keys[row]);
|
|
}
|
|
|
|
void UI::KeysModel::revertKey(int row) {
|
|
std::map<int, Shared::KeyInfo*>::const_iterator itr = modified.find(row);
|
|
if (itr != modified.end()) {
|
|
modified.erase(itr);
|
|
QModelIndex index = createIndex(row, 0, keys[row]);
|
|
dataChanged(index, index);
|
|
}
|
|
}
|
|
|
|
void UI::KeysModel::setTrustLevel(int row, Shared::TrustLevel level) {
|
|
std::map<int, Shared::KeyInfo*>::const_iterator itr = modified.find(row);
|
|
Shared::KeyInfo* info;
|
|
if (itr == modified.end()) {
|
|
if (row < rowCount()) {
|
|
info = new Shared::KeyInfo(*(keys[row]));
|
|
modified.insert(std::make_pair(row, info));
|
|
} else {
|
|
return;
|
|
}
|
|
} else {
|
|
info = itr->second;
|
|
}
|
|
|
|
info->trustLevel = level;
|
|
|
|
QModelIndex index = createIndex(row, 0, info);
|
|
dataChanged(index, index, {KeysModel::Dirty});
|
|
}
|
|
|
|
|
|
|