tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add DialogManager::check_dialog_access.

Browse Source
This commit is contained in:
levlam 2024-04-15 01:23:14 +03:00
parent c0ed8f09b0
commit e0a7ff6c53
2 changed files with 23 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
td/telegram/DialogManager.cpp
View File

@ -566,6 +566,26 @@ InputDialogId DialogManager::get_input_dialog_id(DialogId dialog_id) const {
}
}
Status DialogManager::check_dialog_access(DialogId dialog_id, bool allow_secret_chats, AccessRights access_rights,
const char *source) const {
if (!have_dialog_force(dialog_id, source)) {
if (!dialog_id.is_valid()) {
return Status::Error(400, "Invalid chat identifier specified");
}
return Status::Error(400, "Chat not found");
}
if (!have_input_peer(dialog_id, allow_secret_chats, access_rights)) {
if (dialog_id.get_type() == DialogType::SecretChat && !allow_secret_chats) {
return Status::Error(400, "Not supported in secret chats");
}
if (access_rights == AccessRights::Write || access_rights == AccessRights::Edit) {
return Status::Error(400, "Have no write access to the chat");
}
return Status::Error(400, "Can't access the chat");
}
return Status::OK();
}
tl_object_ptr<telegram_api::InputPeer> DialogManager::get_input_peer(DialogId dialog_id,
AccessRights access_rights) const {
switch (dialog_id.get_type()) {

3
td/telegram/DialogManager.h
View File

@ -72,6 +72,9 @@ class DialogManager final : public Actor {
tl_object_ptr<telegram_api::inputEncryptedChat> get_input_encrypted_chat(DialogId dialog_id,
AccessRights access_rights) const;
Status check_dialog_access(DialogId dialog_id, bool allow_secret_chats, AccessRights access_rights,
const char *source) const;
bool have_input_peer(DialogId dialog_id, bool allow_secret_chats, AccessRights access_rights) const;
bool have_dialog_force(DialogId dialog_id, const char *source) const;