tdesktop/Telegram/SourceFiles/chat_helpers/emoji_list_widget.h

/*
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 "chat_helpers/tabbed_selector.h"
#include "ui/widgets/tooltip.h"
#include "base/timer.h"

namespace Data {
class StickersSet;
} // namespace Data

namespace tr {
template <typename ...Tags>
struct phrase;
} // namespace tr

namespace Ui {
class RippleAnimation;
} // namespace Ui

namespace Ui::Emoji {
enum class Section;
} // namespace Ui::Emoji

namespace Ui::CustomEmoji {
class Loader;
class Instance;
struct RepaintRequest;
} // namespace Ui::CustomEmoji

namespace Window {
class SessionController;
} // namespace Window

namespace ChatHelpers {

inline constexpr auto kEmojiSectionCount = 8;

struct StickerIcon;
class EmojiColorPicker;
class StickersListFooter;

class EmojiListWidget
	: public TabbedSelector::Inner
	, public Ui::AbstractTooltipShower {
public:
	EmojiListWidget(
		QWidget *parent,
		not_null<Window::SessionController*> controller);
	~EmojiListWidget();

	using Section = Ui::Emoji::Section;

	void refreshRecent() override;
	void clearSelection() override;
	object_ptr<TabbedSelector::InnerFooter> createFooter() override;

	void showEmojiSection(Section section);
	void showSet(uint64 setId);
	[[nodiscard]] uint64 currentSet(int yOffset) const;

	// Ui::AbstractTooltipShower interface.
	QString tooltipText() const override;
	QPoint tooltipPos() const override;
	bool tooltipWindowActive() const override;

	void refreshEmoji();

	[[nodiscard]] rpl::producer<EmojiPtr> chosen() const;
	[[nodiscard]] auto customChosen() const
		-> rpl::producer<TabbedSelector::FileChosen>;

protected:
	void visibleTopBottomUpdated(
		int visibleTop,
		int visibleBottom) override;

	void mousePressEvent(QMouseEvent *e) override;
	void mouseReleaseEvent(QMouseEvent *e) override;
	void mouseMoveEvent(QMouseEvent *e) override;
	void paintEvent(QPaintEvent *e) override;
	void leaveEventHook(QEvent *e) override;
	void leaveToChildEvent(QEvent *e, QWidget *child) override;
	void enterFromChildEvent(QEvent *e, QWidget *child) override;
	bool eventHook(QEvent *e) override;

	TabbedSelector::InnerFooter *getFooter() const override;
	void processHideFinished() override;
	int countDesiredHeight(int newWidth) override;

private:
	struct SectionInfo {
		int section = 0;
		int count = 0;
		int top = 0;
		int rowsCount = 0;
		int rowsTop = 0;
		int rowsBottom = 0;
	};
	struct CustomInstance;
	struct CustomOne {
		not_null<CustomInstance*> instance;
		not_null<DocumentData*> document;
	};
	struct CustomSet {
		uint64 id = 0;
		not_null<Data::StickersSet*> set;
		DocumentData *thumbnailDocument = nullptr;
		QString title;
		std::vector<CustomOne> list;
		std::unique_ptr<Ui::RippleAnimation> ripple;
		bool painted = false;
	};
	struct RecentOne {
		not_null<CustomInstance*> instance;
		std::variant<EmojiPtr, DocumentId> id;
	};
	struct RepaintSet {
		base::flat_set<uint64> ids;
		crl::time when = 0;
	};
	struct OverEmoji {
		int section = 0;
		int index = 0;

		inline bool operator==(OverEmoji other) const {
			return (section == other.section)
				&& (index == other.index);
		}
		inline bool operator!=(OverEmoji other) const {
			return !(*this == other);
		}
	};
	struct OverSet {
		int section = 0;

		inline bool operator==(OverSet other) const {
			return (section == other.section);
		}
		inline bool operator!=(OverSet other) const {
			return !(*this == other);
		}
	};
	struct OverButton {
		int section = 0;

		inline bool operator==(OverButton other) const {
			return (section == other.section);
		}
		inline bool operator!=(OverButton other) const {
			return !(*this == other);
		}
	};
	using OverState = std::variant<
		v::null_t,
		OverEmoji,
		OverSet,
		OverButton>;

	template <typename Callback>
	bool enumerateSections(Callback callback) const;
	[[nodiscard]] SectionInfo sectionInfo(int section) const;
	[[nodiscard]] SectionInfo sectionInfoByOffset(int yOffset) const;
	[[nodiscard]] int sectionsCount() const;

	void showPicker();
	void pickerHidden();
	void colorChosen(EmojiPtr emoji);
	bool checkPickerHide();
	void refreshCustom();
	void unloadNotSeenCustom(int visibleTop, int visibleBottom);

	void ensureLoaded(int section);
	void updateSelected();
	void setSelected(OverState newSelected);
	void setPressed(OverState newPressed);

	[[nodiscard]] EmojiPtr lookupOverEmoji(const OverEmoji *over) const;
	void selectEmoji(EmojiPtr emoji);
	void selectCustom(not_null<DocumentData*> document);
	void drawRecent(
		QPainter &p,
		QPoint position,
		crl::time now,
		bool paused,
		int index);
	void drawEmoji(
		QPainter &p,
		QPoint position,
		int section,
		int index);
	void drawCustom(
		QPainter &p,
		QPoint position,
		crl::time now,
		bool paused,
		int set,
		int index);
	[[nodiscard]] bool hasRemoveButton(int index) const;
	[[nodiscard]] QRect removeButtonRect(int index) const;
	[[nodiscard]] QRect emojiRect(int section, int index) const;
	[[nodiscard]] int emojiRight() const;
	[[nodiscard]] int emojiLeft() const;
	[[nodiscard]] uint64 sectionSetId(int section) const;
	[[nodiscard]] std::vector<StickerIcon> fillIcons();

	void displaySet(uint64 setId);
	void removeSet(uint64 setId);

	[[nodiscard]] std::unique_ptr<Ui::RippleAnimation> createButtonRipple(
		int section);
	[[nodiscard]] QPoint buttonRippleTopLeft(int section) const;

	void repaintLater(
		DocumentId documentId,
		uint64 setId,
		Ui::CustomEmoji::RepaintRequest request);
	template <typename CheckId>
	void repaintCustom(CheckId checkId);
	void scheduleRepaintTimer();
	void invokeRepaints();

	void fillRecent();
	[[nodiscard]] not_null<CustomInstance*> resolveCustomInstance(
		not_null<DocumentData*> document,
		uint64 setId);
	[[nodiscard]] not_null<CustomInstance*> resolveCustomInstance(
		std::variant<EmojiPtr, DocumentId> customId);
	[[nodiscard]] not_null<CustomInstance*> resolveCustomInstance(
		DocumentId fakeId,
		EmojiPtr emoji);
	[[nodiscard]] not_null<CustomInstance*> resolveCustomInstance(
		DocumentId documentId);
	[[nodiscard]] std::unique_ptr<CustomInstance> customInstanceWithLoader(
		std::unique_ptr<Ui::CustomEmoji::Loader> loader,
		DocumentId documentId,
		uint64 setId);

	StickersListFooter *_footer = nullptr;

	int _counts[kEmojiSectionCount];
	std::vector<RecentOne> _recent;
	base::flat_set<DocumentId> _recentCustomIds;
	bool _recentPainted = false;
	QVector<EmojiPtr> _emoji[kEmojiSectionCount];
	std::vector<CustomSet> _custom;
	base::flat_map<DocumentId, std::unique_ptr<CustomInstance>> _instances;

	int _rowsLeft = 0;
	int _columnCount = 1;
	QSize _singleSize;
	int _esize = 0;

	OverState _selected;
	OverState _pressed;
	OverState _pickerSelected;
	QPoint _lastMousePos;

	object_ptr<EmojiColorPicker> _picker;
	base::Timer _showPickerTimer;

	base::flat_map<crl::time, RepaintSet> _repaints;
	bool _repaintTimerScheduled = false;
	base::Timer _repaintTimer;
	crl::time _repaintNext = 0;

	rpl::event_stream<EmojiPtr> _chosen;
	rpl::event_stream<TabbedSelector::FileChosen> _customChosen;

};

tr::phrase<> EmojiCategoryTitle(int index);

} // namespace ChatHelpers