182: fixing SlicePanel compilation after isToShowCloseButton method removal

2018-05-08 18:32:43 +03:00
parent e727e20973
commit 24b8d12eca
1 changed files with 139 additions and 0 deletions
@@ -0,0 +1,139 @@
+/*
+ * Copyright 2010-2017 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.kotlin.idea.slicer
+
+import com.intellij.analysis.AnalysisScope
+import com.intellij.openapi.Disposable
+import com.intellij.openapi.project.Project
+import com.intellij.openapi.util.Disposer
+import com.intellij.openapi.wm.ToolWindowId
+import com.intellij.openapi.wm.ToolWindowManager
+import com.intellij.psi.PsiElement
+import com.intellij.slicer.DuplicateMap
+import com.intellij.slicer.SliceAnalysisParams
+import com.intellij.slicer.SlicePanel
+import com.intellij.slicer.SliceRootNode
+import com.intellij.usageView.UsageInfo
+import com.intellij.usageView.UsageViewBundle
+import com.intellij.usages.PsiElementUsageTarget
+import com.intellij.usages.UsageContextPanel
+import com.intellij.usages.UsageView
+import com.intellij.usages.UsageViewPresentation
+import com.intellij.usages.impl.UsageContextPanelBase
+import com.intellij.usages.impl.UsageViewImpl
+import org.jetbrains.kotlin.psi.KtDeclaration
+import java.awt.BorderLayout
+import javax.swing.JLabel
+import javax.swing.JPanel
+import javax.swing.SwingConstants
+
+sealed class KotlinUsageContextDataFlowPanelBase(
+        project: Project,
+        presentation: UsageViewPresentation,
+        private val isInflow: Boolean
+) : UsageContextPanelBase(project, presentation) {
+    private var panel: JPanel? = null
+
+    abstract class ProviderBase : UsageContextPanel.Provider {
+        override fun isAvailableFor(usageView: UsageView): Boolean {
+            val target = (usageView as UsageViewImpl).targets.firstOrNull() ?: return false
+            val element = (target as? PsiElementUsageTarget)?.element
+            return element is KtDeclaration && element.isValid
+        }
+    }
+
+    private fun createParams(element: PsiElement): SliceAnalysisParams {
+        return SliceAnalysisParams().apply {
+            scope = AnalysisScope(element.project)
+            dataFlowToThis = isInflow
+            showInstanceDereferences = true
+        }
+    }
+
+    protected fun createPanel(element: PsiElement, dataFlowToThis: Boolean): JPanel {
+        val toolWindow = ToolWindowManager.getInstance(myProject).getToolWindow(ToolWindowId.FIND)
+        val params = createParams(element)
+
+        val rootNode = SliceRootNode(myProject, DuplicateMap(), KotlinSliceUsage(element, params))
+
+        return object : SlicePanel(myProject, dataFlowToThis, rootNode, false, toolWindow) {
+            override fun isToShowAutoScrollButton() = false
+
+            override fun isToShowPreviewButton() = false
+
+            override fun isAutoScroll() = false
+
+            override fun setAutoScroll(autoScroll: Boolean) {}
+
+            override fun isPreview() = false
+
+            override fun setPreview(preview: Boolean) {}
+        }
+    }
+
+    public override fun updateLayoutLater(infos: List<UsageInfo>?) {
+        if (infos == null) {
+            removeAll()
+            val title = UsageViewBundle.message("select.the.usage.to.preview", myPresentation.usagesWord)
+            add(JLabel(title, SwingConstants.CENTER), BorderLayout.CENTER)
+        }
+        else {
+            val element = infos.firstOrNull()?.element ?: return
+            if (panel != null) {
+                Disposer.dispose(panel as Disposable)
+            }
+
+            val panel = createPanel(element, isInflow)
+            Disposer.register(this, panel as Disposable)
+            removeAll()
+            add(panel, BorderLayout.CENTER)
+            this.panel = panel
+        }
+        revalidate()
+    }
+
+    override fun dispose() {
+        super.dispose()
+        panel = null
+    }
+}
+
+class KotlinUsageContextDataInflowPanel(
+        project: Project,
+        presentation: UsageViewPresentation
+) : KotlinUsageContextDataFlowPanelBase(project, presentation, true) {
+    class Provider : ProviderBase() {
+        override fun create(usageView: UsageView): UsageContextPanel {
+            return KotlinUsageContextDataInflowPanel((usageView as UsageViewImpl).project, usageView.getPresentation())
+        }
+
+        override fun getTabTitle() = "Dataflow to Here"
+    }
+}
+
+class KotlinUsageContextDataOutflowPanel(
+        project: Project,
+        presentation: UsageViewPresentation
+) : KotlinUsageContextDataFlowPanelBase(project, presentation, false) {
+    class Provider : ProviderBase() {
+        override fun create(usageView: UsageView): UsageContextPanel {
+            return KotlinUsageContextDataOutflowPanel((usageView as UsageViewImpl).project, usageView.getPresentation())
+        }
+
+        override fun getTabTitle() = "Dataflow from Here"
+    }
+}