tdlight/td/telegram/PollManager.h

269 lines
8.9 KiB
C
Raw Normal View History

//
2024-01-01 01:07:21 +01:00
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once
2023-07-01 15:22:01 +02:00
#include "td/telegram/ChannelId.h"
2023-04-25 15:41:04 +02:00
#include "td/telegram/DialogId.h"
#include "td/telegram/MessageEntity.h"
2023-09-21 18:11:17 +02:00
#include "td/telegram/MessageFullId.h"
2023-04-25 15:41:04 +02:00
#include "td/telegram/MinChannel.h"
#include "td/telegram/net/NetQuery.h"
#include "td/telegram/PollId.h"
#include "td/telegram/ReplyMarkup.h"
#include "td/telegram/td_api.h"
#include "td/telegram/telegram_api.h"
#include "td/actor/actor.h"
2022-07-02 22:32:18 +02:00
#include "td/actor/MultiTimeout.h"
#include "td/utils/buffer.h"
#include "td/utils/common.h"
2022-02-07 22:04:34 +01:00
#include "td/utils/FlatHashMap.h"
2022-03-11 19:38:48 +01:00
#include "td/utils/FlatHashSet.h"
2022-06-27 12:30:18 +02:00
#include "td/utils/Promise.h"
#include "td/utils/Status.h"
2022-11-22 10:23:58 +01:00
#include "td/utils/WaitFreeHashMap.h"
#include "td/utils/WaitFreeHashSet.h"
#include <utility>
namespace td {
struct BinlogEvent;
class Td;
class PollManager final : public Actor {
public:
PollManager(Td *td, ActorShared<> parent);
PollManager(const PollManager &) = delete;
PollManager &operator=(const PollManager &) = delete;
PollManager(PollManager &&) = delete;
PollManager &operator=(PollManager &&) = delete;
2021-07-03 22:51:36 +02:00
~PollManager() final;
static bool is_local_poll_id(PollId poll_id);
PollId create_poll(string &&question, vector<string> &&options, bool is_anonymous, bool allow_multiple_answers,
bool is_quiz, int32 correct_option_id, FormattedText &&explanation, int32 open_period,
int32 close_date, bool is_closed);
2023-09-21 18:11:17 +02:00
void register_poll(PollId poll_id, MessageFullId message_full_id, const char *source);
2023-09-21 18:11:17 +02:00
void unregister_poll(PollId poll_id, MessageFullId message_full_id, const char *source);
2023-11-06 18:37:17 +01:00
void register_reply_poll(PollId poll_id);
void unregister_reply_poll(PollId poll_id);
bool get_poll_is_closed(PollId poll_id) const;
bool get_poll_is_anonymous(PollId poll_id) const;
string get_poll_search_text(PollId poll_id) const;
2023-09-21 18:11:17 +02:00
void set_poll_answer(PollId poll_id, MessageFullId message_full_id, vector<int32> &&option_ids,
Promise<Unit> &&promise);
2023-09-21 18:11:17 +02:00
void get_poll_voters(PollId poll_id, MessageFullId message_full_id, int32 option_id, int32 offset, int32 limit,
2023-04-25 15:41:04 +02:00
Promise<td_api::object_ptr<td_api::messageSenders>> &&promise);
2023-09-21 18:11:17 +02:00
void stop_poll(PollId poll_id, MessageFullId message_full_id, unique_ptr<ReplyMarkup> &&reply_markup,
Promise<Unit> &&promise);
void stop_local_poll(PollId poll_id);
2021-09-05 22:11:25 +02:00
PollId dup_poll(PollId poll_id);
bool has_input_media(PollId poll_id) const;
tl_object_ptr<telegram_api::InputMedia> get_input_media(PollId poll_id) const;
PollId on_get_poll(PollId poll_id, tl_object_ptr<telegram_api::poll> &&poll_server,
2022-05-29 18:31:36 +02:00
tl_object_ptr<telegram_api::pollResults> &&poll_results, const char *source);
2023-04-25 15:41:04 +02:00
void on_get_poll_vote(PollId poll_id, DialogId dialog_id, vector<BufferSlice> &&options);
td_api::object_ptr<td_api::poll> get_poll_object(PollId poll_id) const;
void on_binlog_events(vector<BinlogEvent> &&events);
static vector<int32> get_vote_percentage(const vector<int32> &voter_counts, int32 total_voter_count);
template <class StorerT>
void store_poll(PollId poll_id, StorerT &storer) const;
template <class ParserT>
PollId parse_poll(ParserT &parser);
private:
struct PollOption {
2023-04-25 14:16:29 +02:00
string text_;
string data_;
int32 voter_count_ = 0;
bool is_chosen_ = false;
template <class StorerT>
void store(StorerT &storer) const;
template <class ParserT>
void parse(ParserT &parser);
};
struct Poll {
2023-04-25 14:29:24 +02:00
string question_;
vector<PollOption> options_;
2023-04-25 15:41:04 +02:00
vector<DialogId> recent_voter_dialog_ids_;
vector<std::pair<ChannelId, MinChannel>> recent_voter_min_channels_;
2023-04-25 14:29:24 +02:00
FormattedText explanation_;
int32 total_voter_count_ = 0;
int32 correct_option_id_ = -1;
int32 open_period_ = 0;
int32 close_date_ = 0;
bool is_anonymous_ = true;
bool allow_multiple_answers_ = false;
bool is_quiz_ = false;
bool is_closed_ = false;
bool is_updated_after_close_ = false;
mutable bool was_saved_ = false;
template <class StorerT>
void store(StorerT &storer) const;
template <class ParserT>
void parse(ParserT &parser);
};
struct PollOptionVoters {
2023-04-25 15:41:04 +02:00
vector<DialogId> voter_dialog_ids_;
2023-04-25 14:16:29 +02:00
string next_offset_;
2023-04-25 15:41:04 +02:00
vector<Promise<td_api::object_ptr<td_api::messageSenders>>> pending_queries_;
2023-04-25 14:16:29 +02:00
bool was_invalidated_ = false; // the list needs to be invalidated when voters are changed
};
static constexpr int32 MAX_GET_POLL_VOTERS = 50; // server side limit
2022-05-09 00:46:46 +02:00
static constexpr int32 UNLOAD_POLL_DELAY = 600; // some reasonable value
class SetPollAnswerLogEvent;
class StopPollLogEvent;
2021-07-03 22:51:36 +02:00
void start_up() final;
void tear_down() final;
static void on_update_poll_timeout_callback(void *poll_manager_ptr, int64 poll_id_int);
static void on_close_poll_timeout_callback(void *poll_manager_ptr, int64 poll_id_int);
2022-05-09 00:46:46 +02:00
static void on_unload_poll_timeout_callback(void *poll_manager_ptr, int64 poll_id_int);
static td_api::object_ptr<td_api::pollOption> get_poll_option_object(const PollOption &poll_option);
static telegram_api::object_ptr<telegram_api::pollAnswer> get_input_poll_option(const PollOption &poll_option);
static vector<PollOption> get_poll_options(vector<tl_object_ptr<telegram_api::pollAnswer>> &&poll_options);
bool have_poll(PollId poll_id) const;
bool have_poll_force(PollId poll_id);
const Poll *get_poll(PollId poll_id) const;
2022-05-09 00:46:46 +02:00
const Poll *get_poll(PollId poll_id);
Poll *get_poll_editable(PollId poll_id);
2022-05-09 00:46:46 +02:00
bool can_unload_poll(PollId poll_id);
void schedule_poll_unload(PollId poll_id);
void notify_on_poll_update(PollId poll_id);
static string get_poll_database_key(PollId poll_id);
2021-10-19 17:11:16 +02:00
static void save_poll(const Poll *poll, PollId poll_id);
void on_load_poll_from_database(PollId poll_id, string value);
double get_polling_timeout() const;
void on_update_poll_timeout(PollId poll_id);
void on_close_poll_timeout(PollId poll_id);
2022-05-09 00:46:46 +02:00
void on_unload_poll_timeout(PollId poll_id);
void on_online();
Poll *get_poll_force(PollId poll_id);
td_api::object_ptr<td_api::poll> get_poll_object(PollId poll_id, const Poll *poll) const;
void on_get_poll_results(PollId poll_id, uint64 generation, Result<tl_object_ptr<telegram_api::Updates>> result);
2023-09-21 18:11:17 +02:00
void do_set_poll_answer(PollId poll_id, MessageFullId message_full_id, vector<string> &&options, uint64 log_event_id,
Promise<Unit> &&promise);
void on_set_poll_answer(PollId poll_id, uint64 generation, Result<tl_object_ptr<telegram_api::Updates>> &&result);
void on_set_poll_answer_finished(PollId poll_id, Result<Unit> &&result, uint64 generation);
2020-12-21 23:51:20 +01:00
void invalidate_poll_voters(const Poll *poll, PollId poll_id);
void invalidate_poll_option_voters(const Poll *poll, PollId poll_id, size_t option_index);
PollOptionVoters &get_poll_option_voters(const Poll *poll, PollId poll_id, int32 option_id);
2023-04-25 15:41:04 +02:00
td_api::object_ptr<td_api::messageSenders> get_poll_voters_object(int32 total_count,
const vector<DialogId> &voter_dialog_ids) const;
void on_get_poll_voters(PollId poll_id, int32 option_id, string offset, int32 limit,
Result<tl_object_ptr<telegram_api::messages_votesList>> &&result);
2023-09-21 18:11:17 +02:00
void do_stop_poll(PollId poll_id, MessageFullId message_full_id, unique_ptr<ReplyMarkup> &&reply_markup,
uint64 log_event_id, Promise<Unit> &&promise);
2023-09-21 18:11:17 +02:00
void on_stop_poll_finished(PollId poll_id, MessageFullId message_full_id, uint64 log_event_id, Result<Unit> &&result,
Promise<Unit> &&promise);
void forget_local_poll(PollId poll_id);
MultiTimeout update_poll_timeout_{"UpdatePollTimeout"};
MultiTimeout close_poll_timeout_{"ClosePollTimeout"};
2022-05-09 00:46:46 +02:00
MultiTimeout unload_poll_timeout_{"UnloadPollTimeout"};
2022-11-22 10:23:58 +01:00
WaitFreeHashMap<PollId, unique_ptr<Poll>, PollIdHash> polls_;
2023-09-21 18:11:17 +02:00
WaitFreeHashMap<PollId, WaitFreeHashSet<MessageFullId, MessageFullIdHash>, PollIdHash> server_poll_messages_;
WaitFreeHashMap<PollId, WaitFreeHashSet<MessageFullId, MessageFullIdHash>, PollIdHash> other_poll_messages_;
2023-11-06 18:37:17 +01:00
WaitFreeHashMap<PollId, int32, PollIdHash> reply_poll_counts_;
struct PendingPollAnswer {
vector<string> options_;
vector<Promise<Unit>> promises_;
uint64 generation_ = 0;
uint64 log_event_id_ = 0;
NetQueryRef query_ref_;
bool is_finished_ = false;
};
FlatHashMap<PollId, PendingPollAnswer, PollIdHash> pending_answers_;
FlatHashMap<PollId, vector<PollOptionVoters>, PollIdHash> poll_voters_;
int64 current_local_poll_id_ = 0;
uint64 current_generation_ = 0;
2022-03-11 19:38:48 +01:00
FlatHashSet<PollId, PollIdHash> loaded_from_database_polls_;
2022-03-11 19:38:48 +01:00
FlatHashSet<PollId, PollIdHash> being_closed_polls_;
Td *td_;
ActorShared<> parent_;
};
} // namespace td