diff --git a/idea/src/META-INF/plugin.xml b/idea/src/META-INF/plugin.xml
index c86ddf539a9..fb2ead6a6d3 100644
--- a/idea/src/META-INF/plugin.xml
+++ b/idea/src/META-INF/plugin.xml
@@ -108,6 +108,10 @@
+
+
+
diff --git a/idea/src/org/jetbrains/jet/plugin/actions/internal/CheckPartialBodyResolveAction.kt b/idea/src/org/jetbrains/jet/plugin/actions/internal/CheckPartialBodyResolveAction.kt
new file mode 100644
index 00000000000..7788fcdad9b
--- /dev/null
+++ b/idea/src/org/jetbrains/jet/plugin/actions/internal/CheckPartialBodyResolveAction.kt
@@ -0,0 +1,190 @@
+/*
+ * Copyright 2010-2014 JetBrains s.r.o.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package org.jetbrains.jet.plugin.actions.internal
+
+import com.intellij.openapi.actionSystem.AnAction
+import com.intellij.openapi.actionSystem.AnActionEvent
+import com.intellij.openapi.actionSystem.CommonDataKeys
+import com.intellij.openapi.vfs.VirtualFile
+import com.intellij.openapi.project.Project
+import com.intellij.psi.PsiManager
+import java.util.ArrayList
+import org.jetbrains.jet.lang.psi.JetFile
+import com.intellij.openapi.vfs.VfsUtilCore
+import com.intellij.openapi.vfs.VirtualFileVisitor
+import com.intellij.openapi.progress.ProgressManager
+import org.jetbrains.jet.lang.psi.JetVisitorVoid
+import org.jetbrains.jet.lang.psi.JetElement
+import org.jetbrains.jet.lang.psi.JetReferenceExpression
+import org.jetbrains.jet.plugin.caches.resolve.getLazyResolveSession
+import org.jetbrains.jet.lang.resolve.BindingContext
+import org.jetbrains.jet.plugin.project.ResolveSessionForBodies
+import com.intellij.openapi.fileEditor.FileDocumentManager
+import org.jetbrains.jet.lang.descriptors.DeclarationDescriptor
+import org.jetbrains.jet.renderer.DescriptorRenderer
+import com.intellij.openapi.diff.DiffManager
+import com.intellij.openapi.diff.SimpleDiffRequest
+import com.intellij.openapi.diff.SimpleContent
+import com.intellij.openapi.ui.DialogBuilder
+import com.intellij.openapi.diff.impl.DiffPanelImpl
+import com.intellij.openapi.diff.ex.DiffPanelOptions
+import com.intellij.openapi.ui.Messages
+import javax.swing.SwingUtilities
+import org.jetbrains.jet.lang.psi.JetNameReferenceExpression
+import org.jetbrains.jet.lang.psi.JetExpression
+import org.jetbrains.jet.lang.types.JetType
+
+public class CheckPartialBodyResolveAction : AnAction() {
+ override fun actionPerformed(e: AnActionEvent) {
+ val selectedFiles = selectedKotlinFiles(e).toList()
+ val project = CommonDataKeys.PROJECT.getData(e.getDataContext())!!
+
+ ProgressManager.getInstance().runProcessWithProgressSynchronously(
+ Runnable { checkResolve(selectedFiles, project) },
+ "Checking Partial Body Resolve",
+ true,
+ project)
+ }
+
+ private fun checkResolve(files: Collection, project: Project) {
+ //TODO: drop resolve caches if any
+ val progressIndicator = ProgressManager.getInstance().getProgressIndicator()
+ for ((i, file) in files.withIndices()) {
+ progressIndicator?.setText("Checking resolve $i of ${files.size}...")
+ progressIndicator?.setText2(file.getVirtualFile().getPath())
+
+ val partialResolveDump = dumpResolve(file) {(element, resolveSession) ->
+ resolveSession.resolveToElementWithPartialBodyResolve(element)
+ }
+ val goldDump = dumpResolve(file) {(element, resolveSession) -> resolveSession.resolveToElement(element) }
+ if (partialResolveDump != goldDump) {
+ SwingUtilities.invokeLater {
+ val title = "Difference Found in File ${file.getVirtualFile().getPath()}"
+
+ val request = SimpleDiffRequest(project, title)
+ request.setContents(SimpleContent(goldDump), SimpleContent(partialResolveDump))
+ request.setContentTitles("Expected", "Partial Body Resolve")
+ val diffBuilder = DialogBuilder(project)
+ val diffPanel = DiffManager.getInstance().createDiffPanel(diffBuilder.getWindow(), project, diffBuilder, null) as DiffPanelImpl
+ diffPanel.getOptions().setShowSourcePolicy(DiffPanelOptions.ShowSourcePolicy.DONT_SHOW)
+ diffBuilder.setCenterPanel(diffPanel.getComponent())
+ diffBuilder.setDimensionServiceKey("org.jetbrains.jet.plugin.actions.internal.CheckPartialBodyResolveAction.Diff")
+ diffPanel.setDiffRequest(request)
+ diffBuilder.addOkAction().setText("Close")
+ diffBuilder.setTitle(title)
+ diffBuilder.show()
+ //TODO: choose continue or abort
+ }
+ return
+ }
+
+ progressIndicator?.setFraction((i + 1) / files.size.toDouble())
+ }
+
+ SwingUtilities.invokeLater {
+ Messages.showInfoMessage(project, "Analyzed ${files.size} file(s). No resolve difference found. ", "Success")
+ }
+ }
+
+ private fun dumpResolve(file: JetFile, resolver: (JetElement, ResolveSessionForBodies) -> BindingContext): String {
+ val builder = StringBuilder()
+ val resolveSession = file.getLazyResolveSession()
+ val document = FileDocumentManager.getInstance().getDocument(file.getVirtualFile())
+ file.acceptChildren(object : JetVisitorVoid(){
+ override fun visitJetElement(element: JetElement) {
+ ProgressManager.checkCanceled()
+ element.acceptChildren(this)
+ }
+
+ override fun visitReferenceExpression(expression: JetReferenceExpression) {
+ super.visitReferenceExpression(expression)
+
+ val bindingContext = resolver(expression, resolveSession)
+ val target = bindingContext[BindingContext.REFERENCE_TARGET, expression]
+ val offset = expression.getTextOffset()
+ val line = document.getLineNumber(offset)
+ val column = offset - document.getLineStartOffset(line)
+ val exprName = if (expression is JetNameReferenceExpression) expression.getReferencedName() else ""
+ builder.append("$exprName at (${line + 1}:${column + 1}) resolves to ${target?.presentation()}\n")
+ }
+
+ override fun visitExpression(expression: JetExpression) {
+ super.visitExpression(expression)
+
+ val bindingContext = resolver(expression, resolveSession)
+
+ val offset = expression.getTextOffset()
+ val line = document.getLineNumber(offset)
+ val column = offset - document.getLineStartOffset(line)
+ val exprName = if (expression is JetNameReferenceExpression) expression.getReferencedName() else ""
+ builder.append("$exprName at (${line + 1}:${column + 1})")
+
+ if (expression is JetReferenceExpression) {
+ val target = bindingContext[BindingContext.REFERENCE_TARGET, expression]
+ builder.append(" resolves to ${target?.presentation()}")
+ }
+
+ val type = bindingContext[BindingContext.EXPRESSION_TYPE, expression]
+ if (type != null) {
+ builder.append(" has type ${type.presentation()}")
+ }
+
+ builder.append("\n")
+ }
+ })
+ return builder.toString()
+ }
+
+ private fun DeclarationDescriptor.presentation() = DescriptorRenderer.DEBUG_TEXT.render(this)
+ private fun JetType.presentation() = DescriptorRenderer.DEBUG_TEXT.renderType(this)
+
+ override fun update(e: AnActionEvent) {
+ if (!KotlinInternalMode.enabled) {
+ e.getPresentation().setVisible(false)
+ e.getPresentation().setEnabled(false)
+ }
+ e.getPresentation().setVisible(true)
+ e.getPresentation().setEnabled(selectedKotlinFiles(e).any())
+ }
+
+ private fun selectedKotlinFiles(e: AnActionEvent): Stream {
+ val virtualFiles = e.getData(CommonDataKeys.VIRTUAL_FILE_ARRAY) ?: return streamOf()
+ val project = CommonDataKeys.PROJECT.getData(e.getDataContext()) ?: return streamOf()
+ return allKotlinFiles(virtualFiles, project)
+ }
+
+ private fun allKotlinFiles(filesOrDirs: Array, project: Project): Stream {
+ val manager = PsiManager.getInstance(project)
+ return allFiles(filesOrDirs)
+ .stream()
+ .map { manager.findFile(it) as? JetFile }
+ .filterNotNull()
+ }
+
+ private fun allFiles(filesOrDirs: Array): Collection {
+ val result = ArrayList()
+ for (file in filesOrDirs) {
+ VfsUtilCore.visitChildrenRecursively(file, object : VirtualFileVisitor() {
+ override fun visitFile(file: VirtualFile): Boolean {
+ result.add(file)
+ return true
+ }
+ })
+ }
+ return result
+ }
+}