Inherit from FSharpContextActionBase.

JetBrains · Jul 28, 2022 · c8c71c7 · c8c71c7
1 parent cf3210c
commit c8c71c7
Showing 1 changed file with 27 additions and 14 deletions.
diff --git a/ReSharper.FSharp/src/FSharp.Psi.Intentions/src/Intentions/GenerateSignatureFileAction.fs b/ReSharper.FSharp/src/FSharp.Psi.Intentions/src/Intentions/GenerateSignatureFileAction.fs
@@ -1,22 +1,33 @@
 namespace JetBrains.ReSharper.Plugins.FSharp.Psi.Features.ContextActions
 
+open System
 open System.IO
-open JetBrains.Application.UI.ActionsRevised.Menu
+open JetBrains.ReSharper.Feature.Services.ContextActions
 open JetBrains.ReSharper.Plugins.FSharp.Psi.Features.Intentions
+open type JetBrains.ReSharper.Psi.PsiSourceFileExtensions
 
+[<ContextAction(Group = "F#", Name = "Generate signature file for current file", Priority = 1s,
+                Description = "Generate signature file for current file.")>]
 type GenerateSignatureFileAction(dataProvider: FSharpContextActionDataProvider) =
-    interface IExecutableAction with
-        member this.Update(context, presentation, nextUpdate) =
-            let sourceFile = dataProvider.SourceFile.Name
-            let fsiFile = Path.ChangeExtension(sourceFile, ".fsi")
-            not (File.Exists fsiFile)
-
-        member this.Execute(context, nextExecute) =
-            let sourceFile = dataProvider.SourceFile.Name
-            let fsiFile = Path.ChangeExtension(sourceFile, ".fsi")
-            let currentFSharpFile = dataProvider.PsiFile
-            let fcsService = currentFSharpFile.FcsCheckerService
-            let checkResult = fcsService.ParseAndCheckFile(currentFSharpFile.GetSourceFile(), "for signature file", true)
+    inherit FSharpContextActionBase(dataProvider)
+
+    override this.Text = "Generate signature file for current file"
+
+    override this.IsAvailable _ =
+        let currentFSharpFile = dataProvider.PsiFile
+        let fcsService = currentFSharpFile.FcsCheckerService
+        let hasSignature = fcsService.FcsProjectProvider.HasPairFile dataProvider.SourceFile
+        not hasSignature
+
+    override this.ExecutePsiTransaction(solution, _) =
+        let project = dataProvider.SourceFile.GetProject()
+
+        let physicalPath = dataProvider.SourceFile.ToProjectFile().Location.FileAccessPath
+        let fsiFile = Path.ChangeExtension(physicalPath, ".fsi")        
+        let currentFSharpFile = dataProvider.PsiFile
+        let fcsService = currentFSharpFile.FcsCheckerService
+        let checkResult = fcsService.ParseAndCheckFile(currentFSharpFile.GetSourceFile(), "for signature file", true)
+        do
             match checkResult with
             | None -> ()
             | Some { CheckResults = checkResult } ->
@@ -25,4 +36,6 @@ type GenerateSignatureFileAction(dataProvider: FSharpContextActionDataProvider)
             | None -> ()
             | Some signatureSourceText ->
                 let content = string signatureSourceText
-                File.WriteAllText(fsiFile, content)
+                File.WriteAllText(fsiFile, content)
+
+        Action<_>(ignore)