|
| 1 | +// Copyright (c) Microsoft Corporation. All Rights Reserved. See License.txt in the project root for license information. |
| 2 | + |
| 3 | +namespace Microsoft.VisualStudio.FSharp.Editor |
| 4 | + |
| 5 | +open System.Composition |
| 6 | + |
| 7 | +open Microsoft.CodeAnalysis.Text |
| 8 | +open Microsoft.CodeAnalysis.CodeFixes |
| 9 | + |
| 10 | +[<ExportCodeFixProvider(FSharpConstants.FSharpLanguageName, Name = "ConvertCSharpLambdaToFSharpLambda"); Shared>] |
| 11 | +type internal FSharpConvertCSharpLambdaToFSharpLambdaCodeFixProvider |
| 12 | + [<ImportingConstructor>] |
| 13 | + ( |
| 14 | + checkerProvider: FSharpCheckerProvider, |
| 15 | + projectInfoManager: FSharpProjectOptionsManager |
| 16 | + ) = |
| 17 | + inherit CodeFixProvider() |
| 18 | + |
| 19 | + static let userOpName = "ConvertCSharpLambdaToFSharpLambda" |
| 20 | + let fixableDiagnosticIds = set ["FS0039"; "FS0043"] |
| 21 | + |
| 22 | + override _.FixableDiagnosticIds = Seq.toImmutableArray fixableDiagnosticIds |
| 23 | + |
| 24 | + override _.RegisterCodeFixesAsync context = |
| 25 | + asyncMaybe { |
| 26 | + let! sourceText = context.Document.GetTextAsync(context.CancellationToken) |
| 27 | + let! parsingOptions, _ = projectInfoManager.TryGetOptionsForEditingDocumentOrProject(context.Document, context.CancellationToken, userOpName) |
| 28 | + let! parseResults = checkerProvider.Checker.ParseFile(context.Document.FilePath, sourceText.ToFSharpSourceText(), parsingOptions, userOpName) |> liftAsync |
| 29 | + |
| 30 | + let errorRange = RoslynHelpers.TextSpanToFSharpRange(context.Document.FilePath, context.Span, sourceText) |
| 31 | + |
| 32 | + let! fullParenRange, lambdaArgRange, lambdaBodyRange = parseResults.TryRangeOfParenEnclosingOpEqualsGreaterUsage errorRange.Start |
| 33 | + |
| 34 | + let! fullParenSpan = RoslynHelpers.TryFSharpRangeToTextSpan(sourceText, fullParenRange) |
| 35 | + let! lambdaArgSpan = RoslynHelpers.TryFSharpRangeToTextSpan(sourceText, lambdaArgRange) |
| 36 | + let! lambdaBodySpan = RoslynHelpers.TryFSharpRangeToTextSpan(sourceText, lambdaBodyRange) |
| 37 | + |
| 38 | + let replacement = |
| 39 | + let argText = sourceText.GetSubText(lambdaArgSpan).ToString() |
| 40 | + let bodyText = sourceText.GetSubText(lambdaBodySpan).ToString() |
| 41 | + TextChange(fullParenSpan, "fun " + argText + " -> " + bodyText) |
| 42 | + |
| 43 | + let diagnostics = |
| 44 | + context.Diagnostics |
| 45 | + |> Seq.filter (fun x -> fixableDiagnosticIds |> Set.contains x.Id) |
| 46 | + |> Seq.toImmutableArray |
| 47 | + |
| 48 | + let title = SR.UseFSharpLambda() |
| 49 | + |
| 50 | + let codeFix = |
| 51 | + CodeFixHelpers.createTextChangeCodeFix( |
| 52 | + title, |
| 53 | + context, |
| 54 | + (fun () -> asyncMaybe.Return [| replacement |])) |
| 55 | + |
| 56 | + context.RegisterCodeFix(codeFix, diagnostics) |
| 57 | + } |
| 58 | + |> Async.Ignore |
| 59 | + |> RoslynHelpers.StartAsyncUnitAsTask(context.CancellationToken) |
0 commit comments