Migrate LightMemberOriginForCompiledElement from IJ repository

2021-12-20 13:30:13 +03:00
parent 764ad8b9fa
commit b563824b35
8 changed files with 382 additions and 7 deletions
@@ -0,0 +1,17 @@
+plugins {
+    kotlin("jvm")
+    id("jps-compatible")
+}
+
+dependencies {
+    api(project(":compiler:psi"))
+    api(project(":analysis:decompiled:decompiler-to-psi"))
+    api(project(":analysis:decompiled:light-classes-for-decompiled"))
+    api(project(":compiler:light-classes"))
+    implementation(intellijCore())
+}
+
+sourceSets {
+    "main" { projectDefault() }
+    "test" { none() }
+}
@@ -0,0 +1,47 @@
+/*
+ * 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.analysis.decompiled.light.classes.fe10
+
+import com.intellij.psi.PsiMember
+import org.jetbrains.kotlin.analysis.decompiled.light.classes.origin.KotlinDeclarationInCompiledFileSearcher
+import org.jetbrains.kotlin.analysis.decompiler.psi.file.KtClsFile
+import org.jetbrains.kotlin.analysis.decompiler.psi.text.BySignatureIndexer
+import org.jetbrains.kotlin.analysis.decompiler.psi.text.ClassNameAndSignature
+import org.jetbrains.kotlin.analysis.decompiler.psi.text.relativeClassName
+import org.jetbrains.kotlin.load.kotlin.MemberSignature
+import org.jetbrains.kotlin.psi.KtClassOrObject
+import org.jetbrains.kotlin.psi.KtDeclaration
+import org.jetbrains.kotlin.psi.KtDeclarationContainer
+
+class KotlinDeclarationInCompiledFileSearcherFE10Impl: KotlinDeclarationInCompiledFileSearcher() {
+    override fun findDeclarationInCompiledFile(file: KtClsFile, member: PsiMember, signature: MemberSignature): KtDeclaration? {
+        val relativeClassName = member.relativeClassName()
+        val key = ClassNameAndSignature(relativeClassName, signature)
+
+        val memberName = member.name
+
+        if (memberName != null && !file.isContentsLoaded && file.hasDeclarationWithKey(BySignatureIndexer, key)) {
+            val container: KtDeclarationContainer? = if (relativeClassName.isEmpty())
+                file
+            else {
+                val topClassOrObject = file.declarations.singleOrNull() as? KtClassOrObject
+                relativeClassName.fold(topClassOrObject) { classOrObject, name ->
+                    classOrObject?.declarations?.singleOrNull { it.name == name.asString() } as? KtClassOrObject
+                }
+            }
+
+            val declaration = container?.declarations?.singleOrNull {
+                it.name == memberName
+            }
+
+            if (declaration != null) {
+                return declaration
+            }
+        }
+
+        return file.getDeclaration(BySignatureIndexer, key)
+    }
+}