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