Merge pull request #139 from SpecFlowOSS/fix-typing-assist

Fix typing assist #87
SpecFlowOSS · Jun 14, 2021 · b78fad9 · b78fad9
2 parents 42e1553 + 0cc47e1
commit b78fad9
Showing 1 changed file with 41 additions and 6 deletions.
diff --git a/src/dotnet/ReSharperPlugin.SpecflowRiderPlugin/TypingAssist/GherkinTypingAssist.cs b/src/dotnet/ReSharperPlugin.SpecflowRiderPlugin/TypingAssist/GherkinTypingAssist.cs
@@ -98,14 +98,13 @@ private bool HandleEnter([NotNull] IActionContext context)
 
                     if (inTable)
                     {
-                        if (GetFormatSettingsKey(textControl).SmallTableIndent)
-                            extraIdent = "  ";
-                        else
-                            extraIdentSize += GetFormatSettingsKey(textControl).TableIndentSize;
+                        var indentSizeBasedOnPreviousRow = FindPreviousRowFirstPipeIndent(cachingLexer, caret);
+                        var indentText = GetTableIndentText(indentSizeBasedOnPreviousRow);
+                        textControl.Document.InsertText(caret, GetNewLineText(textControl) + indentText);
+                        return true;
                     }
 
                     var currentIndent = ComputeIndentOfCurrentKeyword(cachingLexer) + GetIndentText(textControl, extraIdentSize) + extraIdent;
-
                     textControl.Document.InsertText(caret, GetNewLineText(textControl) + currentIndent);
                     return true;
                 }
@@ -114,6 +113,16 @@ private bool HandleEnter([NotNull] IActionContext context)
             return false;
         }
 
+        private static string GetTableIndentText(int indent)
+        {
+            var sb = new StringBuilder();
+            for (var i = 0; i < indent; i++)
+            {
+                sb.Append(" ");
+            }
+            return sb.ToString();
+        }
+
         private bool HandleTableCellClosing([NotNull] ITypingContext context)
         {
             var textControl = context.TextControl;
@@ -128,7 +137,7 @@ private bool HandleTableCellClosing([NotNull] ITypingContext context)
 
                 PsiServices.Transactions.Execute("Format code", () =>
                 {
-                    GetCodeFormatter(tokenNode).Format(parentTable.firstChild, parentTable.lastChild.LastChild, CodeFormatProfile.SOFT, new AdditionalFormatterParameters(treatTextAfterLastNodeAsIncorrect: false));
+                    GetCodeFormatter(tokenNode).Format(parentTable.firstChild.FirstChild, parentTable.lastChild.LastChild, CodeFormatProfile.SOFT, new AdditionalFormatterParameters(treatTextAfterLastNodeAsIncorrect: false));
                 });
                 return null;
             }));
@@ -188,6 +197,32 @@ private TokenNodeType FindLastKeywordToken(CachingLexer cachingLexer, int caret,
             }
             return cachingLexer.TokenType;
         }
+
+        private int FindPreviousRowFirstPipeIndent(CachingLexer cachingLexer, int caret)
+        {
+            //find previous enter
+            cachingLexer.FindTokenAt(caret - 1);
+            while (GherkinTokenTypes.NEW_LINE != cachingLexer.TokenType)
+            {
+                if (cachingLexer.CurrentPosition == 0)
+                    return 0;
+                cachingLexer.SetCurrentToken(cachingLexer.CurrentPosition - 1);
+            }
+
+            var enterStart = cachingLexer.TokenStart;
+
+            //find the first pipe after the enter
+            while (GherkinTokenTypes.PIPE != cachingLexer.TokenType)
+            {
+                if (cachingLexer.CurrentPosition == 0)
+                    return 0;
+                cachingLexer.SetCurrentToken(cachingLexer.CurrentPosition + 1);
+            }
+
+            var pipeStart = cachingLexer.TokenStart;
+
+            return pipeStart - enterStart -1;
+        }
 
 
         private IndentStyle GetIndentType(ITextControl textControl)