001    /*
002     * Copyright 2010-2014 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;
018    
019    import com.google.common.collect.Lists;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.jet.lang.descriptors.ClassDescriptor;
022    import org.jetbrains.jet.lang.descriptors.ScriptDescriptor;
023    import org.jetbrains.jet.lang.descriptors.ValueParameterDescriptor;
024    import org.jetbrains.jet.lang.descriptors.annotations.Annotations;
025    import org.jetbrains.jet.lang.descriptors.impl.ValueParameterDescriptorImpl;
026    import org.jetbrains.jet.lang.parsing.JetScriptDefinition;
027    import org.jetbrains.jet.lang.parsing.JetScriptDefinitionProvider;
028    import org.jetbrains.jet.lang.psi.JetFile;
029    import org.jetbrains.jet.lang.psi.JetScript;
030    import org.jetbrains.jet.lang.resolve.name.FqName;
031    import org.jetbrains.jet.lang.types.DependencyClassByQualifiedNameResolver;
032    import org.jetbrains.jet.lang.types.JetType;
033    import org.jetbrains.jet.lang.types.TypeUtils;
034    import org.jetbrains.jet.lang.types.ref.JetTypeName;
035    
036    import javax.inject.Inject;
037    import java.util.ArrayList;
038    import java.util.List;
039    
040    // SCRIPT: Resolve script parameters
041    public class ScriptParameterResolver {
042        @NotNull
043        private DependencyClassByQualifiedNameResolver dependencyClassByQualifiedNameResolver;
044    
045        @Inject
046        public void setDependencyClassByQualifiedNameResolver(@NotNull DependencyClassByQualifiedNameResolver dependencyClassByQualifiedNameResolver) {
047            this.dependencyClassByQualifiedNameResolver = dependencyClassByQualifiedNameResolver;
048        }
049    
050        @NotNull
051        public List<ValueParameterDescriptor> resolveScriptParameters(
052                @NotNull JetScript declaration,
053                @NotNull ScriptDescriptor scriptDescriptor
054        ) {
055            List<ValueParameterDescriptor> valueParameters = Lists.newArrayList();
056    
057            JetFile file = declaration.getContainingJetFile();
058            JetScriptDefinition scriptDefinition = JetScriptDefinitionProvider.getInstance(file.getProject()).findScriptDefinition(file);
059    
060            int index = 0;
061            for (AnalyzerScriptParameter scriptParameter : scriptDefinition.getScriptParameters()) {
062                ValueParameterDescriptor parameter = resolveScriptParameter(scriptParameter, index, scriptDescriptor);
063                valueParameters.add(parameter);
064                ++index;
065            }
066            return valueParameters;
067        }
068    
069        @NotNull
070        private ClassDescriptor resolveClass(@NotNull FqName className) {
071            ClassDescriptor classDescriptor = dependencyClassByQualifiedNameResolver.resolveClass(className);
072            if (classDescriptor == null) {
073                throw new IllegalStateException("dependency class not found by name: " + className);
074            }
075            return classDescriptor;
076        }
077    
078        @NotNull
079        public JetType resolveTypeName(@NotNull JetTypeName typeName) {
080            List<JetType> typeArguments = new ArrayList<JetType>();
081            for (JetTypeName typeArgumentName : typeName.getArguments()) {
082                typeArguments.add(resolveTypeName(typeArgumentName));
083            }
084            ClassDescriptor classDescriptor = resolveClass(typeName.getClassName());
085            return TypeUtils.substituteParameters(classDescriptor, typeArguments);
086        }
087    
088    
089        @NotNull
090        private ValueParameterDescriptor resolveScriptParameter(
091                @NotNull AnalyzerScriptParameter scriptParameter,
092                int index,
093                @NotNull ScriptDescriptor script) {
094            JetType type = resolveTypeName(scriptParameter.getType());
095            return new ValueParameterDescriptorImpl(script, null, index, Annotations.EMPTY, scriptParameter.getName(), type, false, null);
096        }
097    }