From 7ab4d9b87bfa5ccc131f44c62dd04af42eabff67 Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Thu, 16 May 2024 13:41:04 +0300
Subject: [PATCH] Add td_api::updateOwnedStarCount.

---
 td/generate/scheme/td_api.tl   |  3 +++
 td/telegram/UpdatesManager.cpp | 13 +++++++++----
 td/telegram/UpdatesManager.h   |  4 ++--
 3 files changed, 14 insertions(+), 6 deletions(-)

diff --git a/td/generate/scheme/td_api.tl b/td/generate/scheme/td_api.tl
index f94ad4cc1..63bc92751 100644
--- a/td/generate/scheme/td_api.tl
+++ b/td/generate/scheme/td_api.tl
@@ -7286,6 +7286,9 @@ updateDefaultReactionType reaction_type:ReactionType = Update;
 //@tags The new tags
 updateSavedMessagesTags saved_messages_topic_id:int53 tags:savedMessagesTags = Update;
 
+//@description The number of stars owned by the current user has changed @star_count The new number of stars owned
+updateOwnedStarCount star_count:int53 = Update;
+
 //@description The revenue earned from sponsored messages in a chat has changed. If chat revenue screen is opened, then getChatRevenueTransactions may be called to fetch new transactions
 //@chat_id Identifier of the chat
 //@revenue_amount New amount of earned revenue
diff --git a/td/telegram/UpdatesManager.cpp b/td/telegram/UpdatesManager.cpp
index ec35337c6..455dc2177 100644
--- a/td/telegram/UpdatesManager.cpp
+++ b/td/telegram/UpdatesManager.cpp
@@ -4560,14 +4560,19 @@ void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateBroadcastRevenu
   promise.set_value(Unit());
 }
 
+void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateStarsBalance> update, Promise<Unit> &&promise) {
+  if (update->balance_ < 0) {
+    LOG(ERROR) << "Receive " << update->balance_ << " stars";
+    update->balance_ = 0;
+  }
+  send_closure(G()->td(), &Td::send_update, td_api::make_object<td_api::updateOwnedStarCount>(update->balance_));
+  promise.set_value(Unit());
+}
+
 // unsupported updates
 
 void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateNewStoryReaction> update, Promise<Unit> &&promise) {
   promise.set_value(Unit());
 }
 
-void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateStarsBalance> update, Promise<Unit> &&promise) {
-  promise.set_value(Unit());
-}
-
 }  // namespace td
diff --git a/td/telegram/UpdatesManager.h b/td/telegram/UpdatesManager.h
index ed8a39485..a98801110 100644
--- a/td/telegram/UpdatesManager.h
+++ b/td/telegram/UpdatesManager.h
@@ -671,11 +671,11 @@ class UpdatesManager final : public Actor {
 
   void on_update(tl_object_ptr<telegram_api::updateBroadcastRevenueTransactions> update, Promise<Unit> &&promise);
 
+  void on_update(tl_object_ptr<telegram_api::updateStarsBalance> update, Promise<Unit> &&promise);
+
   // unsupported updates
 
   void on_update(tl_object_ptr<telegram_api::updateNewStoryReaction> update, Promise<Unit> &&promise);
-
-  void on_update(tl_object_ptr<telegram_api::updateStarsBalance> update, Promise<Unit> &&promise);
 };
 
 }  // namespace td