From 363fdf4045373b718284c7b7ae33f2ef6040819b Mon Sep 17 00:00:00 2001
From: Andrey Breslav <andrey.breslav@jetbrains.com>
Date: Tue, 1 Apr 2014 19:47:59 +0400
Subject: [PATCH] Account for script classes in getAllDescriptors()

---
 .../resolve/lazy/descriptors/AbstractLazyMemberScope.java     | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/compiler/frontend/src/org/jetbrains/jet/lang/resolve/lazy/descriptors/AbstractLazyMemberScope.java b/compiler/frontend/src/org/jetbrains/jet/lang/resolve/lazy/descriptors/AbstractLazyMemberScope.java
index b645affdbe6..7a014f7d810 100644
--- a/compiler/frontend/src/org/jetbrains/jet/lang/resolve/lazy/descriptors/AbstractLazyMemberScope.java
+++ b/compiler/frontend/src/org/jetbrains/jet/lang/resolve/lazy/descriptors/AbstractLazyMemberScope.java
@@ -26,6 +26,7 @@ import org.jetbrains.annotations.Nullable;
 import org.jetbrains.jet.lang.descriptors.*;
 import org.jetbrains.jet.lang.psi.*;
 import org.jetbrains.jet.lang.resolve.BindingTrace;
+import org.jetbrains.jet.lang.resolve.ScriptNameUtil;
 import org.jetbrains.jet.lang.resolve.calls.autocasts.DataFlowInfo;
 import org.jetbrains.jet.lang.resolve.lazy.ResolveSession;
 import org.jetbrains.jet.lang.resolve.lazy.data.JetClassLikeInfo;
@@ -241,6 +242,9 @@ public abstract class AbstractLazyMemberScope<D extends DeclarationDescriptor, D
                 JetParameter parameter = (JetParameter) declaration;
                 result.addAll(getProperties(safeNameForLazyResolve(parameter)));
             }
+            else if (declaration instanceof JetScript) {
+                result.addAll(classDescriptors.invoke(ScriptNameUtil.classNameForScript((JetScript) declaration).shortName()));
+            }
             else if (declaration instanceof JetTypedef || declaration instanceof JetMultiDeclaration) {
                 // Do nothing for typedefs as they are not supported.
                 // MultiDeclarations are not supported on global level too.