[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
+12
@@ -9,7 +9,10 @@ import org.jetbrains.kotlin.analysis.api.KtSymbolBasedReference
|
||||
import org.jetbrains.kotlin.analysis.api.components.KtReferenceResolveProvider
|
||||
import org.jetbrains.kotlin.analysis.api.fir.KtFirAnalysisSession
|
||||
import org.jetbrains.kotlin.analysis.api.symbols.KtSymbol
|
||||
import org.jetbrains.kotlin.analysis.low.level.api.fir.api.getOrBuildFirSafe
|
||||
import org.jetbrains.kotlin.fir.expressions.FirResolvedQualifier
|
||||
import org.jetbrains.kotlin.idea.references.KtReference
|
||||
import org.jetbrains.kotlin.idea.references.KtSimpleNameReference
|
||||
|
||||
internal class KtFirReferenceResolveProvider(
|
||||
override val analysisSession: KtFirAnalysisSession
|
||||
@@ -20,4 +23,13 @@ internal class KtFirReferenceResolveProvider(
|
||||
return analysisSession.resolveToSymbols()
|
||||
}
|
||||
}
|
||||
|
||||
override fun isImplicitReferenceToCompanion(reference: KtReference): Boolean {
|
||||
if (reference !is KtSimpleNameReference) {
|
||||
return false
|
||||
}
|
||||
val referenceElement = reference.element
|
||||
val qualifier = referenceElement.getOrBuildFirSafe<FirResolvedQualifier>(analysisSession.firResolveSession) ?: return false
|
||||
return qualifier.resolvedToCompanionObject
|
||||
}
|
||||
}
|
||||
+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.fir.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.fir.test.configurators.AnalysisApiFirTestConfiguratorFactory;
|
||||
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 FirIdeDependentAnalysisSourceModuleIsImplicitCompanionReferenceTestGenerated extends AbstractIsImplicitCompanionReferenceTest {
|
||||
@NotNull
|
||||
@Override
|
||||
public AnalysisApiTestConfigurator getConfigurator() {
|
||||
return AnalysisApiFirTestConfiguratorFactory.INSTANCE.createConfigurator(
|
||||
new AnalysisApiTestConfiguratorFactoryData(
|
||||
FrontendKind.Fir,
|
||||
TestModuleKind.Source,
|
||||
AnalysisSessionMode.Dependent,
|
||||
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");
|
||||
}
|
||||
}
|
||||
+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.fir.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.fir.test.configurators.AnalysisApiFirTestConfiguratorFactory;
|
||||
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 FirIdeNormalAnalysisSourceModuleIsImplicitCompanionReferenceTestGenerated extends AbstractIsImplicitCompanionReferenceTest {
|
||||
@NotNull
|
||||
@Override
|
||||
public AnalysisApiTestConfigurator getConfigurator() {
|
||||
return AnalysisApiFirTestConfiguratorFactory.INSTANCE.createConfigurator(
|
||||
new AnalysisApiTestConfiguratorFactoryData(
|
||||
FrontendKind.Fir,
|
||||
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