amnezia-client/client/ui/models/protocols_model.cpp
2021-09-19 14:31:38 +03:00

65 lines
1.6 KiB
C++

#include "protocols_model.h"
ProtocolsModel::ProtocolsModel(QObject *parent) :
QAbstractListModel(parent)
{
}
int ProtocolsModel::rowCount(const QModelIndex &parent) const
{
Q_UNUSED(parent);
return amnezia::allProtocols().size();
}
QHash<int, QByteArray> ProtocolsModel::roleNames() const {
QHash<int, QByteArray> roles;
roles[NameRole] = "name_role";
roles[DescRole] = "desc_role";
roles[isVpnTypeRole] = "is_vpn_role";
roles[isOtherTypeRole] = "is_other_role";
roles[isInstalledRole] = "is_installed_role";
return roles;
}
QVariant ProtocolsModel::data(const QModelIndex &index, int role) const
{
if (!index.isValid() || index.row() < 0
|| index.row() >= amnezia::allContainers().size()) {
return QVariant();
}
Protocol p = amnezia::allProtocols().at(index.row());
if (role == NameRole) {
return protocolHumanNames().value(p);
}
if (role == DescRole) {
return protocolDescriptions().value(p);
}
if (role == isVpnTypeRole) {
return isProtocolVpnType(p);
}
// if (role == isOtherTypeRole) {
// return isContainerVpnType(c)
// }
if (role == isInstalledRole) {
return protocolsForContainer(m_selectedDockerContainer).contains(p);
}
return QVariant();
}
void ProtocolsModel::setSelectedServerIndex(int index)
{
beginResetModel();
m_selectedServerIndex = index;
endResetModel();
}
void ProtocolsModel::setSelectedDockerContainer(DockerContainer c)
{
beginResetModel();
m_selectedDockerContainer = c;
endResetModel();
}