handling removal of \n after pressing tab on autocomplete option. Fixes

#68
getgauge · Aug 11, 2015 · b626336 · b626336
1 parent 821fbe5
commit b626336
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/com/thoughtworks/gauge/autocomplete/StepCompletionProvider.java b/src/com/thoughtworks/gauge/autocomplete/StepCompletionProvider.java
@@ -23,6 +23,7 @@
 import com.intellij.codeInsight.template.TemplateBuilder;
 import com.intellij.codeInsight.template.TemplateBuilderFactory;
 import com.intellij.openapi.module.Module;
+import com.intellij.psi.PsiDocumentManager;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.util.PsiTreeUtil;
 import com.intellij.util.ProcessingContext;
@@ -69,6 +70,10 @@ public void addCompletions(@NotNull final CompletionParameters parameters, Proce
             element = element.withInsertHandler(new InsertHandler<LookupElement>() {
                 @Override
                 public void handleInsert(InsertionContext context, LookupElement item) {
+                    if (context.getCompletionChar() == '\t') {
+                        context.getDocument().insertString(context.getEditor().getCaretModel().getOffset(), "\n");
+                        PsiDocumentManager.getInstance(context.getProject()).commitDocument(context.getDocument());
+                    }
                     PsiElement stepElement = context.getFile().findElementAt(context.getStartOffset()).getParent();
                     TemplateBuilder templateBuilder = getTemplateBuilder(stepElement, prefix);
                     templateBuilder.run(context.getEditor(), false);