tdlight/td/telegram/DialogListId.h
2023-08-21 16:23:11 +03:00

141 lines
3.9 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023
//
// 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
#include "td/telegram/DialogFilterId.h"
#include "td/telegram/FolderId.h"
#include "td/telegram/td_api.h"
#include "td/utils/common.h"
#include "td/utils/HashTableUtils.h"
#include "td/utils/StringBuilder.h"
#include <limits>
#include <type_traits>
namespace td {
class DialogListId {
int64 id = 0;
static constexpr int64 FILTER_ID_SHIFT = static_cast<int64>(1) << 32;
public:
DialogListId() = default;
explicit constexpr DialogListId(int64 dialog_list_id) : id(dialog_list_id) {
if (is_folder() && get_folder_id() != FolderId::archive()) {
id = FolderId::main().get();
} else if (is_filter()) {
CHECK(get_filter_id().is_valid());
}
}
template <class T, typename = std::enable_if_t<std::is_convertible<T, int32>::value>>
DialogListId(T dialog_list_id) = delete;
explicit DialogListId(DialogFilterId dialog_filter_id) : id(dialog_filter_id.get() + FILTER_ID_SHIFT) {
}
explicit DialogListId(FolderId folder_id) : id(folder_id.get()) {
}
explicit DialogListId(const td_api::object_ptr<td_api::ChatList> &chat_list) {
if (chat_list == nullptr) {
CHECK(id == FolderId::main().get());
return;
}
switch (chat_list->get_id()) {
case td_api::chatListArchive::ID:
id = FolderId::archive().get();
break;
case td_api::chatListMain::ID:
CHECK(id == FolderId::main().get());
break;
case td_api::chatListFolder::ID: {
DialogFilterId filter_id(static_cast<const td_api::chatListFolder *>(chat_list.get())->chat_folder_id_);
if (filter_id.is_valid()) {
*this = DialogListId(filter_id);
}
break;
}
default:
UNREACHABLE();
break;
}
}
td_api::object_ptr<td_api::ChatList> get_chat_list_object() const {
if (is_folder()) {
auto folder_id = get_folder_id();
if (folder_id == FolderId::archive()) {
return td_api::make_object<td_api::chatListArchive>();
}
return td_api::make_object<td_api::chatListMain>();
}
if (is_filter()) {
return td_api::make_object<td_api::chatListFolder>(get_filter_id().get());
}
UNREACHABLE();
return nullptr;
}
int64 get() const {
return id;
}
bool operator==(const DialogListId &other) const {
return id == other.id;
}
bool operator!=(const DialogListId &other) const {
return id != other.id;
}
constexpr bool is_folder() const {
return std::numeric_limits<int32>::min() <= id && id <= std::numeric_limits<int32>::max();
}
bool is_filter() const {
return std::numeric_limits<int32>::min() + FILTER_ID_SHIFT <= id &&
id <= std::numeric_limits<int32>::max() + FILTER_ID_SHIFT;
}
FolderId get_folder_id() const {
CHECK(is_folder());
return FolderId(static_cast<int32>(id));
}
DialogFilterId get_filter_id() const {
CHECK(is_filter());
return DialogFilterId(static_cast<int32>(id - FILTER_ID_SHIFT));
}
};
struct DialogListIdHash {
uint32 operator()(DialogListId dialog_list_id) const {
return Hash<int64>()(dialog_list_id.get());
}
};
inline StringBuilder &operator<<(StringBuilder &string_builder, DialogListId dialog_list_id) {
if (dialog_list_id.is_folder()) {
auto folder_id = dialog_list_id.get_folder_id();
if (folder_id == FolderId::archive()) {
return string_builder << "Archive chat list";
}
if (folder_id == FolderId::main()) {
return string_builder << "Main chat list";
}
return string_builder << "chat list " << folder_id;
}
if (dialog_list_id.is_filter()) {
return string_builder << "chat list " << dialog_list_id.get_filter_id();
}
return string_builder << "unknown chat list " << dialog_list_id.get();
}
} // namespace td