Testing pom wrapping

Camelcade · Apr 9, 2016 · 7b458ff · 7b458ff
1 parent c57a6f9
commit 7b458ff
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 5 deletions.
diff --git a/src/com/perl5/lang/perl/idea/formatter/PerlPreFormatter.java b/src/com/perl5/lang/perl/idea/formatter/PerlPreFormatter.java
@@ -19,11 +19,18 @@
 import com.intellij.openapi.editor.Document;
 import com.intellij.openapi.project.Project;
 import com.intellij.openapi.util.TextRange;
+import com.intellij.pom.PomManager;
+import com.intellij.pom.PomModel;
+import com.intellij.pom.PomTransaction;
+import com.intellij.pom.event.PomModelEvent;
+import com.intellij.pom.impl.PomTransactionBase;
+import com.intellij.pom.tree.TreeAspect;
 import com.intellij.psi.PsiDocumentManager;
 import com.intellij.psi.PsiElement;
 import com.intellij.psi.codeStyle.CodeStyleSettings;
 import com.intellij.psi.codeStyle.CodeStyleSettingsManager;
 import com.intellij.psi.util.PsiTreeUtil;
+import com.intellij.util.IncorrectOperationException;
 import com.perl5.lang.perl.idea.formatter.operations.*;
 import com.perl5.lang.perl.idea.formatter.settings.PerlCodeStyleSettings;
 import com.perl5.lang.perl.lexer.PerlElementTypes;
@@ -33,6 +40,7 @@
 import com.perl5.lang.perl.psi.utils.PerlPsiUtil;
 import com.perl5.lang.perl.util.PerlPackageUtil;
 import org.jetbrains.annotations.NotNull;
+import org.jetbrains.annotations.Nullable;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -95,7 +103,7 @@ public TextRange process(PsiElement element, TextRange range)
 		myRange = range;
 		final PsiDocumentManager manager = PsiDocumentManager.getInstance(myProject);
 		final Document document = manager.getDocument(element.getContainingFile());
-		int myDelta = 0;
+		final int[] myDelta = new int[]{0};
 		if (document != null)
 		{
 			manager.doPostponedOperationsAndUnblockDocument(document);
@@ -105,9 +113,32 @@ public TextRange process(PsiElement element, TextRange range)
 				// scan document
 				element.accept(this);
 
-				for (int i = myFormattingOperations.size() - 1; i >= 0; i--)
+				if( !myFormattingOperations.isEmpty())
 				{
-					myDelta += myFormattingOperations.get(i).apply();
+					final PomModel pomModel = PomManager.getModel(element.getProject());
+					final PomTransaction transaction = new PomTransactionBase(element, pomModel.getModelAspect(TreeAspect.class))
+					{
+						@Nullable
+						@Override
+						public PomModelEvent runInner() throws IncorrectOperationException
+						{
+							int counter = 0;
+							for (int i = myFormattingOperations.size() - 1; i >= 0 && counter++ < 100; i--)
+							{
+								long start = System.currentTimeMillis();
+								myDelta[0] += myFormattingOperations.get(i).apply();
+								System.err.println("Applied " + counter + " in " + (System.currentTimeMillis() - start) );
+							}
+							return null;
+						}
+					};
+					try{
+						pomModel.runTransaction(transaction);
+					}
+					catch (IncorrectOperationException e)
+					{
+						throw new RuntimeException(e);
+					}
 				}
 
 			} finally
@@ -116,7 +147,7 @@ public TextRange process(PsiElement element, TextRange range)
 			}
 		}
 
-		return TextRange.create(range.getStartOffset(), range.getEndOffset() + myDelta);
+		return TextRange.create(range.getStartOffset(), range.getEndOffset() + myDelta[0]);
 	}
 
 	protected void removeElement(PsiElement o)

diff --git a/src/com/perl5/lang/perl/psi/PerlRecursiveVisitor.java b/src/com/perl5/lang/perl/psi/PerlRecursiveVisitor.java
@@ -23,7 +23,7 @@
  */
 public class PerlRecursiveVisitor extends PerlVisitor
 {
-	public StringBuilder log = new StringBuilder();
+//	public StringBuilder log = new StringBuilder();
 	@Override
 	public void visitElement(PsiElement element)
 	{