1 files changed, 112 insertions, 0 deletions

diff --git a/td/telegram/CommunityManager.h b/td/telegram/CommunityManager.h
index 6e3b2ce46..e9df07d79 100644
--- a/td/telegram/CommunityManager.h
+++ b/td/telegram/CommunityManager.h
@@ -6,23 +6,135 @@
 //
 #pragma once
 
+#include "td/telegram/CommunityId.h"
+#include "td/telegram/DialogParticipant.h"
+#include "td/telegram/DialogPhoto.h"
+
 #include "td/actor/actor.h"
 
 #include "td/utils/common.h"
+#include "td/utils/FlatHashMap.h"
+#include "td/utils/FlatHashSet.h"
+#include "td/utils/Promise.h"
+#include "td/utils/WaitFreeHashMap.h"
 
 namespace td {
 
+struct BinlogEvent;
 class Td;
 
 class CommunityManager final : public Actor {
  public:
   CommunityManager(Td *td, ActorShared<> parent);
 
+  bool have_community(CommunityId community_id) const;
+
+  bool have_community_force(CommunityId community_id, const char *source);
+
+  void reload_community(CommunityId community_id, Promise<Unit> &&promise, const char *source);
+
+  void on_get_community(telegram_api::community &community, const char *source);
+
+  void on_get_community_forbidden(telegram_api::communityForbidden &community, const char *source);
+
+  td_api::object_ptr<td_api::community> get_community_object(CommunityId community_id) const;
+
+  void on_binlog_community_event(BinlogEvent &&event);
+
  private:
+  struct Community {
+    int64 access_hash = 0;
+    string title;
+    DialogPhoto photo;
+    DialogParticipantStatus status = DialogParticipantStatus::Banned(0, string());
+    RestrictedRights default_permissions = RestrictedRights::restrict_all();
+    int32 date = 0;
+    bool collapsed_in_dialogs = false;
+
+    static constexpr uint32 CACHE_VERSION = 1;
+    uint32 cache_version = 0;
+
+    bool is_being_updated = false;
+    bool is_changed = true;             // have new changes that need to be sent to the client and database
+    bool need_save_to_database = true;  // have new changes that need only to be saved to the database
+    bool is_update_community_sent = false;
+
+    bool is_repaired = false;  // whether cached value is rechecked
+
+    bool is_saved = false;        // is current community version being saved/is saved to the database
+    bool is_being_saved = false;  // is current community being saved to the database
+
+    bool is_received_from_server = false;  // true, if the community was received from the server and not the database
+
+    uint64 log_event_id = 0;
+
+    template <class StorerT>
+    void store(StorerT &storer) const;
+
+    template <class ParserT>
+    void parse(ParserT &parser);
+  };
+
+  class CommunityLogEvent;
+
   void tear_down() final;
 
+  void save_community(Community *c, CommunityId community_id, bool from_binlog);
+
+  static string get_community_database_key(CommunityId community_id);
+
+  static string get_community_database_value(const Community *c);
+
+  void save_community_to_database(Community *c, CommunityId community_id);
+
+  void save_community_to_database_impl(Community *c, CommunityId community_id, string value);
+
+  void on_save_community_to_database(CommunityId community_id, bool success);
+
+  void load_community_from_database(Community *c, CommunityId community_id, Promise<Unit> promise);
+
+  void load_community_from_database_impl(CommunityId community_id, bool is_recursive, Promise<Unit> promise);
+
+  void on_load_community_from_database(CommunityId community_id, string value, bool force, bool is_recursive);
+
+  void update_community(Community *c, CommunityId community_id, bool from_binlog = false, bool from_database = false);
+
+  const Community *get_community(CommunityId community_id) const;
+
+  Community *get_community(CommunityId community_id);
+
+  Community *get_community_force(CommunityId community_id, const char *source, bool is_recursive = false);
+
+  Community *add_community(CommunityId community_id, const char *source);
+
+  void on_update_community_photo(Community *c, CommunityId community_id,
+                                 telegram_api::object_ptr<telegram_api::ChatPhoto> &&chat_photo_ptr);
+
+  void on_update_community_photo(Community *c, CommunityId community_id, DialogPhoto &&photo,
+                                 bool invalidate_photo_cache);
+
+  static void on_update_community_title(Community *c, CommunityId community_id, string &&title);
+
+  void on_update_community_status(Community *c, CommunityId community_id, DialogParticipantStatus &&status);
+
+  static void on_update_community_default_permissions(Community *c, CommunityId community_id,
+                                                      RestrictedRights default_permissions);
+
+  td_api::object_ptr<td_api::updateCommunity> get_update_community_object(CommunityId community_id,
+                                                                          const Community *c) const;
+
+  td_api::object_ptr<td_api::updateCommunity> get_update_unknown_community_object(CommunityId community_id) const;
+
+  td_api::object_ptr<td_api::community> get_community_object(CommunityId Community_id, const Community *c) const;
+
   Td *td_;
   ActorShared<> parent_;
+
+  WaitFreeHashMap<CommunityId, unique_ptr<Community>, CommunityIdHash> communities_;
+  mutable FlatHashSet<CommunityId, CommunityIdHash> unknown_communities_;
+
+  FlatHashMap<CommunityId, vector<Promise<Unit>>, CommunityIdHash> load_community_from_database_queries_;
+  FlatHashSet<CommunityId, CommunityIdHash> loaded_from_database_communities_;
 };
 
 }  // namespace td