Upload to Azure strorage blobs (#296)

* Implemen AzureStorageService handes uploading files to azure blob * Correct one-shot size * Add azureStorage.service abstraction * Rename azure upload method * Prefer abstractions in DI * Abstract file upload to a single service handling uploads * Fallback to legacy upload method * Linter fix * Limit legacy upload to 404 error
2025-12-18 09:13:33 +00:00 · 2021-03-25 10:20:38 -05:00
parent 7f3bbd6e51
commit cfc7687815
11 changed files with 388 additions and 40 deletions
--- a/src/services/fileUpload.service.ts
+++ b/src/services/fileUpload.service.ts
@@ -0,0 +1,45 @@
+import { ApiService } from '../abstractions/api.service';
+import { FileUploadService as FileUploadServiceAbstraction } from '../abstractions/fileUpload.service';
+import { LogService } from '../abstractions/log.service';
+
+import { FileUploadType } from '../enums/fileUploadType';
+
+import { CipherString } from '../models/domain';
+import { SendFileUploadDataResponse } from '../models/response/sendFileUploadDataResponse';
+
+import { AzureFileUploadService } from './azureFileUpload.service';
+import { BitwardenFileUploadService } from './bitwardenFileUpload.service';
+
+export class FileUploadService implements FileUploadServiceAbstraction {
+    private azureFileUploadService: AzureFileUploadService;
+    private bitwardenFileUploadService: BitwardenFileUploadService;
+
+    constructor(private logService: LogService, private apiService: ApiService) {
+        this.azureFileUploadService = new AzureFileUploadService(logService);
+        this.bitwardenFileUploadService = new BitwardenFileUploadService(apiService);
+    }
+
+    async uploadSendFile(uploadData: SendFileUploadDataResponse, fileName: CipherString, encryptedFileData: ArrayBuffer) {
+        try {
+            switch (uploadData.fileUploadType) {
+                case FileUploadType.Direct:
+                    await this.bitwardenFileUploadService.upload(uploadData.sendResponse, fileName, encryptedFileData);
+                    break;
+                case FileUploadType.Azure:
+                    const renewalCallback = async () => {
+                        const renewalResponse = await this.apiService.renewFileUploadUrl(uploadData.sendResponse.id,
+                            uploadData.sendResponse.file.id);
+                        return renewalResponse.url;
+                    };
+                    await this.azureFileUploadService.upload(uploadData.url, encryptedFileData,
+                        renewalCallback);
+                    break;
+                default:
+                    throw new Error('Unknown file upload type');
+            }
+        } catch (e) {
+            this.apiService.deleteSend(uploadData.sendResponse.id);
+            throw e;
+        }
+    }
+}