Scratch: report errors at the correct line

^KT-24364 Fixed

idea/idea-jvm/src/org/jetbrains/kotlin/idea/scratch/compile/KtCompilingExecutor.kt

@@ -207,9 +207,8 @@ class KtCompilingExecutor(file: ScratchFile) : ScratchExecutor(file) {
     }
 
     private fun ScratchFile.findExpression(psiElement: PsiElement): ScratchExpression? {
-        val lineStart = psiElement.getLineNumber(true)
+        val elementLine = psiElement.getLineNumber()
-        val lineEnd = psiElement.getLineNumber(false)
+        return getExpressions().firstOrNull { elementLine in it.lineStart..it.lineEnd }
-        return getExpressions().firstOrNull { it.lineStart == lineStart || it.lineEnd == lineEnd }
     }
 
     private fun ScratchFile.findExpression(lineStart: Int, lineEnd: Int): ScratchExpression? {

idea/testData/scratch/unresolvedMultiline.comp.after

@@ -1,3 +1,7 @@
 foo.forEach {    // ERROR: Unresolved reference: foo; Cannot choose among the following candidates wi...
     1 + 1
 }
+
+fun goo(a: String) {    // ERROR: Unresolved reference: goo
+    super.goo(a)
+}

idea/testData/scratch/unresolvedMultiline.kts

@@ -1,3 +1,7 @@
 foo.forEach {
     1 + 1
 }
+
+fun goo(a: String) {
+    super.goo(a)
+}

idea/testData/scratch/unresolvedMultiline.repl.after

@@ -1,3 +1,7 @@
 foo.forEach {    // ERROR: error: unresolved reference: foo
     1 + 1
 }
+
+fun goo(a: String) {    // ERROR: error: unresolved reference: goo
+    super.goo(a)
+}