fix bug in syncONeTrustASsessmentToDisk

transcend-io · Jan 24, 2025 · 0ad390c · 0ad390c
1 parent 8b9075c
commit 0ad390c
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 9 deletions.
diff --git a/src/oneTrust/helpers/enrichOneTrustAssessment.ts b/src/oneTrust/helpers/enrichOneTrustAssessment.ts
@@ -45,6 +45,7 @@ export const enrichOneTrustAssessment = ({
         return {
           ...risk,
           ...details,
+          impactLevel: risk.impactLevel ?? 0,
         };
       });
       return {

diff --git a/src/oneTrust/helpers/syncOneTrustAssessmentToDisk.ts b/src/oneTrust/helpers/syncOneTrustAssessmentToDisk.ts
@@ -39,15 +39,30 @@ export const syncOneTrustAssessmentToDisk = ({
   );
 
   if (fileFormat === OneTrustFileFormat.Json) {
-    fs.appendFileSync(
-      file,
-      oneTrustAssessmentToJson({
-        assessment,
-        index,
-        total,
-      }),
-    );
+    if (index === 0) {
+      fs.writeFileSync(
+        file,
+        oneTrustAssessmentToJson({
+          assessment,
+          index,
+          total,
+        }),
+      );
+    } else {
+      fs.appendFileSync(
+        file,
+        oneTrustAssessmentToJson({
+          assessment,
+          index,
+          total,
+        }),
+      );
+    }
   } else if (fileFormat === OneTrustFileFormat.Csv) {
-    fs.appendFileSync(file, oneTrustAssessmentToCsv({ assessment, index }));
+    if (index === 0) {
+      fs.writeFileSync(file, oneTrustAssessmentToCsv({ assessment, index }));
+    } else {
+      fs.appendFileSync(file, oneTrustAssessmentToCsv({ assessment, index }));
+    }
   }
 };