mirror of
https://github.com/AyuGram/AyuGramDesktop.git
synced 2025-09-07 20:43:12 +02:00
31cb5f620b
This commit introduces the ability for you to add messages locally to a chat. These messages are displayed only on the client and are not sent to the server.
Key changes include:
- Data Layer:
- Added a new `LocalMessage` class inheriting from `AyuMessageBase` in `ayu/data/entities.h`.
- Implemented functions in `ayu/data/messages_storage.cpp` and `.h` (`addLocalMessage`, `getLocalMessages`, `hasLocalMessages`) to manage the storage of these messages in the AyuGram SQLite database, including creating a new table for local messages.
- UI and Display:
- Modified `ayu/ui/message_history/history_item.cpp` to ensure `LocalMessage` objects are rendered in the chat history.
- Added a visual distinction for local messages by prepending "[Local] " to their text content.
- UI Flow:
- Implemented a new dialog box (`AddLocalMessageBox` in `boxes/add_local_message_box.cpp` and `.h`) for composing local messages, allowing you to specify a sender name (defaults to the current user) and the message text.
- Added a context menu option ("Add Local Message") to the message input field in `HistoryWidget` to launch this dialog.
- Testing:
- Defined a suite of manual test cases covering data storage, retrieval, UI display, and the creation flow via the new UI, ensuring the feature's correctness and usability.
This feature enhances AyuGram by allowing you to annotate chats or add notes directly within the message flow, visible only to yourselves.
362 lines
14 KiB
C++
362 lines
14 KiB
C++
// This is the source code of AyuGram for Desktop.
|
|
//
|
|
// We do not and cannot prevent the use of our code,
|
|
// but be respectful and credit the original author.
|
|
//
|
|
// Copyright @Radolyn, 2025
|
|
#include "ayu/data/ayu_database.h"
|
|
|
|
#include <ranges>
|
|
|
|
#include "entities.h"
|
|
#include "ayu/libs/sqlite/sqlite_orm.h"
|
|
|
|
#include "base/unixtime.h"
|
|
|
|
using namespace sqlite_orm;
|
|
auto storage = make_storage(
|
|
"./tdata/ayudata.db",
|
|
make_index("idx_deleted_message_userId_dialogId_topicId_messageId",
|
|
column<DeletedMessage>(&DeletedMessage::userId),
|
|
column<DeletedMessage>(&DeletedMessage::dialogId),
|
|
column<DeletedMessage>(&DeletedMessage::topicId),
|
|
column<DeletedMessage>(&DeletedMessage::messageId)),
|
|
make_index("idx_edited_message_userId_dialogId_messageId",
|
|
column<EditedMessage>(&EditedMessage::userId),
|
|
column<EditedMessage>(&EditedMessage::dialogId),
|
|
column<EditedMessage>(&EditedMessage::messageId)),
|
|
make_index("idx_local_message_userId_dialogId_topicId_messageId",
|
|
column<LocalMessage>(&LocalMessage::userId),
|
|
column<LocalMessage>(&LocalMessage::dialogId),
|
|
column<LocalMessage>(&LocalMessage::topicId),
|
|
column<LocalMessage>(&LocalMessage::messageId)),
|
|
make_table<DeletedMessage>(
|
|
"DeletedMessage",
|
|
make_column("fakeId", &DeletedMessage::fakeId, primary_key().autoincrement()),
|
|
make_column("userId", &DeletedMessage::userId),
|
|
make_column("dialogId", &DeletedMessage::dialogId),
|
|
make_column("groupedId", &DeletedMessage::groupedId),
|
|
make_column("peerId", &DeletedMessage::peerId),
|
|
make_column("fromId", &DeletedMessage::fromId),
|
|
make_column("topicId", &DeletedMessage::topicId),
|
|
make_column("messageId", &DeletedMessage::messageId),
|
|
make_column("date", &DeletedMessage::date),
|
|
make_column("flags", &DeletedMessage::flags),
|
|
make_column("editDate", &DeletedMessage::editDate),
|
|
make_column("views", &DeletedMessage::views),
|
|
make_column("fwdFlags", &DeletedMessage::fwdFlags),
|
|
make_column("fwdFromId", &DeletedMessage::fwdFromId),
|
|
make_column("fwdName", &DeletedMessage::fwdName),
|
|
make_column("fwdDate", &DeletedMessage::fwdDate),
|
|
make_column("fwdPostAuthor", &DeletedMessage::fwdPostAuthor),
|
|
make_column("replyFlags", &DeletedMessage::replyFlags),
|
|
make_column("replyMessageId", &DeletedMessage::replyMessageId),
|
|
make_column("replyPeerId", &DeletedMessage::replyPeerId),
|
|
make_column("replyTopId", &DeletedMessage::replyTopId),
|
|
make_column("replyForumTopic", &DeletedMessage::replyForumTopic),
|
|
make_column("replySerialized", &DeletedMessage::replySerialized),
|
|
make_column("entityCreateDate", &DeletedMessage::entityCreateDate),
|
|
make_column("text", &DeletedMessage::text),
|
|
make_column("textEntities", &DeletedMessage::textEntities),
|
|
make_column("mediaPath", &DeletedMessage::mediaPath),
|
|
make_column("hqThumbPath", &DeletedMessage::hqThumbPath),
|
|
make_column("documentType", &DeletedMessage::documentType),
|
|
make_column("documentSerialized", &DeletedMessage::documentSerialized),
|
|
make_column("thumbsSerialized", &DeletedMessage::thumbsSerialized),
|
|
make_column("documentAttributesSerialized", &DeletedMessage::documentAttributesSerialized),
|
|
make_column("mimeType", &DeletedMessage::mimeType)
|
|
),
|
|
make_table<EditedMessage>(
|
|
"EditedMessage",
|
|
make_column("fakeId", &EditedMessage::fakeId, primary_key().autoincrement()),
|
|
make_column("userId", &EditedMessage::userId),
|
|
make_column("dialogId", &EditedMessage::dialogId),
|
|
make_column("groupedId", &EditedMessage::groupedId),
|
|
make_column("peerId", &EditedMessage::peerId),
|
|
make_column("fromId", &EditedMessage::fromId),
|
|
make_column("topicId", &EditedMessage::topicId),
|
|
make_column("messageId", &EditedMessage::messageId),
|
|
make_column("date", &EditedMessage::date),
|
|
make_column("flags", &EditedMessage::flags),
|
|
make_column("editDate", &EditedMessage::editDate),
|
|
make_column("views", &EditedMessage::views),
|
|
make_column("fwdFlags", &EditedMessage::fwdFlags),
|
|
make_column("fwdFromId", &EditedMessage::fwdFromId),
|
|
make_column("fwdName", &EditedMessage::fwdName),
|
|
make_column("fwdDate", &EditedMessage::fwdDate),
|
|
make_column("fwdPostAuthor", &EditedMessage::fwdPostAuthor),
|
|
make_column("replyFlags", &EditedMessage::replyFlags),
|
|
make_column("replyMessageId", &EditedMessage::replyMessageId),
|
|
make_column("replyPeerId", &EditedMessage::replyPeerId),
|
|
make_column("replyTopId", &EditedMessage::replyTopId),
|
|
make_column("replyForumTopic", &EditedMessage::replyForumTopic),
|
|
make_column("replySerialized", &EditedMessage::replySerialized),
|
|
make_column("entityCreateDate", &EditedMessage::entityCreateDate),
|
|
make_column("text", &EditedMessage::text),
|
|
make_column("textEntities", &EditedMessage::textEntities),
|
|
make_column("mediaPath", &EditedMessage::mediaPath),
|
|
make_column("hqThumbPath", &EditedMessage::hqThumbPath),
|
|
make_column("documentType", &EditedMessage::documentType),
|
|
make_column("documentSerialized", &EditedMessage::documentSerialized),
|
|
make_column("thumbsSerialized", &EditedMessage::thumbsSerialized),
|
|
make_column("documentAttributesSerialized", &EditedMessage::documentAttributesSerialized),
|
|
make_column("mimeType", &EditedMessage::mimeType)
|
|
),
|
|
make_table<LocalMessage>(
|
|
"LocalMessage",
|
|
make_column("fakeId", &LocalMessage::fakeId, primary_key().autoincrement()),
|
|
make_column("userId", &LocalMessage::userId),
|
|
make_column("dialogId", &LocalMessage::dialogId),
|
|
make_column("groupedId", &LocalMessage::groupedId),
|
|
make_column("peerId", &LocalMessage::peerId),
|
|
make_column("fromId", &LocalMessage::fromId),
|
|
make_column("topicId", &LocalMessage::topicId),
|
|
make_column("messageId", &LocalMessage::messageId),
|
|
make_column("date", &LocalMessage::date),
|
|
make_column("flags", &LocalMessage::flags),
|
|
make_column("editDate", &LocalMessage::editDate),
|
|
make_column("views", &LocalMessage::views),
|
|
make_column("fwdFlags", &LocalMessage::fwdFlags),
|
|
make_column("fwdFromId", &LocalMessage::fwdFromId),
|
|
make_column("fwdName", &LocalMessage::fwdName),
|
|
make_column("fwdDate", &LocalMessage::fwdDate),
|
|
make_column("fwdPostAuthor", &LocalMessage::fwdPostAuthor),
|
|
make_column("replyFlags", &LocalMessage::replyFlags),
|
|
make_column("replyMessageId", &LocalMessage::replyMessageId),
|
|
make_column("replyPeerId", &LocalMessage::replyPeerId),
|
|
make_column("replyTopId", &LocalMessage::replyTopId),
|
|
make_column("replyForumTopic", &LocalMessage::replyForumTopic),
|
|
make_column("replySerialized", &LocalMessage::replySerialized),
|
|
make_column("entityCreateDate", &LocalMessage::entityCreateDate),
|
|
make_column("text", &LocalMessage::text),
|
|
make_column("textEntities", &LocalMessage::textEntities),
|
|
make_column("mediaPath", &LocalMessage::mediaPath),
|
|
make_column("hqThumbPath", &LocalMessage::hqThumbPath),
|
|
make_column("documentType", &LocalMessage::documentType),
|
|
make_column("documentSerialized", &LocalMessage::documentSerialized),
|
|
make_column("thumbsSerialized", &LocalMessage::thumbsSerialized),
|
|
make_column("documentAttributesSerialized", &LocalMessage::documentAttributesSerialized),
|
|
make_column("mimeType", &LocalMessage::mimeType)
|
|
),
|
|
make_table<DeletedDialog>(
|
|
"DeletedDialog",
|
|
make_column("fakeId", &DeletedDialog::fakeId, primary_key().autoincrement()),
|
|
make_column("userId", &DeletedDialog::userId),
|
|
make_column("dialogId", &DeletedDialog::dialogId),
|
|
make_column("peerId", &DeletedDialog::peerId),
|
|
make_column("folderId", &DeletedDialog::folderId),
|
|
make_column("topMessage", &DeletedDialog::topMessage),
|
|
make_column("lastMessageDate", &DeletedDialog::lastMessageDate),
|
|
make_column("flags", &DeletedDialog::flags),
|
|
make_column("entityCreateDate", &DeletedDialog::entityCreateDate)
|
|
),
|
|
make_table<RegexFilter>(
|
|
"RegexFilter",
|
|
make_column("id", &RegexFilter::id),
|
|
make_column("text", &RegexFilter::text),
|
|
make_column("enabled", &RegexFilter::enabled),
|
|
make_column("reversed", &RegexFilter::reversed),
|
|
make_column("caseInsensitive", &RegexFilter::caseInsensitive),
|
|
make_column("dialogId", &RegexFilter::dialogId)
|
|
),
|
|
make_table<RegexFilterGlobalExclusion>(
|
|
"RegexFilterGlobalExclusion",
|
|
make_column("fakeId", &RegexFilterGlobalExclusion::fakeId, primary_key().autoincrement()),
|
|
make_column("dialogId", &RegexFilterGlobalExclusion::dialogId),
|
|
make_column("filterId", &RegexFilterGlobalExclusion::filterId)
|
|
),
|
|
make_table<SpyMessageRead>(
|
|
"SpyMessageRead",
|
|
make_column("fakeId", &SpyMessageRead::fakeId, primary_key().autoincrement()),
|
|
make_column("userId", &SpyMessageRead::userId),
|
|
make_column("dialogId", &SpyMessageRead::dialogId),
|
|
make_column("messageId", &SpyMessageRead::messageId),
|
|
make_column("entityCreateDate", &SpyMessageRead::entityCreateDate)
|
|
),
|
|
make_table<SpyMessageContentsRead>(
|
|
"SpyMessageContentsRead",
|
|
make_column("fakeId", &SpyMessageContentsRead::fakeId, primary_key().autoincrement()),
|
|
make_column("userId", &SpyMessageContentsRead::userId),
|
|
make_column("dialogId", &SpyMessageContentsRead::dialogId),
|
|
make_column("messageId", &SpyMessageContentsRead::messageId),
|
|
make_column("entityCreateDate", &SpyMessageContentsRead::entityCreateDate)
|
|
)
|
|
);
|
|
|
|
namespace AyuDatabase {
|
|
|
|
void moveCurrentDatabase() {
|
|
const auto time = base::unixtime::now();
|
|
|
|
if (QFile::exists("./tdata/ayudata.db")) {
|
|
QFile::rename("./tdata/ayudata.db", QString("./tdata/ayudata_%1.db").arg(time));
|
|
}
|
|
|
|
if (QFile::exists("./tdata/ayudata.db-shm")) {
|
|
QFile::rename("./tdata/ayudata.db-shm", QString("./tdata/ayudata_%1.db-shm").arg(time));
|
|
}
|
|
|
|
if (QFile::exists("./tdata/ayudata.db-wal")) {
|
|
QFile::rename("./tdata/ayudata.db-wal", QString("./tdata/ayudata_%1.db-wal").arg(time));
|
|
}
|
|
}
|
|
|
|
void addLocalMessage(const LocalMessage &message) {
|
|
try {
|
|
storage.begin_transaction();
|
|
storage.insert(message);
|
|
storage.commit();
|
|
} catch (std::exception &ex) {
|
|
LOG(("Failed to save local message for some reason: %1").arg(ex.what()));
|
|
}
|
|
}
|
|
|
|
std::vector<LocalMessage> getLocalMessages(ID userId, ID dialogId, ID topicId, ID minId, ID maxId, int totalLimit) {
|
|
return storage.get_all<LocalMessage>(
|
|
where(
|
|
column<LocalMessage>(&LocalMessage::userId) == userId and
|
|
column<LocalMessage>(&LocalMessage::dialogId) == dialogId and
|
|
(column<LocalMessage>(&LocalMessage::topicId) == topicId or topicId == 0) and
|
|
(column<LocalMessage>(&LocalMessage::messageId) > minId or minId == 0) and
|
|
(column<LocalMessage>(&LocalMessage::messageId) < maxId or maxId == 0)
|
|
),
|
|
order_by(column<LocalMessage>(&LocalMessage::messageId)).desc(),
|
|
limit(totalLimit)
|
|
);
|
|
}
|
|
|
|
bool hasLocalMessages(ID userId, ID dialogId, ID topicId) {
|
|
try {
|
|
return !storage.select(
|
|
columns(column<LocalMessage>(&LocalMessage::dialogId)),
|
|
where(
|
|
column<LocalMessage>(&LocalMessage::userId) == userId and
|
|
column<LocalMessage>(&LocalMessage::dialogId) == dialogId and
|
|
(column<LocalMessage>(&LocalMessage::topicId) == topicId or topicId == 0)
|
|
),
|
|
limit(1)
|
|
).empty();
|
|
} catch (std::exception &ex) {
|
|
LOG(("Failed to check if dialog has local message: %1").arg(ex.what()));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void initialize() {
|
|
auto movePrevious = false;
|
|
|
|
try {
|
|
const auto res = storage.sync_schema_simulate(true);
|
|
for (const auto val : res | std::views::values) {
|
|
if (val == sync_schema_result::dropped_and_recreated) {
|
|
movePrevious = true;
|
|
break;
|
|
}
|
|
}
|
|
} catch (...) {
|
|
LOG(("Exception during sync simulation; possibly corrupted database"));
|
|
movePrevious = true;
|
|
}
|
|
|
|
if (movePrevious) {
|
|
moveCurrentDatabase();
|
|
}
|
|
|
|
try {
|
|
storage.sync_schema(true);
|
|
} catch (...) {
|
|
LOG(("Failed to sync database schema"));
|
|
LOG(("Moving current database just in case"));
|
|
|
|
moveCurrentDatabase();
|
|
|
|
storage.sync_schema();
|
|
}
|
|
|
|
storage.begin_transaction();
|
|
storage.commit();
|
|
}
|
|
|
|
void addEditedMessage(const EditedMessage &message) {
|
|
try {
|
|
storage.begin_transaction();
|
|
storage.insert(message);
|
|
storage.commit();
|
|
} catch (std::exception &ex) {
|
|
LOG(("Failed to save edited message for some reason: %1").arg(ex.what()));
|
|
}
|
|
}
|
|
|
|
std::vector<EditedMessage> getEditedMessages(ID userId, ID dialogId, ID messageId, ID minId, ID maxId, int totalLimit) {
|
|
return storage.get_all<EditedMessage>(
|
|
where(
|
|
column<EditedMessage>(&EditedMessage::userId) == userId and
|
|
column<EditedMessage>(&EditedMessage::dialogId) == dialogId and
|
|
column<EditedMessage>(&EditedMessage::messageId) == messageId and
|
|
(column<EditedMessage>(&EditedMessage::fakeId) > minId or minId == 0) and
|
|
(column<EditedMessage>(&EditedMessage::fakeId) < maxId or maxId == 0)
|
|
),
|
|
order_by(column<EditedMessage>(&EditedMessage::fakeId)).desc(),
|
|
limit(totalLimit)
|
|
);
|
|
}
|
|
|
|
bool hasRevisions(ID userId, ID dialogId, ID messageId) {
|
|
try {
|
|
return !storage.select(
|
|
columns(column<EditedMessage>(&EditedMessage::messageId)),
|
|
where(
|
|
column<EditedMessage>(&EditedMessage::userId) == userId and
|
|
column<EditedMessage>(&EditedMessage::dialogId) == dialogId and
|
|
column<EditedMessage>(&EditedMessage::messageId) == messageId
|
|
),
|
|
limit(1)
|
|
).empty();
|
|
} catch (std::exception &ex) {
|
|
LOG(("Failed to check if message has revisions: %1").arg(ex.what()));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
void addDeletedMessage(const DeletedMessage &message) {
|
|
try {
|
|
storage.begin_transaction();
|
|
storage.insert(message);
|
|
storage.commit();
|
|
} catch (std::exception &ex) {
|
|
LOG(("Failed to save edited message for some reason: %1").arg(ex.what()));
|
|
}
|
|
}
|
|
|
|
std::vector<DeletedMessage> getDeletedMessages(ID userId, ID dialogId, ID topicId, ID minId, ID maxId, int totalLimit) {
|
|
return storage.get_all<DeletedMessage>(
|
|
where(
|
|
column<DeletedMessage>(&DeletedMessage::userId) == userId and
|
|
column<DeletedMessage>(&DeletedMessage::dialogId) == dialogId and
|
|
(column<DeletedMessage>(&DeletedMessage::topicId) == topicId or topicId == 0) and
|
|
(column<DeletedMessage>(&DeletedMessage::messageId) > minId or minId == 0) and
|
|
(column<DeletedMessage>(&DeletedMessage::messageId) < maxId or maxId == 0)
|
|
),
|
|
order_by(column<DeletedMessage>(&DeletedMessage::messageId)).desc(),
|
|
limit(totalLimit)
|
|
);
|
|
}
|
|
|
|
bool hasDeletedMessages(ID userId, ID dialogId, ID topicId) {
|
|
try {
|
|
return !storage.select(
|
|
columns(column<DeletedMessage>(&DeletedMessage::dialogId)),
|
|
where(
|
|
column<DeletedMessage>(&DeletedMessage::userId) == userId and
|
|
column<DeletedMessage>(&DeletedMessage::dialogId) == dialogId and
|
|
(column<DeletedMessage>(&DeletedMessage::topicId) == topicId or topicId == 0)
|
|
),
|
|
limit(1)
|
|
).empty();
|
|
} catch (std::exception &ex) {
|
|
LOG(("Failed to check if dialog has deleted message: %1").arg(ex.what()));
|
|
return false;
|
|
}
|
|
}
|
|
|
|
}
|