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.descriptors;
018
019 import org.jetbrains.annotations.NotNull;
020 import org.jetbrains.jet.lang.descriptors.annotations.AnnotationDescriptor;
021 import org.jetbrains.jet.lang.descriptors.impl.*;
022 import org.jetbrains.jet.lang.resolve.DescriptorFactory;
023 import org.jetbrains.jet.lang.resolve.name.Name;
024 import org.jetbrains.jet.lang.resolve.scopes.JetScope;
025 import org.jetbrains.jet.lang.resolve.scopes.RedeclarationHandler;
026 import org.jetbrains.jet.lang.resolve.scopes.WritableScope;
027 import org.jetbrains.jet.lang.resolve.scopes.WritableScopeImpl;
028 import org.jetbrains.jet.lang.resolve.scopes.receivers.ScriptReceiver;
029 import org.jetbrains.jet.lang.types.JetType;
030 import org.jetbrains.jet.lang.types.TypeSubstitutor;
031 import org.jetbrains.jet.lang.types.lang.KotlinBuiltIns;
032
033 import java.util.Collections;
034 import java.util.HashSet;
035 import java.util.List;
036
037 public class ScriptDescriptor extends DeclarationDescriptorNonRootImpl {
038 public static final String LAST_EXPRESSION_VALUE_FIELD_NAME = "rv";
039 private static final Name NAME = Name.special("<script>");
040
041 private final int priority;
042
043 private JetType returnType;
044 private List<ValueParameterDescriptor> valueParameters;
045
046 private final ScriptCodeDescriptor scriptCodeDescriptor = new ScriptCodeDescriptor(this);
047 private final ReceiverParameterDescriptor implicitReceiver = new ReceiverParameterDescriptorImpl(this,
048 // Putting Any here makes no sense,
049 // it is simply copied from someplace else
050 // during a refactoring
051 KotlinBuiltIns.getInstance().getAnyType(),
052 new ScriptReceiver(this));
053
054 private final ClassDescriptorImpl classDescriptor;
055
056 private final WritableScopeImpl classScope;
057
058 public ScriptDescriptor(
059 @NotNull DeclarationDescriptor containingDeclaration,
060 int priority,
061 @NotNull JetScope scriptScope,
062 @NotNull Name className
063 ) {
064 super(containingDeclaration, Collections.<AnnotationDescriptor>emptyList(), NAME);
065 this.priority = priority;
066
067 classDescriptor = new ClassDescriptorImpl(containingDeclaration, className, Modality.FINAL,
068 Collections.singleton(KotlinBuiltIns.getInstance().getAnyType()));
069 classScope = new WritableScopeImpl(scriptScope, containingDeclaration, RedeclarationHandler.DO_NOTHING, "script members");
070 classScope.changeLockLevel(WritableScope.LockLevel.BOTH);
071 classDescriptor.initialize(classScope, new HashSet<ConstructorDescriptor>(), null);
072 }
073
074 public void initialize(
075 @NotNull JetType returnType,
076 @NotNull List<? extends PropertyDescriptorImpl> properties,
077 @NotNull List<? extends FunctionDescriptor> functions
078 ) {
079 this.returnType = returnType;
080 scriptCodeDescriptor.initialize(implicitReceiver, valueParameters, returnType);
081
082 PropertyDescriptorImpl propertyDescriptor = new PropertyDescriptorImpl(classDescriptor,
083 Collections.<AnnotationDescriptor>emptyList(),
084 Modality.FINAL,
085 Visibilities.PUBLIC,
086 false,
087 Name.identifier(LAST_EXPRESSION_VALUE_FIELD_NAME),
088 CallableMemberDescriptor.Kind.DECLARATION);
089 propertyDescriptor.setType(
090 returnType,
091 Collections.<TypeParameterDescriptor>emptyList(),
092 classDescriptor.getThisAsReceiverParameter(),
093 ReceiverParameterDescriptor.NO_RECEIVER_PARAMETER);
094 propertyDescriptor.initialize(null, null);
095 classScope.addPropertyDescriptor(propertyDescriptor);
096
097 for (PropertyDescriptorImpl property : properties) {
098 initializeWithDefaultGetterSetter(property);
099 classScope.addPropertyDescriptor(property);
100 }
101
102 for (FunctionDescriptor function : functions) {
103 classScope.addFunctionDescriptor(function);
104 }
105 }
106
107 public static void initializeWithDefaultGetterSetter(PropertyDescriptorImpl propertyDescriptor) {
108 PropertyGetterDescriptorImpl getter = propertyDescriptor.getGetter();
109 if (getter == null && propertyDescriptor.getVisibility() != Visibilities.PRIVATE) {
110 getter = DescriptorFactory.createDefaultGetter(propertyDescriptor);
111 getter.initialize(propertyDescriptor.getType());
112 }
113
114 PropertySetterDescriptor setter = propertyDescriptor.getSetter();
115 if (setter == null && propertyDescriptor.isVar()) {
116 setter = DescriptorFactory.createDefaultSetter(propertyDescriptor);
117 }
118 propertyDescriptor.initialize(getter, setter);
119 }
120
121 public int getPriority() {
122 return priority;
123 }
124
125 @NotNull
126 public JetType getReturnType() {
127 return returnType;
128 }
129
130 @NotNull
131 public List<ValueParameterDescriptor> getValueParameters() {
132 return valueParameters;
133 }
134
135 @NotNull
136 public ScriptCodeDescriptor getScriptCodeDescriptor() {
137 return scriptCodeDescriptor;
138 }
139
140 @NotNull
141 public ReceiverParameterDescriptor getThisAsReceiverParameter() {
142 return implicitReceiver;
143 }
144
145 @Override
146 public DeclarationDescriptor substitute(@NotNull TypeSubstitutor substitutor) {
147 throw new IllegalStateException("nothing to substitute in script");
148 }
149
150 @Override
151 public <R, D> R accept(DeclarationDescriptorVisitor<R, D> visitor, D data) {
152 return visitor.visitScriptDescriptor(this, data);
153 }
154
155 public void setValueParameters(@NotNull List<ValueParameterDescriptor> valueParameters) {
156 this.valueParameters = valueParameters;
157 ConstructorDescriptorImpl constructorDescriptor =
158 new ConstructorDescriptorImpl(classDescriptor, Collections.<AnnotationDescriptor>emptyList(), true)
159 .initialize(Collections.<TypeParameterDescriptor>emptyList(), valueParameters, Visibilities.PUBLIC);
160 constructorDescriptor.setReturnType(classDescriptor.getDefaultType());
161
162 classDescriptor.getConstructors().add(constructorDescriptor);
163 classDescriptor.setPrimaryConstructor(constructorDescriptor);
164
165 for (ValueParameterDescriptor parameter : valueParameters) {
166 PropertyDescriptorImpl propertyDescriptor = new PropertyDescriptorImpl(classDescriptor,
167 Collections.<AnnotationDescriptor>emptyList(),
168 Modality.FINAL,
169 Visibilities.PUBLIC,
170 false,
171 parameter.getName(),
172 CallableMemberDescriptor.Kind.DECLARATION);
173 propertyDescriptor.setType(
174 parameter.getType(),
175 Collections.<TypeParameterDescriptor>emptyList(),
176 classDescriptor.getThisAsReceiverParameter(), ReceiverParameterDescriptor.NO_RECEIVER_PARAMETER);
177 //PropertyGetterDescriptor getter = DescriptorResolver.createDefaultGetter(propertyDescriptor);
178 //getter.initialize(propertyDescriptor.getType());
179 propertyDescriptor.initialize(null, null);
180 classScope.addPropertyDescriptor(propertyDescriptor);
181 }
182 }
183
184 @NotNull
185 public ClassDescriptor getClassDescriptor() {
186 return classDescriptor;
187 }
188 }