From 0107ccfe28b67522b21f7a44f48d7ced6db0f4fd Mon Sep 17 00:00:00 2001
From: Ilya Laktyushin <ilya@laktyush.in>
Date: Thu, 30 Jan 2025 12:07:05 +0400
Subject: [PATCH] Allow video covers in Saved Messages

---
 submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift | 2 ++
 submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift b/submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift
index 54ce4dfa40..c88ee2e026 100644
--- a/submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift
+++ b/submodules/LegacyMediaPickerUI/Sources/LegacyMediaPickers.swift
@@ -38,6 +38,8 @@ public func configureLegacyAssetPicker(_ controller: TGMediaAssetsController, co
     controller.reminder = peer.id == context.account.peerId
     if let channel = peer as? TelegramChannel, case .broadcast = channel.info {
         controller.hasCoverButton = true
+    } else if peer.id == context.account.peerId {
+        controller.hasCoverButton = true
     }
     controller.presentScheduleController = { media, done in
         presentSchedulePicker(media, { time in
diff --git a/submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift b/submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift
index fc022bc9cb..507e1d781f 100644
--- a/submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift
+++ b/submodules/MediaPickerUI/Sources/LegacyMediaPickerGallery.swift
@@ -107,6 +107,8 @@ func presentLegacyMediaPickerGallery(context: AccountContext, peer: EnginePeer?,
     var hasCoverButton = false
     if case let .channel(channel) = peer, case .broadcast = channel.info {
         hasCoverButton = true
+    } else if peer?.id == context.account.peerId {
+        hasCoverButton = true
     }
     
     let legacyController = LegacyController(presentation: .custom, theme: presentationData.theme, initialLayout: nil)