From 4a5e61eaf4a0faa811fd491531c2e0fe360aac53 Mon Sep 17 00:00:00 2001
From: Daniel Gasienica <daniel@signal.org>
Date: Wed, 25 Apr 2018 13:27:28 -0400
Subject: [PATCH] Implement `Attachment.save`

---
 ts/test/types/Attachment_test.ts | 60 ++++++++++++++++++++++++++++++++
 ts/types/Attachment.ts           | 56 +++++++++++++++++++++++++++++
 2 files changed, 116 insertions(+)
 create mode 100644 ts/test/types/Attachment_test.ts

diff --git a/ts/test/types/Attachment_test.ts b/ts/test/types/Attachment_test.ts
new file mode 100644
index 000000000..343373d4b
--- /dev/null
+++ b/ts/test/types/Attachment_test.ts
@@ -0,0 +1,60 @@
+/**
+ * @prettier
+ */
+import 'mocha';
+import { assert } from 'chai';
+
+import * as Attachment from '../../types/Attachment';
+import { MIMEType } from '../../types/MIME';
+// @ts-ignore
+import { stringToArrayBuffer } from '../../../js/modules/string_to_array_buffer';
+
+describe('Attachment', () => {
+  describe('getFileExtension', () => {
+    it('should return file extension from content type', () => {
+      const input: Attachment.Attachment = {
+        data: stringToArrayBuffer('foo'),
+        contentType: 'image/gif' as MIMEType,
+      };
+      assert.strictEqual(Attachment.getFileExtension(input), 'gif');
+    });
+
+    it('should return file extension for QuickTime videos', () => {
+      const input: Attachment.Attachment = {
+        data: stringToArrayBuffer('foo'),
+        contentType: 'video/quicktime' as MIMEType,
+      };
+      assert.strictEqual(Attachment.getFileExtension(input), 'mov');
+    });
+  });
+
+  describe('getSuggestedFilename', () => {
+    context('for attachment with filename', () => {
+      it('should return existing filename if present', () => {
+        const attachment: Attachment.Attachment = {
+          fileName: 'funny-cat.mov',
+          data: stringToArrayBuffer('foo'),
+          contentType: 'video/quicktime' as MIMEType,
+        };
+        const actual = Attachment.getSuggestedFilename({ attachment });
+        const expected = 'funny-cat.mov';
+        assert.strictEqual(actual, expected);
+      });
+    });
+    context('for attachment without filename', () => {
+      it('should generate a filename based on timestamp', () => {
+        const attachment: Attachment.Attachment = {
+          data: stringToArrayBuffer('foo'),
+          contentType: 'video/quicktime' as MIMEType,
+        };
+        const timestamp = new Date(new Date(0).getTimezoneOffset() * 60 * 1000);
+        const actual = Attachment.getSuggestedFilename({
+          attachment,
+          timestamp,
+        });
+        const expected = 'signal-attachment-1970-01-01-000000.mov';
+        assert.strictEqual(actual, expected);
+      });
+    });
+  });
+});
diff --git a/ts/types/Attachment.ts b/ts/types/Attachment.ts
index fa4e09106..b71421557 100644
--- a/ts/types/Attachment.ts
+++ b/ts/types/Attachment.ts
@@ -2,8 +2,10 @@
  * @prettier
  */
 import is from '@sindresorhus/is';
+import moment from 'moment';
 
 import * as GoogleChrome from '../util/GoogleChrome';
+import { arrayBufferToObjectURL } from '../util/arrayBufferToObjectURL';
 import { MIMEType } from './MIME';
 
 export interface Attachment {
@@ -23,6 +25,8 @@ export interface Attachment {
   // flags?: number;
 }
 
+const SAVE_CONTENT_TYPE = 'application/octet-stream' as MIMEType;
+
 export const isVisualMedia = (attachment: Attachment): boolean => {
   const { contentType } = attachment;
 
@@ -34,3 +38,55 @@ export const isVisualMedia = (attachment: Attachment): boolean => {
   const isSupportedVideoType = GoogleChrome.isVideoTypeSupported(contentType);
   return isSupportedImageType || isSupportedVideoType;
 };
+
+export const save = ({
+  attachment,
+  timestamp,
+}: {
+  attachment: Attachment;
+  timestamp?: number;
+}): void => {
+  const url = arrayBufferToObjectURL({
+    data: attachment.data,
+    type: SAVE_CONTENT_TYPE,
+  });
+  const anchorElement = document.createElement('a');
+  anchorElement.href = url;
+  anchorElement.download = getSuggestedFilename({ attachment, timestamp });
+  anchorElement.click();
+  URL.revokeObjectURL(url);
+};
+
+export const getSuggestedFilename = ({
+  attachment,
+  timestamp,
+}: {
+  attachment: Attachment;
+  timestamp?: number | Date;
+}): string => {
+  if (attachment.fileName) {
+    return attachment.fileName;
+  }
+
+  const prefix = 'signal-attachment';
+  const suffix = timestamp
+    ? moment(timestamp).format('-YYYY-MM-DD-HHmmss')
+    : '';
+  const fileType = getFileExtension(attachment);
+  const extension = fileType ? `.${fileType}` : '';
+  return `${prefix}${suffix}${extension}`;
+};
+
+export const getFileExtension = (attachment: Attachment): string | null => {
+  if (!attachment.contentType) {
+    return null;
+  }
+
+  switch (attachment.contentType) {
+    case 'video/quicktime':
+      return 'mov';
+    default:
+      // TODO: Use better MIME --> file extension mapping:
+      return attachment.contentType.split('/')[1];
+  }
+};