[Analysis API] add method for implicit companion object reference detection
Add a new method `isImplicitReferenceToCompanion` to determine if the given reference is an implicit reference to a companion object. The method is needed for the rename refactoring in IJ. ^KTIJ-25863
This commit is contained in:
committed by
Space Team
parent
f5bbf2b4fe
commit
8b0eb0488f
+11
-1
@@ -12,10 +12,12 @@ import org.jetbrains.kotlin.analysis.api.descriptors.components.base.Fe10KtAnaly
|
||||
import org.jetbrains.kotlin.analysis.api.descriptors.symbols.descriptorBased.base.toKtSymbol
|
||||
import org.jetbrains.kotlin.analysis.api.symbols.KtSymbol
|
||||
import org.jetbrains.kotlin.idea.references.KtReference
|
||||
import org.jetbrains.kotlin.idea.references.KtSimpleNameReference
|
||||
import org.jetbrains.kotlin.references.fe10.base.KtFe10Reference
|
||||
import org.jetbrains.kotlin.resolve.BindingContext
|
||||
|
||||
internal class KtFe10ReferenceResolveProvider(
|
||||
override val analysisSession: KtFe10AnalysisSession
|
||||
override val analysisSession: KtFe10AnalysisSession,
|
||||
) : KtReferenceResolveProvider(), Fe10KtAnalysisSessionComponent {
|
||||
override fun resolveToSymbols(reference: KtReference): Collection<KtSymbol> {
|
||||
require(reference is KtFe10Reference)
|
||||
@@ -24,4 +26,12 @@ internal class KtFe10ReferenceResolveProvider(
|
||||
descriptor.toKtSymbol(analysisContext)
|
||||
}
|
||||
}
|
||||
|
||||
override fun isImplicitReferenceToCompanion(reference: KtReference): Boolean {
|
||||
if (reference !is KtSimpleNameReference) {
|
||||
return false
|
||||
}
|
||||
val bindingContext = analysisContext.analyze(reference.element, Fe10AnalysisFacade.AnalysisMode.PARTIAL)
|
||||
return bindingContext[BindingContext.SHORT_REFERENCE_TO_COMPANION_OBJECT, reference.element] != null
|
||||
}
|
||||
}
|
||||
+96
@@ -0,0 +1,96 @@
|
||||
/*
|
||||
* Copyright 2010-2023 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.api.fe10.test.cases.generated.cases.components.referenceResolveProvider;
|
||||
|
||||
import com.intellij.testFramework.TestDataPath;
|
||||
import org.jetbrains.kotlin.test.util.KtTestUtil;
|
||||
import org.jetbrains.annotations.NotNull;
|
||||
import org.jetbrains.kotlin.analysis.api.fe10.test.configurator.AnalysisApiFe10TestConfiguratorFactory;
|
||||
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfiguratorFactoryData;
|
||||
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiTestConfigurator;
|
||||
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.TestModuleKind;
|
||||
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.FrontendKind;
|
||||
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisSessionMode;
|
||||
import org.jetbrains.kotlin.analysis.test.framework.test.configurators.AnalysisApiMode;
|
||||
import org.jetbrains.kotlin.analysis.api.impl.base.test.cases.components.referenceResolveProvider.AbstractIsImplicitCompanionReferenceTest;
|
||||
import org.jetbrains.kotlin.test.TestMetadata;
|
||||
import org.junit.jupiter.api.Nested;
|
||||
import org.junit.jupiter.api.Test;
|
||||
|
||||
import java.io.File;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
/** This class is generated by {@link org.jetbrains.kotlin.generators.tests.analysis.api.GenerateAnalysisApiTestsKt}. DO NOT MODIFY MANUALLY */
|
||||
@SuppressWarnings("all")
|
||||
@TestMetadata("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion")
|
||||
@TestDataPath("$PROJECT_ROOT")
|
||||
public class Fe10IdeNormalAnalysisSourceModuleIsImplicitCompanionReferenceTestGenerated extends AbstractIsImplicitCompanionReferenceTest {
|
||||
@NotNull
|
||||
@Override
|
||||
public AnalysisApiTestConfigurator getConfigurator() {
|
||||
return AnalysisApiFe10TestConfiguratorFactory.INSTANCE.createConfigurator(
|
||||
new AnalysisApiTestConfiguratorFactoryData(
|
||||
FrontendKind.Fe10,
|
||||
TestModuleKind.Source,
|
||||
AnalysisSessionMode.Normal,
|
||||
AnalysisApiMode.Ide
|
||||
)
|
||||
);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testAllFilesPresentInIsImplicitReferenceToCompanion() throws Exception {
|
||||
KtTestUtil.assertAllTestsPresentByMetadataWithExcluded(this.getClass(), new File("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion"), Pattern.compile("^(.+)\\.kt$"), null, true);
|
||||
}
|
||||
|
||||
@Test
|
||||
@TestMetadata("classAccessWithExplicitReferenceToCompanion.kt")
|
||||
public void testClassAccessWithExplicitReferenceToCompanion() throws Exception {
|
||||
runTest("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion/classAccessWithExplicitReferenceToCompanion.kt");
|
||||
}
|
||||
|
||||
@Test
|
||||
@TestMetadata("classAccessWithExplicitReferenceToNamedCompanion.kt")
|
||||
public void testClassAccessWithExplicitReferenceToNamedCompanion() throws Exception {
|
||||
runTest("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion/classAccessWithExplicitReferenceToNamedCompanion.kt");
|
||||
}
|
||||
|
||||
@Test
|
||||
@TestMetadata("constructorCall.kt")
|
||||
public void testConstructorCall() throws Exception {
|
||||
runTest("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion/constructorCall.kt");
|
||||
}
|
||||
|
||||
@Test
|
||||
@TestMetadata("constructorCallWithInvokeInCompanion.kt")
|
||||
public void testConstructorCallWithInvokeInCompanion() throws Exception {
|
||||
runTest("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion/constructorCallWithInvokeInCompanion.kt");
|
||||
}
|
||||
|
||||
@Test
|
||||
@TestMetadata("explicitReferenceToCompanion.kt")
|
||||
public void testExplicitReferenceToCompanion() throws Exception {
|
||||
runTest("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion/explicitReferenceToCompanion.kt");
|
||||
}
|
||||
|
||||
@Test
|
||||
@TestMetadata("explicitReferenceToNamedCompanion.kt")
|
||||
public void testExplicitReferenceToNamedCompanion() throws Exception {
|
||||
runTest("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion/explicitReferenceToNamedCompanion.kt");
|
||||
}
|
||||
|
||||
@Test
|
||||
@TestMetadata("impicitReferenceToCompanion.kt")
|
||||
public void testImpicitReferenceToCompanion() throws Exception {
|
||||
runTest("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion/impicitReferenceToCompanion.kt");
|
||||
}
|
||||
|
||||
@Test
|
||||
@TestMetadata("qualifierReferenceToClassWithCompanion.kt")
|
||||
public void testQualifierReferenceToClassWithCompanion() throws Exception {
|
||||
runTest("analysis/analysis-api/testData/components/referenceResolveProvider/isImplicitReferenceToCompanion/qualifierReferenceToClassWithCompanion.kt");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user