Merge pull request #6012 from awb95/dev-5929-Indent_Single_Line_If_then_Colon

Fix Indent if-then-colon #5929

Author: retailcoder

Rubberduck.SmartIndenter/AbsoluteCodeLine.cs

@@ -26,6 +26,7 @@ public class AbsoluteCodeLine
         private static readonly Regex PrecompilerInRegex = new Regex(@"^#(Else)?If\s.+Then$|^#Else$", RegexOptions.IgnoreCase);
         private static readonly Regex PrecompilerOutRegex = new Regex(@"^#ElseIf\s.+Then|^#Else$|#End\sIf$", RegexOptions.IgnoreCase);
         private static readonly Regex SingleLineElseIfRegex = new Regex(@"^ElseIf\s.*\sThen\s.*", RegexOptions.IgnoreCase);
+        private static readonly Regex IfThenWithColonNoElseRegex = new Regex(@"If\s.*\sThen:\s(?!.*:\sElse(If)?)", RegexOptions.IgnoreCase);
 
         private readonly IIndenterSettings _settings;
         private int _lineNumber;
@@ -192,6 +193,8 @@ public bool IsProcedureEnd
 
         public bool IsEmpty => Original.Trim().Length == 0;
 
+        public bool ContainsIfThenWithColonNoElse => IfThenWithColonNoElseRegex.IsMatch(_code);
+
         public int NextLineIndents
         {
             get

Rubberduck.SmartIndenter/LogicalCodeLine.cs

@@ -51,9 +51,16 @@ public int NextLineIndents
                 {
                     return 0;
                 }
-                return _rebuilt.Segments.Count() < 2
+                var indents = _rebuilt.Segments.Count() < 2
                     ? _rebuilt.NextLineIndents
                     : _rebuilt.Segments.Select(s => new AbsoluteCodeLine(s, _settings)).Select(a => a.NextLineIndents).Sum();
+
+                if (_rebuilt.ContainsIfThenWithColonNoElse)
+                {
+                    indents--;
+                }
+                return indents;
+
             }
         }
 

RubberduckTests/SmartIndenter/MultiSegmentLineTests.cs

@@ -83,25 +83,24 @@ public void UnmatchedEnumsNotIndent()
             Assert.IsTrue(code.SequenceEqual(actual));
         }
 
+        // https://github.com/rubberduck-vba/Rubberduck/issues/5929
         [Test]        // Broken in VB6 SmartIndenter.
         [Category("Indenter")]
         public void IfThenElseOnSameLineWorks()
         {
             var code = new[]
             {
                 "Public Sub Test()",
-                "If Foo = 42 Then: Bar = Foo: Else",
+                "If Foo = 42 Then: Bar = Foo: Else Baz = Bar",
                 "Baz = Foo",
-                "End If",
                 "End Sub"
             };
 
             var expected = new[]
             {
                 "Public Sub Test()",
-                "    If Foo = 42 Then: Bar = Foo: Else",
-                "        Baz = Foo",
-                "    End If",
+                "    If Foo = 42 Then: Bar = Foo: Else Baz = Bar",
+                "    Baz = Foo",
                 "End Sub"
             };