-
Notifications
You must be signed in to change notification settings - Fork 295
/
VariableTypeNotDeclaredInspectionResult.cs
107 lines (88 loc) · 3.75 KB
/
VariableTypeNotDeclaredInspectionResult.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
using System;
using System.Collections.Generic;
using Antlr4.Runtime;
using Rubberduck.Parsing.Grammar;
using Rubberduck.UI;
using Rubberduck.VBEditor;
namespace Rubberduck.Inspections
{
public class VariableTypeNotDeclaredInspectionResult : CodeInspectionResultBase
{
private readonly IEnumerable<CodeInspectionQuickFix> _quickFixes;
public VariableTypeNotDeclaredInspectionResult(IInspection inspection, string result, ParserRuleContext context, QualifiedModuleName qualifiedName)
: base(inspection, result, qualifiedName, context)
{
_quickFixes = new[]
{
new DeclareAsExplicitVariantQuickFix(Context, QualifiedSelection),
};
}
public override IEnumerable<CodeInspectionQuickFix> QuickFixes { get {return _quickFixes; } }
}
public class DeclareAsExplicitVariantQuickFix : CodeInspectionQuickFix
{
public DeclareAsExplicitVariantQuickFix(ParserRuleContext context, QualifiedSelection selection)
: base(context, selection, RubberduckUI.Inspections_DeclareAsExplicitVariant)
{
}
public override void Fix()
{
var codeModule = Selection.QualifiedName.Component.CodeModule;
var codeLine = codeModule.Lines[Selection.Selection.StartLine, Selection.Selection.LineCount];
// methods return empty string if soft-cast context is null - just concat results:
string originalInstruction;
var fix = DeclareExplicitVariant(Context as VBAParser.VariableSubStmtContext, out originalInstruction);
if (string.IsNullOrEmpty(originalInstruction))
{
fix = DeclareExplicitVariant(Context.Parent as VBAParser.ConstSubStmtContext, out originalInstruction);
}
if (string.IsNullOrEmpty(originalInstruction))
{
fix = DeclareExplicitVariant(Context as VBAParser.ArgContext, out originalInstruction);
}
if (string.IsNullOrEmpty(originalInstruction))
{
return;
}
var fixedCodeLine = codeLine.Replace(originalInstruction, fix);
codeModule.ReplaceLine(Selection.Selection.StartLine, fixedCodeLine);
}
private string DeclareExplicitVariant(VBAParser.VariableSubStmtContext context, out string instruction)
{
if (context == null)
{
instruction = null;
return null;
}
instruction = context.GetText();
return instruction + ' ' + Tokens.As + ' ' + Tokens.Variant;
}
private string DeclareExplicitVariant(VBAParser.ArgContext context, out string instruction)
{
if (context == null)
{
instruction = null;
return null;
}
instruction = context.GetText();
return instruction + ' ' + Tokens.As + ' ' + Tokens.Variant;
}
private string DeclareExplicitVariant(VBAParser.ConstSubStmtContext context, out string instruction)
{
if (context == null)
{
instruction = null;
return null;
}
var parent = (VBAParser.ConstStmtContext)context.Parent;
instruction = parent.GetText();
var constant = context.GetText();
var replacement = context.ambiguousIdentifier().GetText() + ' '
+ Tokens.As + ' ' + Tokens.Variant + ' '
+ context.EQ().GetText() + ' '
+ context.valueStmt().GetText();
var result = instruction.Replace(constant, replacement);
return result;
}
}
}