Skip to content

Commit 3174f11

Browse files
committed
Make Implement Interface use the rewriter
1 parent 90a0ae8 commit 3174f11

File tree

3 files changed

+99
-176
lines changed

3 files changed

+99
-176
lines changed

RetailCoder.VBE/Refactorings/ImplementInterface/ImplementInterfaceRefactoring.cs

Lines changed: 8 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -105,13 +105,15 @@ private void ImplementMissingMembers()
105105
AddItems(nonImplementedMembers);
106106
}
107107

108-
private void AddItems(List<Declaration> members)
108+
private void AddItems(List<Declaration> missingMembers)
109109
{
110-
var module = _targetClass.QualifiedSelection.QualifiedName.Component.CodeModule;
111-
{
112-
var missingMembersText = members.Aggregate(string.Empty, (current, member) => current + Environment.NewLine + GetInterfaceMember(member));
113-
module.InsertLines(module.CountOfDeclarationLines + 1, missingMembersText);
114-
}
110+
var missingMembersText = missingMembers.Aggregate(string.Empty,
111+
(current, member) => current + Environment.NewLine + GetInterfaceMember(member));
112+
113+
var rewriter = _state.GetRewriter(_targetClass);
114+
rewriter.InsertAfter(rewriter.TokenStream.Size, Environment.NewLine + missingMembersText);
115+
116+
rewriter.Rewrite();
115117
}
116118

117119
private string GetInterfaceMember(Declaration member)

RetailCoder.VBE/Refactorings/MoveCloserToUsage/MoveCloserToUsageRefactoring.cs

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,11 +1,9 @@
11
using System;
22
using System.Collections.Generic;
33
using System.Linq;
4-
using System.Text.RegularExpressions;
54
using System.Windows.Forms;
65
using Antlr4.Runtime;
76
using Rubberduck.Common;
8-
using Rubberduck.Parsing;
97
using Rubberduck.Parsing.Grammar;
108
using Rubberduck.Parsing.PostProcessing;
119
using Rubberduck.Parsing.Symbols;

0 commit comments

Comments
 (0)