Allopen: Handle allopen annotations on classes (KT-12149)
This commit is contained in:
committed by
Yan Zhulanow
parent
58f408e173
commit
f57df272ac
@@ -131,38 +131,39 @@ public class ImplementationBodyCodegen extends ClassBodyCodegen {
|
||||
boolean isAnnotation = false;
|
||||
boolean isEnum = false;
|
||||
|
||||
if (myClass instanceof KtClass) {
|
||||
KtClass ktClass = (KtClass) myClass;
|
||||
if (ktClass.hasModifier(KtTokens.ABSTRACT_KEYWORD) || ktClass.isSealed()) {
|
||||
ClassKind kind = descriptor.getKind();
|
||||
|
||||
if (kind == ClassKind.OBJECT) {
|
||||
isStatic = isCompanionObject(descriptor);
|
||||
isFinal = true;
|
||||
}
|
||||
else {
|
||||
Modality modality = descriptor.getModality();
|
||||
|
||||
if (modality == Modality.ABSTRACT || modality == Modality.SEALED) {
|
||||
isAbstract = true;
|
||||
}
|
||||
if (ktClass.isInterface()) {
|
||||
|
||||
if (kind == ClassKind.INTERFACE) {
|
||||
isAbstract = true;
|
||||
isInterface = true;
|
||||
}
|
||||
else if (descriptor.getKind() == ClassKind.ANNOTATION_CLASS) {
|
||||
else if (kind == ClassKind.ANNOTATION_CLASS) {
|
||||
isAbstract = true;
|
||||
isInterface = true;
|
||||
isAnnotation = true;
|
||||
}
|
||||
else if (ktClass.isEnum()) {
|
||||
else if (kind == ClassKind.ENUM_CLASS) {
|
||||
isAbstract = hasAbstractMembers(descriptor);
|
||||
isEnum = true;
|
||||
}
|
||||
|
||||
if (isObject(descriptor)) {
|
||||
isFinal = true;
|
||||
if (modality != Modality.OPEN && !isAbstract) {
|
||||
// Light-class mode: Do not make enum classes final since PsiClass corresponding to enum is expected to be inheritable from
|
||||
isFinal = !(kind == ClassKind.ENUM_CLASS && !state.getClassBuilderMode().generateBodies);
|
||||
}
|
||||
|
||||
if (!ktClass.hasModifier(KtTokens.OPEN_KEYWORD) && !isAbstract) {
|
||||
// Light-class mode: Do not make enum classes final since PsiClass corresponding to enum is expected to be inheritable from
|
||||
isFinal = !(ktClass.isEnum() && !state.getClassBuilderMode().generateBodies);
|
||||
}
|
||||
isStatic = !ktClass.isInner();
|
||||
}
|
||||
else {
|
||||
isStatic = isCompanionObject(descriptor);
|
||||
isFinal = true;
|
||||
isStatic = !descriptor.isInner();
|
||||
}
|
||||
|
||||
int access = 0;
|
||||
|
||||
@@ -82,6 +82,7 @@ import org.jetbrains.kotlin.codegen.extensions.ClassBuilderInterceptorExtension
|
||||
import org.jetbrains.kotlin.codegen.extensions.ExpressionCodegenExtension
|
||||
import org.jetbrains.kotlin.compiler.plugin.ComponentRegistrar
|
||||
import org.jetbrains.kotlin.config.*
|
||||
import org.jetbrains.kotlin.extensions.DeclarationAttributeAltererExtension
|
||||
import org.jetbrains.kotlin.extensions.StorageComponentContainerContributor
|
||||
import org.jetbrains.kotlin.idea.KotlinFileType
|
||||
import org.jetbrains.kotlin.load.kotlin.JvmVirtualFileFinderFactory
|
||||
@@ -170,6 +171,7 @@ class KotlinCoreEnvironment private constructor(
|
||||
AnalysisHandlerExtension.registerExtensionPoint(project)
|
||||
PackageFragmentProviderExtension.registerExtensionPoint(project)
|
||||
StorageComponentContainerContributor.registerExtensionPoint(project)
|
||||
DeclarationAttributeAltererExtension.registerExtensionPoint(project)
|
||||
|
||||
for (registrar in configuration.getList(ComponentRegistrar.PLUGIN_COMPONENT_REGISTRARS)) {
|
||||
registrar.registerProjectComponents(project, configuration)
|
||||
|
||||
+40
@@ -0,0 +1,40 @@
|
||||
/*
|
||||
* Copyright 2010-2016 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.extensions
|
||||
|
||||
import org.jetbrains.kotlin.descriptors.DeclarationDescriptor
|
||||
import org.jetbrains.kotlin.descriptors.Modality
|
||||
import org.jetbrains.kotlin.psi.KtModifierListOwner
|
||||
import org.jetbrains.kotlin.resolve.BindingContext
|
||||
|
||||
interface DeclarationAttributeAltererExtension {
|
||||
companion object : ProjectExtensionDescriptor<DeclarationAttributeAltererExtension>(
|
||||
"org.jetbrains.kotlin.declarationAttributeAltererExtension",
|
||||
DeclarationAttributeAltererExtension::class.java
|
||||
)
|
||||
|
||||
/**
|
||||
* Returns the new modality for the [declaration], or null if the [currentModality] is good enough.
|
||||
*/
|
||||
fun refineDeclarationModality(
|
||||
modifierListOwner: KtModifierListOwner,
|
||||
declaration: DeclarationDescriptor?,
|
||||
containingDeclaration: DeclarationDescriptor?,
|
||||
currentModality: Modality,
|
||||
bindingContext: BindingContext
|
||||
): Modality? = null
|
||||
}
|
||||
@@ -780,7 +780,8 @@ public class DescriptorResolver {
|
||||
boolean hasBody = hasBody(property);
|
||||
Visibility visibility = resolveVisibilityFromModifiers(property, getDefaultVisibility(property, containingDeclaration));
|
||||
Modality modality = containingDeclaration instanceof ClassDescriptor
|
||||
? resolveMemberModalityFromModifiers(property, getDefaultModality(containingDeclaration, visibility, hasBody))
|
||||
? resolveMemberModalityFromModifiers(property, getDefaultModality(containingDeclaration, visibility, hasBody),
|
||||
trace.getBindingContext(), containingDeclaration)
|
||||
: Modality.FINAL;
|
||||
|
||||
final AnnotationSplitter.PropertyWrapper wrapper = new AnnotationSplitter.PropertyWrapper(property);
|
||||
@@ -923,7 +924,7 @@ public class DescriptorResolver {
|
||||
@NotNull KtProperty property,
|
||||
@NotNull PropertyDescriptor propertyDescriptor,
|
||||
@NotNull AnnotationSplitter annotationSplitter,
|
||||
BindingTrace trace
|
||||
@NotNull BindingTrace trace
|
||||
) {
|
||||
KtPropertyAccessor setter = property.getSetter();
|
||||
PropertySetterDescriptorImpl setterDescriptor = null;
|
||||
@@ -935,7 +936,8 @@ public class DescriptorResolver {
|
||||
|
||||
setterDescriptor = new PropertySetterDescriptorImpl(
|
||||
propertyDescriptor, annotations,
|
||||
resolveMemberModalityFromModifiers(setter, propertyDescriptor.getModality()),
|
||||
resolveMemberModalityFromModifiers(setter, propertyDescriptor.getModality(),
|
||||
trace.getBindingContext(), propertyDescriptor.getContainingDeclaration()),
|
||||
resolveVisibilityFromModifiers(setter, propertyDescriptor.getVisibility()),
|
||||
/* isDefault = */ false, setter.hasModifier(EXTERNAL_KEYWORD),
|
||||
property.hasModifier(KtTokens.INLINE_KEYWORD) || setter.hasModifier(KtTokens.INLINE_KEYWORD),
|
||||
@@ -1001,7 +1003,7 @@ public class DescriptorResolver {
|
||||
@NotNull KtProperty property,
|
||||
@NotNull PropertyDescriptor propertyDescriptor,
|
||||
@NotNull AnnotationSplitter annotationSplitter,
|
||||
BindingTrace trace,
|
||||
@NotNull BindingTrace trace,
|
||||
@Nullable KotlinType propertyTypeIfKnown
|
||||
) {
|
||||
PropertyGetterDescriptorImpl getterDescriptor;
|
||||
@@ -1014,7 +1016,8 @@ public class DescriptorResolver {
|
||||
|
||||
getterDescriptor = new PropertyGetterDescriptorImpl(
|
||||
propertyDescriptor, getterAnnotations,
|
||||
resolveMemberModalityFromModifiers(getter, propertyDescriptor.getModality()),
|
||||
resolveMemberModalityFromModifiers(getter, propertyDescriptor.getModality(),
|
||||
trace.getBindingContext(), propertyDescriptor.getContainingDeclaration()),
|
||||
resolveVisibilityFromModifiers(getter, propertyDescriptor.getVisibility()),
|
||||
/* isDefault = */ false, getter.hasModifier(EXTERNAL_KEYWORD),
|
||||
property.hasModifier(KtTokens.INLINE_KEYWORD) || getter.hasModifier(KtTokens.INLINE_KEYWORD),
|
||||
@@ -1120,7 +1123,7 @@ public class DescriptorResolver {
|
||||
PropertyDescriptorImpl propertyDescriptor = PropertyDescriptorImpl.create(
|
||||
classDescriptor,
|
||||
propertyAnnotations,
|
||||
resolveMemberModalityFromModifiers(parameter, Modality.FINAL),
|
||||
resolveMemberModalityFromModifiers(parameter, Modality.FINAL, trace.getBindingContext(), classDescriptor),
|
||||
resolveVisibilityFromModifiers(parameter, getDefaultVisibility(parameter, classDescriptor)),
|
||||
isMutable,
|
||||
name,
|
||||
|
||||
@@ -168,7 +168,8 @@ class FunctionDescriptorResolver(
|
||||
val returnType = function.typeReference?.let { typeResolver.resolveType(headerScope, it, trace, true) }
|
||||
|
||||
val visibility = resolveVisibilityFromModifiers(function, getDefaultVisibility(function, containingDescriptor))
|
||||
val modality = resolveMemberModalityFromModifiers(function, getDefaultModality(containingDescriptor, visibility, function.hasBody()))
|
||||
val modality = resolveMemberModalityFromModifiers(function, getDefaultModality(containingDescriptor, visibility, function.hasBody()),
|
||||
trace.bindingContext, containingDescriptor)
|
||||
functionDescriptor.initialize(
|
||||
receiverType,
|
||||
getDispatchReceiverParameterIfNeeded(containingDescriptor),
|
||||
|
||||
@@ -24,6 +24,7 @@ import org.jetbrains.annotations.Nullable;
|
||||
import org.jetbrains.kotlin.config.LanguageVersionSettings;
|
||||
import org.jetbrains.kotlin.descriptors.*;
|
||||
import org.jetbrains.kotlin.diagnostics.DiagnosticFactory1;
|
||||
import org.jetbrains.kotlin.extensions.DeclarationAttributeAltererExtension;
|
||||
import org.jetbrains.kotlin.lexer.KtKeywordToken;
|
||||
import org.jetbrains.kotlin.lexer.KtModifierKeywordToken;
|
||||
import org.jetbrains.kotlin.lexer.KtTokens;
|
||||
@@ -102,15 +103,50 @@ public class ModifiersChecker {
|
||||
|
||||
@NotNull
|
||||
public static Modality resolveMemberModalityFromModifiers(
|
||||
@NotNull KtModifierListOwner modifierListOwner,
|
||||
@NotNull Modality defaultModality
|
||||
@Nullable KtModifierListOwner modifierListOwner,
|
||||
@NotNull Modality defaultModality,
|
||||
@NotNull BindingContext bindingContext,
|
||||
@Nullable DeclarationDescriptor containingDescriptor
|
||||
) {
|
||||
return resolveModalityFromModifiers(modifierListOwner.getModifierList(), defaultModality, /* allowSealed = */ false);
|
||||
return resolveModalityFromModifiers(modifierListOwner, defaultModality,
|
||||
bindingContext, containingDescriptor, /* allowSealed = */ false);
|
||||
}
|
||||
|
||||
@NotNull
|
||||
public static Modality resolveModalityFromModifiers(
|
||||
@Nullable KtModifierList modifierList, @NotNull Modality defaultModality, boolean allowSealed
|
||||
@Nullable KtModifierListOwner modifierListOwner,
|
||||
@NotNull Modality defaultModality,
|
||||
@NotNull BindingContext bindingContext,
|
||||
@Nullable DeclarationDescriptor containingDescriptor,
|
||||
boolean allowSealed
|
||||
) {
|
||||
KtModifierList modifierList = (modifierListOwner != null) ? modifierListOwner.getModifierList() : null;
|
||||
Modality modality = resolveModalityFromModifiers(modifierList, defaultModality, allowSealed);
|
||||
|
||||
if (modifierListOwner != null) {
|
||||
Collection<DeclarationAttributeAltererExtension> extensions =
|
||||
DeclarationAttributeAltererExtension.Companion.getInstances(modifierListOwner.getProject());
|
||||
|
||||
DeclarationDescriptor descriptor = bindingContext.get(BindingContext.DECLARATION_TO_DESCRIPTOR, modifierListOwner);
|
||||
for (DeclarationAttributeAltererExtension extension : extensions) {
|
||||
Modality newModality = extension.refineDeclarationModality(
|
||||
modifierListOwner, descriptor, containingDescriptor, modality, bindingContext);
|
||||
|
||||
if (newModality != null) {
|
||||
modality = newModality;
|
||||
break;
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return modality;
|
||||
}
|
||||
|
||||
@NotNull
|
||||
private static Modality resolveModalityFromModifiers(
|
||||
@Nullable KtModifierList modifierList,
|
||||
@NotNull Modality defaultModality,
|
||||
boolean allowSealed
|
||||
) {
|
||||
if (modifierList == null) return defaultModality;
|
||||
boolean hasAbstractModifier = modifierList.hasModifier(ABSTRACT_KEYWORD);
|
||||
|
||||
+20
-7
@@ -83,7 +83,7 @@ public class LazyClassDescriptor extends ClassDescriptorBase implements ClassDes
|
||||
private final ClassMemberDeclarationProvider declarationProvider;
|
||||
|
||||
private final LazyClassTypeConstructor typeConstructor;
|
||||
private final Modality modality;
|
||||
private final NotNullLazyValue<Modality> modality;
|
||||
private final Visibility visibility;
|
||||
private final ClassKind kind;
|
||||
private final boolean isInner;
|
||||
@@ -120,7 +120,7 @@ public class LazyClassDescriptor extends ClassDescriptorBase implements ClassDes
|
||||
);
|
||||
this.c = c;
|
||||
|
||||
KtClassOrObject classOrObject = classLikeInfo.getCorrespondingClassOrObject();
|
||||
final KtClassOrObject classOrObject = classLikeInfo.getCorrespondingClassOrObject();
|
||||
if (classOrObject != null) {
|
||||
this.c.getTrace().record(BindingContext.CLASS, classOrObject, this);
|
||||
}
|
||||
@@ -138,13 +138,26 @@ public class LazyClassDescriptor extends ClassDescriptorBase implements ClassDes
|
||||
|
||||
this.isCompanionObject = classLikeInfo instanceof KtObjectInfo && ((KtObjectInfo) classLikeInfo).isCompanionObject();
|
||||
|
||||
KtModifierList modifierList = classLikeInfo.getModifierList();
|
||||
final KtModifierList modifierList = classLikeInfo.getModifierList();
|
||||
if (kind.isSingleton()) {
|
||||
this.modality = Modality.FINAL;
|
||||
this.modality = storageManager.createLazyValue(new Function0<Modality>() {
|
||||
@Override
|
||||
public Modality invoke() {
|
||||
return Modality.FINAL;
|
||||
}
|
||||
});
|
||||
}
|
||||
else {
|
||||
Modality defaultModality = kind == ClassKind.INTERFACE ? Modality.ABSTRACT : Modality.FINAL;
|
||||
this.modality = resolveModalityFromModifiers(modifierList, defaultModality, /* allowSealed = */ true);
|
||||
final Modality defaultModality = kind == ClassKind.INTERFACE ? Modality.ABSTRACT : Modality.FINAL;
|
||||
this.modality = storageManager.createLazyValue(new Function0<Modality>() {
|
||||
@Override
|
||||
public Modality invoke() {
|
||||
return resolveModalityFromModifiers(classOrObject, defaultModality,
|
||||
c.getTrace().getBindingContext(),
|
||||
null,
|
||||
/* allowSealed = */ true);
|
||||
}
|
||||
});
|
||||
}
|
||||
|
||||
boolean isLocal = classOrObject != null && KtPsiUtil.isLocal(classOrObject);
|
||||
@@ -473,7 +486,7 @@ public class LazyClassDescriptor extends ClassDescriptorBase implements ClassDes
|
||||
@NotNull
|
||||
@Override
|
||||
public Modality getModality() {
|
||||
return modality;
|
||||
return modality.invoke();
|
||||
}
|
||||
|
||||
@NotNull
|
||||
|
||||
Reference in New Issue
Block a user