mirror of https://github.com/Nheko-Reborn/nheko
Weblate
4 years ago
commit
b5669310e5
File diff suppressed because it is too large
Load Diff
@ -0,0 +1,20 @@ |
||||
#pragma once |
||||
|
||||
// Class for showing a limited amount of completions at a time
|
||||
|
||||
#include <QSortFilterProxyModel> |
||||
|
||||
class CompletionModel : public QSortFilterProxyModel |
||||
{ |
||||
public: |
||||
CompletionModel(QAbstractItemModel *model, QObject *parent = nullptr) |
||||
: QSortFilterProxyModel(parent) |
||||
{ |
||||
setSourceModel(model); |
||||
} |
||||
int rowCount(const QModelIndex &parent) const override |
||||
{ |
||||
auto row_count = QSortFilterProxyModel::rowCount(parent); |
||||
return (row_count < 7) ? row_count : 7; |
||||
} |
||||
}; |
||||
@ -0,0 +1,37 @@ |
||||
#pragma once |
||||
|
||||
#include "EmojiModel.h" |
||||
|
||||
#include <QDebug> |
||||
#include <QEvent> |
||||
#include <QSortFilterProxyModel> |
||||
|
||||
namespace emoji { |
||||
|
||||
// Map emoji data to searchable data
|
||||
class EmojiSearchModel : public QSortFilterProxyModel |
||||
{ |
||||
public: |
||||
EmojiSearchModel(QObject *parent = nullptr) |
||||
: QSortFilterProxyModel(parent) |
||||
{ |
||||
setSourceModel(new EmojiModel(this)); |
||||
} |
||||
QVariant data(const QModelIndex &index, int role = Qt::UserRole + 1) const override |
||||
{ |
||||
if (role == Qt::DisplayRole) { |
||||
auto emoji = QSortFilterProxyModel::data(index, role).toString(); |
||||
return emoji + " :" + |
||||
toShortcode(data(index, EmojiModel::ShortName).toString()) + ":"; |
||||
} |
||||
return QSortFilterProxyModel::data(index, role); |
||||
} |
||||
|
||||
private: |
||||
QString toShortcode(QString shortname) const |
||||
{ |
||||
return shortname.replace(" ", "-").replace(":", "-").replace("--", "-").toLower(); |
||||
} |
||||
}; |
||||
|
||||
} |
||||
@ -0,0 +1,570 @@ |
||||
#include "EventStore.h" |
||||
|
||||
#include <QThread> |
||||
#include <QTimer> |
||||
|
||||
#include "Cache.h" |
||||
#include "Cache_p.h" |
||||
#include "EventAccessors.h" |
||||
#include "Logging.h" |
||||
#include "MatrixClient.h" |
||||
#include "Olm.h" |
||||
|
||||
Q_DECLARE_METATYPE(Reaction) |
||||
|
||||
QCache<EventStore::IdIndex, mtx::events::collections::TimelineEvents> EventStore::decryptedEvents_{ |
||||
1000}; |
||||
QCache<EventStore::IdIndex, mtx::events::collections::TimelineEvents> EventStore::events_by_id_{ |
||||
1000}; |
||||
QCache<EventStore::Index, mtx::events::collections::TimelineEvents> EventStore::events_{1000}; |
||||
|
||||
EventStore::EventStore(std::string room_id, QObject *) |
||||
: room_id_(std::move(room_id)) |
||||
{ |
||||
static auto reactionType = qRegisterMetaType<Reaction>(); |
||||
(void)reactionType; |
||||
|
||||
auto range = cache::client()->getTimelineRange(room_id_); |
||||
|
||||
if (range) { |
||||
this->first = range->first; |
||||
this->last = range->last; |
||||
} |
||||
|
||||
connect( |
||||
this, |
||||
&EventStore::eventFetched, |
||||
this, |
||||
[this](std::string id, |
||||
std::string relatedTo, |
||||
mtx::events::collections::TimelineEvents timeline) { |
||||
cache::client()->storeEvent(room_id_, id, {timeline}); |
||||
|
||||
if (!relatedTo.empty()) { |
||||
auto idx = idToIndex(relatedTo); |
||||
if (idx) |
||||
emit dataChanged(*idx, *idx); |
||||
} |
||||
}, |
||||
Qt::QueuedConnection); |
||||
|
||||
connect( |
||||
this, |
||||
&EventStore::oldMessagesRetrieved, |
||||
this, |
||||
[this](const mtx::responses::Messages &res) { |
||||
//
|
||||
uint64_t newFirst = cache::client()->saveOldMessages(room_id_, res); |
||||
if (newFirst == first && !res.chunk.empty()) |
||||
fetchMore(); |
||||
else { |
||||
emit beginInsertRows(toExternalIdx(newFirst), |
||||
toExternalIdx(this->first - 1)); |
||||
this->first = newFirst; |
||||
emit endInsertRows(); |
||||
emit fetchedMore(); |
||||
} |
||||
}, |
||||
Qt::QueuedConnection); |
||||
|
||||
connect(this, &EventStore::processPending, this, [this]() { |
||||
if (!current_txn.empty()) { |
||||
nhlog::ui()->debug("Already processing {}", current_txn); |
||||
return; |
||||
} |
||||
|
||||
auto event = cache::client()->firstPendingMessage(room_id_); |
||||
|
||||
if (!event) { |
||||
nhlog::ui()->debug("No event to send"); |
||||
return; |
||||
} |
||||
|
||||
std::visit( |
||||
[this](auto e) { |
||||
auto txn_id = e.event_id; |
||||
this->current_txn = txn_id; |
||||
|
||||
if (txn_id.empty() || txn_id[0] != 'm') { |
||||
nhlog::ui()->debug("Invalid txn id '{}'", txn_id); |
||||
cache::client()->removePendingStatus(room_id_, txn_id); |
||||
return; |
||||
} |
||||
|
||||
if constexpr (mtx::events::message_content_to_type<decltype(e.content)> != |
||||
mtx::events::EventType::Unsupported) |
||||
http::client()->send_room_message( |
||||
room_id_, |
||||
txn_id, |
||||
e.content, |
||||
[this, txn_id](const mtx::responses::EventId &event_id, |
||||
mtx::http::RequestErr err) { |
||||
if (err) { |
||||
const int status_code = |
||||
static_cast<int>(err->status_code); |
||||
nhlog::net()->warn( |
||||
"[{}] failed to send message: {} {}", |
||||
txn_id, |
||||
err->matrix_error.error, |
||||
status_code); |
||||
emit messageFailed(txn_id); |
||||
return; |
||||
} |
||||
emit messageSent(txn_id, event_id.event_id.to_string()); |
||||
}); |
||||
}, |
||||
event->data); |
||||
}); |
||||
|
||||
connect( |
||||
this, |
||||
&EventStore::messageFailed, |
||||
this, |
||||
[this](std::string txn_id) { |
||||
if (current_txn == txn_id) { |
||||
current_txn_error_count++; |
||||
if (current_txn_error_count > 10) { |
||||
nhlog::ui()->debug("failing txn id '{}'", txn_id); |
||||
cache::client()->removePendingStatus(room_id_, txn_id); |
||||
current_txn_error_count = 0; |
||||
} |
||||
} |
||||
QTimer::singleShot(1000, this, [this]() { |
||||
nhlog::ui()->debug("timeout"); |
||||
this->current_txn = ""; |
||||
emit processPending(); |
||||
}); |
||||
}, |
||||
Qt::QueuedConnection); |
||||
|
||||
connect( |
||||
this, |
||||
&EventStore::messageSent, |
||||
this, |
||||
[this](std::string txn_id, std::string event_id) { |
||||
nhlog::ui()->debug("sent {}", txn_id); |
||||
|
||||
http::client()->read_event( |
||||
room_id_, event_id, [this, event_id](mtx::http::RequestErr err) { |
||||
if (err) { |
||||
nhlog::net()->warn( |
||||
"failed to read_event ({}, {})", room_id_, event_id); |
||||
} |
||||
}); |
||||
|
||||
cache::client()->removePendingStatus(room_id_, txn_id); |
||||
this->current_txn = ""; |
||||
this->current_txn_error_count = 0; |
||||
emit processPending(); |
||||
}, |
||||
Qt::QueuedConnection); |
||||
} |
||||
|
||||
void |
||||
EventStore::addPending(mtx::events::collections::TimelineEvents event) |
||||
{ |
||||
if (this->thread() != QThread::currentThread()) |
||||
nhlog::db()->warn("{} called from a different thread!", __func__); |
||||
|
||||
cache::client()->savePendingMessage(this->room_id_, {event}); |
||||
mtx::responses::Timeline events; |
||||
events.limited = false; |
||||
events.events.emplace_back(event); |
||||
handleSync(events); |
||||
|
||||
emit processPending(); |
||||
} |
||||
|
||||
void |
||||
EventStore::clearTimeline() |
||||
{ |
||||
emit beginResetModel(); |
||||
|
||||
cache::client()->clearTimeline(room_id_); |
||||
auto range = cache::client()->getTimelineRange(room_id_); |
||||
if (range) { |
||||
nhlog::db()->info("Range {} {}", range->last, range->first); |
||||
this->last = range->last; |
||||
this->first = range->first; |
||||
} else { |
||||
this->first = std::numeric_limits<uint64_t>::max(); |
||||
this->last = std::numeric_limits<uint64_t>::max(); |
||||
} |
||||
nhlog::ui()->info("Range {} {}", this->last, this->first); |
||||
|
||||
emit endResetModel(); |
||||
} |
||||
|
||||
void |
||||
EventStore::handleSync(const mtx::responses::Timeline &events) |
||||
{ |
||||
if (this->thread() != QThread::currentThread()) |
||||
nhlog::db()->warn("{} called from a different thread!", __func__); |
||||
|
||||
auto range = cache::client()->getTimelineRange(room_id_); |
||||
if (!range) |
||||
return; |
||||
|
||||
if (events.limited) { |
||||
emit beginResetModel(); |
||||
this->last = range->last; |
||||
this->first = range->first; |
||||
emit endResetModel(); |
||||
|
||||
} else if (range->last > this->last) { |
||||
emit beginInsertRows(toExternalIdx(this->last + 1), toExternalIdx(range->last)); |
||||
this->last = range->last; |
||||
emit endInsertRows(); |
||||
} |
||||
|
||||
for (const auto &event : events.events) { |
||||
std::string relates_to; |
||||
if (auto redaction = |
||||
std::get_if<mtx::events::RedactionEvent<mtx::events::msg::Redaction>>( |
||||
&event)) { |
||||
// fixup reactions
|
||||
auto redacted = events_by_id_.object({room_id_, redaction->redacts}); |
||||
if (redacted) { |
||||
auto id = mtx::accessors::relates_to_event_id(*redacted); |
||||
if (!id.empty()) { |
||||
auto idx = idToIndex(id); |
||||
if (idx) { |
||||
events_by_id_.remove( |
||||
{room_id_, redaction->redacts}); |
||||
events_.remove({room_id_, toInternalIdx(*idx)}); |
||||
emit dataChanged(*idx, *idx); |
||||
} |
||||
} |
||||
} |
||||
|
||||
relates_to = redaction->redacts; |
||||
} else if (auto reaction = |
||||
std::get_if<mtx::events::RoomEvent<mtx::events::msg::Reaction>>( |
||||
&event)) { |
||||
relates_to = reaction->content.relates_to.event_id; |
||||
} else { |
||||
relates_to = mtx::accessors::in_reply_to_event(event); |
||||
} |
||||
|
||||
if (!relates_to.empty()) { |
||||
auto idx = cache::client()->getTimelineIndex(room_id_, relates_to); |
||||
if (idx) { |
||||
events_by_id_.remove({room_id_, relates_to}); |
||||
decryptedEvents_.remove({room_id_, relates_to}); |
||||
events_.remove({room_id_, *idx}); |
||||
emit dataChanged(toExternalIdx(*idx), toExternalIdx(*idx)); |
||||
} |
||||
} |
||||
|
||||
if (auto txn_id = mtx::accessors::transaction_id(event); !txn_id.empty()) { |
||||
auto idx = cache::client()->getTimelineIndex( |
||||
room_id_, mtx::accessors::event_id(event)); |
||||
if (idx) { |
||||
Index index{room_id_, *idx}; |
||||
events_.remove(index); |
||||
emit dataChanged(toExternalIdx(*idx), toExternalIdx(*idx)); |
||||
} |
||||
} |
||||
} |
||||
} |
||||
|
||||
QVariantList |
||||
EventStore::reactions(const std::string &event_id) |
||||
{ |
||||
auto event_ids = cache::client()->relatedEvents(room_id_, event_id); |
||||
|
||||
struct TempReaction |
||||
{ |
||||
int count = 0; |
||||
std::vector<std::string> users; |
||||
std::string reactedBySelf; |
||||
}; |
||||
std::map<std::string, TempReaction> aggregation; |
||||
std::vector<Reaction> reactions; |
||||
|
||||
auto self = http::client()->user_id().to_string(); |
||||
for (const auto &id : event_ids) { |
||||
auto related_event = get(id, event_id); |
||||
if (!related_event) |
||||
continue; |
||||
|
||||
if (auto reaction = std::get_if<mtx::events::RoomEvent<mtx::events::msg::Reaction>>( |
||||
related_event)) { |
||||
auto &agg = aggregation[reaction->content.relates_to.key]; |
||||
|
||||
if (agg.count == 0) { |
||||
Reaction temp{}; |
||||
temp.key_ = |
||||
QString::fromStdString(reaction->content.relates_to.key); |
||||
reactions.push_back(temp); |
||||
} |
||||
|
||||
agg.count++; |
||||
agg.users.push_back(cache::displayName(room_id_, reaction->sender)); |
||||
if (reaction->sender == self) |
||||
agg.reactedBySelf = reaction->event_id; |
||||
} |
||||
} |
||||
|
||||
QVariantList temp; |
||||
for (auto &reaction : reactions) { |
||||
const auto &agg = aggregation[reaction.key_.toStdString()]; |
||||
reaction.count_ = agg.count; |
||||
reaction.selfReactedEvent_ = QString::fromStdString(agg.reactedBySelf); |
||||
|
||||
bool firstReaction = true; |
||||
for (const auto &user : agg.users) { |
||||
if (firstReaction) |
||||
firstReaction = false; |
||||
else |
||||
reaction.users_ += ", "; |
||||
|
||||
reaction.users_ += QString::fromStdString(user); |
||||
} |
||||
|
||||
nhlog::db()->debug("key: {}, count: {}, users: {}", |
||||
reaction.key_.toStdString(), |
||||
reaction.count_, |
||||
reaction.users_.toStdString()); |
||||
temp.append(QVariant::fromValue(reaction)); |
||||
} |
||||
|
||||
return temp; |
||||
} |
||||
|
||||
mtx::events::collections::TimelineEvents * |
||||
EventStore::get(int idx, bool decrypt) |
||||
{ |
||||
if (this->thread() != QThread::currentThread()) |
||||
nhlog::db()->warn("{} called from a different thread!", __func__); |
||||
|
||||
Index index{room_id_, toInternalIdx(idx)}; |
||||
if (index.idx > last || index.idx < first) |
||||
return nullptr; |
||||
|
||||
auto event_ptr = events_.object(index); |
||||
if (!event_ptr) { |
||||
auto event_id = cache::client()->getTimelineEventId(room_id_, index.idx); |
||||
if (!event_id) |
||||
return nullptr; |
||||
|
||||
auto event = cache::client()->getEvent(room_id_, *event_id); |
||||
if (!event) |
||||
return nullptr; |
||||
else |
||||
event_ptr = |
||||
new mtx::events::collections::TimelineEvents(std::move(event->data)); |
||||
events_.insert(index, event_ptr); |
||||
} |
||||
|
||||
if (decrypt) |
||||
if (auto encrypted = |
||||
std::get_if<mtx::events::EncryptedEvent<mtx::events::msg::Encrypted>>( |
||||
event_ptr)) |
||||
return decryptEvent({room_id_, encrypted->event_id}, *encrypted); |
||||
|
||||
return event_ptr; |
||||
} |
||||
|
||||
std::optional<int> |
||||
EventStore::idToIndex(std::string_view id) const |
||||
{ |
||||
if (this->thread() != QThread::currentThread()) |
||||
nhlog::db()->warn("{} called from a different thread!", __func__); |
||||
|
||||
auto idx = cache::client()->getTimelineIndex(room_id_, id); |
||||
if (idx) |
||||
return toExternalIdx(*idx); |
||||
else |
||||
return std::nullopt; |
||||
} |
||||
std::optional<std::string> |
||||
EventStore::indexToId(int idx) const |
||||
{ |
||||
if (this->thread() != QThread::currentThread()) |
||||
nhlog::db()->warn("{} called from a different thread!", __func__); |
||||
|
||||
return cache::client()->getTimelineEventId(room_id_, toInternalIdx(idx)); |
||||
} |
||||
|
||||
mtx::events::collections::TimelineEvents * |
||||
EventStore::decryptEvent(const IdIndex &idx, |
||||
const mtx::events::EncryptedEvent<mtx::events::msg::Encrypted> &e) |
||||
{ |
||||
if (auto cachedEvent = decryptedEvents_.object(idx)) |
||||
return cachedEvent; |
||||
|
||||
MegolmSessionIndex index; |
||||
index.room_id = room_id_; |
||||
index.session_id = e.content.session_id; |
||||
index.sender_key = e.content.sender_key; |
||||
|
||||
auto asCacheEntry = [&idx](mtx::events::collections::TimelineEvents &&event) { |
||||
auto event_ptr = new mtx::events::collections::TimelineEvents(std::move(event)); |
||||
decryptedEvents_.insert(idx, event_ptr); |
||||
return event_ptr; |
||||
}; |
||||
|
||||
auto decryptionResult = olm::decryptEvent(index, e); |
||||
|
||||
if (decryptionResult.error) { |
||||
mtx::events::RoomEvent<mtx::events::msg::Notice> dummy; |
||||
dummy.origin_server_ts = e.origin_server_ts; |
||||
dummy.event_id = e.event_id; |
||||
dummy.sender = e.sender; |
||||
switch (*decryptionResult.error) { |
||||
case olm::DecryptionErrorCode::MissingSession: |
||||
dummy.content.body = |
||||
tr("-- Encrypted Event (No keys found for decryption) --", |
||||
"Placeholder, when the message was not decrypted yet or can't be " |
||||
"decrypted.") |
||||
.toStdString(); |
||||
nhlog::crypto()->info("Could not find inbound megolm session ({}, {}, {})", |
||||
index.room_id, |
||||
index.session_id, |
||||
e.sender); |
||||
// TODO: Check if this actually works and look in key backup
|
||||
olm::send_key_request_for(room_id_, e); |
||||
break; |
||||
case olm::DecryptionErrorCode::DbError: |
||||
nhlog::db()->critical( |
||||
"failed to retrieve megolm session with index ({}, {}, {})", |
||||
index.room_id, |
||||
index.session_id, |
||||
index.sender_key, |
||||
decryptionResult.error_message.value_or("")); |
||||
dummy.content.body = |
||||
tr("-- Decryption Error (failed to retrieve megolm keys from db) --", |
||||
"Placeholder, when the message can't be decrypted, because the DB " |
||||
"access " |
||||
"failed.") |
||||
.toStdString(); |
||||
break; |
||||
case olm::DecryptionErrorCode::DecryptionFailed: |
||||
nhlog::crypto()->critical( |
||||
"failed to decrypt message with index ({}, {}, {}): {}", |
||||
index.room_id, |
||||
index.session_id, |
||||
index.sender_key, |
||||
decryptionResult.error_message.value_or("")); |
||||
dummy.content.body = |
||||
tr("-- Decryption Error (%1) --", |
||||
"Placeholder, when the message can't be decrypted. In this case, the " |
||||
"Olm " |
||||
"decrytion returned an error, which is passed as %1.") |
||||
.arg( |
||||
QString::fromStdString(decryptionResult.error_message.value_or(""))) |
||||
.toStdString(); |
||||
break; |
||||
case olm::DecryptionErrorCode::ParsingFailed: |
||||
dummy.content.body = |
||||
tr("-- Encrypted Event (Unknown event type) --", |
||||
"Placeholder, when the message was decrypted, but we couldn't parse " |
||||
"it, because " |
||||
"Nheko/mtxclient don't support that event type yet.") |
||||
.toStdString(); |
||||
break; |
||||
case olm::DecryptionErrorCode::ReplayAttack: |
||||
nhlog::crypto()->critical( |
||||
"Reply attack while decryptiong event {} in room {} from {}!", |
||||
e.event_id, |
||||
room_id_, |
||||
index.sender_key); |
||||
dummy.content.body = |
||||
tr("-- Reply attack! This message index was reused! --").toStdString(); |
||||
break; |
||||
case olm::DecryptionErrorCode::UnknownFingerprint: |
||||
// TODO: don't fail, just show in UI.
|
||||
nhlog::crypto()->critical("Message by unverified fingerprint {}", |
||||
index.sender_key); |
||||
dummy.content.body = |
||||
tr("-- Message by unverified device! --").toStdString(); |
||||
break; |
||||
} |
||||
return asCacheEntry(std::move(dummy)); |
||||
} |
||||
|
||||
auto encInfo = mtx::accessors::file(decryptionResult.event.value()); |
||||
if (encInfo) |
||||
emit newEncryptedImage(encInfo.value()); |
||||
|
||||
return asCacheEntry(std::move(decryptionResult.event.value())); |
||||
} |
||||
|
||||
mtx::events::collections::TimelineEvents * |
||||
EventStore::get(std::string_view id, std::string_view related_to, bool decrypt) |
||||
{ |
||||
if (this->thread() != QThread::currentThread()) |
||||
nhlog::db()->warn("{} called from a different thread!", __func__); |
||||
|
||||
if (id.empty()) |
||||
return nullptr; |
||||
|
||||
IdIndex index{room_id_, std::string(id.data(), id.size())}; |
||||
|
||||
auto event_ptr = events_by_id_.object(index); |
||||
if (!event_ptr) { |
||||
auto event = cache::client()->getEvent(room_id_, index.id); |
||||
if (!event) { |
||||
http::client()->get_event( |
||||
room_id_, |
||||
index.id, |
||||
[this, |
||||
relatedTo = std::string(related_to.data(), related_to.size()), |
||||
id = index.id](const mtx::events::collections::TimelineEvents &timeline, |
||||
mtx::http::RequestErr err) { |
||||
if (err) { |
||||
nhlog::net()->error( |
||||
"Failed to retrieve event with id {}, which was " |
||||
"requested to show the replyTo for event {}", |
||||
relatedTo, |
||||
id); |
||||
return; |
||||
} |
||||
emit eventFetched(id, relatedTo, timeline); |
||||
}); |
||||
return nullptr; |
||||
} |
||||
event_ptr = new mtx::events::collections::TimelineEvents(std::move(event->data)); |
||||
events_by_id_.insert(index, event_ptr); |
||||
} |
||||
|
||||
if (decrypt) |
||||
if (auto encrypted = |
||||
std::get_if<mtx::events::EncryptedEvent<mtx::events::msg::Encrypted>>( |
||||
event_ptr)) |
||||
return decryptEvent(index, *encrypted); |
||||
|
||||
return event_ptr; |
||||
} |
||||
|
||||
void |
||||
EventStore::fetchMore() |
||||
{ |
||||
mtx::http::MessagesOpts opts; |
||||
opts.room_id = room_id_; |
||||
opts.from = cache::client()->previousBatchToken(room_id_); |
||||
|
||||
nhlog::ui()->debug("Paginating room {}, token {}", opts.room_id, opts.from); |
||||
|
||||
http::client()->messages( |
||||
opts, [this, opts](const mtx::responses::Messages &res, mtx::http::RequestErr err) { |
||||
if (cache::client()->previousBatchToken(room_id_) != opts.from) { |
||||
nhlog::net()->warn("Cache cleared while fetching more messages, dropping " |
||||
"/messages response"); |
||||
emit fetchedMore(); |
||||
return; |
||||
} |
||||
if (err) { |
||||
nhlog::net()->error("failed to call /messages ({}): {} - {} - {}", |
||||
opts.room_id, |
||||
mtx::errors::to_string(err->matrix_error.errcode), |
||||
err->matrix_error.error, |
||||
err->parse_error); |
||||
emit fetchedMore(); |
||||
return; |
||||
} |
||||
|
||||
emit oldMessagesRetrieved(std::move(res)); |
||||
}); |
||||
} |
||||
@ -0,0 +1,122 @@ |
||||
#pragma once |
||||
|
||||
#include <limits> |
||||
#include <string> |
||||
|
||||
#include <QCache> |
||||
#include <QObject> |
||||
#include <QVariant> |
||||
#include <qhashfunctions.h> |
||||
|
||||
#include <mtx/events/collections.hpp> |
||||
#include <mtx/responses/messages.hpp> |
||||
#include <mtx/responses/sync.hpp> |
||||
|
||||
#include "Reaction.h" |
||||
|
||||
class EventStore : public QObject |
||||
{ |
||||
Q_OBJECT |
||||
|
||||
public: |
||||
EventStore(std::string room_id, QObject *parent); |
||||
|
||||
struct Index |
||||
{ |
||||
std::string room; |
||||
uint64_t idx; |
||||
|
||||
friend uint qHash(const Index &i, uint seed = 0) noexcept |
||||
{ |
||||
QtPrivate::QHashCombine hash; |
||||
seed = hash(seed, QByteArray::fromRawData(i.room.data(), i.room.size())); |
||||
seed = hash(seed, i.idx); |
||||
return seed; |
||||
} |
||||
|
||||
friend bool operator==(const Index &a, const Index &b) noexcept |
||||
{ |
||||
return a.idx == b.idx && a.room == b.room; |
||||
} |
||||
}; |
||||
struct IdIndex |
||||
{ |
||||
std::string room, id; |
||||
|
||||
friend uint qHash(const IdIndex &i, uint seed = 0) noexcept |
||||
{ |
||||
QtPrivate::QHashCombine hash; |
||||
seed = hash(seed, QByteArray::fromRawData(i.room.data(), i.room.size())); |
||||
seed = hash(seed, QByteArray::fromRawData(i.id.data(), i.id.size())); |
||||
return seed; |
||||
} |
||||
|
||||
friend bool operator==(const IdIndex &a, const IdIndex &b) noexcept |
||||
{ |
||||
return a.id == b.id && a.room == b.room; |
||||
} |
||||
}; |
||||
|
||||
void fetchMore(); |
||||
void handleSync(const mtx::responses::Timeline &events); |
||||
|
||||
// optionally returns the event or nullptr and fetches it, after which it emits a
|
||||
// relatedFetched event
|
||||
mtx::events::collections::TimelineEvents *get(std::string_view id, |
||||
std::string_view related_to, |
||||
bool decrypt = true); |
||||
// always returns a proper event as long as the idx is valid
|
||||
mtx::events::collections::TimelineEvents *get(int idx, bool decrypt = true); |
||||
|
||||
QVariantList reactions(const std::string &event_id); |
||||
|
||||
int size() const |
||||
{ |
||||
return last != std::numeric_limits<uint64_t>::max() |
||||
? static_cast<int>(last - first) + 1 |
||||
: 0; |
||||
} |
||||
int toExternalIdx(uint64_t idx) const { return static_cast<int>(idx - first); } |
||||
uint64_t toInternalIdx(int idx) const { return first + idx; } |
||||
|
||||
std::optional<int> idToIndex(std::string_view id) const; |
||||
std::optional<std::string> indexToId(int idx) const; |
||||
|
||||
signals: |
||||
void beginInsertRows(int from, int to); |
||||
void endInsertRows(); |
||||
void beginResetModel(); |
||||
void endResetModel(); |
||||
void dataChanged(int from, int to); |
||||
void newEncryptedImage(mtx::crypto::EncryptedFile encryptionInfo); |
||||
void eventFetched(std::string id, |
||||
std::string relatedTo, |
||||
mtx::events::collections::TimelineEvents timeline); |
||||
void oldMessagesRetrieved(const mtx::responses::Messages &); |
||||
void fetchedMore(); |
||||
|
||||
void processPending(); |
||||
void messageSent(std::string txn_id, std::string event_id); |
||||
void messageFailed(std::string txn_id); |
||||
|
||||
public slots: |
||||
void addPending(mtx::events::collections::TimelineEvents event); |
||||
void clearTimeline(); |
||||
|
||||
private: |
||||
mtx::events::collections::TimelineEvents *decryptEvent( |
||||
const IdIndex &idx, |
||||
const mtx::events::EncryptedEvent<mtx::events::msg::Encrypted> &e); |
||||
|
||||
std::string room_id_; |
||||
|
||||
uint64_t first = std::numeric_limits<uint64_t>::max(), |
||||
last = std::numeric_limits<uint64_t>::max(); |
||||
|
||||
static QCache<IdIndex, mtx::events::collections::TimelineEvents> decryptedEvents_; |
||||
static QCache<Index, mtx::events::collections::TimelineEvents> events_; |
||||
static QCache<IdIndex, mtx::events::collections::TimelineEvents> events_by_id_; |
||||
|
||||
std::string current_txn; |
||||
int current_txn_error_count = 0; |
||||
}; |
||||
@ -0,0 +1 @@ |
||||
#include "Reaction.h" |
||||
@ -0,0 +1,24 @@ |
||||
#pragma once |
||||
|
||||
#include <QObject> |
||||
#include <QString> |
||||
|
||||
struct Reaction |
||||
{ |
||||
Q_GADGET |
||||
Q_PROPERTY(QString key READ key) |
||||
Q_PROPERTY(QString users READ users) |
||||
Q_PROPERTY(QString selfReactedEvent READ selfReactedEvent) |
||||
Q_PROPERTY(int count READ count) |
||||
|
||||
public: |
||||
QString key() const { return key_; } |
||||
QString users() const { return users_; } |
||||
QString selfReactedEvent() const { return selfReactedEvent_; } |
||||
int count() const { return count_; } |
||||
|
||||
QString key_; |
||||
QString users_; |
||||
QString selfReactedEvent_; |
||||
int count_; |
||||
}; |
||||
@ -1,98 +0,0 @@ |
||||
#include "ReactionsModel.h" |
||||
|
||||
#include <Cache.h> |
||||
#include <MatrixClient.h> |
||||
|
||||
QHash<int, QByteArray> |
||||
ReactionsModel::roleNames() const |
||||
{ |
||||
return { |
||||
{Key, "key"}, |
||||
{Count, "counter"}, |
||||
{Users, "users"}, |
||||
{SelfReactedEvent, "selfReactedEvent"}, |
||||
}; |
||||
} |
||||
|
||||
int |
||||
ReactionsModel::rowCount(const QModelIndex &) const |
||||
{ |
||||
return static_cast<int>(reactions.size()); |
||||
} |
||||
|
||||
QVariant |
||||
ReactionsModel::data(const QModelIndex &index, int role) const |
||||
{ |
||||
const int i = index.row(); |
||||
if (i < 0 || i >= static_cast<int>(reactions.size())) |
||||
return {}; |
||||
|
||||
switch (role) { |
||||
case Key: |
||||
return QString::fromStdString(reactions[i].key); |
||||
case Count: |
||||
return static_cast<int>(reactions[i].reactions.size()); |
||||
case Users: { |
||||
QString users; |
||||
bool first = true; |
||||
for (const auto &reaction : reactions[i].reactions) { |
||||
if (!first) |
||||
users += ", "; |
||||
else |
||||
first = false; |
||||
users += QString::fromStdString( |
||||
cache::displayName(room_id_, reaction.second.sender)); |
||||
} |
||||
return users; |
||||
} |
||||
case SelfReactedEvent: |
||||
for (const auto &reaction : reactions[i].reactions) |
||||
if (reaction.second.sender == http::client()->user_id().to_string()) |
||||
return QString::fromStdString(reaction.second.event_id); |
||||
return QStringLiteral(""); |
||||
default: |
||||
return {}; |
||||
} |
||||
} |
||||
|
||||
void |
||||
ReactionsModel::addReaction(const std::string &room_id, |
||||
const mtx::events::RoomEvent<mtx::events::msg::Reaction> &reaction) |
||||
{ |
||||
room_id_ = room_id; |
||||
|
||||
int idx = 0; |
||||
for (auto &storedReactions : reactions) { |
||||
if (storedReactions.key == reaction.content.relates_to.key) { |
||||
storedReactions.reactions[reaction.event_id] = reaction; |
||||
emit dataChanged(index(idx, 0), index(idx, 0)); |
||||
return; |
||||
} |
||||
idx++; |
||||
} |
||||
|
||||
beginInsertRows(QModelIndex(), idx, idx); |
||||
reactions.push_back( |
||||
KeyReaction{reaction.content.relates_to.key, {{reaction.event_id, reaction}}}); |
||||
endInsertRows(); |
||||
} |
||||
|
||||
void |
||||
ReactionsModel::removeReaction(const mtx::events::RoomEvent<mtx::events::msg::Reaction> &reaction) |
||||
{ |
||||
int idx = 0; |
||||
for (auto &storedReactions : reactions) { |
||||
if (storedReactions.key == reaction.content.relates_to.key) { |
||||
storedReactions.reactions.erase(reaction.event_id); |
||||
|
||||
if (storedReactions.reactions.size() == 0) { |
||||
beginRemoveRows(QModelIndex(), idx, idx); |
||||
reactions.erase(reactions.begin() + idx); |
||||
endRemoveRows(); |
||||
} else |
||||
emit dataChanged(index(idx, 0), index(idx, 0)); |
||||
return; |
||||
} |
||||
idx++; |
||||
} |
||||
} |
||||
@ -1,41 +0,0 @@ |
||||
#pragma once |
||||
|
||||
#include <QAbstractListModel> |
||||
#include <QHash> |
||||
|
||||
#include <utility> |
||||
#include <vector> |
||||
|
||||
#include <mtx/events/collections.hpp> |
||||
|
||||
class ReactionsModel : public QAbstractListModel |
||||
{ |
||||
Q_OBJECT |
||||
public: |
||||
explicit ReactionsModel(QObject *parent = nullptr) { Q_UNUSED(parent); } |
||||
enum Roles |
||||
{ |
||||
Key, |
||||
Count, |
||||
Users, |
||||
SelfReactedEvent, |
||||
}; |
||||
|
||||
QHash<int, QByteArray> roleNames() const override; |
||||
int rowCount(const QModelIndex &parent = QModelIndex()) const override; |
||||
QVariant data(const QModelIndex &index, int role = Qt::DisplayRole) const override; |
||||
|
||||
public slots: |
||||
void addReaction(const std::string &room_id, |
||||
const mtx::events::RoomEvent<mtx::events::msg::Reaction> &reaction); |
||||
void removeReaction(const mtx::events::RoomEvent<mtx::events::msg::Reaction> &reaction); |
||||
|
||||
private: |
||||
struct KeyReaction |
||||
{ |
||||
std::string key; |
||||
std::map<std::string, mtx::events::RoomEvent<mtx::events::msg::Reaction>> reactions; |
||||
}; |
||||
std::string room_id_; |
||||
std::vector<KeyReaction> reactions; |
||||
}; |
||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in new issue