Calculate warning properly for all methods of attachment download

signalapp · Oct 23, 2018 · 7789bee · 7789bee
1 parent f01c5f3
commit 7789bee
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/ts/components/conversation/Message.tsx b/ts/components/conversation/Message.tsx
@@ -819,6 +819,8 @@ export class Message extends React.Component<Props, State> {
     } = this.props;
 
     const showRetry = status === 'error' && direction === 'outgoing';
+    const fileName = attachment ? attachment.fileName : null;
+    const isDangerous = isFileDangerous(fileName || '');
 
     return (
       <ContextMenu id={triggerId}>
@@ -827,7 +829,11 @@ export class Message extends React.Component<Props, State> {
             attributes={{
               className: 'module-message__context__download',
             }}
-            onClick={onDownload}
+            onClick={() => {
+              if (onDownload) {
+                onDownload(isDangerous);
+              }
+            }}
           >
             {i18n('downloadAttachment')}
           </MenuItem>