From 8daf20eafa8d800466b6be10bdf87b73114845f9 Mon Sep 17 00:00:00 2001
From: "kirill.safonov" <kirill.safonov@jetbrains.com>
Date: Fri, 23 Nov 2012 20:03:24 +0400
Subject: [PATCH] Flash extract superclass: compute source root if target
 folder has been chosen by neighbour class (IDEA-95730)

---
 .../src/com/intellij/ide/util/PlatformPackageUtil.java          | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/platform/lang-impl/src/com/intellij/ide/util/PlatformPackageUtil.java b/platform/lang-impl/src/com/intellij/ide/util/PlatformPackageUtil.java
index a63e725e25c8..71737deaca97 100644
--- a/platform/lang-impl/src/com/intellij/ide/util/PlatformPackageUtil.java
+++ b/platform/lang-impl/src/com/intellij/ide/util/PlatformPackageUtil.java
@@ -129,6 +129,8 @@ public class PlatformPackageUtil {
         psiDirectory = DirectoryChooserUtil.selectDirectory(project, dirs.toArray(new PsiDirectory[dirs.size()]), baseDir,
                                                             File.separatorChar + packageName.replace('.', File.separatorChar));
         if (psiDirectory == null) return null;
+        final VirtualFile sourceRoot = ProjectRootManager.getInstance(project).getFileIndex().getSourceRootForFile(psiDirectory.getVirtualFile());
+        psiDirectory = PsiManager.getInstance(project).findDirectory(sourceRoot);
       }
     }