@tmlmobilidade/interfaces 20250906.141.55 → 20250908.1551.9

package/dist/src/interfaces/files/files.d.ts

@@ -48,7 +48,11 @@ declare class FilesClass extends MongoCollectionClass<File, CreateFileDto, Updat
      * @param createFileDto - The file type to create.
      * @returns The file that was uploaded.
      */
-    upload(file: Buffer, createFileDto: CreateFileDto, options?: InsertOneOptions): Promise<File>;
+    upload(file: Buffer, createFileDto: CreateFileDto & {
+        _id?: string;
+    }, options?: InsertOneOptions & {
+        override?: boolean;
+    }): Promise<File>;
     protected getCollectionIndexes(): IndexDescription[];
     protected getCollectionName(): string;
     protected getEnvName(): string;

package/dist/src/interfaces/files/files.js

@@ -156,9 +156,51 @@ class FilesClass extends MongoCollectionClass {
      * @returns The file that was uploaded.
      */
     async upload(file, createFileDto, options) {
-        const _id = generateRandomString({ length: 5 });
-        await this.storageService.uploadFile(`${createFileDto.scope}/${createFileDto.resource_id}/${_id}.${Files.getFileExtension(createFileDto.name)}`, file, Files.getMimeTypeFromFileExtension(createFileDto.name));
-        return await this.insertOne({ ...createFileDto, _id }, { options });
+        //
+        //
+        // A. Define variables
+        if (createFileDto._id && !options?.override) {
+            throw new HttpException(HttpStatus.INTERNAL_SERVER_ERROR, 'When File ID is provided, override must be true');
+        }
+        const fileId = createFileDto._id || generateRandomString({ length: 5 });
+        const fileExtension = Files.getFileExtension(createFileDto.name);
+        const mimeType = Files.getMimeTypeFromFileExtension(createFileDto.name);
+        const filePath = `${createFileDto.scope}/${createFileDto.resource_id}/${fileId}.${fileExtension}`;
+        //
+        // C. Handle database transaction
+        const session = this.getMongoConnector().client.startSession();
+        let result;
+        try {
+            session.startTransaction();
+            //
+            // C.1. Handle file override if specified
+            if (options?.override) {
+                const existingFile = await this.findOne({ _id: fileId });
+                if (existingFile) {
+                    const existingFileExtension = Files.getFileExtension(existingFile.name);
+                    const existingFilePath = `${existingFile.scope}/${existingFile.resource_id}/${existingFile._id}.${existingFileExtension}`;
+                    if (existingFilePath !== filePath) {
+                        throw new HttpException(HttpStatus.INTERNAL_SERVER_ERROR, 'File ID is provided, but the file path is different from the existing file', { cause: { existingFilePath, filePath } });
+                    }
+                    await super.deleteById(fileId);
+                }
+            }
+            //
+            // C.2. Upload file to storage
+            await this.storageService.uploadFile(filePath, file, mimeType);
+            //
+            // C.3. Insert file record
+            result = await this.insertOne({ ...createFileDto, _id: fileId }, { options });
+            await session.commitTransaction();
+        }
+        catch (error) {
+            await session.abortTransaction();
+            throw error;
+        }
+        finally {
+            session.endSession();
+        }
+        return result;
     }
     getCollectionIndexes() {
         return [

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@tmlmobilidade/interfaces",
-	"version": "20250906.141.55",
+	"version": "20250908.1551.9",
 	"author": "João de Vasconcelos & Jusi Monteiro",
 	"license": "AGPL-3.0-or-later",
 	"homepage": "https://github.com/tmlmobilidade/services#readme",