Preload next stories inside the media viewer.

Telegram/SourceFiles/data/data_story.h

@@ -0,0 +1,158 @@
+/*
+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
+*/
+#pragma once
+
+#include "base/weak_ptr.h"
+
+class Image;
+class PhotoData;
+class DocumentData;
+
+namespace Main {
+class Session;
+} // namespace Main
+
+namespace Data {
+
+class Session;
+class Thread;
+class PhotoMedia;
+
+struct StoryIdDates {
+	StoryId id = 0;
+	TimeId date = 0;
+	TimeId expires = 0;
+
+	[[nodiscard]] bool valid() const {
+		return id != 0;
+	}
+	explicit operator bool() const {
+		return valid();
+	}
+
+	friend inline auto operator<=>(StoryIdDates, StoryIdDates) = default;
+	friend inline bool operator==(StoryIdDates, StoryIdDates) = default;
+};
+
+struct StoryMedia {
+	std::variant<
+		v::null_t,
+		not_null<PhotoData*>,
+		not_null<DocumentData*>> data;
+
+	friend inline bool operator==(StoryMedia, StoryMedia) = default;
+};
+
+struct StoryView {
+	not_null<PeerData*> peer;
+	TimeId date = 0;
+
+	friend inline bool operator==(StoryView, StoryView) = default;
+};
+
+class Story final {
+public:
+	Story(
+		StoryId id,
+		not_null<PeerData*> peer,
+		StoryMedia media,
+		TimeId date,
+		TimeId expires);
+
+	[[nodiscard]] Session &owner() const;
+	[[nodiscard]] Main::Session &session() const;
+	[[nodiscard]] not_null<PeerData*> peer() const;
+
+	[[nodiscard]] StoryId id() const;
+	[[nodiscard]] bool mine() const;
+	[[nodiscard]] StoryIdDates idDates() const;
+	[[nodiscard]] FullStoryId fullId() const;
+	[[nodiscard]] TimeId date() const;
+	[[nodiscard]] TimeId expires() const;
+	[[nodiscard]] bool unsupported() const;
+	[[nodiscard]] bool expired(TimeId now = 0) const;
+	[[nodiscard]] const StoryMedia &media() const;
+	[[nodiscard]] PhotoData *photo() const;
+	[[nodiscard]] DocumentData *document() const;
+
+	[[nodiscard]] bool hasReplyPreview() const;
+	[[nodiscard]] Image *replyPreview() const;
+	[[nodiscard]] TextWithEntities inReplyText() const;
+
+	void setPinned(bool pinned);
+	[[nodiscard]] bool pinned() const;
+	void setIsPublic(bool isPublic);
+	[[nodiscard]] bool isPublic() const;
+	void setCloseFriends(bool closeFriends);
+	[[nodiscard]] bool closeFriends() const;
+
+	[[nodiscard]] bool canDownload() const;
+	[[nodiscard]] bool canShare() const;
+	[[nodiscard]] bool canDelete() const;
+	[[nodiscard]] bool canReport() const;
+
+	[[nodiscard]] bool hasDirectLink() const;
+	[[nodiscard]] std::optional<QString> errorTextForForward(
+		not_null<Thread*> to) const;
+
+	void setCaption(TextWithEntities &&caption);
+	[[nodiscard]] const TextWithEntities &caption() const;
+
+	void setViewsData(std::vector<not_null<PeerData*>> recent, int total);
+	[[nodiscard]] auto recentViewers() const
+		-> const std::vector<not_null<PeerData*>> &;
+	[[nodiscard]] const std::vector<StoryView> &viewsList() const;
+	[[nodiscard]] int views() const;
+	void applyViewsSlice(
+		const std::optional<StoryView> &offset,
+		const std::vector<StoryView> &slice,
+		int total);
+
+	bool applyChanges(StoryMedia media, const MTPDstoryItem &data);
+
+private:
+	const StoryId _id = 0;
+	const not_null<PeerData*> _peer;
+	StoryMedia _media;
+	TextWithEntities _caption;
+	std::vector<not_null<PeerData*>> _recentViewers;
+	std::vector<StoryView> _viewsList;
+	int _views = 0;
+	const TimeId _date = 0;
+	const TimeId _expires = 0;
+	bool _pinned : 1 = false;
+	bool _isPublic : 1 = false;
+	bool _closeFriends : 1 = false;
+
+};
+
+class StoryPreload final : public base::has_weak_ptr {
+public:
+	StoryPreload(not_null<Story*> story, Fn<void()> done);
+	~StoryPreload();
+
+	[[nodiscard]] FullStoryId id() const;
+	[[nodiscard]] not_null<Story*> story() const;
+
+private:
+	class LoadTask;
+
+	void start();
+	void load();
+	void callDone();
+
+	const not_null<Story*> _story;
+	Fn<void()> _done;
+
+	std::shared_ptr<Data::PhotoMedia> _photo;
+	std::unique_ptr<LoadTask> _task;
+	rpl::lifetime _lifetime;
+
+};
+
+} // namespace Data