Various improvements

TelegramMessenger · May 28, 2024 · 0137836 · 0137836
1 parent 51b262a
commit 0137836
Showing 1 changed file with 20 additions and 13 deletions.
diff --git a/submodules/TelegramUI/Components/Stars/StarsTransferScreen/Sources/StarsTransferScreen.swift b/submodules/TelegramUI/Components/Stars/StarsTransferScreen/Sources/StarsTransferScreen.swift
@@ -414,19 +414,26 @@ private final class SheetContent: CombinedComponent {
                             }
                         }, completion: { [weak controller] in
                             let presentationData = accountContext.sharedContext.currentPresentationData.with { $0 }
-                            let resultController = UndoOverlayController(
-                                presentationData: presentationData,
-                                content: .image(
-                                    image: UIImage(bundleImageName: "Premium/Stars/StarLarge")!,
-                                    title: presentationData.strings.Stars_Transfer_PurchasedTitle,
-                                    text: presentationData.strings.Stars_Transfer_PurchasedText(invoice.title, botTitle, presentationData.strings.Stars_Transfer_Purchased_Stars(Int32(invoice.totalAmount))).string,
-                                    round: false,
-                                    undoText: nil
-                                ),
-                                elevatedLayout: true,
-                                action: { _ in return true})
-                            controller?.present(resultController, in: .window(.root))
-
+                            if let navigationController = controller?.navigationController {
+                                Queue.mainQueue().after(0.5) {
+                                    if let lastController = navigationController.viewControllers.last as? ViewController {
+                                        let resultController = UndoOverlayController(
+                                            presentationData: presentationData,
+                                            content: .image(
+                                                image: UIImage(bundleImageName: "Premium/Stars/StarLarge")!,
+                                                title: presentationData.strings.Stars_Transfer_PurchasedTitle,
+                                                text: presentationData.strings.Stars_Transfer_PurchasedText(invoice.title, botTitle, presentationData.strings.Stars_Transfer_Purchased_Stars(Int32(invoice.totalAmount))).string,
+                                                round: false,
+                                                undoText: nil
+                                            ),
+                                            elevatedLayout: lastController is ChatController,
+                                            action: { _ in return true}
+                                        )
+                                        lastController.present(resultController, in: .window(.root))
+                                    }
+                                }
+                            }
+
                             controller?.complete(paid: true)
                             controller?.dismissAnimated()