From 4c083eeb92d599bd8b85fb88ec5d435079b79395 Mon Sep 17 00:00:00 2001
From: Kyle Spearrin <kyle.spearrin@gmail.com>
Date: Fri, 8 Jun 2018 12:14:02 -0400
Subject: [PATCH] download attachments function from component

---
 .../components/attachments.component.ts       | 34 ++++++++++++++++++-
 1 file changed, 33 insertions(+), 1 deletion(-)

diff --git a/src/angular/components/attachments.component.ts b/src/angular/components/attachments.component.ts
index 6a02a7c259..0d13a91305 100644
--- a/src/angular/components/attachments.component.ts
+++ b/src/angular/components/attachments.component.ts
@@ -32,7 +32,7 @@ export class AttachmentsComponent implements OnInit {
     constructor(protected cipherService: CipherService, protected analytics: Angulartics2,
         protected toasterService: ToasterService, protected i18nService: I18nService,
         protected cryptoService: CryptoService, protected tokenService: TokenService,
-        protected platformUtilsService: PlatformUtilsService) { }
+        protected platformUtilsService: PlatformUtilsService, protected win: Window) { }
 
     async ngOnInit() {
         this.cipherDomain = await this.cipherService.get(this.cipherId);
@@ -122,4 +122,36 @@ export class AttachmentsComponent implements OnInit {
 
         this.deletePromises[attachment.id] = null;
     }
+
+    async download(attachment: AttachmentView) {
+        const a = (attachment as any);
+        if (a.downloading) {
+            return;
+        }
+
+        if (!this.canAccessAttachments) {
+            this.toasterService.popAsync('error', this.i18nService.t('premiumRequired'),
+                this.i18nService.t('premiumRequiredDesc'));
+            return;
+        }
+
+        a.downloading = true;
+        const response = await fetch(new Request(attachment.url, { cache: 'no-cache' }));
+        if (response.status !== 200) {
+            this.toasterService.popAsync('error', null, this.i18nService.t('errorOccurred'));
+            a.downloading = false;
+            return;
+        }
+
+        try {
+            const buf = await response.arrayBuffer();
+            const key = await this.cryptoService.getOrgKey(this.cipher.organizationId);
+            const decBuf = await this.cryptoService.decryptFromBytes(buf, key);
+            this.platformUtilsService.saveFile(this.win, decBuf, null, attachment.fileName);
+        } catch (e) {
+            this.toasterService.popAsync('error', null, this.i18nService.t('errorOccurred'));
+        }
+
+        a.downloading = false;
+    }
 }