Fixed compilation issue

Author: cx-rahul-pidde

ast-visual-studio-extension/CxExtension/CxAssist/Core/CopilotIntegration.cs

@@ -377,64 +377,45 @@ private static IVsWindowFrame TryCaptureAssistDocumentFrame()
         private static void ShowCopilotNotAgentModeUserMessage(IVsWindowFrame assistDocumentFrame)
         {
             ThreadHelper.ThrowIfNotOnUIThread();
-            AssistDocumentInfoBar.TryShowWarning(
-                assistDocumentFrame,
+            ShowAssistNotification(
                 CxAssistConstants.CopilotNotAgentModeInfoBarMessage,
-                () => ShowAssistNotification(
-                    CxAssistConstants.CopilotNotAgentModeInfoBarMessage,
-                    isError: false,
-                    useWarningSeverity: true));
+                isError: false,
+                useWarningSeverity: true);
         }
 
         /// <summary>
         /// Shows Copilot not installed warning in the info bar.
         /// </summary>
         private static void ShowCopilotNotInstalledMessage(IVsWindowFrame assistDocumentFrame)
         {
-            if (assistDocumentFrame == null) return;
             ThreadHelper.ThrowIfNotOnUIThread();
-            AssistDocumentInfoBar.TryShowWarning(
-                assistDocumentFrame,
+            ShowAssistNotification(
                 CxAssistConstants.CopilotNotInstalledInfoBarMessage,
-                () => ShowAssistNotification(
-                    CxAssistConstants.CopilotNotInstalledInfoBarMessage,
-                    isError: false,
-                    useWarningSeverity: true));
+                isError: false,
+                useWarningSeverity: true);
         }
 
         /// <summary>
         /// Shows Copilot Chat failed to open warning in the info bar.
         /// </summary>
         private static void ShowCopilotChatOpenFailedMessage(IVsWindowFrame assistDocumentFrame)
         {
-            if (assistDocumentFrame == null) return;
             ThreadHelper.ThrowIfNotOnUIThread();
-            AssistDocumentInfoBar.TryShowWarning(
-                assistDocumentFrame,
+            ShowAssistNotification(
                 CxAssistConstants.CopilotChatOpenFailedInfoBarMessage,
-                () => ShowAssistNotification(
-                    CxAssistConstants.CopilotChatOpenFailedInfoBarMessage,
-                    isError: false,
-                    useWarningSeverity: true));
+                isError: false,
+                useWarningSeverity: true);
         }
 
         /// <summary>
         /// Shows Copilot prompt preparation failed error in the info bar (as warning with error fallback).
         /// </summary>
         private static void ShowCopilotPromptPrepareFailedMessage(IVsWindowFrame assistDocumentFrame)
         {
-            if (assistDocumentFrame == null)
-            {
-                ShowAssistNotification(CxAssistConstants.CopilotPromptPrepareFailedInfoBarMessage, isError: true);
-                return;
-            }
             ThreadHelper.ThrowIfNotOnUIThread();
-            AssistDocumentInfoBar.TryShowWarning(
-                assistDocumentFrame,
+            ShowAssistNotification(
                 CxAssistConstants.CopilotPromptPrepareFailedInfoBarMessage,
-                () => ShowAssistNotification(
-                    CxAssistConstants.CopilotPromptPrepareFailedInfoBarMessage,
-                    isError: true));
+                isError: true);
         }
 
         /// <summary>
@@ -443,15 +424,11 @@ private static void ShowCopilotPromptPrepareFailedMessage(IVsWindowFrame assistD
         /// </summary>
         private static void ShowCopilotPasteOnlyVs2026Message(IVsWindowFrame assistDocumentFrame)
         {
-            if (assistDocumentFrame == null) return;
             ThreadHelper.ThrowIfNotOnUIThread();
-            AssistDocumentInfoBar.TryShowWarning(
-                assistDocumentFrame,
+            ShowAssistNotification(
                 CxAssistConstants.CopilotPasteOnlyVs2026InfoBarMessage,
-                () => ShowAssistNotification(
-                    CxAssistConstants.CopilotPasteOnlyVs2026InfoBarMessage,
-                    isError: false,
-                    useWarningSeverity: true));
+                isError: false,
+                useWarningSeverity: true);
         }
 
         private static void ScheduleAutomatedPromptEntry(string prompt, IVsWindowFrame assistDocumentFrame)

ast-visual-studio-extension/CxExtension/CxAssist/Core/CxAssistDisplayCoordinator.cs

@@ -357,5 +357,39 @@ public static void RefreshProblemWindow(
                 findingsControl.SetAllFileNodes(fileNodes);
             }, "Coordinator.RefreshProblemWindow");
         }
+
+        /// <summary>
+        /// Clears all findings from the coordinator.
+        /// Used when user logs out or disables all scanners.
+        /// </summary>
+        public static void ClearAllFindings()
+        {
+            lock (_lock)
+            {
+                _fileToIssues.Clear();
+            }
+            IssuesUpdated?.Invoke(new Dictionary<string, List<Vulnerability>>());
+        }
+
+        /// <summary>
+        /// Clears findings from disabled scanners only.
+        /// Called when user toggles a scanner off via preferences.
+        /// </summary>
+        public static void ClearFindingsFromDisabledScanners()
+        {
+            lock (_lock)
+            {
+                foreach (var filePath in _fileToIssues.Keys.ToList())
+                {
+                    _fileToIssues[filePath] = _fileToIssues[filePath]
+                        .Where(v => v != null && v.Scanner != 0)
+                        .ToList();
+
+                    if (_fileToIssues[filePath].Count == 0)
+                        _fileToIssues.Remove(filePath);
+                }
+            }
+            IssuesUpdated?.Invoke(GetAllIssuesByFile());
+        }
     }
 }

ast-visual-studio-extension/ast-visual-studio-extension.csproj

@@ -121,7 +121,6 @@
     <Compile Include="CxExtension\Toolbar\ProjectsCombobox.cs" />
     <Compile Include="CxExtension\Toolbar\ScansCombobox.cs" />
     <Compile Include="CxExtension\Toolbar\CxToolbar.cs" />
-    <Compile Include="CxExtension\CxAssist\Core\EditorBufferResolver.cs" />
     <Compile Include="CxExtension\Utils\OutputPaneUtils.cs" />
     <Compile Include="CxExtension\Utils\OutputPaneWriter.cs" />
     <Compile Include="CxExtension\Utils\RepositoryInformation.cs" />
@@ -207,7 +206,6 @@
     <Compile Include="CxExtension\CxAssist\Core\AssistIconLoader.cs" />
     <Compile Include="CxExtension\CxAssist\Core\CxAssistOutputPane.cs" />
     <Compile Include="CxExtension\CxAssist\Core\CxAssistErrorHandler.cs" />
-    <Compile Include="CxExtension\CxAssist\Core\AssistDocumentInfoBar.cs" />
     <Compile Include="CxExtension\CxAssist\Core\CopilotIntegration.cs" />
     <Compile Include="CxExtension\CxAssist\Core\CxAssistCopilotActions.cs" />
     <Compile Include="CxExtension\CxAssist\Core\Prompts\CxOneAssistFixPrompts.cs" />