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.resolver;
018    
019    import com.intellij.psi.PsiClass;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
022    import org.jetbrains.jet.lang.resolve.java.DescriptorSearchRule;
023    import org.jetbrains.jet.lang.resolve.java.JvmAbi;
024    import org.jetbrains.jet.lang.resolve.java.kt.JetClassObjectAnnotation;
025    import org.jetbrains.jet.lang.resolve.java.provider.ClassPsiDeclarationProvider;
026    import org.jetbrains.jet.lang.resolve.name.FqName;
027    
028    import javax.inject.Inject;
029    import java.util.ArrayList;
030    import java.util.Collections;
031    import java.util.List;
032    
033    public final class JavaInnerClassResolver {
034    
035        private JavaClassResolver classResolver;
036    
037        public JavaInnerClassResolver() {
038        }
039    
040        @Inject
041        public void setClassResolver(JavaClassResolver classResolver) {
042            this.classResolver = classResolver;
043        }
044    
045        @NotNull
046        public List<ClassDescriptor> resolveInnerClasses(@NotNull ClassPsiDeclarationProvider declarationProvider) {
047            if (declarationProvider.isStaticMembers()) {
048                return Collections.emptyList();
049            }
050    
051            return resolveInnerClasses(declarationProvider.getPsiClass());
052        }
053    
054        @NotNull
055        private List<ClassDescriptor> resolveInnerClasses(@NotNull PsiClass psiClass) {
056            PsiClass[] innerPsiClasses = psiClass.getInnerClasses();
057            List<ClassDescriptor> result = new ArrayList<ClassDescriptor>(innerPsiClasses.length);
058            for (PsiClass innerPsiClass : innerPsiClasses) {
059                if (shouldBeIgnored(innerPsiClass)) {
060                    continue;
061                }
062                ClassDescriptor classDescriptor = resolveInnerClass(innerPsiClass);
063                result.add(classDescriptor);
064            }
065            return result;
066        }
067    
068        private static boolean shouldBeIgnored(PsiClass innerPsiClass) {
069            return JetClassObjectAnnotation.get(innerPsiClass).isDefined();
070        }
071    
072        @NotNull
073        private ClassDescriptor resolveInnerClass(@NotNull PsiClass innerPsiClass) {
074            String name = innerPsiClass.getQualifiedName();
075            assert name != null : "Inner class has no qualified name";
076            ClassDescriptor classDescriptor = classResolver.resolveClass(new FqName(name), DescriptorSearchRule.IGNORE_IF_FOUND_IN_KOTLIN);
077            assert classDescriptor != null : "Couldn't resolve class " + name;
078            return classDescriptor;
079        }
080    }