Fix Obsolete Type Hint bugs.

Author: Hosch250

RetailCoder.VBE/Inspections/ObsoleteTypeHintInspectionResult.cs

@@ -1,5 +1,6 @@
 using System;
 using System.Collections.Generic;
+using System.Linq;
 using System.Text.RegularExpressions;
 using Antlr4.Runtime;
 using Microsoft.Vbe.Interop;
@@ -78,13 +79,30 @@ public override void Fix()
 
         private void FixTypeHintUsage(string hint, CodeModule module, Selection selection, bool isDeclaration = false)
         {
-            var line = module.get_Lines(selection.StartLine, 1);
+            var line = module.Lines[selection.StartLine, 1];
 
             var asTypeClause = ' ' + Tokens.As + ' ' + TypeHints[hint];
-            var pattern = "\\b" + _declaration.IdentifierName + "\\" + hint;
-            var fix = Regex.Replace(line, pattern, _declaration.IdentifierName + (isDeclaration ? asTypeClause : String.Empty));
 
-            module.ReplaceLine(selection.StartLine, fix);
+            string fix;
+
+            if (isDeclaration && (Context is VBAParser.FunctionStmtContext || Context is VBAParser.PropertyGetStmtContext))
+            {
+                var typeHint = (ParserRuleContext)Context.children.First(c => c is VBAParser.TypeHintContext);
+                var argList = (ParserRuleContext) Context.children.First(c => c is VBAParser.ArgListContext);
+                var endLine = argList.Stop.Line;
+                var endColumn = argList.Stop.Column;
+
+                var oldLine = module.Lines[endLine, selection.LineCount];
+                fix = oldLine.Insert(endColumn + 1, asTypeClause).Remove(typeHint.Start.Column, 1);  // adjust for VBA 0-based indexing
+
+                module.ReplaceLine(endLine, fix);
+            }
+            else
+            {
+                var pattern = "\\b" + _declaration.IdentifierName + "\\" + hint;
+                fix = Regex.Replace(line, pattern, _declaration.IdentifierName + (isDeclaration ? asTypeClause : string.Empty));
+                module.ReplaceLine(selection.StartLine, fix);
+            }
         }
     }
 }

RubberduckTests/Inspections/ObsoleteTypeHintInspectionTests.cs

@@ -435,10 +435,10 @@ public void ObsoleteCallStatement_QuickFixWorks_Field_SingleTypeHint()
         public void ObsoleteCallStatement_QuickFixWorks_Field_DecimalTypeHint()
         {
             const string inputCode =
-@"Public Foo&";
+@"Public Foo@";
 
             const string expectedCode =
-@"Public Foo As Decimal ' This doesn't compile, should it be float?";
+@"Public Foo As Decimal";
 
             //Arrange
             var builder = new MockVbeBuilder();