Fix that ViFindBrace doesn't search for brace when current char is not a brace

Author: sharpchen

PSReadLine/Movement.vi.cs

@@ -254,7 +254,29 @@ private int ViFindBrace(int i)
                 case ')':
                     return ViFindBackward(i, '(', withoutPassing: ')');
                 default:
-                    return i;
+                    ReadOnlySpan<char> parenthese = stackalloc char[] { '{', '}', '(', ')', '[', ']' };
+                    int nextParen = i;
+                    for (; nextParen < _buffer.Length; nextParen++)
+                    {
+                        for (int idx = 0; idx < parenthese.Length; idx++)
+                            if (parenthese[idx] == _buffer[nextParen]) goto Outer;
+                    }
+
+                Outer:
+                    int match = _buffer[nextParen] switch
+                    {
+                        // if next is opening, find forward
+                        '{' => ViFindForward(nextParen, '}', withoutPassing: '{'),
+                        '[' => ViFindForward(nextParen, ']', withoutPassing: '['),
+                        '(' => ViFindForward(nextParen, ')', withoutPassing: '('),
+                        // if next is closing, find backward
+                        '}' => ViFindBackward(nextParen, '}', withoutPassing: '}'),
+                        ']' => ViFindBackward(nextParen, '[', withoutPassing: ']'),
+                        ')' => ViFindBackward(nextParen, '(', withoutPassing: ')'),
+                        _ => nextParen
+                    };
+
+                    return match == nextParen ? i : match;
             }
         }
 

test/MovementTest.VI.cs

@@ -370,7 +370,7 @@ public void ViGlobMovement_EmptyBuffer_Defect1195()
             TestSetup(KeyMode.Vi);
 
             TestMustDing("", Keys(
-                _.Escape, "W" 
+                _.Escape, "W"
             ));
         }
 
@@ -454,6 +454,23 @@ public void ViGotoBrace()
                     ));
             }
 
+            Test("0[2[4foo]", Keys(
+                "0[2[4foo]",
+                CheckThat(() => AssertCursorLeftIs(9)),
+                _.Escape, CheckThat(() => AssertCursorLeftIs(7)),
+                "0ff", CheckThat(() => AssertCursorLeftIs(5)),
+                _.Percent, CheckThat(() => AssertCursorLeftIs(3)),
+                _.Percent, CheckThat(() => AssertCursorLeftIs(8))
+            ));
+
+            Test("0[2[4foo)}", Keys(
+                "0[2[4foo)}",
+                CheckThat(() => AssertCursorLeftIs(7)),
+                _.Escape, CheckThat(() => AssertCursorLeftIs(6)),
+                "0ff", CheckThat(() => AssertCursorLeftIs(5)),
+                _.Percent, CheckThat(() => AssertCursorLeftIs(5)), // stay still
+                _.Percent, CheckThat(() => AssertCursorLeftIs(5))
+            ));
             // <%> with empty text buffer should work fine.
             Test("", Keys(
                 _.Escape, _.Percent,