Force refactorings to rewrite changes to the code pane

Author: Hosch250

RetailCoder.VBE/Refactorings/EncapsulateField/EncapsulateFieldRefactoring.cs

@@ -16,7 +16,7 @@ public class EncapsulateFieldRefactoring : IRefactoring
         private readonly IRefactoringPresenterFactory<IEncapsulateFieldPresenter> _factory;
         private EncapsulateFieldModel _model;
 
-        private HashSet<IModuleRewriter> _referenceRewriters = new HashSet<IModuleRewriter>();
+        private readonly HashSet<IModuleRewriter> _referenceRewriters = new HashSet<IModuleRewriter>();
 
         public EncapsulateFieldRefactoring(IVBE vbe, IIndenter indenter, IRefactoringPresenterFactory<IEncapsulateFieldPresenter> factory)
         {

RetailCoder.VBE/Refactorings/IntroduceField/IntroduceFieldRefactoring.cs

@@ -95,6 +95,8 @@ private void PromoteVariable(IModuleRewriter rewriter, Declaration target)
                     pane.Selection = oldSelection.Value.Selection;
                 }
             }
+
+            rewriter.Rewrite();
         }
 
         private void AddField(IModuleRewriter rewriter, Declaration target)

RetailCoder.VBE/Refactorings/IntroduceParameter/IntroduceParameterRefactoring.cs

@@ -3,6 +3,7 @@
 using System.Windows.Forms;
 using Rubberduck.Common;
 using Rubberduck.Parsing.Grammar;
+using Rubberduck.Parsing.PostProcessing;
 using Rubberduck.Parsing.Symbols;
 using Rubberduck.Parsing.VBA;
 using Rubberduck.UI;
@@ -18,6 +19,8 @@ public class IntroduceParameterRefactoring : IRefactoring
         private readonly IList<Declaration> _declarations;
         private readonly IMessageBox _messageBox;
 
+        private readonly HashSet<IModuleRewriter> _rewriters = new HashSet<IModuleRewriter>();
+
         private static readonly DeclarationType[] ValidDeclarationTypes =
         {
             DeclarationType.Function,
@@ -94,6 +97,7 @@ private void PromoteVariable(Declaration target)
             }
 
             var rewriter = _state.GetRewriter(target);
+            _rewriters.Add(rewriter);
 
             QualifiedSelection? oldSelection = null;
             var pane = _vbe.ActiveCodePane;
@@ -110,6 +114,11 @@ private void PromoteVariable(Declaration target)
             {
                 pane.Selection = oldSelection.Value.Selection;
             }
+
+            foreach (var tokenRewriter in _rewriters)
+            {
+                tokenRewriter.Rewrite();
+            }
         }
 
         private bool PromptIfMethodImplementsInterface(Declaration targetVariable)
@@ -188,6 +197,7 @@ private void UpdateSignature(Declaration targetMethod, Declaration targetVariabl
         private void AddParameter(Declaration targetMethod, Declaration targetVariable, VBAParser.ArgListContext paramList)
         {
             var rewriter = _state.GetRewriter(targetMethod);
+            _rewriters.Add(rewriter);
 
             var argList = paramList.arg();
             var newParameter = Tokens.ByVal + " " + targetVariable.IdentifierName + " "+ Tokens.As + " " + targetVariable.AsTypeName;