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

Add StarManager::reload_star_transaction.

Browse Source
This commit is contained in:
levlam 2024-06-28 13:29:22 +03:00
parent 218877c0ac
commit 02bd5a2498
2 changed files with 39 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
td/telegram/StarManager.cpp
View File

@ -29,6 +29,8 @@
#include "td/utils/misc.h"
#include "td/utils/ScopeGuard.h"
#include <type_traits>
namespace td {
class GetStarsTopupOptionsQuery final : public Td::ResultHandler {
@ -101,7 +103,27 @@ class GetStarsTransactionsQuery final : public Td::ResultHandler {
flags, false /*ignored*/, false /*ignored*/, false /*ignored*/, std::move(input_peer), offset, limit)));
}
void send(DialogId dialog_id, const string &transaction_id, bool is_refund) {
dialog_id_ = dialog_id;
auto input_peer = td_->dialog_manager_->get_input_peer(dialog_id, AccessRights::Write);
if (input_peer == nullptr) {
return on_error(Status::Error(400, "Have no access to the chat"));
}
int32 flags = 0;
if (is_refund) {
flags |= telegram_api::inputStarsTransaction::REFUND_MASK;
}
vector<telegram_api::object_ptr<telegram_api::inputStarsTransaction>> transaction_ids;
transaction_ids.push_back(
telegram_api::make_object<telegram_api::inputStarsTransaction>(flags, false /*ignored*/, transaction_id));
send_query(G()->net_query_creator().create(
telegram_api::payments_getStarsTransactionsByID(std::move(input_peer), std::move(transaction_ids))));
}
void on_result(BufferSlice packet) final {
static_assert(std::is_same<telegram_api::payments_getStarsTransactionsByID::ReturnType,
telegram_api::payments_getStarsTransactions::ReturnType>::value,
"");
auto result_ptr = fetch_result<telegram_api::payments_getStarsTransactions>(packet);
if (result_ptr.is_error()) {
return on_error(result_ptr.move_as_error());
@ -497,6 +519,20 @@ void StarManager::get_star_ad_account_url(const td_api::object_ptr<td_api::Messa
td_->create_handler<GetStarsRevenueAdsAccountUrlQuery>(std::move(promise))->send(dialog_id);
}
void StarManager::reload_star_transaction(DialogId dialog_id, const string &transaction_id, bool is_refund,
Promise<Unit> &&promise) {
TRY_STATUS_PROMISE(promise, can_manage_stars(dialog_id, true));
auto query_promise = PromiseCreator::lambda(
[promise = std::move(promise)](Result<td_api::object_ptr<td_api::starTransactions>> r_transactions) mutable {
if (r_transactions.is_error()) {
promise.set_error(r_transactions.move_as_error());
} else {
promise.set_value(Unit());
}
});
td_->create_handler<GetStarsTransactionsQuery>(std::move(query_promise))->send(dialog_id, transaction_id, is_refund);
}
void StarManager::on_update_stars_revenue_status(
telegram_api::object_ptr<telegram_api::updateStarsRevenueStatus> &&update) {
DialogId dialog_id(update->peer_);

3
td/telegram/StarManager.h
View File

@ -41,6 +41,9 @@ class StarManager final : public Actor {
void get_star_ad_account_url(const td_api::object_ptr<td_api::MessageSender> &owner_id, Promise<string> &&promise);
void reload_star_transaction(DialogId dialog_id, const string &transaction_id, bool is_refund,
Promise<Unit> &&promise);
void on_update_stars_revenue_status(telegram_api::object_ptr<telegram_api::updateStarsRevenueStatus> &&update);
static int64 get_star_count(int64 amount, bool allow_negative = false);