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.openapi.project.Project;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.annotations.Nullable;
022    import org.jetbrains.jet.lang.descriptors.CallableMemberDescriptor;
023    import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
024    import org.jetbrains.jet.lang.descriptors.TypeParameterDescriptor;
025    import org.jetbrains.jet.lang.descriptors.ValueParameterDescriptor;
026    import org.jetbrains.jet.lang.resolve.BindingTrace;
027    import org.jetbrains.jet.lang.resolve.java.JavaBindingContext;
028    import org.jetbrains.jet.lang.resolve.java.kotlinSignature.AlternativeFieldSignatureData;
029    import org.jetbrains.jet.lang.resolve.java.kotlinSignature.AlternativeMethodSignatureData;
030    import org.jetbrains.jet.lang.resolve.java.kotlinSignature.SignaturesPropagationData;
031    import org.jetbrains.jet.lang.resolve.java.structure.JavaField;
032    import org.jetbrains.jet.lang.resolve.java.structure.JavaMethod;
033    import org.jetbrains.jet.lang.types.JetType;
034    
035    import javax.inject.Inject;
036    import java.util.Collections;
037    import java.util.List;
038    
039    public class TraceBasedExternalSignatureResolver implements ExternalSignatureResolver {
040        private BindingTrace trace;
041        private ExternalAnnotationResolver externalAnnotationResolver;
042        private Project project;
043    
044        @Inject
045        public void setTrace(BindingTrace trace) {
046            this.trace = trace;
047        }
048    
049        @Inject
050        public void setExternalAnnotationResolver(ExternalAnnotationResolver externalAnnotationResolver) {
051            this.externalAnnotationResolver = externalAnnotationResolver;
052        }
053    
054        @Inject
055        public void setProject(Project project) {
056            this.project = project;
057        }
058    
059        @Override
060        @NotNull
061        public PropagatedMethodSignature resolvePropagatedSignature(
062                @NotNull JavaMethod method,
063                @NotNull ClassDescriptor owner,
064                @NotNull JetType returnType,
065                @Nullable JetType receiverType,
066                @NotNull List<ValueParameterDescriptor> valueParameters,
067                @NotNull List<TypeParameterDescriptor> typeParameters
068        ) {
069            SignaturesPropagationData data =
070                    new SignaturesPropagationData(owner, returnType, receiverType, valueParameters, typeParameters, method);
071            return new PropagatedMethodSignature(data.getModifiedReturnType(), data.getModifiedReceiverType(),
072                                                 data.getModifiedValueParameters(), data.getModifiedTypeParameters(), data.getSignatureErrors(),
073                                                 data.getModifiedHasStableParameterNames(), data.getSuperFunctions());
074        }
075    
076        @Override
077        @NotNull
078        public AlternativeMethodSignature resolveAlternativeMethodSignature(
079                @NotNull JavaMethod method,
080                boolean hasSuperMethods,
081                @Nullable JetType returnType,
082                @Nullable JetType receiverType,
083                @NotNull List<ValueParameterDescriptor> valueParameters,
084                @NotNull List<TypeParameterDescriptor> typeParameters,
085                boolean hasStableParameterNames
086        ) {
087            AlternativeMethodSignatureData data =
088                    new AlternativeMethodSignatureData(externalAnnotationResolver, method, receiverType, project, valueParameters, returnType,
089                                                       typeParameters, hasSuperMethods);
090    
091            if (data.isAnnotated() && !data.hasErrors()) {
092                return new AlternativeMethodSignature(data.getReturnType(), receiverType, data.getValueParameters(), data.getTypeParameters(),
093                                                      Collections.<String>emptyList(), true);
094            }
095    
096            List<String> error = data.hasErrors() ? Collections.singletonList(data.getError()) : Collections.<String>emptyList();
097            return new AlternativeMethodSignature(returnType, receiverType, valueParameters, typeParameters, error, hasStableParameterNames);
098        }
099    
100        @Override
101        @NotNull
102        public AlternativeFieldSignature resolveAlternativeFieldSignature(
103                @NotNull JavaField field,
104                @NotNull JetType returnType,
105                boolean isVar
106        ) {
107            AlternativeFieldSignatureData data =
108                    new AlternativeFieldSignatureData(externalAnnotationResolver, field, returnType, project, isVar);
109    
110            if (data.isAnnotated() && !data.hasErrors()) {
111                return new AlternativeFieldSignature(data.getReturnType(), null);
112            }
113    
114            String error = data.hasErrors() ? data.getError() : null;
115            return new AlternativeFieldSignature(returnType, error);
116        }
117    
118        @Override
119        public void reportSignatureErrors(@NotNull CallableMemberDescriptor descriptor, @NotNull List<String> signatureErrors) {
120            trace.record(JavaBindingContext.LOAD_FROM_JAVA_SIGNATURE_ERRORS, descriptor, signatureErrors);
121        }
122    }