From cbc1ab437bba3574ade31103414b9b0883f6a554 Mon Sep 17 00:00:00 2001 From: Morgan Pretty Date: Fri, 8 Sep 2023 09:09:28 +1000 Subject: [PATCH] Fixed an issue where tapping All Media on the media detail screen wasn't working --- .../MediaPageViewController.swift | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/Session/Media Viewing & Editing/MediaPageViewController.swift b/Session/Media Viewing & Editing/MediaPageViewController.swift index 1e14b2bdd..abb9ff23f 100644 --- a/Session/Media Viewing & Editing/MediaPageViewController.swift +++ b/Session/Media Viewing & Editing/MediaPageViewController.swift @@ -469,8 +469,17 @@ class MediaPageViewController: UIPageViewController, UIPageViewControllerDataSou // If the screen wasn't presented or it was presented from a location which isn't the // MediaTileViewController then just pop/dismiss the screen + let parentNavController: UINavigationController? = { + switch self.presentingViewController { + case let topBannerController as TopBannerController: + return topBannerController.children.first as? UINavigationController + + default: return self.presentingViewController as? UINavigationController + } + }() + guard - let presentingNavController: UINavigationController = (self.presentingViewController as? UINavigationController), + let presentingNavController: UINavigationController = parentNavController, !(presentingNavController.viewControllers.last is AllMediaViewController) else { guard self.navigationController?.viewControllers.count == 1 else {