From c772a7b2ebe3c1cc5f32e20c18582711fdee4681 Mon Sep 17 00:00:00 2001
From: Alexander Zinchuk <alx.zinchuk@gmail.com>
Date: Wed, 20 Jul 2022 16:02:11 +0200
Subject: [PATCH] Audio: Remove download button in protected chats (#1960)

---
 src/components/common/Audio.tsx             | 4 +++-
 src/components/left/search/AudioResults.tsx | 1 +
 src/components/middle/message/Message.tsx   | 1 +
 src/components/right/Profile.tsx            | 2 ++
 4 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/src/components/common/Audio.tsx b/src/components/common/Audio.tsx
index dea1506c0..cb79a5f90 100644
--- a/src/components/common/Audio.tsx
+++ b/src/components/common/Audio.tsx
@@ -57,6 +57,7 @@ type OwnProps = {
   isDownloading: boolean;
   isTranscribing?: boolean;
   isTranscribed?: boolean;
+  canDownload?: boolean;
   canTranscribe?: boolean;
   isTranscriptionHidden?: boolean;
   isTranscriptionError?: boolean;
@@ -92,6 +93,7 @@ const Audio: FC<OwnProps> = ({
   isTranscriptionHidden,
   isTranscribed,
   isTranscriptionError,
+  canDownload,
   canTranscribe,
   onHideTranscription,
   onPlay,
@@ -365,7 +367,7 @@ const Audio: FC<OwnProps> = ({
           />
         </div>
       )}
-      {audio && !isUploading && (
+      {audio && canDownload && !isUploading && (
         <Button
           round
           size="tiny"
diff --git a/src/components/left/search/AudioResults.tsx b/src/components/left/search/AudioResults.tsx
index 8cba05389..0b3d0aebd 100644
--- a/src/components/left/search/AudioResults.tsx
+++ b/src/components/left/search/AudioResults.tsx
@@ -104,6 +104,7 @@ const AudioResults: FC<OwnProps & StateProps> = ({
             className="scroll-item"
             onPlay={handlePlayAudio}
             onDateClick={handleMessageFocus}
+            canDownload={!chatsById[message.chatId]?.isProtected && !message.isProtected}
             isDownloading={activeDownloads[message.chatId]?.includes(message.id)}
           />
         </div>
diff --git a/src/components/middle/message/Message.tsx b/src/components/middle/message/Message.tsx
index 5fc80768e..4f1a35e7f 100644
--- a/src/components/middle/message/Message.tsx
+++ b/src/components/middle/message/Message.tsx
@@ -772,6 +772,7 @@ const Message: FC<OwnProps & StateProps> = ({
             isTranscriptionHidden={isTranscriptionHidden}
             isTranscribed={Boolean(transcribedText)}
             isTranscriptionError={isTranscriptionError}
+            canDownload={!isProtected}
             onHideTranscription={setTranscriptionHidden}
             canTranscribe={isPremium}
           />
diff --git a/src/components/right/Profile.tsx b/src/components/right/Profile.tsx
index be5802547..47760df43 100644
--- a/src/components/right/Profile.tsx
+++ b/src/components/right/Profile.tsx
@@ -389,6 +389,7 @@ const Profile: FC<OwnProps & StateProps> = ({
               className="scroll-item"
               onPlay={handlePlayAudio}
               onDateClick={handleMessageFocus}
+              canDownload={!isChatProtected && !chatMessages[id].isProtected}
               isDownloading={activeDownloadIds.includes(id)}
             />
           ))
@@ -405,6 +406,7 @@ const Profile: FC<OwnProps & StateProps> = ({
               className="scroll-item"
               onPlay={handlePlayAudio}
               onDateClick={handleMessageFocus}
+              canDownload={!isChatProtected && !chatMessages[id].isProtected}
               isDownloading={activeDownloadIds.includes(id)}
             />
           ))