diff --git a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/AbstractCodeTransformationIntention.java b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/AbstractCodeTransformationIntention.java index f30ceb3a156..4358b53dc57 100644 --- a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/AbstractCodeTransformationIntention.java +++ b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/AbstractCodeTransformationIntention.java @@ -26,6 +26,7 @@ import com.intellij.util.IncorrectOperationException; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; import org.jetbrains.jet.lang.psi.JetElement; +import org.jetbrains.jet.lang.psi.JetFile; import org.jetbrains.jet.lang.psi.JetPsiUtil; import org.jetbrains.jet.plugin.JetBundle; import org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransformations.core.Transformer; @@ -63,6 +64,6 @@ public abstract class AbstractCodeTransformationIntention assert target != null : "Intention is not applicable"; - transformer.transform(target, editor); + transformer.transform(target, editor, (JetFile) file); } } diff --git a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/FoldableKind.java b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/FoldableKind.java index e8503d216c0..4f1cfabb32a 100644 --- a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/FoldableKind.java +++ b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/FoldableKind.java @@ -19,6 +19,7 @@ package org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransfo import com.intellij.openapi.editor.Editor; import com.intellij.psi.PsiElement; import org.jetbrains.annotations.NotNull; +import org.jetbrains.jet.lang.psi.JetFile; import org.jetbrains.jet.lang.psi.JetIfExpression; import org.jetbrains.jet.lang.psi.JetWhenExpression; import org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransformations.core.Transformer; @@ -26,31 +27,31 @@ import org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransfor public enum FoldableKind implements Transformer { IF_TO_ASSIGNMENT("fold.if.to.assignment") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, @NotNull JetFile file) { BranchedFoldingUtils.foldIfExpressionWithAssignments((JetIfExpression) element); } }, IF_TO_RETURN("fold.if.to.return") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, @NotNull JetFile file) { BranchedFoldingUtils.foldIfExpressionWithReturns((JetIfExpression) element); } }, IF_TO_RETURN_ASYMMETRICALLY("fold.if.to.return") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, @NotNull JetFile file) { BranchedFoldingUtils.foldIfExpressionWithAsymmetricReturns((JetIfExpression) element); } }, WHEN_TO_ASSIGNMENT("fold.when.to.assignment") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, @NotNull JetFile file) { BranchedFoldingUtils.foldWhenExpressionWithAssignments((JetWhenExpression) element); } }, WHEN_TO_RETURN("fold.when.to.return") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, @NotNull JetFile file) { BranchedFoldingUtils.foldWhenExpressionWithReturns((JetWhenExpression) element); } }; diff --git a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/UnfoldableKind.java b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/UnfoldableKind.java index 07bf3f3134f..86ed5e5b5e6 100644 --- a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/UnfoldableKind.java +++ b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/UnfoldableKind.java @@ -20,31 +20,32 @@ import com.intellij.openapi.editor.Editor; import com.intellij.psi.PsiElement; import org.jetbrains.annotations.NotNull; import org.jetbrains.jet.lang.psi.JetBinaryExpression; +import org.jetbrains.jet.lang.psi.JetFile; import org.jetbrains.jet.lang.psi.JetReturnExpression; import org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransformations.core.Transformer; public enum UnfoldableKind implements Transformer { ASSIGNMENT_TO_IF("unfold.assignment.to.if") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, JetFile file) { BranchedUnfoldingUtils.unfoldAssignmentToIf((JetBinaryExpression) element, editor); } }, RETURN_TO_IF("unfold.return.to.if") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, JetFile file) { BranchedUnfoldingUtils.unfoldReturnToIf((JetReturnExpression) element); } }, ASSIGNMENT_TO_WHEN("unfold.assignment.to.when") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, JetFile file) { BranchedUnfoldingUtils.unfoldAssignmentToWhen((JetBinaryExpression) element, editor); } }, RETURN_TO_WHEN("unfold.return.to.when") { @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, JetFile file) { BranchedUnfoldingUtils.unfoldReturnToWhen((JetReturnExpression) element); } }; diff --git a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/core/Transformer.java b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/core/Transformer.java index 82f5154f2aa..956ee756b1a 100644 --- a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/core/Transformer.java +++ b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/core/Transformer.java @@ -19,9 +19,10 @@ package org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransfo import com.intellij.openapi.editor.Editor; import com.intellij.psi.PsiElement; import org.jetbrains.annotations.NotNull; +import org.jetbrains.jet.lang.psi.JetFile; public interface Transformer { @NotNull String getKey(); - void transform(@NotNull PsiElement element, @NotNull Editor editor); + void transform(@NotNull PsiElement element, @NotNull Editor editor, @NotNull JetFile file); } diff --git a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/intentions/EliminateWhenSubjectIntention.java b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/intentions/EliminateWhenSubjectIntention.java index 225a36b770a..9c2dd1c00af 100644 --- a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/intentions/EliminateWhenSubjectIntention.java +++ b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/intentions/EliminateWhenSubjectIntention.java @@ -5,6 +5,7 @@ import com.intellij.openapi.editor.Editor; import com.intellij.psi.PsiElement; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; +import org.jetbrains.jet.lang.psi.JetFile; import org.jetbrains.jet.lang.psi.JetWhenExpression; import org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransformations.AbstractCodeTransformationIntention; import org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransformations.WhenUtils; @@ -19,7 +20,7 @@ public class EliminateWhenSubjectIntention extends AbstractCodeTransformationInt } @Override - public void transform(@NotNull PsiElement element, @NotNull Editor editor) { + public void transform(@NotNull PsiElement element, @NotNull Editor editor, @NotNull JetFile file) { WhenUtils.eliminateWhenSubject((JetWhenExpression) element); } }; diff --git a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/intentions/FlattenWhenIntention.java b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/intentions/FlattenWhenIntention.java index 6cc9dc00320..bd2a4ac55e7 100644 --- a/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/intentions/FlattenWhenIntention.java +++ b/idea/src/org/jetbrains/jet/plugin/codeInsight/codeTransformations/branchedTransformations/intentions/FlattenWhenIntention.java @@ -5,6 +5,7 @@ import com.intellij.openapi.editor.Editor; import com.intellij.psi.PsiElement; import org.jetbrains.annotations.NotNull; import org.jetbrains.annotations.Nullable; +import org.jetbrains.jet.lang.psi.JetFile; import org.jetbrains.jet.lang.psi.JetWhenExpression; import org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransformations.AbstractCodeTransformationIntention; import org.jetbrains.jet.plugin.codeInsight.codeTransformations.branchedTransformations.WhenUtils; @@ -19,7 +20,7 @@ public class FlattenWhenIntention extends AbstractCodeTransformationIntention