afb81b8278
Some checks failed
Docker. / Ubuntu (push) Has been cancelled
User-agent updater. / User-agent (push) Failing after 15s
Lock Threads / lock (push) Failing after 10s
Waiting for answer. / waiting-for-answer (push) Failing after 22s
Close stale issues and PRs / stale (push) Successful in 13s
Needs user action. / needs-user-action (push) Failing after 8s
Can't reproduce. / cant-reproduce (push) Failing after 8s
116 lines
2.7 KiB
C++
116 lines
2.7 KiB
C++
/*
|
|
This file is part of Telegram Desktop,
|
|
the official desktop application for the Telegram messaging service.
|
|
|
|
For license and copyright information please follow this link:
|
|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
|
|
*/
|
|
#include "api/api_messages_search_merged.h"
|
|
|
|
#include "history/history.h"
|
|
|
|
namespace Api {
|
|
|
|
MessagesSearchMerged::MessagesSearchMerged(not_null<History*> history)
|
|
: _apiSearch(history) {
|
|
if (const auto migrated = history->migrateFrom()) {
|
|
_migratedSearch.emplace(migrated);
|
|
}
|
|
const auto checkWaitingForTotal = [=] {
|
|
if (_waitingForTotal) {
|
|
if (_concatedFound.total >= 0 && _migratedFirstFound.total >= 0) {
|
|
_waitingForTotal = false;
|
|
_concatedFound.total += _migratedFirstFound.total;
|
|
_newFounds.fire({});
|
|
}
|
|
} else {
|
|
_newFounds.fire({});
|
|
}
|
|
};
|
|
|
|
const auto checkFull = [=](const FoundMessages &data) {
|
|
if (data.total == int(_concatedFound.messages.size())) {
|
|
_isFull = true;
|
|
addFound(_migratedFirstFound);
|
|
}
|
|
};
|
|
|
|
_apiSearch.messagesFounds(
|
|
) | rpl::on_next([=](const FoundMessages &data) {
|
|
if (data.nextToken == _concatedFound.nextToken) {
|
|
addFound(data);
|
|
checkFull(data);
|
|
_nextFounds.fire({});
|
|
} else {
|
|
_concatedFound = data;
|
|
checkFull(data);
|
|
checkWaitingForTotal();
|
|
}
|
|
}, _lifetime);
|
|
|
|
if (_migratedSearch) {
|
|
_migratedSearch->messagesFounds(
|
|
) | rpl::on_next([=](const FoundMessages &data) {
|
|
if (_isFull) {
|
|
addFound(data);
|
|
}
|
|
if (data.nextToken == _migratedFirstFound.nextToken) {
|
|
_nextFounds.fire({});
|
|
} else {
|
|
_migratedFirstFound = data;
|
|
checkWaitingForTotal();
|
|
}
|
|
}, _lifetime);
|
|
}
|
|
}
|
|
|
|
void MessagesSearchMerged::disableMigrated() {
|
|
_migratedSearch = std::nullopt;
|
|
}
|
|
|
|
void MessagesSearchMerged::addFound(const FoundMessages &data) {
|
|
for (const auto &message : data.messages) {
|
|
_concatedFound.messages.push_back(message);
|
|
}
|
|
}
|
|
|
|
const FoundMessages &MessagesSearchMerged::messages() const {
|
|
return _concatedFound;
|
|
}
|
|
|
|
const MessagesSearch::Request &MessagesSearchMerged::request() const {
|
|
return _request;
|
|
}
|
|
|
|
void MessagesSearchMerged::clear() {
|
|
_concatedFound = {};
|
|
_migratedFirstFound = {};
|
|
}
|
|
|
|
void MessagesSearchMerged::search(const Request &search) {
|
|
_request = search;
|
|
if (_migratedSearch) {
|
|
_waitingForTotal = true;
|
|
_migratedSearch->searchMessages(search);
|
|
}
|
|
_apiSearch.searchMessages(search);
|
|
}
|
|
|
|
void MessagesSearchMerged::searchMore() {
|
|
if (_migratedSearch && _isFull) {
|
|
_migratedSearch->searchMore();
|
|
} else {
|
|
_apiSearch.searchMore();
|
|
}
|
|
}
|
|
|
|
rpl::producer<> MessagesSearchMerged::newFounds() const {
|
|
return _newFounds.events();
|
|
}
|
|
|
|
rpl::producer<> MessagesSearchMerged::nextFounds() const {
|
|
return _nextFounds.events();
|
|
}
|
|
|
|
} // namespace Api
|