82 lines
1.9 KiB
C++
82 lines
1.9 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 "data/data_pinned_messages.h"
|
|
|
|
#include "data/data_peer.h"
|
|
#include "main/main_session.h"
|
|
#include "storage/storage_facade.h"
|
|
#include "storage/storage_shared_media.h"
|
|
|
|
namespace Data {
|
|
namespace {
|
|
|
|
constexpr auto PinnedType = Storage::SharedMediaType::Pinned;
|
|
|
|
using Storage::SharedMediaQuery;
|
|
using Storage::SharedMediaKey;
|
|
using Storage::SharedMediaResult;
|
|
|
|
} // namespace
|
|
|
|
PinnedMessages::PinnedMessages(not_null<PeerData*> peer)
|
|
: _peer(peer)
|
|
, _storage(_peer->session().storage()) {
|
|
}
|
|
|
|
bool PinnedMessages::empty() const {
|
|
return _storage.empty(SharedMediaKey(_peer->id, PinnedType, 0));
|
|
}
|
|
|
|
MsgId PinnedMessages::topId() const {
|
|
const auto slice = _storage.snapshot(
|
|
SharedMediaQuery(
|
|
SharedMediaKey(_peer->id, PinnedType, ServerMaxMsgId),
|
|
1,
|
|
1));
|
|
return slice.messageIds.empty() ? 0 : slice.messageIds.back();
|
|
}
|
|
|
|
rpl::producer<PinnedAroundId> PinnedMessages::viewer(
|
|
MsgId aroundId,
|
|
int limit) const {
|
|
return _storage.query(
|
|
SharedMediaQuery(
|
|
SharedMediaKey(_peer->id, PinnedType, aroundId),
|
|
limit,
|
|
limit)
|
|
) | rpl::map([](const SharedMediaResult &result) {
|
|
auto data = PinnedAroundId();
|
|
data.fullCount = result.count;
|
|
data.skippedBefore = result.skippedBefore;
|
|
data.skippedAfter = result.skippedAfter;
|
|
data.ids = result.messageIds | ranges::to_vector;
|
|
return data;
|
|
});
|
|
}
|
|
|
|
void PinnedMessages::setTopId(MsgId messageId) {
|
|
while (true) {
|
|
auto top = topId();
|
|
if (top > messageId) {
|
|
_storage.remove(Storage::SharedMediaRemoveOne(
|
|
_peer->id,
|
|
PinnedType,
|
|
top));
|
|
} else if (top == messageId) {
|
|
return;
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
_storage.add(Storage::SharedMediaAddNew(
|
|
_peer->id,
|
|
PinnedType,
|
|
messageId));
|
|
}
|
|
|
|
} // namespace Data
|