From 3436a0c96ddebab9cf2a7b19004219124e71d692 Mon Sep 17 00:00:00 2001
From: "Maxim.Medvedev" <maxim.medvedev@jetbrains.com>
Date: Tue, 24 Jan 2012 16:27:09 +0400
Subject: [PATCH] introduce parameter in script method

---
 .../org/jetbrains/plugins/groovy/lang/psi/util/PsiUtil.java   | 2 +-
 .../parameter/java2groovy/OldReferencesResolver.java          | 4 ++--
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/util/PsiUtil.java b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/util/PsiUtil.java
index b8d8d2419342..da14556a1cee 100644
--- a/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/util/PsiUtil.java
+++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/lang/psi/util/PsiUtil.java
@@ -543,7 +543,7 @@ public class PsiUtil {
       if (context instanceof GrTypeDefinition) {
         return (PsiClass)context;
       }
-      else if (context instanceof GroovyFileBase) {
+      else if (context instanceof GroovyFileBase && context.isPhysical()) {
         return ((GroovyFileBase)context).getScriptClass();
       }
 
diff --git a/plugins/groovy/src/org/jetbrains/plugins/groovy/refactoring/introduce/parameter/java2groovy/OldReferencesResolver.java b/plugins/groovy/src/org/jetbrains/plugins/groovy/refactoring/introduce/parameter/java2groovy/OldReferencesResolver.java
index 13059b33361a..9947c4b0ca52 100644
--- a/plugins/groovy/src/org/jetbrains/plugins/groovy/refactoring/introduce/parameter/java2groovy/OldReferencesResolver.java
+++ b/plugins/groovy/src/org/jetbrains/plugins/groovy/refactoring/introduce/parameter/java2groovy/OldReferencesResolver.java
@@ -145,7 +145,7 @@ public class OldReferencesResolver {
       //newExpr = ((GrReferenceExpression)newExpr).getReferenceNameElement();
       final GroovyResolveResult adv = oldRef.advancedResolve();
       final PsiElement scope = getClassContainingResolve(adv);
-      final PsiElement owner = PsiTreeUtil.getContextOfType(oldExpr, PsiClass.class);
+      final PsiElement owner = PsiUtil.getContextClass(oldExpr);
 
       if (myToReplaceIn instanceof GrClosableBlock || (owner != null && scope != null && PsiTreeUtil.isContextAncestor(owner, scope, false))) {
 
@@ -368,7 +368,7 @@ public class OldReferencesResolver {
         return ((PsiMember)elem).getContainingClass();
       }
       else {
-        return PsiTreeUtil.getParentOfType(elem, PsiClass.class);
+        return PsiUtil.getContextClass(elem);
       }
     }
     return null;