azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[Analysis API] Extract dangling file module computation

Browse Source 
Use the same code for creating dangling file modules inside tests,
the stand-alone API, and in the IDE.
This commit is contained in:
Yan Zhulanow
2023-11-13 21:19:24 +09:00
committed by Space Team
parent 441f3e0209
commit 2899822102
4 changed files with 90 additions and 45 deletions
Download Patch File Download Diff File Expand all files Collapse all files
analysis/project-structure/src/org/jetbrains/kotlin/analysis/project/structure/ProjectStructureProvider.kt
+47 -1
View File
@@ -7,8 +7,14 @@ package org.jetbrains.kotlin.analysis.project.structure
import com.intellij.openapi.project.Project
import com.intellij.psi.PsiElement
import com.intellij.psi.PsiFile
import org.jetbrains.kotlin.analysis.project.structure.impl.KtDanglingFileModuleImpl
import org.jetbrains.kotlin.config.LanguageVersionSettings
import org.jetbrains.kotlin.config.LanguageVersionSettingsImpl
import org.jetbrains.kotlin.psi.KtCodeFragment
import org.jetbrains.kotlin.psi.KtFile
import org.jetbrains.kotlin.psi.analysisContext
import org.jetbrains.kotlin.psi.doNotAnalyze
public abstract class ProjectStructureProvider {
/**
@@ -34,6 +40,39 @@ public abstract class ProjectStructureProvider {
*/
public abstract fun getModule(element: PsiElement, contextualModule: KtModule?): KtModule
protected abstract fun getNotUnderContentRootModule(project: Project, file: PsiFile?): KtNotUnderContentRootModule
@OptIn(KtModuleStructureInternals::class)
protected fun computeSpecialModule(file: PsiFile): KtModule? {
val virtualFile = file.virtualFile
if (virtualFile != null) {
val contextModule = virtualFile.analysisExtensionFileContextModule
if (contextModule != null) {
return contextModule
}
}
if (file is KtFile && file.isDangling) {
val contextModule = computeContextModule(file)
return KtDanglingFileModuleImpl(file, contextModule)
}
return null
}
@OptIn(KtModuleStructureInternals::class)
private fun computeContextModule(file: KtFile): KtModule {
val contextElement = file.context
?: file.analysisContext
?: file.originalFile.takeIf { it !== file }
if (contextElement != null) {
return getModule(contextElement, contextualModule = null)
}
return getNotUnderContentRootModule(file.project, file = null)
}
/**
* Project-global [LanguageVersionSettings] for source modules lacking explicit settings (such as [KtNotUnderContentRootModule]).
*/
@@ -55,4 +94,11 @@ public abstract class ProjectStructureProvider {
return getInstance(project).getModule(element, contextualModule)
}
}
}
}
@OptIn(KtModuleStructureInternals::class)
private val KtFile.isDangling: Boolean
get() = this is KtCodeFragment
|| !isPhysical
|| analysisContext != null
|| doNotAnalyze != null