Add code cleanup support for open documents in VS

Added RunCodeCleanupAsync to service interfaces, client, and server to enable code cleanup on open documents via DTE command. Introduced the document_cleanup server tool in DocumentTools for external access, with documentation and attributes for discoverability.

Author: Ebersbach, Stephan

src/CodingWithCalvin.MCPServer.Server/RpcClient.cs

@@ -112,6 +112,7 @@ public Task ShutdownAsync()
     public Task<bool> OpenDocumentAsync(string path) => Proxy.OpenDocumentAsync(path);
     public Task<bool> CloseDocumentAsync(string path, bool save) => Proxy.CloseDocumentAsync(path, save);
     public Task<bool> SaveDocumentAsync(string path) => Proxy.SaveDocumentAsync(path);
+    public Task<bool> RunCodeCleanupAsync(string path) => Proxy.RunCodeCleanupAsync(path);
     public Task<string?> ReadDocumentAsync(string path) => Proxy.ReadDocumentAsync(path);
     public Task<bool> WriteDocumentAsync(string path, string content) => Proxy.WriteDocumentAsync(path, content);
     public Task<SelectionInfo?> GetSelectionAsync() => Proxy.GetSelectionAsync();

src/CodingWithCalvin.MCPServer.Server/Tools/DocumentTools.cs

@@ -72,6 +72,15 @@ public async Task<string> SaveDocumentAsync(
         return success ? $"Saved: {path}" : $"Document not found or failed to save: {path}";
     }
 
+    [McpServerTool(Name = "document_cleanup", Destructive = true, Idempotent = true)]
+    [Description("Run code cleanup on an open document in Visual Studio. The document must already be open in VS. Activates the document and executes the 'Edit.RunCodeCleanupOnFile' command.")]
+    public async Task<string> RunCodeCleanupAsync(
+        [Description("The full absolute path to the document. Must be open in VS. Get the path from document_list. Supports forward slashes (/) or backslashes (\\).")] string path)
+    {
+        var success = await _rpcClient.RunCodeCleanupAsync(path);
+        return success ? $"Code cleanup completed: {path}" : $"Document not found or code cleanup failed: {path}";
+    }
+
     [McpServerTool(Name = "document_read", ReadOnly = true)]
     [Description("Read the contents of a document. If the document is open in VS, reads the current editor buffer (including unsaved changes); otherwise reads from disk.")]
     public async Task<string> ReadDocumentAsync(

src/CodingWithCalvin.MCPServer.Shared/RpcContracts.cs

@@ -20,6 +20,7 @@ public interface IVisualStudioRpc
     Task<bool> OpenDocumentAsync(string path);
     Task<bool> CloseDocumentAsync(string path, bool save);
     Task<bool> SaveDocumentAsync(string path);
+    Task<bool> RunCodeCleanupAsync(string path);
     Task<string?> ReadDocumentAsync(string path);
     Task<bool> WriteDocumentAsync(string path, string content);
     Task<SelectionInfo?> GetSelectionAsync();

src/CodingWithCalvin.MCPServer/Services/IVisualStudioService.cs

@@ -16,6 +16,7 @@ public interface IVisualStudioService
     Task<bool> OpenDocumentAsync(string path);
     Task<bool> CloseDocumentAsync(string path, bool save = true);
     Task<bool> SaveDocumentAsync(string path);
+    Task<bool> RunCodeCleanupAsync(string path);
     Task<string?> ReadDocumentAsync(string path);
     Task<bool> WriteDocumentAsync(string path, string content);
     Task<SelectionInfo?> GetSelectionAsync();

src/CodingWithCalvin.MCPServer/Services/RpcServer.cs

@@ -170,6 +170,7 @@ public async Task RequestShutdownAsync()
     public Task<bool> OpenDocumentAsync(string path) => _vsService.OpenDocumentAsync(path);
     public Task<bool> CloseDocumentAsync(string path, bool save) => _vsService.CloseDocumentAsync(path, save);
     public Task<bool> SaveDocumentAsync(string path) => _vsService.SaveDocumentAsync(path);
+    public Task<bool> RunCodeCleanupAsync(string path) => _vsService.RunCodeCleanupAsync(path);
     public Task<string?> ReadDocumentAsync(string path) => _vsService.ReadDocumentAsync(path);
     public Task<bool> WriteDocumentAsync(string path, string content) => _vsService.WriteDocumentAsync(path, content);
     public Task<SelectionInfo?> GetSelectionAsync() => _vsService.GetSelectionAsync();

src/CodingWithCalvin.MCPServer/Services/VisualStudioService.cs

@@ -260,6 +260,31 @@ public async Task<bool> SaveDocumentAsync(string path)
         return false;
     }
 
+    public async Task<bool> RunCodeCleanupAsync(string path)
+    {
+        await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();
+        var dte = await GetDteAsync();
+
+        foreach (Document doc in dte.Documents)
+        {
+            try
+            {
+                if (PathsEqual(doc.FullName, path))
+                {
+                    doc.Activate();
+                    dte.ExecuteCommand("EditorContextMenus.FileHealthIndicator.RunDefaultCodeCleanup");
+                    return true;
+                }
+            }
+            catch (Exception ex)
+            {
+                VsixTelemetry.TrackException(ex);
+            }
+        }
+
+        return false;
+    }
+
     public async Task<string?> ReadDocumentAsync(string path)
     {
         await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync();