Fix CodegenFileClassesProvider problems related to wrong/incomplete code in IDE

This commit is contained in:
Dmitry Petrov
2015-09-09 18:44:43 +03:00
parent bf1c0260f1
commit 0839d78797
3 changed files with 5 additions and 55 deletions
@@ -24,44 +24,12 @@ import org.jetbrains.kotlin.psi.JetFile
import org.jetbrains.kotlin.resolve.BindingContext
import java.util.*
public class CodegenFileClassesProvider private constructor() : JvmFileClassesProvider {
public class CodegenFileClassesProvider() : JvmFileClassesProvider {
private val fileParts = hashMapOf<JetFile, JvmFileClassInfo>()
override fun getFileClassFqName(file: JetFile): FqName =
getFileClassInfo(file).fileClassFqName
public fun getFileClassInfo(file: JetFile): JvmFileClassInfo =
fileParts.getOrPut(file) { createFileClassInfo(file) }
private fun createFileClassInfo(file: JetFile): JvmFileClassInfo =
JvmFileClassUtil.getFileClassInfoNoResolve(file)
internal fun addFileClassInfo(file: JetFile) {
if (fileParts.containsKey(file)) return
fileParts[file] = createFileClassInfo(file)
}
companion object {
public @jvmStatic fun createForCodegenTask(
bindingContext: BindingContext,
files: Collection<JetFile>,
packagesWithObsoleteParts: Collection<FqName>,
multifileFacadesWithObsoleteParts: Collection<FqName>
) : CodegenFileClassesProvider {
val codegenFileClassesManager = CodegenFileClassesProvider()
files.forEach {
codegenFileClassesManager.addFileClassInfo(it)
}
val packagesToProcess = HashSet<FqName>(packagesWithObsoleteParts)
packagesToProcess.addAll(multifileFacadesWithObsoleteParts.map { it.parent() })
for (packageFqName in packagesToProcess) {
bindingContext.get(BindingContext.PACKAGE_TO_FILES, packageFqName)?.forEach {
codegenFileClassesManager.addFileClassInfo(it)
}
}
return codegenFileClassesManager
}
}
}
fileParts.getOrPut(file) { JvmFileClassUtil.getFileClassInfoNoResolve(file) }
}
@@ -82,9 +82,7 @@ public class GenerationState jvmOverloads constructor(
}
}
public val fileClassesProvider: CodegenFileClassesProvider =
CodegenFileClassesProvider.createForCodegenTask(bindingContext, files, packagesWithObsoleteParts,
/* TODO */ multifileFacadesWithObsoleteParts = emptySet())
public val fileClassesProvider: CodegenFileClassesProvider = CodegenFileClassesProvider()
public val classBuilderMode: ClassBuilderMode = builderFactory.getClassBuilderMode()
public val bindingTrace: BindingTrace = DelegatingBindingTrace(bindingContext, "trace in GenerationState")
@@ -17,13 +17,10 @@
package org.jetbrains.kotlin.fileClasses
import org.jetbrains.kotlin.descriptors.annotations.AnnotationDescriptor
import org.jetbrains.kotlin.descriptors.annotations.Annotations
import org.jetbrains.kotlin.descriptors.annotations.AnnotationsImpl
import org.jetbrains.kotlin.load.kotlin.PackagePartClassUtils
import org.jetbrains.kotlin.name.FqName
import org.jetbrains.kotlin.name.Name
import org.jetbrains.kotlin.psi.*
import org.jetbrains.kotlin.resolve.BindingContext
import org.jetbrains.kotlin.resolve.constants.StringValue
public object JvmFileClassUtil {
@@ -58,12 +55,6 @@ public object JvmFileClassUtil {
public @JvmStatic fun manglePartName(facadeName: String, fileName: String): String =
"1${PackagePartClassUtils.getFilePartShortName(fileName)}"
public @JvmStatic fun parseJvmFileClass(annotations: Annotations): ParsedJmvFileClassAnnotations? {
val jvmName = annotations.findAnnotation(JVM_NAME) ?: return null
val jvmMultifileClass = annotations.findAnnotation(JVM_MULTIFILE_CLASS)
return parseJvmFileClass(jvmName, jvmMultifileClass)
}
public @JvmStatic fun parseJvmFileClass(jvmName: AnnotationDescriptor, jvmMultifileClass: AnnotationDescriptor?): ParsedJmvFileClassAnnotations? {
val jvmNameArgument = jvmName.allValueArguments.values().singleOrNull() ?: return null
val name = (jvmNameArgument as? StringValue)?.value ?: return null
@@ -78,6 +69,7 @@ public object JvmFileClassUtil {
val jvmName = findAnnotationEntryOnFileNoResolve(file, JVM_NAME_SHORT) ?: return null
val nameExpr = jvmName.valueArguments.firstOrNull()?.getArgumentExpression() ?: return null
val name = getLiteralStringFromRestrictedConstExpression(nameExpr) ?: return null
if (!Name.isValidIdentifier(name)) return null
val isMultifileClassPart = findAnnotationEntryOnFileNoResolve(file, JVM_MULTIFILE_CLASS_SHORT) != null
return ParsedJmvFileClassAnnotations(name, isMultifileClassPart)
}
@@ -95,14 +87,6 @@ public object JvmFileClassUtil {
return singleEntry.text
}
public @JvmStatic fun collectFileAnnotations(file: JetFile, bindingContext: BindingContext): Annotations {
val fileAnnotationsList = file.fileAnnotationList ?: return Annotations.EMPTY
val annotationDescriptors = arrayListOf<AnnotationDescriptor>()
for (annotationEntry in fileAnnotationsList.annotationEntries) {
bindingContext.get(BindingContext.ANNOTATION, annotationEntry)?.let { annotationDescriptors.add(it) }
}
return AnnotationsImpl(annotationDescriptors)
}
}
public class ParsedJmvFileClassAnnotations(public val name: String, public val multipleFiles: Boolean)