001 /*
002 * Copyright 2010-2013 JetBrains s.r.o.
003 *
004 * Licensed under the Apache License, Version 2.0 (the "License");
005 * you may not use this file except in compliance with the License.
006 * You may obtain a copy of the License at
007 *
008 * http://www.apache.org/licenses/LICENSE-2.0
009 *
010 * Unless required by applicable law or agreed to in writing, software
011 * distributed under the License is distributed on an "AS IS" BASIS,
012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
013 * See the License for the specific language governing permissions and
014 * limitations under the License.
015 */
016
017 package org.jetbrains.jet.lang.resolve.java;
018
019 import org.jetbrains.annotations.NotNull;
020 import org.jetbrains.jet.lang.descriptors.*;
021 import org.jetbrains.jet.lang.resolve.DescriptorUtils;
022 import org.jetbrains.jet.lang.resolve.java.descriptor.JavaPackageFragmentDescriptor;
023
024 public class JavaVisibilities {
025 private JavaVisibilities() {
026 }
027
028 public static final Visibility PACKAGE_VISIBILITY = new Visibility("package", false) {
029 @Override
030 protected boolean isVisible(@NotNull DeclarationDescriptorWithVisibility what, @NotNull DeclarationDescriptor from) {
031 return areInSamePackage(what, from);
032 }
033
034 @Override
035 protected Integer compareTo(@NotNull Visibility visibility) {
036 if (this == visibility) return 0;
037 if (visibility == Visibilities.PRIVATE) return 1;
038 return -1;
039 }
040
041 @Override
042 public String toString() {
043 return "public/*package*/";
044 }
045
046 @NotNull
047 @Override
048 public Visibility normalize() {
049 return Visibilities.INTERNAL;
050 }
051 };
052
053 public static final Visibility PROTECTED_STATIC_VISIBILITY = new Visibility("protected_static", false) {
054 @Override
055 protected boolean isVisible(@NotNull DeclarationDescriptorWithVisibility what, @NotNull DeclarationDescriptor from) {
056 ClassDescriptor fromClass = DescriptorUtils.getParentOfType(from, ClassDescriptor.class, false);
057 if (fromClass == null) return false;
058
059 ClassDescriptor whatClass;
060 // protected static class
061 if (what instanceof ClassDescriptor) {
062 DeclarationDescriptor containingDeclaration = what.getContainingDeclaration();
063 assert containingDeclaration instanceof ClassDescriptor : "Only static nested classes can have protected_static visibility";
064 whatClass = (ClassDescriptor) containingDeclaration;
065 }
066 // protected static function or property
067 else {
068 DeclarationDescriptor whatDeclarationDescriptor = what.getContainingDeclaration();
069
070 assert whatDeclarationDescriptor instanceof JavaPackageFragmentDescriptor : "Only static declarations can have protected_static visibility";
071 JavaPackageFragmentDescriptor javaPackageFragmentDescriptor = (JavaPackageFragmentDescriptor) whatDeclarationDescriptor;
072
073 whatClass = javaPackageFragmentDescriptor.getJavaDescriptorResolver().resolveClass(
074 javaPackageFragmentDescriptor.getFqName());
075 }
076
077 assert whatClass != null : "Couldn't find ClassDescriptor for protected static member " + what;
078
079 if (DescriptorUtils.isSubclass(fromClass, whatClass)) {
080 return true;
081 }
082 return isVisible(what, fromClass.getContainingDeclaration());
083 }
084
085 @Override
086 public String toString() {
087 return "protected/*protected static*/";
088 }
089
090 @NotNull
091 @Override
092 public Visibility normalize() {
093 return Visibilities.PROTECTED;
094 }
095 };
096
097 public static final Visibility PROTECTED_AND_PACKAGE = new Visibility("protected_and_package", false) {
098 @Override
099 protected boolean isVisible(@NotNull DeclarationDescriptorWithVisibility what, @NotNull DeclarationDescriptor from) {
100 if (areInSamePackage(what, from)) {
101 return true;
102 }
103
104 ClassDescriptor whatClass = DescriptorUtils.getParentOfType(what, ClassDescriptor.class, false);
105 if (whatClass == null) return false;
106
107 ClassDescriptor fromClass = DescriptorUtils.getParentOfType(from, ClassDescriptor.class, false);
108 if (fromClass == null) return false;
109
110 if (DescriptorUtils.isSubclass(fromClass, whatClass)) {
111 return true;
112 }
113 return isVisible(what, fromClass.getContainingDeclaration());
114 }
115
116 @Override
117 protected Integer compareTo(@NotNull Visibility visibility) {
118 if (this == visibility) return 0;
119 if (visibility == Visibilities.INTERNAL) return null;
120 if (visibility == Visibilities.PRIVATE) return 1;
121 return -1;
122 }
123
124 @Override
125 public String toString() {
126 return "protected/*protected and package*/";
127 }
128
129 @NotNull
130 @Override
131 public Visibility normalize() {
132 return Visibilities.PROTECTED;
133 }
134 };
135
136 private static boolean areInSamePackage(@NotNull DeclarationDescriptor first, @NotNull DeclarationDescriptor second) {
137 PackageFragmentDescriptor whatPackage = DescriptorUtils.getParentOfType(first, PackageFragmentDescriptor.class, false);
138 PackageFragmentDescriptor fromPackage = DescriptorUtils.getParentOfType(second, PackageFragmentDescriptor.class, false);
139 return fromPackage != null && whatPackage != null && whatPackage.getFqName().equals(fromPackage.getFqName());
140 }
141 }