From 53983774d0521b08ed5e91b1bd80b003fb8915b4 Mon Sep 17 00:00:00 2001
From: Valentin Kipyatkov <valentin@jetbrains.com>
Date: Wed, 1 Oct 2014 15:51:39 +0400
Subject: [PATCH] Moved and renamed class + added check for JetFile

---
 idea/src/META-INF/plugin.xml                   |  2 +-
 .../JoinDeclarationAndAssignmentHandler.kt}    | 18 ++++++++++--------
 2 files changed, 11 insertions(+), 9 deletions(-)
 rename idea/src/org/jetbrains/jet/plugin/{intentions/declarations/JetDeclarationJoinLinesHandler.kt => joinLines/JoinDeclarationAndAssignmentHandler.kt} (89%)

diff --git a/idea/src/META-INF/plugin.xml b/idea/src/META-INF/plugin.xml
index 2ac202ad0c6..614837d5651 100644
--- a/idea/src/META-INF/plugin.xml
+++ b/idea/src/META-INF/plugin.xml
@@ -431,7 +431,7 @@
                           implementation="org.jetbrains.jet.plugin.codeInsight.upDownMover.JetDeclarationMover"
                           order="before jetExpression" />
 
-    <joinLinesHandler implementation="org.jetbrains.jet.plugin.intentions.declarations.JetDeclarationJoinLinesHandler"/>
+    <joinLinesHandler implementation="org.jetbrains.jet.plugin.joinLines.JoinDeclarationAndAssignmentHandler"/>
 
     <targetElementEvaluator
         language="jet"
diff --git a/idea/src/org/jetbrains/jet/plugin/intentions/declarations/JetDeclarationJoinLinesHandler.kt b/idea/src/org/jetbrains/jet/plugin/joinLines/JoinDeclarationAndAssignmentHandler.kt
similarity index 89%
rename from idea/src/org/jetbrains/jet/plugin/intentions/declarations/JetDeclarationJoinLinesHandler.kt
rename to idea/src/org/jetbrains/jet/plugin/joinLines/JoinDeclarationAndAssignmentHandler.kt
index 68819acc529..67e0cf01986 100644
--- a/idea/src/org/jetbrains/jet/plugin/intentions/declarations/JetDeclarationJoinLinesHandler.kt
+++ b/idea/src/org/jetbrains/jet/plugin/joinLines/JoinDeclarationAndAssignmentHandler.kt
@@ -14,25 +14,27 @@
  * limitations under the License.
  */
 
-package org.jetbrains.jet.plugin.intentions.declarations
+package org.jetbrains.jet.plugin.joinLines
 
+import org.jetbrains.jet.lang.psi.psiUtil.siblings
+import org.jetbrains.jet.lang.psi.psiUtil.parents
 import com.intellij.codeInsight.editorActions.JoinRawLinesHandlerDelegate
 import com.intellij.openapi.editor.Document
-import com.intellij.psi.PsiElement
 import com.intellij.psi.PsiFile
-import org.jetbrains.jet.lang.psi.JetPsiUtil
-import org.jetbrains.jet.lang.psi.psiUtil.*
+import org.jetbrains.jet.lang.psi.JetFile
+import com.intellij.psi.PsiElement
 import org.jetbrains.jet.lang.psi.JetProperty
 import org.jetbrains.jet.lang.psi.JetBinaryExpression
 import org.jetbrains.jet.lang.psi.JetSimpleNameExpression
-import org.jetbrains.jet.lexer.JetTokens
 import com.intellij.psi.PsiWhiteSpace
 import com.intellij.psi.PsiComment
-import com.google.common.base.Predicate
+import org.jetbrains.jet.lexer.JetTokens
 
-public class JetDeclarationJoinLinesHandler : JoinRawLinesHandlerDelegate {
+public class JoinDeclarationAndAssignmentHandler : JoinRawLinesHandlerDelegate {
 
     override fun tryJoinRawLines(document: Document, file: PsiFile, start: Int, end: Int): Int {
+        if (file !is JetFile) return -1
+
         val element = file.findElementAt(start)
                               ?.siblings(forward = false, withItself = false)
                               ?.firstOrNull { !isToSkip(it) }  ?: return -1
@@ -67,7 +69,7 @@ public class JetDeclarationJoinLinesHandler : JoinRawLinesHandlerDelegate {
 
     private fun doJoin(property: JetProperty, assignment: JetBinaryExpression) {
         property.setInitializer(assignment.getRight())
-        property.getParent()!!.deleteChildRange(property.getNextSibling(), assignment) //TODO: should we remove range?
+        property.getParent()!!.deleteChildRange(property.getNextSibling(), assignment) //TODO: should we delete range?
     }
 
     private fun isToSkip(element: PsiElement)