File tree Expand file tree Collapse file tree 1 file changed +12
-1
lines changed
Rubberduck.CodeAnalysis/QuickFixes Expand file tree Collapse file tree 1 file changed +12
-1
lines changed Original file line number Diff line number Diff line change 1
1
using Rubberduck . Inspections . Abstract ;
2
2
using Rubberduck . Inspections . Concrete ;
3
+ using Rubberduck . Parsing ;
4
+ using Rubberduck . Parsing . Grammar ;
3
5
using Rubberduck . Parsing . Inspections . Abstract ;
4
6
using Rubberduck . Parsing . Rewriter ;
5
7
@@ -14,7 +16,16 @@ public UseSetKeywordForObjectAssignmentQuickFix()
14
16
public override void Fix ( IInspectionResult result , IRewriteSession rewriteSession )
15
17
{
16
18
var rewriter = rewriteSession . CheckOutModuleRewriter ( result . QualifiedSelection . QualifiedName ) ;
17
- rewriter . InsertBefore ( result . Context . Start . TokenIndex , "Set " ) ;
19
+ var letStmt = result . Context . GetAncestor < VBAParser . LetStmtContext > ( ) ;
20
+ var letToken = letStmt . LET ( ) ;
21
+ if ( letToken != null )
22
+ {
23
+ rewriter . Replace ( letToken , "Set" ) ;
24
+ }
25
+ else
26
+ {
27
+ rewriter . InsertBefore ( letStmt . Start . TokenIndex , "Set " ) ;
28
+ }
18
29
}
19
30
20
31
public override string Description ( IInspectionResult result ) => Resources . Inspections . QuickFixes . SetObjectVariableQuickFix ;
You can’t perform that action at this time.
0 commit comments