diff --git a/server/src/cores/storage.core.ts b/server/src/cores/storage.core.ts index 2ce7348f8d..4cbf963158 100644 --- a/server/src/cores/storage.core.ts +++ b/server/src/cores/storage.core.ts @@ -12,7 +12,7 @@ import { MoveRepository } from 'src/repositories/move.repository'; import { PersonRepository } from 'src/repositories/person.repository'; import { StorageRepository } from 'src/repositories/storage.repository'; import { SystemMetadataRepository } from 'src/repositories/system-metadata.repository'; -import { getAssetFiles } from 'src/utils/asset.util'; +import { getAssetFile } from 'src/utils/asset.util'; import { getConfig } from 'src/utils/config'; export interface MoveRequest { @@ -117,8 +117,7 @@ export class StorageCore { async moveAssetImage(asset: AssetEntity, pathType: GeneratedImageType, format: ImageFormat) { const { id: entityId, files } = asset; - const { thumbnailFile, previewFile } = getAssetFiles(files); - const oldFile = pathType === AssetPathType.PREVIEW ? previewFile : thumbnailFile; + const oldFile = getAssetFile(files, pathType); return this.moveFile({ entityId, pathType, diff --git a/server/src/services/asset.service.ts b/server/src/services/asset.service.ts index 37ec00b0bb..d05bb023f2 100644 --- a/server/src/services/asset.service.ts +++ b/server/src/services/asset.service.ts @@ -233,8 +233,8 @@ export class AssetService extends BaseService { } } - const { thumbnailFile, previewFile } = getAssetFiles(asset.files); - const files = [thumbnailFile?.path, previewFile?.path, asset.encodedVideoPath]; + const { fullsizeFile, previewFile, thumbnailFile } = getAssetFiles(asset.files); + const files = [thumbnailFile?.path, previewFile?.path, fullsizeFile?.path, asset.encodedVideoPath]; if (deleteOnDisk) { files.push(asset.sidecarPath, asset.originalPath); diff --git a/server/src/services/audit.service.ts b/server/src/services/audit.service.ts index 3948469765..a049a9c64b 100644 --- a/server/src/services/audit.service.ts +++ b/server/src/services/audit.service.ts @@ -136,8 +136,14 @@ export class AuditService extends BaseService { for await (const assets of pagination) { assetCount += assets.length; for (const { id, files, originalPath, encodedVideoPath, isExternal, checksum } of assets) { - const { previewFile, thumbnailFile } = getAssetFiles(files); - for (const file of [originalPath, previewFile?.path, encodedVideoPath, thumbnailFile?.path]) { + const { fullsizeFile, previewFile, thumbnailFile } = getAssetFiles(files); + for (const file of [ + originalPath, + fullsizeFile?.path, + previewFile?.path, + encodedVideoPath, + thumbnailFile?.path, + ]) { track(file); } diff --git a/server/src/services/duplicate.service.ts b/server/src/services/duplicate.service.ts index 74b86f8e4e..da6c6794fb 100644 --- a/server/src/services/duplicate.service.ts +++ b/server/src/services/duplicate.service.ts @@ -5,12 +5,12 @@ import { mapAsset } from 'src/dtos/asset-response.dto'; import { AuthDto } from 'src/dtos/auth.dto'; import { DuplicateResponseDto } from 'src/dtos/duplicate.dto'; import { AssetEntity } from 'src/entities/asset.entity'; -import { JobName, JobStatus, QueueName } from 'src/enum'; +import { AssetFileType, JobName, JobStatus, QueueName } from 'src/enum'; import { WithoutProperty } from 'src/repositories/asset.repository'; import { AssetDuplicateResult } from 'src/repositories/search.repository'; import { BaseService } from 'src/services/base.service'; import { JobOf } from 'src/types'; -import { getAssetFiles } from 'src/utils/asset.util'; +import { getAssetFile } from 'src/utils/asset.util'; import { isDuplicateDetectionEnabled } from 'src/utils/misc'; import { usePagination } from 'src/utils/pagination'; @@ -69,7 +69,7 @@ export class DuplicateService extends BaseService { return JobStatus.SKIPPED; } - const { previewFile } = getAssetFiles(asset.files); + const previewFile = getAssetFile(asset.files, AssetFileType.PREVIEW); if (!previewFile) { this.logger.warn(`Asset ${id} is missing preview image`); return JobStatus.FAILED; diff --git a/server/src/services/notification.service.ts b/server/src/services/notification.service.ts index 0b8b332960..dcec865f89 100644 --- a/server/src/services/notification.service.ts +++ b/server/src/services/notification.service.ts @@ -2,12 +2,12 @@ import { BadRequestException, Injectable } from '@nestjs/common'; import { OnEvent, OnJob } from 'src/decorators'; import { SystemConfigSmtpDto } from 'src/dtos/system-config.dto'; import { AlbumEntity } from 'src/entities/album.entity'; -import { JobName, JobStatus, QueueName } from 'src/enum'; +import { AssetFileType, JobName, JobStatus, QueueName } from 'src/enum'; import { ArgOf } from 'src/repositories/event.repository'; import { EmailTemplate } from 'src/repositories/notification.repository'; import { BaseService } from 'src/services/base.service'; import { EmailImageAttachment, IEntityJob, INotifyAlbumUpdateJob, JobItem, JobOf } from 'src/types'; -import { getAssetFiles } from 'src/utils/asset.util'; +import { getAssetFile } from 'src/utils/asset.util'; import { getFilenameExtension } from 'src/utils/file'; import { getExternalDomain } from 'src/utils/misc'; import { isEqualObject } from 'src/utils/object'; @@ -398,7 +398,11 @@ export class NotificationService extends BaseService { } const albumThumbnail = await this.assetRepository.getById(album.albumThumbnailAssetId, { files: true }); - const { thumbnailFile } = getAssetFiles(albumThumbnail?.files); + if (!albumThumbnail) { + return; + } + + const thumbnailFile = getAssetFile(albumThumbnail.files, AssetFileType.THUMBNAIL); if (!thumbnailFile) { return; } diff --git a/server/src/services/person.service.ts b/server/src/services/person.service.ts index 65a4508a3b..b34b0ddcff 100644 --- a/server/src/services/person.service.ts +++ b/server/src/services/person.service.ts @@ -26,6 +26,7 @@ import { AssetEntity } from 'src/entities/asset.entity'; import { FaceSearchEntity } from 'src/entities/face-search.entity'; import { PersonEntity } from 'src/entities/person.entity'; import { + AssetFileType, AssetType, CacheControl, ImageFormat, @@ -42,7 +43,7 @@ import { BoundingBox } from 'src/repositories/machine-learning.repository'; import { UpdateFacesData } from 'src/repositories/person.repository'; import { BaseService } from 'src/services/base.service'; import { CropOptions, ImageDimensions, InputDimensions, JobItem, JobOf } from 'src/types'; -import { getAssetFiles } from 'src/utils/asset.util'; +import { getAssetFile } from 'src/utils/asset.util'; import { ImmichFileResponse } from 'src/utils/file'; import { mimeTypes } from 'src/utils/mime-types'; import { isFaceImportEnabled, isFacialRecognitionEnabled } from 'src/utils/misc'; @@ -300,7 +301,7 @@ export class PersonService extends BaseService { const relations = { exifInfo: true, faces: { person: false, withDeleted: true }, files: true }; const [asset] = await this.assetRepository.getByIds([id], relations); - const { previewFile } = getAssetFiles(asset.files); + const previewFile = getAssetFile(asset.files, AssetFileType.PREVIEW); if (!asset || !previewFile) { return JobStatus.FAILED; } @@ -674,7 +675,7 @@ export class PersonService extends BaseService { throw new Error(`Asset ${asset.id} dimensions are unknown`); } - const { previewFile } = getAssetFiles(asset.files); + const previewFile = getAssetFile(asset.files, AssetFileType.PREVIEW); if (!previewFile) { throw new Error(`Asset ${asset.id} has no preview path`); } diff --git a/server/src/services/smart-info.service.ts b/server/src/services/smart-info.service.ts index 063bb0bd3b..1909d2bb99 100644 --- a/server/src/services/smart-info.service.ts +++ b/server/src/services/smart-info.service.ts @@ -2,12 +2,12 @@ import { Injectable } from '@nestjs/common'; import { SystemConfig } from 'src/config'; import { JOBS_ASSET_PAGINATION_SIZE } from 'src/constants'; import { OnEvent, OnJob } from 'src/decorators'; -import { DatabaseLock, ImmichWorker, JobName, JobStatus, QueueName } from 'src/enum'; +import { AssetFileType, DatabaseLock, ImmichWorker, JobName, JobStatus, QueueName } from 'src/enum'; import { WithoutProperty } from 'src/repositories/asset.repository'; import { ArgOf } from 'src/repositories/event.repository'; import { BaseService } from 'src/services/base.service'; import { JobOf } from 'src/types'; -import { getAssetFiles } from 'src/utils/asset.util'; +import { getAssetFile } from 'src/utils/asset.util'; import { getCLIPModelInfo, isSmartSearchEnabled } from 'src/utils/misc'; import { usePagination } from 'src/utils/pagination'; @@ -116,7 +116,7 @@ export class SmartInfoService extends BaseService { return JobStatus.SKIPPED; } - const { previewFile } = getAssetFiles(asset.files); + const previewFile = getAssetFile(asset.files, AssetFileType.PREVIEW); if (!previewFile) { return JobStatus.FAILED; } diff --git a/server/src/utils/asset.util.ts b/server/src/utils/asset.util.ts index d5b100bf0c..50dcab334c 100644 --- a/server/src/utils/asset.util.ts +++ b/server/src/utils/asset.util.ts @@ -1,5 +1,5 @@ import { BadRequestException } from '@nestjs/common'; -import { StorageCore } from 'src/cores/storage.core'; +import { GeneratedImageType, 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'; @@ -13,14 +13,14 @@ import { PartnerRepository } from 'src/repositories/partner.repository'; import { IBulkAsset, ImmichFile, UploadFile } from 'src/types'; import { checkAccess } from 'src/utils/access'; -const getFileByType = (files: AssetFileEntity[] | undefined, type: AssetFileType) => { - return (files || []).find((file) => file.type === type); +export const getAssetFile = (files: AssetFileEntity[], type: AssetFileType | GeneratedImageType) => { + return files.find((file) => file.type === type); }; -export const getAssetFiles = (files?: AssetFileEntity[]) => ({ - fullsizeFile: getFileByType(files, AssetFileType.FULLSIZE), - previewFile: getFileByType(files, AssetFileType.PREVIEW), - thumbnailFile: getFileByType(files, AssetFileType.THUMBNAIL), +export const getAssetFiles = (files: AssetFileEntity[]) => ({ + fullsizeFile: getAssetFile(files, AssetFileType.FULLSIZE), + previewFile: getAssetFile(files, AssetFileType.PREVIEW), + thumbnailFile: getAssetFile(files, AssetFileType.THUMBNAIL), }); export const addAssets = async (