org.jetbrains.kotlin.psi.KtCodeFragment.kt Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of kotlin-compiler-embeddable Show documentation
Show all versions of kotlin-compiler-embeddable Show documentation
the Kotlin compiler embeddable
/*
* Copyright 2010-2015 JetBrains s.r.o.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.jetbrains.kotlin.psi
import com.intellij.mock.MockProject
import com.intellij.openapi.diagnostic.Logger
import com.intellij.openapi.project.Project
import com.intellij.openapi.util.Key
import com.intellij.psi.*
import com.intellij.psi.JavaCodeFragment.VisibilityChecker
import com.intellij.psi.impl.PsiManagerEx
import com.intellij.psi.impl.source.tree.FileElement
import com.intellij.psi.search.GlobalSearchScope
import com.intellij.psi.tree.IElementType
import com.intellij.testFramework.LightVirtualFile
import com.intellij.util.messages.Topic
import org.jetbrains.kotlin.idea.KotlinFileType
import org.jetbrains.kotlin.psi.psiUtil.getElementTextWithContext
abstract class KtCodeFragment(
private val myProject: Project,
name: String,
text: CharSequence,
imports: String?, // Should be separated by KtCodeFragment.IMPORT_SEPARATOR
elementType: IElementType,
private val context: PsiElement?
) : KtFile(
run {
val psiManager = PsiManager.getInstance(myProject) as PsiManagerEx
psiManager.fileManager.createFileViewProvider(LightVirtualFile(name, KotlinFileType.INSTANCE, text), true)
}, false
), KtCodeFragmentBase {
private var viewProvider = super.getViewProvider() as SingleRootFileViewProvider
private var importDirectiveStrings = LinkedHashSet()
private val fakeContextForJavaFile: PsiElement? by lazy {
this.getCopyableUserData(FAKE_CONTEXT_FOR_JAVA_FILE)?.invoke()
}
init {
@Suppress("LeakingThis")
getViewProvider().forceCachedPsi(this)
init(TokenType.CODE_FRAGMENT, elementType)
if (imports != null) {
appendImports(imports)
}
}
final override fun init(elementType: IElementType, contentElementType: IElementType?) {
super.init(elementType, contentElementType)
}
private var resolveScope: GlobalSearchScope? = null
private var thisType: PsiType? = null
private var superType: PsiType? = null
private var exceptionHandler: JavaCodeFragment.ExceptionHandler? = null
private var isPhysical = true
abstract fun getContentElement(): KtElement?
override fun forceResolveScope(scope: GlobalSearchScope?) {
resolveScope = scope
}
override fun getForcedResolveScope() = resolveScope
override fun isPhysical() = isPhysical
override fun isValid() = true
override fun getContext(): PsiElement? {
if (fakeContextForJavaFile != null) return fakeContextForJavaFile
if (context != null && context !is KtElement) {
val logInfoForContextElement = (context as? PsiFile)?.virtualFile?.path ?: context.getElementTextWithContext()
LOG.warn("CodeFragment with non-kotlin context should have fakeContextForJavaFile set: \noriginalContext = $logInfoForContextElement")
return null
}
return context
}
override fun getResolveScope() = context?.resolveScope ?: super.getResolveScope()
override fun clone(): KtCodeFragment {
val elementClone = calcTreeElement().clone() as FileElement
return (cloneImpl(elementClone) as KtCodeFragment).apply {
isPhysical = false
myOriginalFile = this@KtCodeFragment
importDirectiveStrings = LinkedHashSet([email protected])
viewProvider = SingleRootFileViewProvider(
PsiManager.getInstance(myProject),
LightVirtualFile(name, KotlinFileType.INSTANCE, text),
false
)
viewProvider.forceCachedPsi(this)
}
}
final override fun getViewProvider() = viewProvider
override fun getThisType() = thisType
override fun setThisType(psiType: PsiType?) {
thisType = psiType
}
override fun getSuperType() = superType
override fun setSuperType(superType: PsiType?) {
this.superType = superType
}
override fun importsToString(): String {
return importDirectiveStrings.joinToString(IMPORT_SEPARATOR)
}
override fun addImportsFromString(imports: String?) {
val notifyChanged = viewProvider.isEventSystemEnabled && project !is MockProject
if (imports != null && appendImports(imports)) {
if (notifyChanged) {
// This forces the code fragment to be re-highlighted
add(KtPsiFactory(project).createColon()).delete()
}
// Increment the modification stamp
clearCaches()
if (notifyChanged) {
project.messageBus.syncPublisher(IMPORT_MODIFICATION).onCodeFragmentImportsModification(this)
}
}
}
@Deprecated("Use 'addImportsFromString()w' instead", ReplaceWith("addImportsFromString(import)"), level = DeprecationLevel.WARNING)
fun addImport(import: String) {
addImportsFromString(import)
}
/**
* Parses raw [rawImports] and appends them to the list of code fragment imports.
*
* Import strings must be separated by the [IMPORT_SEPARATOR].
* Each import must be either a qualified name to import (e.g., 'foo.bar'), or a complete text representation of an import directive
* (e.g., 'import foo.bar as baz').
*
* Note that already present import directives will be ignored.
*
* @return `true` if new import directives were added.
*/
private fun appendImports(rawImports: String): Boolean {
if (rawImports.isEmpty()) {
return false
}
var hasNewImports = false
for (rawImport in rawImports.split(IMPORT_SEPARATOR)) {
val importDirectiveString = if (rawImport.startsWith("import ")) rawImport else "import $rawImport"
if (importDirectiveStrings.add(importDirectiveString) && !hasNewImports) {
hasNewImports = true
}
}
return hasNewImports
}
fun importsAsImportList(): KtImportList? {
if (importDirectiveStrings.isNotEmpty() && context != null) {
val ktPsiFactory = KtPsiFactory.contextual(context)
val fileText = importDirectiveStrings.joinToString("\n")
return ktPsiFactory.createFile("imports_for_codeFragment.kt", fileText).importList
}
return null
}
override val importLists: List
get() = listOfNotNull(importsAsImportList())
override val importDirectives: List
get() = importsAsImportList()?.imports ?: emptyList()
override fun setVisibilityChecker(checker: VisibilityChecker?) {}
override fun getVisibilityChecker(): VisibilityChecker = VisibilityChecker.EVERYTHING_VISIBLE
override fun setExceptionHandler(checker: JavaCodeFragment.ExceptionHandler?) {
exceptionHandler = checker
}
override fun getExceptionHandler() = exceptionHandler
fun getContextContainingFile(): KtFile? {
return getOriginalContext()?.takeIf { it.isValid }?.containingKtFile
}
fun getOriginalContext(): KtElement? {
val contextElement = getContext() as? KtElement
val contextFile = contextElement?.containingFile as? KtFile
if (contextFile is KtCodeFragment) {
return contextFile.getOriginalContext()
}
return contextElement
}
companion object {
const val IMPORT_SEPARATOR: String = ","
@Suppress("UnstableApiUsage")
val IMPORT_MODIFICATION: Topic =
Topic(KotlinCodeFragmentImportModificationListener::class.java, Topic.BroadcastDirection.TO_CHILDREN, true)
val FAKE_CONTEXT_FOR_JAVA_FILE: Key> = Key.create("FAKE_CONTEXT_FOR_JAVA_FILE")
private val LOG = Logger.getInstance(KtCodeFragment::class.java)
}
}
fun interface KotlinCodeFragmentImportModificationListener {
fun onCodeFragmentImportsModification(codeFragment: KtCodeFragment)
}