diff --git a/server/src/middleware/file-upload.interceptor.ts b/server/src/middleware/file-upload.interceptor.ts index 075a7f504636a..108a187e67b8c 100644 --- a/server/src/middleware/file-upload.interceptor.ts +++ b/server/src/middleware/file-upload.interceptor.ts @@ -11,6 +11,7 @@ import { RouteKey } from 'src/enum'; import { ILoggerRepository } from 'src/interfaces/logger.interface'; import { AuthRequest } from 'src/middleware/auth.guard'; import { AssetMediaService, UploadFile } from 'src/services/asset-media.service'; +import { asRequest, mapToUploadFile } from 'src/utils/asset.util'; export interface UploadFiles { assetData: ImmichFile[]; @@ -35,16 +36,6 @@ export interface ImmichFile extends Express.Multer.File { checksum: Buffer; } -export function mapToUploadFile(file: ImmichFile): UploadFile { - return { - uuid: file.uuid, - checksum: file.checksum, - originalPath: file.path, - originalName: Buffer.from(file.originalname, 'latin1').toString('utf8'), - size: file.size, - }; -} - type DiskStorageCallback = (error: Error | null, result: string) => void; type ImmichMulterFile = Express.Multer.File & { uuid: string }; @@ -62,14 +53,6 @@ const callbackify = (target: (...arguments_: any[]) => T, callback: Callback< } }; -const asRequest = (request: AuthRequest, file: Express.Multer.File) => { - return { - auth: request.user || null, - fieldName: file.fieldname as UploadFieldName, - file: mapToUploadFile(file as ImmichFile), - }; -}; - @Injectable() export class FileUploadInterceptor implements NestInterceptor { private handlers: { @@ -141,6 +124,12 @@ export class FileUploadInterceptor implements NestInterceptor { private handleFile(request: AuthRequest, file: Express.Multer.File, callback: Callback>) { (file as ImmichMulterFile).uuid = randomUUID(); + + request.on('error', (error) => { + this.logger.warn('Request error while uploading file, cleaning up', error); + this.assetService.onUploadError(request, file).catch(this.logger.error); + }); + if (!this.isAssetUploadFile(file)) { this.defaultStorage._handleFile(request, file, callback); return; diff --git a/server/src/services/asset-media.service.spec.ts b/server/src/services/asset-media.service.spec.ts index c269739935e01..d68140367d702 100644 --- a/server/src/services/asset-media.service.spec.ts +++ b/server/src/services/asset-media.service.spec.ts @@ -14,6 +14,7 @@ import { IAssetRepository } from 'src/interfaces/asset.interface'; import { IJobRepository, JobName } from 'src/interfaces/job.interface'; import { IStorageRepository } from 'src/interfaces/storage.interface'; import { IUserRepository } from 'src/interfaces/user.interface'; +import { AuthRequest } from 'src/middleware/auth.guard'; import { AssetMediaService } from 'src/services/asset-media.service'; import { ImmichFileResponse } from 'src/utils/file'; import { assetStub } from 'test/fixtures/asset.stub'; @@ -879,4 +880,28 @@ describe(AssetMediaService.name, () => { expect(assetMock.getByChecksums).toHaveBeenCalledWith(authStub.admin.user.id, [file1, file2]); }); }); + + describe('onUploadError', () => { + it('should queue a job to delete the uploaded file', async () => { + const request = { user: authStub.user1 } as AuthRequest; + + const file = { + fieldname: UploadFieldName.ASSET_DATA, + originalname: 'image.jpg', + mimetype: 'image/jpeg', + buffer: Buffer.from(''), + size: 1000, + uuid: 'random-uuid', + checksum: Buffer.from('checksum', 'utf8'), + originalPath: 'upload/upload/user-id/ra/nd/random-uuid.jpg', + } as unknown as Express.Multer.File; + + await sut.onUploadError(request, file); + + expect(jobMock.queue).toHaveBeenCalledWith({ + name: JobName.DELETE_FILES, + data: { files: ['upload/upload/user-id/ra/nd/random-uuid.jpg'] }, + }); + }); + }); }); diff --git a/server/src/services/asset-media.service.ts b/server/src/services/asset-media.service.ts index 70f4905de31e4..2424c93e44a15 100644 --- a/server/src/services/asset-media.service.ts +++ b/server/src/services/asset-media.service.ts @@ -23,9 +23,10 @@ import { AuthDto } from 'src/dtos/auth.dto'; import { ASSET_CHECKSUM_CONSTRAINT, AssetEntity } from 'src/entities/asset.entity'; import { AssetStatus, AssetType, CacheControl, Permission, StorageFolder } from 'src/enum'; import { JobName } from 'src/interfaces/job.interface'; +import { AuthRequest } from 'src/middleware/auth.guard'; import { BaseService } from 'src/services/base.service'; import { requireUploadAccess } from 'src/utils/access'; -import { getAssetFiles, onBeforeLink } from 'src/utils/asset.util'; +import { asRequest, getAssetFiles, onBeforeLink } from 'src/utils/asset.util'; import { ImmichFileResponse } from 'src/utils/file'; import { mimeTypes } from 'src/utils/mime-types'; import { fromChecksum } from 'src/utils/request'; @@ -118,6 +119,14 @@ export class AssetMediaService extends BaseService { return folder; } + async onUploadError(request: AuthRequest, file: Express.Multer.File) { + const uploadFilename = this.getUploadFilename(asRequest(request, file)); + const uploadFolder = this.getUploadFolder(asRequest(request, file)); + const uploadPath = `${uploadFolder}/${uploadFilename}`; + + await this.jobRepository.queue({ name: JobName.DELETE_FILES, data: { files: [uploadPath] } }); + } + async uploadAsset( auth: AuthDto, dto: AssetMediaCreateDto, diff --git a/server/src/utils/asset.util.ts b/server/src/utils/asset.util.ts index 44c291e139766..f8bed5485f8b1 100644 --- a/server/src/utils/asset.util.ts +++ b/server/src/utils/asset.util.ts @@ -1,6 +1,7 @@ import { BadRequestException } from '@nestjs/common'; import { StorageCore } from 'src/cores/storage.core'; import { BulkIdErrorReason, BulkIdResponseDto } from 'src/dtos/asset-ids.response.dto'; +import { UploadFieldName } from 'src/dtos/asset-media.dto'; import { AuthDto } from 'src/dtos/auth.dto'; import { AssetFileEntity } from 'src/entities/asset-files.entity'; import { AssetFileType, AssetType, Permission } from 'src/enum'; @@ -8,6 +9,9 @@ import { IAccessRepository } from 'src/interfaces/access.interface'; import { IAssetRepository } from 'src/interfaces/asset.interface'; import { IEventRepository } from 'src/interfaces/event.interface'; import { IPartnerRepository } from 'src/interfaces/partner.interface'; +import { AuthRequest } from 'src/middleware/auth.guard'; +import { ImmichFile } from 'src/middleware/file-upload.interceptor'; +import { UploadFile } from 'src/services/asset-media.service'; import { checkAccess } from 'src/utils/access'; export interface IBulkAsset { @@ -181,3 +185,21 @@ export const onAfterUnlink = async ( await assetRepository.update({ id: livePhotoVideoId, isVisible: true }); await eventRepository.emit('asset.show', { assetId: livePhotoVideoId, userId }); }; + +export function mapToUploadFile(file: ImmichFile): UploadFile { + return { + uuid: file.uuid, + checksum: file.checksum, + originalPath: file.path, + originalName: Buffer.from(file.originalname, 'latin1').toString('utf8'), + size: file.size, + }; +} + +export const asRequest = (request: AuthRequest, file: Express.Multer.File) => { + return { + auth: request.user || null, + fieldName: file.fieldname as UploadFieldName, + file: mapToUploadFile(file as ImmichFile), + }; +};