[REPL] Copy findLabelAndCall to scripting-ide-services to fix problems with embeddable artifact

2021-05-18 19:04:17 +03:00
parent 7caadd87a2
commit 9412fe094b
2 changed files with 40 additions and 1 deletions
@@ -0,0 +1,40 @@
+/*
+ * Copyright 2010-2021 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.idea.util
+
+import org.jetbrains.kotlin.name.Name
+import org.jetbrains.kotlin.psi.*
+
+/**
+ * This extension method was copied from `idea:idea-frontend-independent` module because it's the only method that
+ * included in `idea:idea-frontend-independent` and needed for performing REPL completion
+ */
+@Suppress("unused")
+fun KtFunctionLiteral.findLabelAndCall(): Pair<Name?, KtCallExpression?> {
+    val literalParent = (this.parent as KtLambdaExpression).parent
+
+    fun KtValueArgument.callExpression(): KtCallExpression? {
+        val parent = parent
+        return (if (parent is KtValueArgumentList) parent else this).parent as? KtCallExpression
+    }
+
+    when (literalParent) {
+        is KtLabeledExpression -> {
+            val callExpression = (literalParent.parent as? KtValueArgument)?.callExpression()
+            return Pair(literalParent.getLabelNameAsName(), callExpression)
+        }
+
+        is KtValueArgument -> {
+            val callExpression = literalParent.callExpression()
+            val label = (callExpression?.calleeExpression as? KtSimpleNameExpression)?.getReferencedNameAsName()
+            return Pair(label, callExpression)
+        }
+
+        else -> {
+            return Pair(null, null)
+        }
+    }
+}