mirrors/hedgedoc
1
0
Fork 0
mirror of https://github.com/hedgedoc/hedgedoc.git synced 2025-05-23 11:37:02 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

MediaService: Get media backend from configuration

Browse source 
Signed-off-by: David Mehren <git@herrmehren.de>
This commit is contained in:
David Mehren 2021-01-08 13:06:44 +01:00
parent 0e7845e38f
commit 75b6d3cc2b
No known key found for this signature in database
GPG key ID: 185982BA4C42B7C3
1 changed files with 28 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

36
src/media/media.service.ts
View file

@ -1,13 +1,14 @@
/* /*
* SPDX-FileCopyrightText: 2021 The HedgeDoc developers (see AUTHORS file) * SPDX-FileCopyrightText: 2020 The HedgeDoc developers (see AUTHORS file)
* *
* SPDX-License-Identifier: AGPL-3.0-only * SPDX-License-Identifier: AGPL-3.0-only
*/ */
import { 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 * as FileType from 'file-type'; import * as FileType from 'file-type';
import applicationConfig, { AppConfig } from 'src/config/app.config';
import { Repository } from 'typeorm'; import { Repository } from 'typeorm';
import { ClientError, NotInDBError, PermissionError } from '../errors/errors'; import { ClientError, NotInDBError, PermissionError } from '../errors/errors';
import { ConsoleLoggerService } from '../logger/console-logger.service'; import { ConsoleLoggerService } from '../logger/console-logger.service';
@ -15,10 +16,14 @@ import { NotesService } from '../notes/notes.service';
import { UsersService } from '../users/users.service'; import { UsersService } from '../users/users.service';
import { BackendType } from './backends/backend-type.enum'; import { BackendType } from './backends/backend-type.enum';
import { FilesystemBackend } from './backends/filesystem-backend'; import { FilesystemBackend } from './backends/filesystem-backend';
import { MediaBackend } from './media-backend.interface';
import { MediaUpload } from './media-upload.entity'; import { MediaUpload } from './media-upload.entity';
@Injectable() @Injectable()
export class MediaService { export class MediaService {
mediaBackend: MediaBackend;
mediaBackendType: BackendType;
constructor( constructor(
private readonly logger: ConsoleLoggerService, private readonly logger: ConsoleLoggerService,
@InjectRepository(MediaUpload) @InjectRepository(MediaUpload)
@ -26,8 +31,12 @@ export class MediaService {
private notesService: NotesService, private notesService: NotesService,
private usersService: UsersService, private usersService: UsersService,
private moduleRef: ModuleRef, private moduleRef: ModuleRef,
@Inject(applicationConfig.KEY)
private appConfig: AppConfig,
) { ) {
this.logger.setContext(MediaService.name); this.logger.setContext(MediaService.name);
this.mediaBackendType = this.chooseBackendType();
this.mediaBackend = this.getBackendFromType(this.mediaBackendType);
} }
private static isAllowedMimeType(mimeType: string): boolean { private static isAllowedMimeType(mimeType: string): boolean {
@ -63,16 +72,14 @@ 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.');
} }
//TODO: Choose backend according to config
const mediaUpload = MediaUpload.create( const mediaUpload = MediaUpload.create(
note, note,
user, user,
fileTypeResult.ext, fileTypeResult.ext,
BackendType.FILEYSTEM, this.mediaBackendType,
); );
this.logger.debug(`Generated filename: '${mediaUpload.id}'`, 'saveFile'); this.logger.debug(`Generated filename: '${mediaUpload.id}'`, 'saveFile');
const backend = this.moduleRef.get(FilesystemBackend); const [url, backendData] = await this.mediaBackend.saveFile(
const [url, backendData] = await backend.saveFile(
fileBuffer, fileBuffer,
mediaUpload.id, mediaUpload.id,
); );
@ -96,8 +103,7 @@ export class MediaService {
`File '${filename}' is not owned by '${username}'`, `File '${filename}' is not owned by '${username}'`,
); );
} }
const backend = this.moduleRef.get(FilesystemBackend); await this.mediaBackend.deleteFile(filename, mediaUpload.backendData);
await backend.deleteFile(filename, mediaUpload.backendData);
await this.mediaUploadRepository.remove(mediaUpload); await this.mediaUploadRepository.remove(mediaUpload);
} }
@ -112,4 +118,18 @@ export class MediaService {
} }
return mediaUpload; return mediaUpload;
} }
private chooseBackendType(): BackendType {
switch (this.appConfig.media.backend.use) {
case 'filesystem':
return BackendType.FILEYSTEM;
}
}
private getBackendFromType(type: BackendType): MediaBackend {
switch (type) {
case BackendType.FILEYSTEM:
return this.moduleRef.get(FilesystemBackend);
}
}
} }