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

Allow self in get_star_transactions.

Browse Source
This commit is contained in:
levlam 2024-06-14 23:31:13 +03:00
parent 8460b1fe51
commit 1ebd431791
2 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
td/telegram/StarManager.cpp
View File

@ -298,10 +298,13 @@ void StarManager::tear_down() {
parent_.reset();
}
Status StarManager::can_manage_stars(DialogId dialog_id) const {
Status StarManager::can_manage_stars(DialogId dialog_id, bool allow_self) const {
switch (dialog_id.get_type()) {
case DialogType::User: {
auto user_id = dialog_id.get_user_id();
if (allow_self && user_id == td_->user_manager_->get_my_id()) {
break;
}
TRY_RESULT(bot_data, td_->user_manager_->get_bot_data(user_id));
if (!bot_data.can_be_edited) {
return Status::Error(400, "The bot isn't owned");
@ -332,7 +335,7 @@ void StarManager::get_star_transactions(td_api::object_ptr<td_api::MessageSender
int32 limit, td_api::object_ptr<td_api::StarTransactionDirection> &&direction,
Promise<td_api::object_ptr<td_api::starTransactions>> &&promise) {
TRY_RESULT_PROMISE(promise, dialog_id, get_message_sender_dialog_id(td_, owner_id, true, false));
TRY_STATUS_PROMISE(promise, can_manage_stars(dialog_id));
TRY_STATUS_PROMISE(promise, can_manage_stars(dialog_id, true));
if (limit < 0) {
return promise.set_error(Status::Error(400, "Limit must be non-negative"));
}

2
td/telegram/StarManager.h
View File

@ -45,7 +45,7 @@ class StarManager final : public Actor {
private:
void tear_down() final;
Status can_manage_stars(DialogId dialog_id) const;
Status can_manage_stars(DialogId dialog_id, bool allow_self = false) const;
void send_get_star_withdrawal_url_query(
DialogId dialog_id, int64 star_count,