|
1 | | -namespace JetBrains.ReSharper.Plugins.FSharp.Psi.Features.PostfixTemplates |
2 | | - |
3 | | -open JetBrains.ReSharper.Feature.Services.LiveTemplates.Hotspots |
4 | | -open JetBrains.ReSharper.Feature.Services.LiveTemplates.LiveTemplates |
5 | | -open JetBrains.ReSharper.Feature.Services.LiveTemplates.Templates |
6 | | -open JetBrains.ReSharper.Feature.Services.PostfixTemplates |
7 | | -open JetBrains.ReSharper.Feature.Services.PostfixTemplates.Contexts |
8 | | -open JetBrains.ReSharper.Plugins.FSharp.Psi |
9 | | -open JetBrains.ReSharper.Plugins.FSharp.Psi.Impl |
10 | | -open JetBrains.ReSharper.Plugins.FSharp.Psi.Services.Util |
11 | | -open JetBrains.ReSharper.Plugins.FSharp.Psi.Tree |
12 | | -open JetBrains.ReSharper.Psi |
13 | | -open JetBrains.ReSharper.Psi.ExtensionsAPI |
14 | | -open JetBrains.ReSharper.Psi.ExtensionsAPI.Tree |
15 | | -open JetBrains.ReSharper.Psi.Transactions |
16 | | -open JetBrains.ReSharper.Psi.Tree |
17 | | -open JetBrains.ReSharper.Resources.Shell |
18 | | - |
19 | | -[<PostfixTemplate("new", "Construct new record instance", "{ Field = expr }")>] |
20 | | -type NewRecordPostfixTemplate() = |
21 | | - inherit FSharpPostfixTemplateBase() |
22 | | - |
23 | | - override x.CreateBehavior(info) = NewRecordPostfixTemplateBehavior(info) :> _ |
24 | | - override x.CreateInfo(context) = NewRecordPostfixTemplateInfo(context) :> _ |
25 | | - |
26 | | - override this.IsApplicable(node) = |
27 | | - let refExpr = node.As<IReferenceExpr>() |
28 | | - isNotNull refExpr && |
29 | | - |
30 | | - let expr = node.As<IFSharpExpression>() |
31 | | - FSharpPostfixTemplates.canBecomeStatement false expr && |
32 | | - |
33 | | - match refExpr.Qualifier with |
34 | | - | :? IReferenceExpr as refExpr -> |
35 | | - let typeElement = refExpr.Reference.Resolve().DeclaredElement.As<ITypeElement>() |
36 | | - typeElement.IsRecord() |
37 | | - | _ -> false |
38 | | - |
39 | | - override this.IsEnabled _ = true |
40 | | - |
41 | | - |
42 | | -and NewRecordPostfixTemplateInfo(expressionContext: PostfixExpressionContext) = |
43 | | - inherit PostfixTemplateInfo("new", expressionContext) |
44 | | - |
45 | | -and NewRecordPostfixTemplateBehavior(info) = |
46 | | - inherit FSharpPostfixTemplateBehaviorBase(info) |
47 | | - |
48 | | - override this.ExpandPostfix(context) = |
49 | | - let node = context.Expression |
50 | | - let psiServices = node.GetPsiServices() |
51 | | - |
52 | | - psiServices.Transactions.Execute(this.ExpandCommandName, fun _ -> |
53 | | - let factory = node.CreateElementFactory() |
54 | | - use writeCookie = WriteLockCookie.Create(node.IsPhysical()) |
55 | | - use disableFormatter = new DisableCodeFormatter() |
56 | | - let refExpr = this.GetExpression(context) :?> IReferenceExpr |
57 | | - let typeElement = refExpr.Reference.Resolve().DeclaredElement.As<ITypeElement>() |
58 | | - |
59 | | - let recordExpr = factory.CreateExpr("{ }") :?> IRecordExpr |
60 | | - |
61 | | - let recordExpr = ModificationUtil.ReplaceChild(refExpr, recordExpr) |
62 | | - RecordExprUtil.generateBindings typeElement recordExpr |> ignore |
63 | | - recordExpr |
64 | | - ) |
65 | | - |
66 | | - override this.AfterComplete(textControl, node, _) = |
67 | | - let recordExpr = node :?> IRecordExpr |
68 | | - let endOffset = recordExpr.RightBrace.GetDocumentEndOffset() |
69 | | - |
70 | | - let hotspotInfos = |
71 | | - recordExpr.FieldBindings.ToArray() |
72 | | - |> Array.map (fun binding -> |
73 | | - let templateField = TemplateField(binding.ReferenceName.ShortName, SimpleHotspotExpression(null), 0) |
74 | | - HotspotInfo(templateField, binding.Expression.GetDocumentRange(), KeepExistingText = true)) |
75 | | - |
76 | | - let hotspotSession = |
77 | | - LiveTemplatesManager.Instance.CreateHotspotSessionAtopExistingText( |
78 | | - node.GetSolution(), endOffset, textControl, |
79 | | - LiveTemplatesManager.EscapeAction.LeaveTextAndCaret, hotspotInfos) |
80 | | - |
81 | | - hotspotSession.ExecuteAndForget() |
| 1 | +namespace JetBrains.ReSharper.Plugins.FSharp.Psi.Features.PostfixTemplates |
| 2 | + |
| 3 | +open JetBrains.ReSharper.Feature.Services.LiveTemplates.Hotspots |
| 4 | +open JetBrains.ReSharper.Feature.Services.LiveTemplates.LiveTemplates |
| 5 | +open JetBrains.ReSharper.Feature.Services.LiveTemplates.Templates |
| 6 | +open JetBrains.ReSharper.Feature.Services.PostfixTemplates |
| 7 | +open JetBrains.ReSharper.Feature.Services.PostfixTemplates.Contexts |
| 8 | +open JetBrains.ReSharper.Plugins.FSharp.Psi |
| 9 | +open JetBrains.ReSharper.Plugins.FSharp.Psi.Impl |
| 10 | +open JetBrains.ReSharper.Plugins.FSharp.Psi.Services.Util |
| 11 | +open JetBrains.ReSharper.Plugins.FSharp.Psi.Tree |
| 12 | +open JetBrains.ReSharper.Psi |
| 13 | +open JetBrains.ReSharper.Psi.ExtensionsAPI |
| 14 | +open JetBrains.ReSharper.Psi.ExtensionsAPI.Tree |
| 15 | +open JetBrains.ReSharper.Psi.Transactions |
| 16 | +open JetBrains.ReSharper.Psi.Tree |
| 17 | +open JetBrains.ReSharper.Resources.Shell |
| 18 | + |
| 19 | +[<PostfixTemplate("new", "Construct new record instance", "{ Field = expr }")>] |
| 20 | +type NewRecordPostfixTemplate() = |
| 21 | + inherit FSharpPostfixTemplateBase() |
| 22 | + |
| 23 | + override x.CreateBehavior(info) = NewRecordPostfixTemplateBehavior(info) :> _ |
| 24 | + override x.CreateInfo(context) = NewRecordPostfixTemplateInfo(context) :> _ |
| 25 | + |
| 26 | + override this.IsApplicable(node) = |
| 27 | + let refExpr = node.As<IReferenceExpr>() |
| 28 | + isNotNull refExpr && |
| 29 | + |
| 30 | + let expr = node.As<IFSharpExpression>() |
| 31 | + FSharpPostfixTemplates.canBecomeStatement false expr && |
| 32 | + |
| 33 | + match refExpr.Qualifier with |
| 34 | + | :? IReferenceExpr as refExpr -> |
| 35 | + let typeElement = refExpr.Reference.Resolve().DeclaredElement.As<ITypeElement>() |
| 36 | + typeElement.IsRecord() |
| 37 | + | _ -> false |
| 38 | + |
| 39 | + override this.IsEnabled _ = true |
| 40 | + |
| 41 | + |
| 42 | +and NewRecordPostfixTemplateInfo(expressionContext: PostfixExpressionContext) = |
| 43 | + inherit PostfixTemplateInfo("new", expressionContext) |
| 44 | + |
| 45 | +and NewRecordPostfixTemplateBehavior(info) = |
| 46 | + inherit FSharpPostfixTemplateBehaviorBase(info) |
| 47 | + |
| 48 | + override this.ExpandPostfix(context) = |
| 49 | + let node = context.Expression |
| 50 | + let psiServices = node.GetPsiServices() |
| 51 | + |
| 52 | + psiServices.Transactions.Execute(this.ExpandCommandName, fun _ -> |
| 53 | + let factory = node.CreateElementFactory() |
| 54 | + use writeCookie = WriteLockCookie.Create(node.IsPhysical()) |
| 55 | + use disableFormatter = new DisableCodeFormatter() |
| 56 | + let refExpr = this.GetExpression(context) :?> IReferenceExpr |
| 57 | + let typeElement = refExpr.Reference.Resolve().DeclaredElement.As<ITypeElement>() |
| 58 | + |
| 59 | + let recordExpr = factory.CreateExpr("{ }") :?> IRecordExpr |
| 60 | + |
| 61 | + let recordExpr = ModificationUtil.ReplaceChild(refExpr, recordExpr) |
| 62 | + RecordExprUtil.generateBindings typeElement recordExpr |> ignore |
| 63 | + recordExpr |
| 64 | + ) |
| 65 | + |
| 66 | + override this.AfterComplete(textControl, node, _) = |
| 67 | + let recordExpr = node :?> IRecordExpr |
| 68 | + let endOffset = recordExpr.RightBrace.GetDocumentEndOffset() |
| 69 | + |
| 70 | + let hotspotInfos = |
| 71 | + recordExpr.FieldBindings.ToArray() |
| 72 | + |> Array.map (fun binding -> |
| 73 | + let templateField = TemplateField(binding.ReferenceName.ShortName, SimpleHotspotExpression(null), 0) |
| 74 | + HotspotInfo(templateField, binding.Expression.GetDocumentRange(), KeepExistingText = true)) |
| 75 | + |
| 76 | + let hotspotSession = |
| 77 | + LiveTemplatesManager.Instance.CreateHotspotSessionAtopExistingText( |
| 78 | + node.GetSolution(), endOffset, textControl, |
| 79 | + LiveTemplatesManager.EscapeAction.LeaveTextAndCaret, hotspotInfos) |
| 80 | + |
| 81 | + hotspotSession.ExecuteAndForget() |
0 commit comments