From bf546c83fb14ee6a6e861bc123f037e3b4ba792c Mon Sep 17 00:00:00 2001
From: "Pavel V. Talanov" <talanov.pavel@gmail.com>
Date: Thu, 18 Sep 2014 19:20:06 +0400
Subject: [PATCH] Introduce ModuleSourceInfo

---
 .../jet/plugin/caches/resolve/IdeaModuleInfos.kt       | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/idea/idea-analysis/src/org/jetbrains/jet/plugin/caches/resolve/IdeaModuleInfos.kt b/idea/idea-analysis/src/org/jetbrains/jet/plugin/caches/resolve/IdeaModuleInfos.kt
index f2be6dc5d4f..98996d9c3b1 100644
--- a/idea/idea-analysis/src/org/jetbrains/jet/plugin/caches/resolve/IdeaModuleInfos.kt
+++ b/idea/idea-analysis/src/org/jetbrains/jet/plugin/caches/resolve/IdeaModuleInfos.kt
@@ -80,7 +80,11 @@ fun ideaModelDependencies(module: Module, productionOnly: Boolean): List<IdeaMod
     return result.toList()
 }
 
-public data class ModuleProductionSourceInfo(val module: Module) : IdeaModuleInfo() {
+public abstract class ModuleSourceInfo(): IdeaModuleInfo() {
+    abstract val module: Module
+}
+
+public data class ModuleProductionSourceInfo(override val module: Module) : ModuleSourceInfo() {
     override val name = Name.special("<production sources for module ${module.getName()}>")
 
     override fun contentScope() = module.getModuleScope(false)
@@ -91,7 +95,7 @@ public data class ModuleProductionSourceInfo(val module: Module) : IdeaModuleInf
 }
 
 //TODO: (module refactoring) do not create ModuleTestSourceInfo when there are no test roots for module
-public data class ModuleTestSourceInfo(val module: Module) : IdeaModuleInfo() {
+public data class ModuleTestSourceInfo(override val module: Module) : ModuleSourceInfo() {
     override val name = Name.special("<test sources for module ${module.getName()}>")
 
     override fun contentScope() = module.getModuleScope().intersectWith(GlobalSearchScope.notScope(module.getModuleScope(false)))
@@ -99,6 +103,8 @@ public data class ModuleTestSourceInfo(val module: Module) : IdeaModuleInfo() {
     override fun dependencies() = ideaModelDependencies(module, productionOnly = false)
 }
 
+private fun ModuleSourceInfo.isTests() = this is ModuleTestSourceInfo
+
 public fun Module.productionSourceInfo(): ModuleProductionSourceInfo = ModuleProductionSourceInfo(this)
 public fun Module.testSourceInfo(): ModuleTestSourceInfo = ModuleTestSourceInfo(this)