Merge pull request #1316 from bluewave-labs/hp-may-8-update-files-source-enum

Create migration script for reporting

Author: samuel-coutinho

Servers/database/migrations/20250508061148-update-enum_files_source-reporting.js

@@ -0,0 +1,22 @@
+'use strict';
+
+/** @type {import('sequelize-cli').Migration} */
+module.exports = {
+  async up(queryInterface, Sequelize) {
+    const transaction = await queryInterface.sequelize.transaction();
+    try {
+      await queryInterface.sequelize.query(
+        `ALTER TYPE enum_files_source ADD VALUE 'Report';`,
+        { transaction }
+      );
+      await transaction.commit();
+    } catch (error) {
+      await transaction.rollback();
+      throw error;
+    }
+  },
+
+  async down(queryInterface, Sequelize) {
+    // The enum value won't cause any issues if it is not removed
+  }
+};

Servers/models/file.model.ts

@@ -8,7 +8,7 @@ export interface File {
   project_id: number;
   uploaded_by: number;
   uploaded_time: Date;
-  source: "Assessment tracker group" | "Compliance tracker group";
+  source: "Assessment tracker group" | "Compliance tracker group" | "Report";
 }
 
 export interface FileType {
@@ -18,7 +18,7 @@ export interface FileType {
   uploaded_by: number;
   uploaded_time: Date;
   type: string;
-  source: "Assessment tracker group" | "Compliance tracker group";
+  source: "Assessment tracker group" | "Compliance tracker group" | "Report";
 }
 
 @Table({
@@ -60,9 +60,9 @@ export class FileModel extends Model<File> {
   uploaded_time!: Date;
 
   @Column({
-    type: DataType.ENUM("Assessment tracker group", "Compliance tracker group"),
+    type: DataType.ENUM("Assessment tracker group", "Compliance tracker group", "Report"),
   })
-  source!: "Assessment tracker group" | "Compliance tracker group";
+  source!: "Assessment tracker group" | "Compliance tracker group" | "Report";
 
   @Column({
     type: DataType.STRING

Servers/utils/fileUpload.utils.ts

@@ -10,7 +10,7 @@ export const uploadFile = async (
   file: UploadedFile,
   user_id: number,
   project_framework_id: number,
-  source: "Assessment tracker group" | "Compliance tracker group",
+  source: "Assessment tracker group" | "Compliance tracker group" | "Report",
   transaction: Transaction | null = null
 ) => {
   const project = await sequelize.query(