Register ControlFlowFactory in KotlinCoreProjectEnvironment

#KT-16047 Fixed
2017-01-30 13:22:47 +01:00
parent f0e890797a
commit 3a3cf048a6
1 changed files with 7 additions and 0 deletions
@@ -20,10 +20,17 @@ import com.intellij.core.JavaCoreApplicationEnvironment
 import com.intellij.core.JavaCoreProjectEnvironment
 import com.intellij.openapi.Disposable
 import com.intellij.psi.PsiManager
+import com.intellij.psi.controlFlow.ControlFlowFactory

 open class KotlinCoreProjectEnvironment(
        disposable: Disposable,
        applicationEnvironment: JavaCoreApplicationEnvironment
 ) : JavaCoreProjectEnvironment(disposable, applicationEnvironment) {
+    init {
+        myProject.registerService<ControlFlowFactory>(ControlFlowFactory::class.java,
+                                                      ControlFlowFactory(myPsiManager))
+
+    }
+
    override fun createCoreFileManager() = KotlinCliJavaFileManagerImpl(PsiManager.getInstance(project))
 }