[AA] Make KotlinReferenceProvidersService and KotlinReferenceProviderContributor a project services
There is no need to have those services on application level
This commit is contained in:
+3
-6
@@ -37,14 +37,11 @@ object AnalysisApiFe10TestServiceRegistrar : AnalysisApiTestServiceRegistrar() {
|
||||
registerService(ModuleVisibilityManager::class.java, CliModuleVisibilityManagerImpl(enabled = true))
|
||||
|
||||
registerService(ReadWriteAccessChecker::class.java, ReadWriteAccessCheckerDescriptorsImpl())
|
||||
registerService(KotlinReferenceProviderContributor::class.java, KtFe10KotlinReferenceProviderContributor::class.java)
|
||||
}
|
||||
AnalysisHandlerExtension.registerExtension(project, KtFe10AnalysisHandlerExtension())
|
||||
KotlinCoreEnvironment.registerKotlinLightClassSupport(project)
|
||||
}
|
||||
|
||||
override fun registerApplicationServices(application: MockApplication, testServices: TestServices) {
|
||||
application.apply {
|
||||
registerService(KotlinReferenceProviderContributor::class.java, KtFe10KotlinReferenceProviderContributor::class.java)
|
||||
}
|
||||
}
|
||||
}
|
||||
override fun registerApplicationServices(application: MockApplication, testServices: TestServices) {}
|
||||
}
|
||||
|
||||
+4
-3
@@ -5,6 +5,7 @@
|
||||
|
||||
package org.jetbrains.kotlin.analysis.api.impl.base.references
|
||||
|
||||
import com.intellij.openapi.project.Project
|
||||
import com.intellij.psi.ContributedReferenceHost
|
||||
import com.intellij.psi.PsiElement
|
||||
import com.intellij.psi.PsiReference
|
||||
@@ -21,13 +22,13 @@ import org.jetbrains.kotlin.idea.references.KotlinReferenceProviderContributor
|
||||
import org.jetbrains.kotlin.psi.KotlinReferenceProvidersService
|
||||
import org.jetbrains.kotlin.utils.SmartList
|
||||
|
||||
class HLApiReferenceProviderService : KotlinReferenceProvidersService() {
|
||||
class HLApiReferenceProviderService(val project: Project) : KotlinReferenceProvidersService() {
|
||||
private val originalProvidersBinding: MultiMap<Class<out PsiElement>, KotlinPsiReferenceProvider>
|
||||
private val providersBindingCache: Map<Class<out PsiElement>, List<KotlinPsiReferenceProvider>>
|
||||
|
||||
init {
|
||||
val registrar = KotlinPsiReferenceRegistrar()
|
||||
KotlinReferenceProviderContributor.getInstance().registerReferenceProviders(registrar)
|
||||
KotlinReferenceProviderContributor.getInstance(project).registerReferenceProviders(registrar)
|
||||
originalProvidersBinding = registrar.providers
|
||||
|
||||
providersBindingCache = ConcurrentFactoryMap.createMap<Class<out PsiElement>, List<KotlinPsiReferenceProvider>> { klass ->
|
||||
@@ -69,4 +70,4 @@ class HLApiReferenceProviderService : KotlinReferenceProvidersService() {
|
||||
)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
+4
-9
@@ -28,8 +28,7 @@ import org.jetbrains.kotlin.psi.KtFile
|
||||
import org.jetbrains.kotlin.test.services.TestServices
|
||||
|
||||
object AnalysisApiBaseTestServiceRegistrar: AnalysisApiTestServiceRegistrar() {
|
||||
override fun registerProjectExtensionPoints(project: MockProject, testServices: TestServices) {
|
||||
}
|
||||
override fun registerProjectExtensionPoints(project: MockProject, testServices: TestServices) {}
|
||||
|
||||
@OptIn(KtAnalysisApiInternals::class)
|
||||
override fun registerProjectServices(project: MockProject, testServices: TestServices) {
|
||||
@@ -43,13 +42,9 @@ object AnalysisApiBaseTestServiceRegistrar: AnalysisApiTestServiceRegistrar() {
|
||||
registerService(KotlinAnnotationsResolverFactory::class.java, KotlinStaticAnnotationsResolverFactory(allKtFiles))
|
||||
registerService(KotlinDeclarationProviderFactory::class.java, KotlinStaticDeclarationProviderFactory(allKtFiles))
|
||||
registerService(KotlinPackageProviderFactory::class.java, KotlinStaticPackageProviderFactory(allKtFiles))
|
||||
}
|
||||
}
|
||||
|
||||
override fun registerApplicationServices(application: MockApplication, testServices: TestServices) {
|
||||
application.apply {
|
||||
registerService(KotlinReferenceProvidersService::class.java, HLApiReferenceProviderService::class.java)
|
||||
}
|
||||
|
||||
}
|
||||
}
|
||||
|
||||
override fun registerApplicationServices(application: MockApplication, testServices: TestServices) {}
|
||||
}
|
||||
|
||||
+4
-9
@@ -94,13 +94,6 @@ public class StandaloneAnalysisAPISessionBuilder(
|
||||
)
|
||||
}
|
||||
|
||||
private fun registerApplicationServices() {
|
||||
kotlinCoreProjectEnvironment.environment.application.apply {
|
||||
registerService(KotlinReferenceProvidersService::class.java, HLApiReferenceProviderService::class.java)
|
||||
registerService(KotlinReferenceProviderContributor::class.java, KotlinFirReferenceContributor::class.java)
|
||||
}
|
||||
}
|
||||
|
||||
public fun <T : Any> registerApplicationService(serviceInterface: Class<T>, serviceImplementation: T) {
|
||||
kotlinCoreProjectEnvironment.environment.application.apply {
|
||||
registerService(serviceInterface, serviceImplementation)
|
||||
@@ -140,6 +133,10 @@ public class StandaloneAnalysisAPISessionBuilder(
|
||||
registerService(LLFirBuiltinsSessionFactory::class.java, LLFirBuiltinsSessionFactory(this))
|
||||
RegisterComponentService.registerLLFirLibrarySessionFactory(this)
|
||||
RegisterComponentService.registerLLFirNonUnderContentRootSessionFactory(this)
|
||||
|
||||
registerService(KotlinReferenceProvidersService::class.java, HLApiReferenceProviderService::class.java)
|
||||
registerService(KotlinReferenceProviderContributor::class.java, KotlinFirReferenceContributor::class.java)
|
||||
|
||||
RegisterComponentService.registerLLFirResolveSessionService(this)
|
||||
registerService(
|
||||
PackagePartProviderFactory::class.java,
|
||||
@@ -203,8 +200,6 @@ public class StandaloneAnalysisAPISessionBuilder(
|
||||
allSourceFiles,
|
||||
)
|
||||
|
||||
registerApplicationServices()
|
||||
|
||||
val project = kotlinCoreProjectEnvironment.project
|
||||
val ktFiles = allSourceFiles.filterIsInstance<KtFile>()
|
||||
val libraryRoots = StandaloneProjectFactory.getAllBinaryRoots(modules, kotlinCoreProjectEnvironment)
|
||||
|
||||
+12
@@ -79,6 +79,8 @@ public fun configureApplicationEnvironment(app: MockApplication) {
|
||||
* * [KotlinDeclarationProviderFactory]
|
||||
* * [KotlinPackageProviderFactory]
|
||||
* * [PackagePartProviderFactory]
|
||||
* * [KotlinReferenceProvidersService]
|
||||
* * [KotlinReferenceProviderContributor]
|
||||
*
|
||||
* Note that [ProjectStructureProvider] is built by using
|
||||
* * given [ktFiles] as Kotlin sources
|
||||
@@ -122,6 +124,16 @@ internal fun configureProjectEnvironment(
|
||||
KotlinStaticAnnotationsResolverFactory(ktFiles)
|
||||
)
|
||||
|
||||
project.registerService(
|
||||
KotlinReferenceProvidersService::class.java,
|
||||
HLApiReferenceProviderService::class.java
|
||||
)
|
||||
|
||||
project.registerService(
|
||||
KotlinReferenceProviderContributor::class.java,
|
||||
KotlinFirReferenceContributor::class.java
|
||||
)
|
||||
|
||||
RegisterComponentService.registerLLFirResolveSessionService(project)
|
||||
project.picoContainer.registerComponentInstance(
|
||||
FirSealedClassInheritorsProcessorFactory::class.qualifiedName,
|
||||
|
||||
+4
-3
@@ -6,6 +6,7 @@
|
||||
package org.jetbrains.kotlin.idea.references
|
||||
|
||||
import com.intellij.openapi.components.ServiceManager
|
||||
import com.intellij.openapi.project.Project
|
||||
import com.intellij.psi.PsiElement
|
||||
import com.intellij.psi.PsiReference
|
||||
import com.intellij.util.containers.MultiMap
|
||||
@@ -19,8 +20,8 @@ interface KotlinReferenceProviderContributor {
|
||||
fun registerReferenceProviders(registrar: KotlinPsiReferenceRegistrar)
|
||||
|
||||
companion object {
|
||||
fun getInstance(): KotlinReferenceProviderContributor =
|
||||
ServiceManager.getService(KotlinReferenceProviderContributor::class.java)
|
||||
fun getInstance(project: Project): KotlinReferenceProviderContributor =
|
||||
ServiceManager.getService(project, KotlinReferenceProviderContributor::class.java)
|
||||
}
|
||||
}
|
||||
|
||||
@@ -47,4 +48,4 @@ class KotlinPsiReferenceRegistrar {
|
||||
fun registerMultiProvider(klass: Class<out PsiElement>, provider: KotlinPsiReferenceProvider) {
|
||||
providers.putValue(klass, provider)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
+2
-5
@@ -54,6 +54,7 @@ object AnalysisApiFirTestServiceRegistrar : AnalysisApiTestServiceRegistrar() {
|
||||
registerService(KotlinAsJavaSupport::class.java, SymbolKotlinAsJavaSupport(project))
|
||||
registerService(KtCompilerPluginsProvider::class.java, KtCompilerPluginsProviderForTests(project))
|
||||
registerService(ReadWriteAccessChecker::class.java, ReadWriteAccessCheckerFirImpl())
|
||||
registerService(KotlinReferenceProviderContributor::class.java, KotlinFirReferenceContributor::class.java)
|
||||
}
|
||||
|
||||
with(PsiElementFinder.EP.getPoint(project)) {
|
||||
@@ -62,9 +63,5 @@ object AnalysisApiFirTestServiceRegistrar : AnalysisApiTestServiceRegistrar() {
|
||||
}
|
||||
}
|
||||
|
||||
override fun registerApplicationServices(application: MockApplication, testServices: TestServices) {
|
||||
application.apply {
|
||||
registerService(KotlinReferenceProviderContributor::class.java, KotlinFirReferenceContributor::class.java)
|
||||
}
|
||||
}
|
||||
override fun registerApplicationServices(application: MockApplication, testServices: TestServices) {}
|
||||
}
|
||||
|
||||
@@ -6,6 +6,7 @@
|
||||
package org.jetbrains.kotlin.psi
|
||||
|
||||
import com.intellij.openapi.components.ServiceManager
|
||||
import com.intellij.openapi.project.Project
|
||||
import com.intellij.psi.PsiElement
|
||||
import com.intellij.psi.PsiReference
|
||||
|
||||
@@ -16,13 +17,13 @@ open class KotlinReferenceProvidersService {
|
||||
private val NO_REFERENCES_SERVICE = KotlinReferenceProvidersService()
|
||||
|
||||
@JvmStatic
|
||||
fun getInstance(): KotlinReferenceProvidersService {
|
||||
return ServiceManager.getService(KotlinReferenceProvidersService::class.java) ?: NO_REFERENCES_SERVICE
|
||||
fun getInstance(project: Project): KotlinReferenceProvidersService {
|
||||
return project.getService(KotlinReferenceProvidersService::class.java) ?: NO_REFERENCES_SERVICE
|
||||
}
|
||||
|
||||
@JvmStatic
|
||||
fun getReferencesFromProviders(psiElement: PsiElement): Array<PsiReference> {
|
||||
return getInstance().getReferences(psiElement)
|
||||
return getInstance(psiElement.project).getReferences(psiElement)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user