From efd1947722587d15bf1c81a0fd7ca722a7bc6fa8 Mon Sep 17 00:00:00 2001
From: Ilya Ryzhenkov <orangy@jetbrains.com>
Date: Mon, 13 Oct 2014 19:30:42 +0400
Subject: Migrate to lazy resolve session

---
 src/Analysis/AnalysisEnvironment.kt |  5 +++--
 src/Analysis/CompilerAPI.kt         | 33 +++++++++++++++++++--------------
 2 files changed, 22 insertions(+), 16 deletions(-)

(limited to 'src')

diff --git a/src/Analysis/AnalysisEnvironment.kt b/src/Analysis/AnalysisEnvironment.kt
index 1d51b8a2..90231b84 100644
--- a/src/Analysis/AnalysisEnvironment.kt
+++ b/src/Analysis/AnalysisEnvironment.kt
@@ -32,8 +32,9 @@ public class AnalysisEnvironment(val messageCollector: MessageCollector, body: A
      */
     public fun withContext<T>(processor: (JetCoreEnvironment, ModuleDescriptor, BindingContext) -> T): T {
         val environment = JetCoreEnvironment.createForProduction(this, configuration)
-        val exhaust = environment.analyze()
-        return processor(environment, exhaust.getModuleDescriptor(), exhaust.getBindingContext())
+        val resolveSession = environment.analyze()
+        resolveSession.forceResolveAll()
+        return processor(environment, resolveSession.getModuleDescriptor(), resolveSession.getBindingContext())
     }
 
     /**
diff --git a/src/Analysis/CompilerAPI.kt b/src/Analysis/CompilerAPI.kt
index 03d4dde6..43d0304e 100644
--- a/src/Analysis/CompilerAPI.kt
+++ b/src/Analysis/CompilerAPI.kt
@@ -1,19 +1,19 @@
 package org.jetbrains.dokka
 
 import org.jetbrains.jet.cli.common.arguments.*
-import org.jetbrains.jet.cli.common.messages.*
-import org.jetbrains.jet.cli.jvm.*
 import org.jetbrains.jet.cli.jvm.compiler.*
 import org.jetbrains.jet.utils.*
 import java.io.*
 import org.jetbrains.jet.lang.resolve.java.*
-import com.google.common.base.*
-import com.intellij.psi.*
 import org.jetbrains.jet.lang.resolve.*
 import org.jetbrains.jet.lang.psi.*
 import org.jetbrains.jet.analyzer.*
 import org.jetbrains.jet.lang.descriptors.*
 import org.jetbrains.jet.lang.resolve.scopes.*
+import org.jetbrains.jet.context.GlobalContext
+import com.intellij.psi.search.GlobalSearchScope
+import org.jetbrains.jet.lang.resolve.name.Name
+import org.jetbrains.jet.lang.resolve.lazy.ResolveSession
 
 private fun getAnnotationsPath(paths: KotlinPaths, arguments: K2JVMCompilerArguments): MutableList<File> {
     val annotationsPath = arrayListOf<File>()
@@ -27,18 +27,23 @@ private fun getAnnotationsPath(paths: KotlinPaths, arguments: K2JVMCompilerArgum
     return annotationsPath
 }
 
-fun JetCoreEnvironment.analyze(): AnalyzeExhaust {
+fun JetCoreEnvironment.analyze(): ResolveSession {
+    val globalContext = GlobalContext()
     val project = getProject()
     val sourceFiles = getSourceFiles()
-    val support = CliLightClassGenerationSupport.getInstanceForCli(project)!!
-    val sharedTrace = support.getTrace()
-    val sharedModule = support.newModule()
-    val exhaust = TopDownAnalyzerFacadeForJVM.analyzeFilesWithJavaIntegration(project, sourceFiles, sharedTrace,
-                                                         Predicates.alwaysFalse<PsiFile>(),
-                                                         sharedModule,
-                                                         null,
-                                                         null)
-    return exhaust
+
+    val module = object : ModuleInfo {
+        override val name: Name = Name.special("<module>")
+        override fun dependencies(): List<ModuleInfo> = listOf(this)
+    }
+    val resolverForProject = JvmAnalyzerFacade.setupResolverForProject(
+            globalContext,
+            project,
+            listOf(module),
+            { ModuleContent(sourceFiles, GlobalSearchScope.allScope(project)) },
+            JvmPlatformParameters { module }
+    )
+    return resolverForProject.resolverForModule(module).lazyResolveSession
 }
 
 fun BindingContext.getPackageFragment(file: JetFile): PackageFragmentDescriptor? = get(BindingContext.FILE_TO_PACKAGE_FRAGMENT, file)
-- 
cgit