Avoid eagerly warming csharpier, doesn't seem to be a good way to del…

…ay that and it is possibly slowing down VS startup. references #1182
belav · Dec 21, 2024 · c7e1545 · c7e1545
1 parent 5cf6488
commit c7e1545
Show file tree

Hide file tree

Showing 4 changed files with 25 additions and 22 deletions.
diff --git a/Src/CSharpier.VisualStudio/CSharpier.VisualStudioShared/CSharpierPackage.cs b/Src/CSharpier.VisualStudio/CSharpier.VisualStudioShared/CSharpierPackage.cs
@@ -35,23 +35,6 @@ IProgress<ServiceProgressData> progress
             await ReformatWithCSharpier.InitializeAsync(this);
             await InstallerService.InitializeAsync(this);
 
-            try
-            {
-#pragma warning disable VSSDK006
-                var dte = await this.GetServiceAsync(typeof(DTE)) as DTE;
-#pragma warning restore
-                if (dte?.ActiveDocument != null)
-                {
-                    CSharpierProcessProvider
-                        .GetInstance(this)
-                        .FindAndWarmProcess(dte.ActiveDocument.FullName);
-                }
-            }
-            catch (Exception ex)
-            {
-                Logger.Instance.Error(ex);
-            }
-
             SolutionEvents.OnAfterOpenSolution += this.HandleOpenSolution;
         }
 

diff --git a/Src/CSharpier.VisualStudio/CSharpier.VisualStudioShared/CSharpierProcessProvider.cs b/Src/CSharpier.VisualStudio/CSharpier.VisualStudioShared/CSharpierProcessProvider.cs
@@ -69,6 +69,12 @@ public void FindAndWarmProcess(string filePath)
             this.warmingByDirectory.Remove(directory);
         }
 
+        public bool HasWarmedProcessFor(string filePath)
+        {
+            var directory = new FileInfo(filePath).DirectoryName;
+            return this.csharpierVersionByDirectory.TryGetValue(directory, out _);
+        }
+
         public ICSharpierProcess GetProcessFor(string filePath)
         {
             var directory = new FileInfo(filePath).DirectoryName;

diff --git a/Src/CSharpier.VisualStudio/CSharpier.VisualStudioShared/ReformatWithCSharpier.cs b/Src/CSharpier.VisualStudio/CSharpier.VisualStudioShared/ReformatWithCSharpier.cs
@@ -14,13 +14,12 @@ internal sealed class ReformatWithCSharpier
 
         private readonly DTE dte;
         private readonly FormattingService formattingService;
+        private readonly CSharpierProcessProvider cSharpierProcessProvider;
 
         public static ReformatWithCSharpier Instance { get; private set; } = default!;
 
         public static async Task InitializeAsync(CSharpierPackage package)
         {
-            await ThreadHelper.JoinableTaskFactory.SwitchToMainThreadAsync(package.DisposalToken);
-
             var commandService =
                 await package.GetServiceAsync(typeof(IMenuCommandService)) as OleMenuCommandService;
 
@@ -42,17 +41,30 @@ DTE dte
             menuItem.BeforeQueryStatus += this.QueryStatus;
             commandService.AddCommand(menuItem);
             this.formattingService = FormattingService.GetInstance(package);
+            this.cSharpierProcessProvider = CSharpierProcessProvider.GetInstance(package);
         }
 
         private void QueryStatus(object sender, EventArgs e)
         {
             ThreadHelper.ThrowIfNotOnUIThread();
             var button = (OleMenuCommand)sender;
 
-            button.Visible = this.dte.ActiveDocument.Name.EndsWith(".cs");
-            button.Enabled = this.formattingService.ProcessSupportsFormatting(
-                this.dte.ActiveDocument
+            var hasWarmedProcess = this.cSharpierProcessProvider.HasWarmedProcessFor(
+                this.dte.ActiveDocument.FullName
             );
+            button.Visible = this.dte.ActiveDocument.Name.EndsWith(".cs");
+
+            if (!hasWarmedProcess)
+            {
+                // default to assuming they can format, that way if they do format we can start everything up properly
+                button.Enabled = true;
+            }
+            else
+            {
+                button.Enabled = this.formattingService.ProcessSupportsFormatting(
+                    this.dte.ActiveDocument
+                );
+            }
         }
 
         private void Execute(object sender, EventArgs e)

diff --git a/Src/CSharpier.VisualStudio/CSharpier.VisualStudioShared/ReformatWithCSharpierOnSave.cs b/Src/CSharpier.VisualStudio/CSharpier.VisualStudioShared/ReformatWithCSharpierOnSave.cs
@@ -76,6 +76,8 @@ public int OnBeforeDocumentWindowShow(uint docCookie, int fFirstShow, IVsWindowF
                 var documentInfo = this.runningDocumentTable.GetDocumentInfo(docCookie);
                 var documentPath = documentInfo.Moniker;
 
+                Logger.Instance.Debug("Trying to find - " + documentPath);
+
                 return this.dte.ActiveDocument?.FullName == documentPath
                     ? this.dte.ActiveDocument
                     : this.dte.Documents?.Item(documentPath);