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;
018    
019    import com.intellij.openapi.util.Key;
020    import org.jetbrains.annotations.NotNull;
021    import org.jetbrains.jet.lang.descriptors.PackageFragmentDescriptor;
022    import org.jetbrains.jet.lang.descriptors.ScriptDescriptor;
023    import org.jetbrains.jet.lang.descriptors.ScriptDescriptorImpl;
024    import org.jetbrains.jet.lang.descriptors.ValueParameterDescriptor;
025    import org.jetbrains.jet.lang.psi.JetFile;
026    import org.jetbrains.jet.lang.psi.JetScript;
027    import org.jetbrains.jet.lang.resolve.name.FqName;
028    import org.jetbrains.jet.lang.resolve.name.Name;
029    import org.jetbrains.jet.lang.resolve.scopes.RedeclarationHandler;
030    import org.jetbrains.jet.lang.resolve.scopes.WritableScope;
031    import org.jetbrains.jet.lang.resolve.scopes.WritableScopeImpl;
032    
033    import javax.inject.Inject;
034    import java.util.List;
035    import java.util.Map;
036    
037    // SCRIPT: Resolve declarations in scripts
038    public class ScriptHeaderResolver {
039    
040        public static final Key<Integer> PRIORITY_KEY = Key.create(JetScript.class.getName() + ".priority");
041    
042        @NotNull
043        private MutablePackageFragmentProvider packageFragmentProvider;
044        @NotNull
045        private BindingTrace trace;
046    
047        @Inject
048        public void setPackageFragmentProvider(@NotNull MutablePackageFragmentProvider packageFragmentProvider) {
049            this.packageFragmentProvider = packageFragmentProvider;
050        }
051    
052        @Inject
053        public void setTrace(@NotNull BindingTrace trace) {
054            this.trace = trace;
055        }
056    
057    
058    
059        public void processScriptHierarchy(@NotNull TopDownAnalysisContext c, @NotNull JetScript script, @NotNull WritableScope outerScope) {
060            JetFile file = script.getContainingJetFile();
061            FqName fqName = file.getPackageFqName();
062            PackageFragmentDescriptor ns = packageFragmentProvider.getOrCreateFragment(fqName);
063    
064            Integer priority = getScriptPriority(script);
065    
066            FqName nameForScript = ScriptNameUtil.classNameForScript(script);
067            Name className = nameForScript.shortName();
068            ScriptDescriptorImpl scriptDescriptor = new ScriptDescriptorImpl(ns, priority, outerScope, className);
069    
070            WritableScopeImpl scriptScope = new WritableScopeImpl(outerScope, scriptDescriptor, RedeclarationHandler.DO_NOTHING, "script");
071            scriptScope.changeLockLevel(WritableScope.LockLevel.BOTH);
072            scriptDescriptor.setScopeForBodyResolution(scriptScope);
073    
074            c.getScripts().put(script, scriptDescriptor);
075    
076            trace.record(BindingContext.SCRIPT, script, scriptDescriptor);
077    
078            outerScope.addClassifierDescriptor(scriptDescriptor.getClassDescriptor());
079        }
080    
081        public static int getScriptPriority(@NotNull JetScript script) {
082            Integer priority = script.getUserData(PRIORITY_KEY);
083            return priority == null ? 0 : priority;
084        }
085    
086        public static void resolveScriptDeclarations(@NotNull TopDownAnalysisContext c) {
087            for (Map.Entry<JetScript, ScriptDescriptor> e : c.getScripts().entrySet()) {
088                JetScript declaration = e.getKey();
089                ScriptDescriptorImpl descriptor = (ScriptDescriptorImpl) e.getValue();
090    
091                List<ValueParameterDescriptor> valueParameters = ScriptParameterResolver.resolveScriptParameters(
092                        declaration,
093                        descriptor
094                );
095    
096                descriptor.setValueParameters(valueParameters);
097    
098                WritableScope scope = descriptor.getScopeForBodyResolution();
099                scope.setImplicitReceiver(descriptor.getThisAsReceiverParameter());
100                for (ValueParameterDescriptor valueParameterDescriptor : valueParameters) {
101                    scope.addVariableDescriptor(valueParameterDescriptor);
102                }
103            }
104        }
105    
106    }