fix(media-upload): rework create and media services saveFile

This was done to make the create method more concise.

Signed-off-by: Philip Molares <philip.molares@udo.edu>
This commit is contained in:
Philip Molares 2021-11-11 22:06:30 +01:00
parent d5c77613bb
commit 09f0d7c389
2 changed files with 24 additions and 16 deletions

View file

@ -3,7 +3,6 @@
* *
* SPDX-License-Identifier: AGPL-3.0-only * SPDX-License-Identifier: AGPL-3.0-only
*/ */
import * as crypto from 'crypto';
import { import {
Column, Column,
CreateDateColumn, CreateDateColumn,
@ -53,24 +52,32 @@ export class MediaUpload {
// eslint-disable-next-line @typescript-eslint/no-empty-function // eslint-disable-next-line @typescript-eslint/no-empty-function
private constructor() {} private constructor() {}
/**
* Create a new media upload enity
* @param id the id of the upload
* @param note the note the upload should be associated with. This is required despite the fact the note field is optional, because it's possible to delete a note without also deleting the associated media uploads, but a note is required for the initial creation.
* @param user the user that owns the upload
* @param extension which file extension the upload has
* @param backendType on which type of media backend the upload is saved
* @param backendData the backend data returned by the media backend
* @param fileUrl the url where the upload can be accessed
*/
public static create( public static create(
id: string,
note: Note, note: Note,
user: User, user: User,
extension: string, extension: string,
backendType: BackendType, backendType: BackendType,
backendData?: string, backendData: BackendData | null,
fileUrl: string,
): Omit<MediaUpload, 'createdAt'> { ): Omit<MediaUpload, 'createdAt'> {
const upload = new MediaUpload(); const upload = new MediaUpload();
const randomBytes = crypto.randomBytes(16); upload.id = id;
upload.id = randomBytes.toString('hex') + '.' + extension;
upload.note = note; upload.note = note;
upload.user = user; upload.user = user;
upload.backendType = backendType; upload.backendType = backendType;
if (backendData) { upload.backendData = backendData;
upload.backendData = backendData; upload.fileUrl = fileUrl;
} else {
upload.backendData = null;
}
return upload; return upload;
} }
} }

View file

@ -6,6 +6,7 @@
import { Inject, Injectable } from '@nestjs/common'; import { Inject, Injectable } from '@nestjs/common';
import { ModuleRef } from '@nestjs/core'; import { ModuleRef } from '@nestjs/core';
import { InjectRepository } from '@nestjs/typeorm'; import { InjectRepository } from '@nestjs/typeorm';
import crypto from 'crypto';
import * as FileType from 'file-type'; import * as FileType from 'file-type';
import { Repository } from 'typeorm'; import { Repository } from 'typeorm';
@ -88,19 +89,19 @@ export class MediaService {
if (!MediaService.isAllowedMimeType(fileTypeResult.mime)) { if (!MediaService.isAllowedMimeType(fileTypeResult.mime)) {
throw new ClientError('MIME type not allowed.'); throw new ClientError('MIME type not allowed.');
} }
const randomBytes = crypto.randomBytes(16);
const id = randomBytes.toString('hex') + '.' + fileTypeResult.ext;
this.logger.debug(`Generated filename: '${id}'`, 'saveFile');
const [url, backendData] = await this.mediaBackend.saveFile(fileBuffer, id);
const mediaUpload = MediaUpload.create( const mediaUpload = MediaUpload.create(
id,
note, note,
user, user,
fileTypeResult.ext, fileTypeResult.ext,
this.mediaBackendType, this.mediaBackendType,
backendData,
url,
); );
this.logger.debug(`Generated filename: '${mediaUpload.id}'`, 'saveFile');
const [url, backendData] = await this.mediaBackend.saveFile(
fileBuffer,
mediaUpload.id,
);
mediaUpload.backendData = backendData;
mediaUpload.fileUrl = url;
await this.mediaUploadRepository.save(mediaUpload); await this.mediaUploadRepository.save(mediaUpload);
return url; return url;
} }