From 0de92f5c5b3bc3a038d265b37c388c3398d437d7 Mon Sep 17 00:00:00 2001
From: sarahCx <sarah.chen@checkmarx.com>
Date: Mon, 30 Sep 2024 15:24:39 +0300
Subject: [PATCH] get current path

---
 .../CxExtension/Toolbar/CxToolbar.cs          | 23 +++++++++++++++++--
 1 file changed, 21 insertions(+), 2 deletions(-)

diff --git a/ast-visual-studio-extension/CxExtension/Toolbar/CxToolbar.cs b/ast-visual-studio-extension/CxExtension/Toolbar/CxToolbar.cs
index eee1495..70c6814 100644
--- a/ast-visual-studio-extension/CxExtension/Toolbar/CxToolbar.cs
+++ b/ast-visual-studio-extension/CxExtension/Toolbar/CxToolbar.cs
@@ -16,6 +16,8 @@
 using Microsoft.VisualStudio.Imaging;
 using ast_visual_studio_extension.CxWrapper.Exceptions;
 using System.Linq;
+using EnvDTE;
+using EnvDTE80;
 
 namespace ast_visual_studio_extension.CxExtension.Toolbar
 {
@@ -361,9 +363,11 @@ private async Task StartScanAsync()
             var scanId = SettingsUtils.GetToolbarValue(Package, SettingsUtils.createdScanIdProperty);
             if (cxWrapper == null || !string.IsNullOrWhiteSpace(scanId)) return;
 
+            string currentPath = await GetCurrentWorkingDirAsync();
+
             Dictionary<string, string> parameters = new Dictionary<string, string>
             {
-                { CxCLI.CxConstants.FLAG_SOURCE, "." },
+                { CxCLI.CxConstants.FLAG_SOURCE, currentPath },
                 { CxCLI.CxConstants.FLAG_PROJECT_NAME, ProjectsCombo.Text },
                 { CxCLI.CxConstants.FLAG_BRANCH, BranchesCombo.Text },
                 { CxCLI.CxConstants.FLAG_AGENT, CxCLI.CxConstants.EXTENSION_AGENT }
@@ -384,7 +388,22 @@ private async Task StartScanAsync()
             }
         }
 
-        private async Task PollScanStartedAsync()
+        private static async Task<string> GetCurrentWorkingDirAsync()
+        {
+            await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();
+
+            DTE2 dte = (DTE2)ServiceProvider.GlobalProvider.GetService(typeof(DTE));
+
+            var solutionExplorer = dte.ToolWindows.SolutionExplorer;
+
+            if ((solutionExplorer.DTE.ActiveSolutionProjects as Array)?.Length > 0)
+            {
+                return System.IO.Path.GetDirectoryName(dte.Solution.FullName);
+            }
+            return ".";
+        }
+
+            private async Task PollScanStartedAsync()
         {
             ScanStartButton.IsEnabled = false;
             var tsc = Microsoft.VisualStudio.Shell.Package.GetGlobalService(typeof(SVsTaskStatusCenterService)) as IVsTaskStatusCenterService;