139 lines
4.4 KiB
C++
139 lines
4.4 KiB
C++
// SPDX-FileCopyrightText: 2023 Tobias Fella <tobias.fella@kde.org>
|
|
// SPDX-License-Identifier: GPL-2.0-or-later
|
|
|
|
#include "locationsmodel.h"
|
|
|
|
#include <QGuiApplication>
|
|
|
|
using namespace Quotient;
|
|
|
|
LocationsModel::LocationsModel(QObject *parent)
|
|
: QAbstractListModel(parent)
|
|
{
|
|
connect(this, &LocationsModel::roomChanged, this, [this]() {
|
|
for (const auto &event : m_room->messageEvents()) {
|
|
if (!is<RoomMessageEvent>(*event)) {
|
|
continue;
|
|
}
|
|
if (event->contentJson()["msgtype"_ls] == "m.location"_ls) {
|
|
const auto &e = *event;
|
|
addLocation(eventCast<const RoomMessageEvent>(&e));
|
|
}
|
|
}
|
|
connect(m_room, &NeoChatRoom::aboutToAddHistoricalMessages, this, [this](const auto &events) {
|
|
for (const auto &event : events) {
|
|
if (!is<RoomMessageEvent>(*event)) {
|
|
continue;
|
|
}
|
|
if (event->contentJson()["msgtype"_ls] == "m.location"_ls) {
|
|
const auto &e = *event;
|
|
addLocation(eventCast<const RoomMessageEvent>(&e));
|
|
}
|
|
}
|
|
});
|
|
connect(m_room, &NeoChatRoom::aboutToAddNewMessages, this, [this](const auto &events) {
|
|
for (const auto &event : events) {
|
|
if (!is<RoomMessageEvent>(*event)) {
|
|
continue;
|
|
}
|
|
if (event->contentJson()["msgtype"_ls] == "m.location"_ls) {
|
|
const auto &e = *event;
|
|
addLocation(eventCast<const RoomMessageEvent>(&e));
|
|
}
|
|
}
|
|
});
|
|
});
|
|
|
|
connect(this, &LocationsModel::rowsInserted, this, &LocationsModel::boundingBoxChanged);
|
|
|
|
connect(static_cast<QGuiApplication *>(QGuiApplication::instance()), &QGuiApplication::paletteChanged, this, [this] {
|
|
Q_EMIT dataChanged(index(0, 0), index(rowCount() - 1, 0), {AuthorRole});
|
|
});
|
|
}
|
|
|
|
void LocationsModel::addLocation(const RoomMessageEvent *event)
|
|
{
|
|
const auto uri = event->contentJson()["org.matrix.msc3488.location"_ls]["uri"_ls].toString();
|
|
const auto parts = uri.mid(4).split(QLatin1Char(','));
|
|
if (parts.size() < 2) {
|
|
qWarning() << "invalid geo: URI" << uri;
|
|
return;
|
|
}
|
|
const auto latitude = parts[0].toFloat();
|
|
const auto longitude = parts[1].toFloat();
|
|
beginInsertRows(QModelIndex(), m_locations.size(), m_locations.size() + 1);
|
|
m_locations += LocationData{
|
|
.eventId = event->id(),
|
|
.latitude = latitude,
|
|
.longitude = longitude,
|
|
.content = event->contentJson(),
|
|
.author = m_room->user(event->senderId()),
|
|
};
|
|
endInsertRows();
|
|
}
|
|
|
|
NeoChatRoom *LocationsModel::room() const
|
|
{
|
|
return m_room;
|
|
}
|
|
|
|
void LocationsModel::setRoom(NeoChatRoom *room)
|
|
{
|
|
if (m_room) {
|
|
disconnect(this, nullptr, m_room, nullptr);
|
|
}
|
|
m_room = room;
|
|
Q_EMIT roomChanged();
|
|
}
|
|
|
|
QHash<int, QByteArray> LocationsModel::roleNames() const
|
|
{
|
|
return {
|
|
{LongitudeRole, "longitude"},
|
|
{LatitudeRole, "latitude"},
|
|
{TextRole, "text"},
|
|
{AssetRole, "asset"},
|
|
{AuthorRole, "author"},
|
|
};
|
|
}
|
|
|
|
QVariant LocationsModel::data(const QModelIndex &index, int roleName) const
|
|
{
|
|
auto row = index.row();
|
|
if (roleName == LongitudeRole) {
|
|
return m_locations[row].longitude;
|
|
} else if (roleName == LatitudeRole) {
|
|
return m_locations[row].latitude;
|
|
} else if (roleName == TextRole) {
|
|
return m_locations[row].content["body"_ls].toString();
|
|
} else if (roleName == AssetRole) {
|
|
return m_locations[row].content["org.matrix.msc3488.asset"_ls].toObject()["type"_ls].toString();
|
|
} else if (roleName == AuthorRole) {
|
|
return m_room->getUser(m_locations[row].author);
|
|
}
|
|
return {};
|
|
}
|
|
|
|
int LocationsModel::rowCount(const QModelIndex &parent) const
|
|
{
|
|
Q_UNUSED(parent);
|
|
return m_locations.size();
|
|
}
|
|
|
|
QRectF LocationsModel::boundingBox() const
|
|
{
|
|
QRectF bbox(QPointF(180.0, 90.0), QPointF(-180.0, -90.0));
|
|
for (auto i = 0; i < rowCount(); ++i) {
|
|
const auto lat = data(index(i, 0), LatitudeRole).toDouble();
|
|
const auto lon = data(index(i, 0), LongitudeRole).toDouble();
|
|
|
|
bbox.setLeft(std::min(bbox.left(), lon));
|
|
bbox.setRight(std::max(bbox.right(), lon));
|
|
bbox.setTop(std::min(bbox.top(), lat));
|
|
bbox.setBottom(std::max(bbox.bottom(), lat));
|
|
}
|
|
return bbox;
|
|
}
|
|
|
|
#include "moc_locationsmodel.cpp"
|