79 lines
3.9 KiB
TypeScript
79 lines
3.9 KiB
TypeScript
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: string, encryptedFileData: EncArrayBuffer) {
|
|
const response = admin ? uploadData.cipherMiniResponse : uploadData.cipherResponse;
|
|
try {
|
|
switch (uploadData.fileUploadType) {
|
|
case FileUploadType.Direct:
|
|
await this.bitwardenFileUploadService.upload(encryptedFileName, 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;
|
|
}
|
|
}
|
|
}
|