From 1df7382b4cff477aa607f4fec7afb560e50e105f Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Micka=C3=ABl=20VILLERS?=
 <mvillers.externe@bedrockstreaming.com>
Date: Thu, 3 Nov 2022 11:31:24 +0100
Subject: [PATCH] feat(input): allow to disable annotations

---
 README.md   | 1 +
 action.yml  | 5 +++++
 src/main.js | 6 ++++--
 3 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index b1d740e..2d728b0 100644
--- a/README.md
+++ b/README.md
@@ -74,6 +74,7 @@ And ensure that you're using the <a href="https://github.com/Checkmarx/kics-gith
 | Variable                                  | Example Value &nbsp;                                   | Description &nbsp;                                                                                                                                          | Type    | Required | Default                                                |
 |-------------------------------------------|--------------------------------------------------------|-------------------------------------------------------------------------------------------------------------------------------------------------------------|---------| -------- |--------------------------------------------------------|
 | enable_comment                            | true                                                   | Enable pull request report comments                                                                                                                         | Boolean | No       | false                                                  |
+| disable_annotations                       | true                                                   | Disable annotations report                                                                                                                                  | Boolean | No       | false                                                  |
 | comments_with_queries                     | true                                                   | Add queries in th pull request report comments (available when enable_comments = true)                                                                      | Boolean | No       | false                                                  |
 | excluded_column_for_comments_with_queries | description_id,similarity_id,search_line,search_value  | Excluded columns for the comment with queries, accepts a comma separated list                                                                               | String  | No       | description_id,similarity_id,search_line,search_value  |
 | path                                      | terraform/main.tf,Dockerfile                           | paths to a file or directories to scan, comma separated list                                                                                                | String  | Yes      | N/A                                                    |
diff --git a/action.yml b/action.yml
index 833bff5..b5f366f 100644
--- a/action.yml
+++ b/action.yml
@@ -6,6 +6,10 @@ inputs:
     description: "The GITHUB_TOKEN for the current workflow run"
     required: false
     default: ${{github.token}}
+  disable_annotations:
+    required: false
+    default: "false"
+    description: "Disable annotations report"
   enable_comments:
     required: false
     default: "false"
@@ -105,6 +109,7 @@ runs:
   env:
     INPUT_TOKEN: ${{ inputs.token }}
     INPUT_OUTPUT_PATH: ${{ inputs.output_path }}
+    INPUT_DISABLE_ANNOTATIONS: ${{ inputs.disable_annotations }}
     INPUT_ENABLE_COMMENTS: ${{ inputs.enable_comments }}
     INPUT_COMMENTS_WITH_QUERIES: ${{ inputs.comments_with_queries }}
     INPUT_EXCLUDED_COLUMNS_FOR_COMMENTS_WITH_QUERIES: ${{ inputs.excluded_column_for_comments_with_queries }}
diff --git a/src/main.js b/src/main.js
index deb2bd7..bf362b9 100644
--- a/src/main.js
+++ b/src/main.js
@@ -47,6 +47,7 @@ async function main() {
 
     // Get ENV variables
     const githubToken = process.env.INPUT_TOKEN;
+    const disableAnnotations = process.env.INPUT_DISABLE_ANNOTATIONS;
     const enableComments = process.env.INPUT_ENABLE_COMMENTS;
     const commentsWithQueries = process.env.INPUT_COMMENTS_WITH_QUERIES;
     const excludedColumnsForCommentsWithQueries = process.env.INPUT_EXCLUDED_COLUMNS_FOR_COMMENTS_WITH_QUERIES.split(',');
@@ -71,12 +72,13 @@ async function main() {
         }
 
         const parsedResults = readJSON(outputPath.resultsJSONFile);
+        if (disableAnnotations.toLocaleLowerCase() === "false") {
+            annotator.annotateChangesWithResults(parsedResults);
+        }
         if (enableComments.toLocaleLowerCase() === "true") {
             await commenter.postPRComment(parsedResults, repo, prNumber, octokit, commentsWithQueries.toLocaleLowerCase() === "true", excludedColumnsForCommentsWithQueries);
         }
 
-        annotator.annotateChangesWithResults(parsedResults);
-
         setWorkflowStatus(exitCode);
         cleanupOutput(outputPath.resultsJSONFile, outputFormats);
     } catch (e) {