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.kotlin;
018
019 import org.jetbrains.annotations.NotNull;
020 import org.jetbrains.annotations.Nullable;
021 import org.jetbrains.jet.descriptors.serialization.ClassData;
022 import org.jetbrains.jet.descriptors.serialization.ClassId;
023 import org.jetbrains.jet.descriptors.serialization.DescriptorFinder;
024 import org.jetbrains.jet.descriptors.serialization.JavaProtoBufUtil;
025 import org.jetbrains.jet.descriptors.serialization.descriptors.DeserializedClassDescriptor;
026 import org.jetbrains.jet.descriptors.serialization.descriptors.DeserializedPackageMemberScope;
027 import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
028 import org.jetbrains.jet.lang.descriptors.PackageFragmentDescriptor;
029 import org.jetbrains.jet.lang.resolve.java.JavaDescriptorResolver;
030 import org.jetbrains.jet.lang.resolve.java.resolver.ErrorReporter;
031 import org.jetbrains.jet.lang.resolve.java.resolver.JavaPackageFragmentProvider;
032 import org.jetbrains.jet.lang.resolve.kotlin.header.KotlinClassHeader;
033 import org.jetbrains.jet.lang.resolve.name.FqName;
034 import org.jetbrains.jet.lang.resolve.name.Name;
035 import org.jetbrains.jet.lang.resolve.scopes.JetScope;
036 import org.jetbrains.jet.storage.StorageManager;
037
038 import javax.inject.Inject;
039 import java.util.Collection;
040
041 import static org.jetbrains.jet.lang.resolve.kotlin.DeserializedResolverUtils.kotlinFqNameToJavaFqName;
042 import static org.jetbrains.jet.lang.resolve.kotlin.header.KotlinClassHeader.Kind.CLASS;
043 import static org.jetbrains.jet.lang.resolve.kotlin.header.KotlinClassHeader.Kind.PACKAGE_FACADE;
044
045 public final class DeserializedDescriptorResolver {
046 private AnnotationDescriptorDeserializer annotationDeserializer;
047
048 private StorageManager storageManager;
049
050 private JavaPackageFragmentProvider javaPackageFragmentProvider;
051
052 private JavaDescriptorResolver javaDescriptorResolver;
053
054 private ErrorReporter errorReporter;
055
056 @NotNull
057 private final DescriptorFinder javaDescriptorFinder = new DescriptorFinder() {
058 @Nullable
059 @Override
060 public ClassDescriptor findClass(@NotNull ClassId classId) {
061 return javaDescriptorResolver.resolveClass(kotlinFqNameToJavaFqName(classId.asSingleFqName()));
062 }
063
064 @NotNull
065 @Override
066 public Collection<Name> getClassNames(@NotNull FqName packageName) {
067 return javaPackageFragmentProvider.getClassNamesInPackage(packageName);
068 }
069 };
070
071 @Inject
072 public void setAnnotationDeserializer(AnnotationDescriptorDeserializer annotationDeserializer) {
073 this.annotationDeserializer = annotationDeserializer;
074 }
075
076 @Inject
077 public void setJavaPackageFragmentProvider(JavaPackageFragmentProvider javaPackageFragmentProvider) {
078 this.javaPackageFragmentProvider = javaPackageFragmentProvider;
079 }
080
081 @Inject
082 public void setJavaDescriptorResolver(JavaDescriptorResolver javaDescriptorResolver) {
083 this.javaDescriptorResolver = javaDescriptorResolver;
084 }
085
086 @Inject
087 public void setErrorReporter(ErrorReporter errorReporter) {
088 this.errorReporter = errorReporter;
089 }
090
091 @Inject
092 public void setStorageManager(StorageManager storageManager) {
093 this.storageManager = storageManager;
094 }
095
096 @Nullable
097 public ClassDescriptor resolveClass(@NotNull KotlinJvmBinaryClass kotlinClass) {
098 String[] data = readData(kotlinClass, CLASS);
099 if (data != null) {
100 ClassData classData = JavaProtoBufUtil.readClassDataFrom(data);
101 return new DeserializedClassDescriptor(storageManager, annotationDeserializer, javaDescriptorFinder,
102 javaPackageFragmentProvider, classData.getNameResolver(), classData.getClassProto());
103 }
104 return null;
105 }
106
107 @Nullable
108 public JetScope createKotlinPackageScope(@NotNull PackageFragmentDescriptor descriptor, @NotNull KotlinJvmBinaryClass kotlinClass) {
109 String[] data = readData(kotlinClass, PACKAGE_FACADE);
110 if (data != null) {
111 return new DeserializedPackageMemberScope(storageManager, descriptor, annotationDeserializer, javaDescriptorFinder,
112 JavaProtoBufUtil.readPackageDataFrom(data));
113 }
114 return null;
115 }
116
117 @Nullable
118 private String[] readData(@NotNull KotlinJvmBinaryClass kotlinClass, @NotNull KotlinClassHeader.Kind expectedKind) {
119 KotlinClassHeader header = kotlinClass.getClassHeader();
120 if (header == null) return null;
121
122 if (header.getKind() == KotlinClassHeader.Kind.INCOMPATIBLE_ABI_VERSION) {
123 errorReporter.reportIncompatibleAbiVersion(kotlinClass, header.getVersion());
124 }
125 else if (header.getKind() == expectedKind) {
126 return header.getAnnotationData();
127 }
128
129 return null;
130 }
131 }