From 628ba5cdfca4f7ac15a99fb70063cb023ac561f8 Mon Sep 17 00:00:00 2001
From: Chanan Merari <cmerari@paloaltonetworks.com>
Date: Sun, 8 Sep 2024 13:56:56 +0300
Subject: [PATCH] [PCSUP-24582] QA Testing fixes

---
 src/services/filesService.ts                       | 14 ++++++++++++--
 src/utils/index.ts                                 |  6 ++++++
 .../checkovResult/messages/focusString.ts          |  7 ++++---
 src/views/interface/checkovResult/webviewPanel.ts  |  2 +-
 .../dataProviders/resultTreeDataProvider.ts        |  6 +++++-
 static/webviews/result/index.html                  |  4 ++--
 6 files changed, 30 insertions(+), 9 deletions(-)

diff --git a/src/services/filesService.ts b/src/services/filesService.ts
index 8eb9196..34a93f2 100644
--- a/src/services/filesService.ts
+++ b/src/services/filesService.ts
@@ -1,4 +1,7 @@
 import * as vscode from 'vscode';
+import logger from '../logger';
+import { CheckovResult } from '../types';
+import { formatWindowsAbsoluteFilePath, isWindows } from '../utils';
 
 export class FilesService {
     private static context: vscode.ExtensionContext;
@@ -7,13 +10,20 @@ export class FilesService {
         FilesService.context = context;
     }
 
-    public static async openFile(file: string, line: number = 1) {
+    public static async openResult(result: CheckovResult, line: number = 1) {
         if (line < 1) {
             line = 1;
         }
         if (!vscode.window.activeTextEditor) {
             vscode.commands.executeCommand('workbench.action.previousEditor');
         }
-        return vscode.window.showTextDocument(vscode.Uri.file(file), { selection: new vscode.Range(line - 1, 0, line - 1, 0) });
+        let filePath;
+        if (isWindows()) {
+            filePath = formatWindowsAbsoluteFilePath(result.file_abs_path);
+        } else {
+            filePath = result.file_abs_path;
+        }
+        logger.info(`Opening file at ${filePath}`);
+        return vscode.window.showTextDocument(vscode.Uri.file(filePath), { selection: new vscode.Range(line - 1, 0, line - 1, 0) });
     }
 }
diff --git a/src/utils/index.ts b/src/utils/index.ts
index d96a991..bf2d5cf 100644
--- a/src/utils/index.ts
+++ b/src/utils/index.ts
@@ -24,6 +24,12 @@ export const formatWindowsFilePath = (path: string): string => {
     return path.replace(':', '').replace(/\\/g, '/');
 };
 
+export const formatWindowsAbsoluteFilePath = (path: string): string => {
+    const splitPath = path.replace(/\\/g, '/').replace(/^\/+/g, '').split('/');
+    splitPath[0] = splitPath[0].toLocaleUpperCase() + ':';
+    return splitPath.join('/');
+};
+
 export const getOsNameAndVersion = async () => {
     const operatingSystem = os.type();
     if ("Darwin" === operatingSystem) {
diff --git a/src/views/interface/checkovResult/messages/focusString.ts b/src/views/interface/checkovResult/messages/focusString.ts
index 4fa0797..abde020 100644
--- a/src/views/interface/checkovResult/messages/focusString.ts
+++ b/src/views/interface/checkovResult/messages/focusString.ts
@@ -1,9 +1,10 @@
 import { FilesService } from '../../../../services';
+import { CheckovResult } from '../../../../types';
 
 export class FocusString {
-    public static async handle({ fileAbsPath, row }: { fileAbsPath: string, row: number }) {
-        if (fileAbsPath && row) {
-            await  FilesService.openFile(fileAbsPath, row);
+    public static async handle({ result, row }: { result: CheckovResult, row: number }) {
+        if (result && row) {
+            await  FilesService.openResult(result, row);
             return;
         }
 
diff --git a/src/views/interface/checkovResult/webviewPanel.ts b/src/views/interface/checkovResult/webviewPanel.ts
index 99169ed..c039a24 100644
--- a/src/views/interface/checkovResult/webviewPanel.ts
+++ b/src/views/interface/checkovResult/webviewPanel.ts
@@ -309,7 +309,7 @@ export class CheckovResultWebviewPanel {
     private static getDataFlowItemString(dataFlow: DataFlow, result: CheckovResult): string {
         const splitPath = dataFlow.path.split('/');
         return `<div class="details">
-                    <a target="_blank" onclick="onSastStepClick('${result.file_abs_path}', ${dataFlow.start.row})">${splitPath[splitPath.length - 1]}: ${dataFlow.start.row}</a><span>${dataFlow.code_block}</span>
+                    <a target="_blank" onclick="onSastStepClick('${result}', ${dataFlow.start.row})">${splitPath[splitPath.length - 1]}: ${dataFlow.start.row}</a><span>${dataFlow.code_block}</span>
                 </div>`;
     }
 
diff --git a/src/views/interface/primarySidebar/dataProviders/resultTreeDataProvider.ts b/src/views/interface/primarySidebar/dataProviders/resultTreeDataProvider.ts
index ca9a12c..bd626b2 100644
--- a/src/views/interface/primarySidebar/dataProviders/resultTreeDataProvider.ts
+++ b/src/views/interface/primarySidebar/dataProviders/resultTreeDataProvider.ts
@@ -12,6 +12,9 @@ export class ResultTreeDataProvider implements vscode.TreeDataProvider<ResultTre
 
   readonly category: CHECKOV_RESULT_CATEGORY;
 
+  private _onDidChangeTreeData: vscode.EventEmitter<ResultTreeItem | void> = new vscode.EventEmitter<ResultTreeItem | void>();
+  readonly onDidChangeTreeData: vscode.Event<ResultTreeItem | void> = this._onDidChangeTreeData.event;
+
   private data: ResultTreeItem[] = [];
   private treeService: TreeService;
 
@@ -27,6 +30,7 @@ export class ResultTreeDataProvider implements vscode.TreeDataProvider<ResultTre
   public refresh() {
     const checkovResults = this.getCheckovResults();
     this.data = this.treeService.getTreeData(this.category, checkovResults);
+    this._onDidChangeTreeData.fire();
   }
 
   public getTreeItem(element: ResultTreeItem): vscode.TreeItem|Thenable<vscode.TreeItem> {
@@ -63,7 +67,7 @@ export class ResultTreeDataProvider implements vscode.TreeDataProvider<ResultTre
         result.description = fetchedDescription;
       }
     }
-    const openedTextEditor = await FilesService.openFile(result.file_abs_path, result.file_line_range[0]);
+    const openedTextEditor = await FilesService.openResult(result, result.file_line_range[0]);
     await CheckovResultWebviewPanel.show(this.category, result, openedTextEditor);
   }
 
diff --git a/static/webviews/result/index.html b/static/webviews/result/index.html
index 80f7f07..0726ba6 100644
--- a/static/webviews/result/index.html
+++ b/static/webviews/result/index.html
@@ -127,8 +127,8 @@
     function onDocumentationClick(url) {
         vscode.postMessage({ type: 'documentationClick', url: url });
     }
-    function onSastStepClick(fileAbsPath, row) {
-        vscode.postMessage({ type: 'sastStepClick', fileAbsPath, row });
+    function onSastStepClick(result, row) {
+        vscode.postMessage({ type: 'sastStepClick', result, row });
     }
   </script>
 </body>