mirror of
https://github.com/AyuGram/AyuGramDesktop.git
synced 2025-09-09 05:23:17 +02:00
651 lines
18 KiB
C++
651 lines
18 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/ui/context_menu/context_menu.h"
|
||
|
||
#include "apiwrap.h"
|
||
#include "lang_auto.h"
|
||
#include "mainwidget.h"
|
||
#include "mainwindow.h"
|
||
#include "ayu/ayu_settings.h"
|
||
#include "ayu/ayu_state.h"
|
||
#include "ayu/data/messages_storage.h"
|
||
#include "ayu/ui/context_menu/menu_item_subtext.h"
|
||
#include "ayu/utils/qt_key_modifiers_extended.h"
|
||
#include "history/history_item_components.h"
|
||
|
||
#include "core/mime_type.h"
|
||
#include "styles/style_ayu_icons.h"
|
||
#include "styles/style_menu_icons.h"
|
||
#include "ui/widgets/popup_menu.h"
|
||
#include "ui/widgets/menu/menu_add_action_callback_factory.h"
|
||
#include "ui/widgets/fields/input_field.h"
|
||
#include "ui/boxes/generic_box.h"
|
||
#include "window/window_peer_menu.h"
|
||
|
||
#include "ayu/ui/message_history/history_section.h"
|
||
#include "ayu/utils/telegram_helpers.h"
|
||
#include "base/unixtime.h"
|
||
#include "data/data_channel.h"
|
||
#include "data/data_user.h"
|
||
#include "data/data_chat.h"
|
||
#include "data/data_forum_topic.h"
|
||
#include "data/data_session.h"
|
||
#include "history/view/history_view_context_menu.h"
|
||
#include "history/view/history_view_element.h"
|
||
#include "history/history.h"
|
||
#include "window/window_controller.h"
|
||
#include "window/window_session_controller.h"
|
||
#include "base/unixtime.h"
|
||
|
||
namespace AyuUi {
|
||
|
||
bool needToShowItem(int state) {
|
||
return state == 1 || (state == 2 && base::IsExtendedContextMenuModifierPressed());
|
||
}
|
||
|
||
void AddDeletedMessagesActions(PeerData *peerData,
|
||
Data::Thread *thread,
|
||
not_null<Window::SessionController*> sessionController,
|
||
const Window::PeerMenuCallback &addCallback) {
|
||
if (!peerData) {
|
||
return;
|
||
}
|
||
|
||
const auto topic = peerData->isForum() ? thread->asTopic() : nullptr;
|
||
const auto topicId = topic ? topic->rootId().bare : 0;
|
||
|
||
// const auto has = AyuMessages::hasDeletedMessages(peerData, topicId);
|
||
// if (!has) {
|
||
// return;
|
||
// }
|
||
|
||
addCallback(
|
||
tr::ayu_ViewDeletedMenuText(tr::now),
|
||
[=]
|
||
{
|
||
sessionController->session().tryResolveWindow()
|
||
->showSection(std::make_shared<MessageHistory::SectionMemento>(peerData, nullptr, topicId));
|
||
},
|
||
&st::menuIconArchive);
|
||
}
|
||
|
||
void AddJumpToBeginningAction(PeerData *peerData,
|
||
Data::Thread *thread,
|
||
not_null<Window::SessionController*> sessionController,
|
||
const Window::PeerMenuCallback &addCallback) {
|
||
const auto user = peerData->asUser();
|
||
const auto group = peerData->isChat() ? peerData->asChat() : nullptr;
|
||
const auto chat = peerData->isMegagroup()
|
||
? peerData->asMegagroup()
|
||
: peerData->isChannel()
|
||
? peerData->asChannel()
|
||
: nullptr;
|
||
const auto topic = peerData->isForum() ? thread->asTopic() : nullptr;
|
||
if (!user && !group && !chat && !topic) {
|
||
return;
|
||
}
|
||
if (topic && topic->creating()) {
|
||
return;
|
||
}
|
||
|
||
const auto controller = sessionController;
|
||
const auto jumpToDate = [=](auto history, auto callback)
|
||
{
|
||
const auto weak = base::make_weak(controller);
|
||
controller->session().api().resolveJumpToDate(
|
||
history,
|
||
QDate(2013, 8, 1),
|
||
[=](not_null<PeerData*> peer, MsgId id)
|
||
{
|
||
if (const auto strong = weak.get()) {
|
||
callback(peer, id);
|
||
}
|
||
});
|
||
};
|
||
|
||
const auto showPeerHistory = [=](auto peer, MsgId id)
|
||
{
|
||
controller->showPeerHistory(
|
||
peer,
|
||
Window::SectionShow::Way::Forward,
|
||
id);
|
||
};
|
||
|
||
const auto showTopic = [=](auto topic, MsgId id)
|
||
{
|
||
controller->showTopic(
|
||
topic,
|
||
id,
|
||
Window::SectionShow::Way::Forward);
|
||
};
|
||
|
||
addCallback(
|
||
tr::ayu_JumpToBeginning(tr::now),
|
||
[=]
|
||
{
|
||
if (user) {
|
||
jumpToDate(controller->session().data().history(user), showPeerHistory);
|
||
} else if (group && !chat) {
|
||
jumpToDate(controller->session().data().history(group), showPeerHistory);
|
||
} else if (chat && !topic) {
|
||
if (!chat->migrateFrom() && chat->availableMinId() == 1) {
|
||
showPeerHistory(chat, 1);
|
||
} else {
|
||
jumpToDate(controller->session().data().history(chat), showPeerHistory);
|
||
}
|
||
} else if (topic) {
|
||
if (topic->isGeneral()) {
|
||
showTopic(topic, 1);
|
||
} else {
|
||
jumpToDate(
|
||
topic,
|
||
[=](not_null<PeerData*>, MsgId id)
|
||
{
|
||
showTopic(topic, id);
|
||
});
|
||
}
|
||
}
|
||
},
|
||
&st::ayuMenuIconToBeginning);
|
||
}
|
||
|
||
void AddOpenChannelAction(PeerData *peerData,
|
||
not_null<Window::SessionController*> sessionController,
|
||
const Window::PeerMenuCallback &addCallback) {
|
||
if (!peerData || !peerData->isMegagroup()) {
|
||
return;
|
||
}
|
||
|
||
const auto chat = peerData->asMegagroup()->discussionLink();
|
||
if (!chat) {
|
||
return;
|
||
}
|
||
|
||
addCallback(
|
||
tr::lng_context_open_channel(tr::now),
|
||
[=]
|
||
{
|
||
sessionController->showPeerHistory(chat, Window::SectionShow::Way::Forward);
|
||
},
|
||
&st::menuIconChannel);
|
||
}
|
||
|
||
void AddHistoryAction(not_null<Ui::PopupMenu*> menu, HistoryItem *item) {
|
||
if (item->hideEditedBadge()) {
|
||
return;
|
||
}
|
||
|
||
const auto edited = item->Get<HistoryMessageEdited>();
|
||
if (!edited) {
|
||
return;
|
||
}
|
||
|
||
const auto has = AyuMessages::hasRevisions(item);
|
||
if (!has) {
|
||
return;
|
||
}
|
||
|
||
menu->addAction(
|
||
tr::ayu_EditsHistoryMenuText(tr::now),
|
||
[=]
|
||
{
|
||
item->history()->session().tryResolveWindow()
|
||
->showSection(
|
||
std::make_shared<MessageHistory::SectionMemento>(item->history()->peer, item, 0));
|
||
},
|
||
&st::ayuEditsHistoryIcon);
|
||
}
|
||
|
||
void AddHideMessageAction(not_null<Ui::PopupMenu*> menu, HistoryItem *item) {
|
||
const auto& settings = AyuSettings::getInstance();
|
||
if (!needToShowItem(settings.showHideMessageInContextMenu)) {
|
||
return;
|
||
}
|
||
|
||
if (item->history()->peer->isSelf()) {
|
||
return;
|
||
}
|
||
|
||
const auto history = item->history();
|
||
menu->addAction(
|
||
tr::ayu_ContextHideMessage(tr::now),
|
||
[=]()
|
||
{
|
||
item->destroy();
|
||
history->requestChatListMessage();
|
||
|
||
AyuState::hide(item);
|
||
},
|
||
&st::menuIconClear);
|
||
}
|
||
|
||
void AddUserMessagesAction(not_null<Ui::PopupMenu*> menu, HistoryItem *item) {
|
||
const auto& settings = AyuSettings::getInstance();
|
||
if (!needToShowItem(settings.showUserMessagesInContextMenu)) {
|
||
return;
|
||
}
|
||
|
||
if (item->history()->peer->isChat() || item->history()->peer->isMegagroup()) {
|
||
menu->addAction(
|
||
tr::ayu_UserMessagesMenuText(tr::now),
|
||
[=]
|
||
{
|
||
if (const auto controller = item->history()->session().tryResolveWindow()) {
|
||
const auto peer = item->history()->peer;
|
||
const auto key = (peer && !peer->isUser())
|
||
? item->topic()
|
||
? Dialogs::Key{item->topic()}
|
||
: Dialogs::Key{item->history()}
|
||
: Dialogs::Key{item->history()};
|
||
controller->searchInChat(key, item->from());
|
||
}
|
||
},
|
||
&st::menuIconTTL);
|
||
}
|
||
}
|
||
|
||
void AddMessageDetailsAction(not_null<Ui::PopupMenu*> menu, HistoryItem *item) {
|
||
const auto& settings = AyuSettings::getInstance();
|
||
if (!needToShowItem(settings.showMessageDetailsInContextMenu)) {
|
||
return;
|
||
}
|
||
|
||
if (item->isLocal()) {
|
||
return;
|
||
}
|
||
|
||
const auto view = item->mainView();
|
||
const auto forwarded = item->Get<HistoryMessageForwarded>();
|
||
const auto views = item->Get<HistoryMessageViews>();
|
||
const auto media = item->media();
|
||
|
||
const auto isSticker = media && media->document() && media->document()->sticker();
|
||
|
||
const auto emojiPacks = HistoryView::CollectEmojiPacks(item, HistoryView::EmojiPacksSource::Message);
|
||
auto containsSingleCustomEmojiPack = emojiPacks.size() == 1;
|
||
if (!containsSingleCustomEmojiPack && emojiPacks.size() > 1) {
|
||
const auto author = emojiPacks.front().id >> 32;
|
||
auto sameAuthor = true;
|
||
for (const auto &pack : emojiPacks) {
|
||
if (pack.id >> 32 != author) {
|
||
sameAuthor = false;
|
||
break;
|
||
}
|
||
}
|
||
|
||
containsSingleCustomEmojiPack = sameAuthor;
|
||
}
|
||
|
||
const auto isForwarded = forwarded && !forwarded->story && forwarded->psaType.isEmpty();
|
||
|
||
const auto messageId = QString::number(item->id.bare);
|
||
const auto messageDate = base::unixtime::parse(item->date());
|
||
const auto messageEditDate = base::unixtime::parse(view ? view->displayedEditDate() : TimeId(0));
|
||
|
||
const auto messageForwardedDate =
|
||
isForwarded && forwarded
|
||
? base::unixtime::parse(forwarded->originalDate)
|
||
: QDateTime();
|
||
|
||
const auto
|
||
messageViews = item->hasViews() && item->viewsCount() > 0 ? QString::number(item->viewsCount()) : QString();
|
||
const auto messageForwards = views && views->forwardsCount > 0 ? QString::number(views->forwardsCount) : QString();
|
||
|
||
const auto mediaSize = media ? getMediaSize(item) : QString();
|
||
const auto mediaMime = media ? getMediaMime(item) : QString();
|
||
// todo: bitrate (?)
|
||
const auto mediaName = media ? getMediaName(item) : QString();
|
||
const auto mediaResolution = media ? getMediaResolution(item) : QString();
|
||
const auto mediaDC = media ? getMediaDC(item) : QString();
|
||
|
||
const auto hasAnyPostField =
|
||
!messageViews.isEmpty() ||
|
||
!messageForwards.isEmpty();
|
||
|
||
const auto hasAnyMediaField =
|
||
!mediaSize.isEmpty() ||
|
||
!mediaMime.isEmpty() ||
|
||
!mediaName.isEmpty() ||
|
||
!mediaResolution.isEmpty() ||
|
||
!mediaDC.isEmpty();
|
||
|
||
const auto callback = Ui::Menu::CreateAddActionCallback(menu);
|
||
|
||
callback(Window::PeerMenuCallback::Args{
|
||
.text = tr::ayu_MessageDetailsPC(tr::now),
|
||
.handler = nullptr,
|
||
.icon = &st::menuIconInfo,
|
||
.fillSubmenu = [&](not_null<Ui::PopupMenu*> menu2)
|
||
{
|
||
if (hasAnyPostField) {
|
||
if (!messageViews.isEmpty()) {
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconShowInChat,
|
||
tr::ayu_MessageDetailsViewsPC(tr::now),
|
||
messageViews
|
||
));
|
||
}
|
||
|
||
if (!messageForwards.isEmpty()) {
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconViewReplies,
|
||
tr::ayu_MessageDetailsSharesPC(tr::now),
|
||
messageForwards
|
||
));
|
||
}
|
||
|
||
menu2->addSeparator();
|
||
}
|
||
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconInfo,
|
||
QString("ID"),
|
||
messageId
|
||
));
|
||
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconSchedule,
|
||
tr::ayu_MessageDetailsDatePC(tr::now),
|
||
formatDateTime(messageDate)
|
||
));
|
||
|
||
if (view && view->displayedEditDate()) {
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconEdit,
|
||
tr::ayu_MessageDetailsEditedDatePC(tr::now),
|
||
formatDateTime(messageEditDate)
|
||
));
|
||
}
|
||
|
||
if (isForwarded) {
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconTTL,
|
||
tr::ayu_MessageDetailsForwardedDatePC(tr::now),
|
||
formatDateTime(messageForwardedDate)
|
||
));
|
||
}
|
||
|
||
if (media && hasAnyMediaField) {
|
||
menu2->addSeparator();
|
||
|
||
if (!mediaSize.isEmpty()) {
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconDownload,
|
||
tr::ayu_MessageDetailsFileSizePC(tr::now),
|
||
mediaSize
|
||
));
|
||
}
|
||
|
||
if (!mediaMime.isEmpty()) {
|
||
const auto mime = Core::MimeTypeForName(mediaMime);
|
||
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconShowAll,
|
||
tr::ayu_MessageDetailsMimeTypePC(tr::now),
|
||
mime.name()
|
||
));
|
||
}
|
||
|
||
if (!mediaName.isEmpty()) {
|
||
auto const shortified = mediaName.length() > 20 ? "…" + mediaName.right(20) : mediaName;
|
||
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::ayuEditsHistoryIcon,
|
||
tr::ayu_MessageDetailsFileNamePC(tr::now),
|
||
shortified,
|
||
[=]
|
||
{
|
||
QGuiApplication::clipboard()->setText(mediaName);
|
||
}
|
||
));
|
||
}
|
||
|
||
if (!mediaResolution.isEmpty()) {
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconStats,
|
||
tr::ayu_MessageDetailsResolutionPC(tr::now),
|
||
mediaResolution
|
||
));
|
||
}
|
||
|
||
if (!mediaDC.isEmpty()) {
|
||
menu2->addAction(Ui::ContextActionWithSubText(
|
||
menu2->menu(),
|
||
st::menuIconBoosts,
|
||
tr::ayu_MessageDetailsDatacenterPC(tr::now),
|
||
mediaDC
|
||
));
|
||
}
|
||
|
||
if (isSticker) {
|
||
const auto authorId = getUserIdFromPackId(media->document()->sticker()->set.id);
|
||
|
||
if (authorId != 0) {
|
||
menu2->addAction(Ui::ContextActionStickerAuthor(
|
||
menu2->menu(),
|
||
&item->history()->session(),
|
||
authorId
|
||
));
|
||
}
|
||
}
|
||
}
|
||
|
||
if (containsSingleCustomEmojiPack) {
|
||
const auto authorId = getUserIdFromPackId(emojiPacks.front().id);
|
||
|
||
if (authorId != 0) {
|
||
menu2->addAction(Ui::ContextActionStickerAuthor(
|
||
menu2->menu(),
|
||
&item->history()->session(),
|
||
authorId
|
||
));
|
||
}
|
||
}
|
||
},
|
||
});
|
||
}
|
||
|
||
void AddReadUntilAction(not_null<Ui::PopupMenu*> menu, HistoryItem *item) {
|
||
if (item->isLocal() || item->isService() || item->out() || item->isDeleted()) {
|
||
return;
|
||
}
|
||
|
||
if (item->history()->peer->isSelf()) {
|
||
return;
|
||
}
|
||
|
||
const auto& settings = AyuSettings::getInstance();
|
||
if (settings.sendReadMessages) {
|
||
return;
|
||
}
|
||
|
||
menu->addAction(
|
||
tr::ayu_ReadUntilMenuText(tr::now),
|
||
[=]()
|
||
{
|
||
readHistory(item);
|
||
if (item->media() && item->media()->ttlSeconds() <= 0 && item->unsupportedTTL() <= 0 && !item->out() && item
|
||
->isUnreadMedia()) {
|
||
const auto ids = MTP_vector<MTPint>(1, MTP_int(item->id));
|
||
if (const auto channel = item->history()->peer->asChannel()) {
|
||
item->history()->session().api().request(MTPchannels_ReadMessageContents(
|
||
channel->inputChannel,
|
||
ids
|
||
)).send();
|
||
} else {
|
||
item->history()->session().api().request(MTPmessages_ReadMessageContents(
|
||
ids
|
||
)).done([=](const MTPmessages_AffectedMessages &result)
|
||
{
|
||
item->history()->session().api().applyAffectedMessages(
|
||
item->history()->peer,
|
||
result);
|
||
}).send();
|
||
}
|
||
item->markContentsRead();
|
||
}
|
||
},
|
||
&st::menuIconShowInChat);
|
||
}
|
||
|
||
void AddBurnAction(not_null<Ui::PopupMenu*> menu, HistoryItem *item) {
|
||
if (!item->media() || item->media()->ttlSeconds() <= 0 && item->unsupportedTTL() <= 0 || item->out() ||
|
||
!item->isUnreadMedia()) {
|
||
return;
|
||
}
|
||
|
||
menu->addAction(
|
||
tr::ayu_ExpireMediaContextMenuText(tr::now),
|
||
[=]()
|
||
{
|
||
const auto ids = MTP_vector<MTPint>(1, MTP_int(item->id));
|
||
const auto callback = [=]()
|
||
{
|
||
if (const auto window = Core::App().activeWindow()) {
|
||
if (const auto controller = window->sessionController()) {
|
||
controller->showToast(tr::lng_box_ok(tr::now));
|
||
}
|
||
}
|
||
};
|
||
|
||
if (const auto channel = item->history()->peer->asChannel()) {
|
||
item->history()->session().api().request(MTPchannels_ReadMessageContents(
|
||
channel->inputChannel,
|
||
ids
|
||
)).done([=]()
|
||
{
|
||
callback();
|
||
}).send();
|
||
} else {
|
||
item->history()->session().api().request(MTPmessages_ReadMessageContents(
|
||
ids
|
||
)).done([=](const MTPmessages_AffectedMessages &result)
|
||
{
|
||
item->history()->session().api().applyAffectedMessages(
|
||
item->history()->peer,
|
||
result);
|
||
callback();
|
||
}).send();
|
||
}
|
||
item->markContentsRead();
|
||
},
|
||
&st::menuIconTTLAny);
|
||
}
|
||
|
||
void AddLocalMessageAction(not_null<Ui::PopupMenu*> menu, HistoryItem *item) {
|
||
if (!item) {
|
||
return;
|
||
}
|
||
|
||
const auto& settings = AyuSettings::getInstance();
|
||
if (!needToShowItem(settings.showAddLocalMessageInContextMenu)) {
|
||
return;
|
||
}
|
||
|
||
const auto history = item->history();
|
||
const auto controller = history->session().tryResolveWindow();
|
||
if (!controller) {
|
||
return;
|
||
}
|
||
|
||
menu->addAction(
|
||
tr::ayu_AddLocalMessage(tr::now),
|
||
[=]() {
|
||
const auto showBox = [=](const QString &text, PeerId fromId) {
|
||
if (text.isEmpty()) {
|
||
return;
|
||
}
|
||
|
||
// Создаем локальное сообщение
|
||
const auto localId = history->session().data().nextLocalMessageId();
|
||
const auto localItem = history->addNewLocalMessage({
|
||
.id = localId,
|
||
.flags = MessageFlag::HasFromId,
|
||
.from = fromId,
|
||
.date = base::unixtime::now(),
|
||
}, TextWithEntities{ text }, MTP_messageMediaEmpty());
|
||
|
||
// Добавляем в базу данных локальных сообщений
|
||
AyuMessages::addLocalMessage(localItem);
|
||
};
|
||
|
||
controller->show(Box([=](not_null<Ui::GenericBox*> box) {
|
||
box->setTitle(tr::ayu_AddLocalMessageTitle());
|
||
|
||
// Поле для ввода текста сообщения
|
||
const auto textField = box->addRow(object_ptr<Ui::InputField>(
|
||
box,
|
||
st::defaultInputField,
|
||
tr::ayu_AddLocalMessagePlaceholder(),
|
||
QString()));
|
||
textField->setMaxLength(4096);
|
||
|
||
// Поле для выбора отправителя
|
||
const auto fromField = box->addRow(object_ptr<Ui::InputField>(
|
||
box,
|
||
st::defaultInputField,
|
||
tr::ayu_AddLocalMessageFromPlaceholder(),
|
||
QString()));
|
||
|
||
box->setFocusCallback([=] {
|
||
textField->setFocusFast();
|
||
});
|
||
|
||
box->addButton(tr::lng_box_ok(), [=] {
|
||
const auto text = textField->getLastText().trimmed();
|
||
const auto fromText = fromField->getLastText().trimmed();
|
||
|
||
PeerId fromId = history->peer->id;
|
||
if (!fromText.isEmpty()) {
|
||
// Пытаемся найти пользователя по имени или username
|
||
const auto session = &history->session();
|
||
const auto &owner = session->data();
|
||
|
||
// Сначала пытаемся найти по username
|
||
if (const auto user = owner.userByUsername(fromText)) {
|
||
fromId = user->id;
|
||
} else {
|
||
// Затем ищем по имени среди участников чата
|
||
if (const auto chat = history->peer->asChat()) {
|
||
for (const auto &participant : chat->participants) {
|
||
if (const auto user = owner.user(participant.userId())) {
|
||
if (user->name().toLower().contains(fromText.toLower())) {
|
||
fromId = user->id;
|
||
break;
|
||
}
|
||
}
|
||
}
|
||
} else if (const auto channel = history->peer->asChannel()) {
|
||
// Для каналов используем текущего пользователя, если не найден
|
||
// В реальном приложении можно было бы реализовать поиск по участникам
|
||
}
|
||
}
|
||
}
|
||
|
||
showBox(text, fromId);
|
||
box->closeBox();
|
||
});
|
||
|
||
box->addButton(tr::lng_cancel(), [=] {
|
||
box->closeBox();
|
||
});
|
||
}));
|
||
},
|
||
&st::menuIconEdit);
|
||
}
|
||
|
||
} // namespace AyuUi
|