tdesktop/Telegram/SourceFiles/dialogs/ui/dialogs_stories_content.cpp

268 lines
7.4 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 "dialogs/ui/dialogs_stories_content.h"
#include "data/data_changes.h"
#include "data/data_document.h"
#include "data/data_document_media.h"
#include "data/data_file_origin.h"
#include "data/data_photo.h"
#include "data/data_photo_media.h"
#include "data/data_session.h"
#include "data/data_stories.h"
#include "data/data_user.h"
#include "dialogs/ui/dialogs_stories_list.h"
#include "info/stories/info_stories_widget.h"
#include "info/info_controller.h"
#include "info/info_memento.h"
#include "main/main_session.h"
#include "lang/lang_keys.h"
#include "ui/dynamic_image.h"
#include "ui/dynamic_thumbnails.h"
#include "ui/painter.h"
#include "window/window_session_controller.h"
#include "styles/style_menu_icons.h"
namespace Dialogs::Stories {
namespace {
constexpr auto kShownLastCount = 3;
class State final {
public:
State(not_null<Data::Stories*> data, Data::StorySourcesList list);
[[nodiscard]] Content next();
private:
const not_null<Data::Stories*> _data;
const Data::StorySourcesList _list;
base::flat_map<
not_null<PeerData*>,
std::shared_ptr<Ui::DynamicImage>> _userpics;
};
State::State(not_null<Data::Stories*> data, Data::StorySourcesList list)
: _data(data)
, _list(list) {
}
Content State::next() {
const auto &sources = _data->sources(_list);
auto result = Content{ .total = int(sources.size()) };
result.elements.reserve(sources.size());
for (const auto &info : sources) {
const auto source = _data->source(info.id);
Assert(source != nullptr);
auto userpic = std::shared_ptr<Ui::DynamicImage>();
const auto peer = source->peer;
if (const auto i = _userpics.find(peer); i != end(_userpics)) {
userpic = i->second;
} else {
userpic = Ui::MakeUserpicThumbnail(peer, true);
_userpics.emplace(peer, userpic);
}
result.elements.push_back({
.id = uint64(peer->id.value),
.name = peer->shortName(),
.thumbnail = std::move(userpic),
.count = info.count,
.unreadCount = info.unreadCount,
.skipSmall = peer->isSelf() ? 1U : 0U,
});
}
return result;
}
} // namespace
rpl::producer<Content> ContentForSession(
not_null<Main::Session*> session,
Data::StorySourcesList list) {
return [=](auto consumer) {
auto result = rpl::lifetime();
const auto stories = &session->data().stories();
const auto state = result.make_state<State>(stories, list);
rpl::single(
rpl::empty
) | rpl::then(
stories->sourcesChanged(list)
) | rpl::start_with_next([=] {
consumer.put_next(state->next());
}, result);
return result;
};
}
rpl::producer<Content> LastForPeer(not_null<PeerData*> peer) {
using namespace rpl::mappers;
const auto stories = &peer->owner().stories();
const auto peerId = peer->id;
return rpl::single(
peerId
) | rpl::then(
stories->sourceChanged() | rpl::filter(_1 == peerId)
) | rpl::map([=] {
auto ids = std::vector<StoryId>();
auto readTill = StoryId();
auto total = 0;
if (const auto source = stories->source(peerId)) {
readTill = source->readTill;
total = int(source->ids.size());
ids = ranges::views::all(source->ids)
| ranges::views::reverse
| ranges::views::take(kShownLastCount)
| ranges::views::transform(&Data::StoryIdDates::id)
| ranges::to_vector;
}
return rpl::make_producer<Content>([=](auto consumer) {
auto lifetime = rpl::lifetime();
if (ids.empty()) {
consumer.put_next(Content());
consumer.put_done();
return lifetime;
}
struct State {
Fn<void()> check;
base::has_weak_ptr guard;
int readTill = StoryId();
bool pushed = false;
};
const auto state = lifetime.make_state<State>();
state->readTill = readTill;
state->check = [=] {
if (state->pushed) {
return;
}
auto done = true;
auto resolving = false;
auto result = Content{ .total = total };
for (const auto id : ids) {
const auto storyId = FullStoryId{ peerId, id };
const auto maybe = stories->lookup(storyId);
if (maybe) {
if (!resolving) {
const auto unread = (id > state->readTill);
result.elements.reserve(ids.size());
result.elements.push_back({
.id = uint64(id),
.thumbnail = Ui::MakeStoryThumbnail(*maybe),
.count = 1U,
.unreadCount = unread ? 1U : 0U,
});
if (unread) {
done = false;
}
}
} else if (maybe.error() == Data::NoStory::Unknown) {
resolving = true;
stories->resolve(
storyId,
crl::guard(&state->guard, state->check));
}
}
if (resolving) {
return;
}
state->pushed = true;
consumer.put_next(std::move(result));
if (done) {
consumer.put_done();
}
};
rpl::single(peerId) | rpl::then(
stories->itemsChanged() | rpl::filter(_1 == peerId)
) | rpl::start_with_next(state->check, lifetime);
stories->session().changes().storyUpdates(
Data::StoryUpdate::Flag::MarkRead
) | rpl::start_with_next([=](const Data::StoryUpdate &update) {
if (update.story->peer()->id == peerId) {
if (update.story->id() > state->readTill) {
state->readTill = update.story->id();
if (ranges::contains(ids, state->readTill)
|| state->readTill > ids.front()) {
state->pushed = false;
state->check();
}
}
}
}, lifetime);
return lifetime;
});
}) | rpl::flatten_latest();
}
void FillSourceMenu(
not_null<Window::SessionController*> controller,
const ShowMenuRequest &request) {
const auto owner = &controller->session().data();
const auto peer = owner->peer(PeerId(request.id));
const auto &add = request.callback;
if (peer->isSelf()) {
add(tr::lng_stories_archive_button(tr::now), [=] {
controller->showSection(Info::Stories::Make(
peer,
Info::Stories::Tab::Archive));
}, &st::menuIconStoriesArchiveSection);
add(tr::lng_stories_my_title(tr::now), [=] {
controller->showSection(Info::Stories::Make(peer));
}, &st::menuIconStoriesSavedSection);
} else {
const auto group = peer->isMegagroup();
const auto channel = peer->isChannel();
const auto showHistoryText = group
? tr::lng_context_open_group(tr::now)
: channel
? tr::lng_context_open_channel(tr::now)
: tr::lng_profile_send_message(tr::now);
add(showHistoryText, [=] {
controller->showPeerHistory(peer);
}, channel ? &st::menuIconChannel : &st::menuIconChatBubble);
const auto viewProfileText = group
? tr::lng_context_view_group(tr::now)
: channel
? tr::lng_context_view_channel(tr::now)
: tr::lng_context_view_profile(tr::now);
add(viewProfileText, [=] {
controller->showPeerInfo(peer);
}, channel ? &st::menuIconInfo : &st::menuIconProfile);
const auto in = [&](Data::StorySourcesList list) {
return ranges::contains(
owner->stories().sources(list),
peer->id,
&Data::StoriesSourceInfo::id);
};
const auto toggle = [=](bool shown) {
owner->stories().toggleHidden(
peer->id,
!shown,
controller->uiShow());
};
if (in(Data::StorySourcesList::NotHidden)) {
add(tr::lng_stories_archive(tr::now), [=] {
toggle(false);
}, &st::menuIconArchive);
}
if (in(Data::StorySourcesList::Hidden)) {
add(tr::lng_stories_unarchive(tr::now), [=] {
toggle(true);
}, &st::menuIconUnarchive);
}
}
}
} // namespace Dialogs::Stories