Move to libs

2025-12-19 17:53:39 +00:00 · 2022-06-03 16:24:40 +02:00
parent 28d15bfe2a
commit d7492e3cf3
878 changed files with 0 additions and 0 deletions
--- a/libs/common/src/services/fileUpload.service.ts
+++ b/libs/common/src/services/fileUpload.service.ts
@@ -0,0 +1,108 @@
+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 { EncArrayBuffer } from "../models/domain/encArrayBuffer";
+import { EncString } from "../models/domain/encString";
+import { AttachmentUploadDataResponse } from "../models/response/attachmentUploadDataResponse";
+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: EncString,
+    encryptedFileData: EncArrayBuffer
+  ) {
+    try {
+      switch (uploadData.fileUploadType) {
+        case FileUploadType.Direct:
+          await this.bitwardenFileUploadService.upload(
+            fileName.encryptedString,
+            encryptedFileData,
+            (fd) =>
+              this.apiService.postSendFile(
+                uploadData.sendResponse.id,
+                uploadData.sendResponse.file.id,
+                fd
+              )
+          );
+          break;
+        case FileUploadType.Azure: {
+          const renewalCallback = async () => {
+            const renewalResponse = await this.apiService.renewSendFileUploadUrl(
+              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) {
+      await this.apiService.deleteSend(uploadData.sendResponse.id);
+      throw e;
+    }
+  }
+
+  async uploadCipherAttachment(
+    admin: boolean,
+    uploadData: AttachmentUploadDataResponse,
+    encryptedFileName: EncString,
+    encryptedFileData: EncArrayBuffer
+  ) {
+    const response = admin ? uploadData.cipherMiniResponse : uploadData.cipherResponse;
+    try {
+      switch (uploadData.fileUploadType) {
+        case FileUploadType.Direct:
+          await this.bitwardenFileUploadService.upload(
+            encryptedFileName.encryptedString,
+            encryptedFileData,
+            (fd) => this.apiService.postAttachmentFile(response.id, uploadData.attachmentId, fd)
+          );
+          break;
+        case FileUploadType.Azure: {
+          const renewalCallback = async () => {
+            const renewalResponse = await this.apiService.renewAttachmentUploadUrl(
+              response.id,
+              uploadData.attachmentId
+            );
+            return renewalResponse.url;
+          };
+          await this.azureFileUploadService.upload(
+            uploadData.url,
+            encryptedFileData,
+            renewalCallback
+          );
+          break;
+        }
+        default:
+          throw new Error("Unknown file upload type.");
+      }
+    } catch (e) {
+      if (admin) {
+        await this.apiService.deleteCipherAttachmentAdmin(response.id, uploadData.attachmentId);
+      } else {
+        await this.apiService.deleteCipherAttachment(response.id, uploadData.attachmentId);
+      }
+      throw e;
+    }
+  }
+}