refactor: get lang from live stack trace

Author: BFergerson

plugin/src/main/kotlin/spp/jetbrains/sourcemarker/service/instrument/breakpoint/tree/VariableRootSimpleNode.kt

@@ -16,18 +16,13 @@
  */
 package spp.jetbrains.sourcemarker.service.instrument.breakpoint.tree
 
-import com.intellij.lang.Language
-import com.intellij.openapi.application.ApplicationInfo
-import com.intellij.openapi.fileEditor.FileEditorManager
-import com.intellij.openapi.project.ProjectManager
-import com.intellij.psi.PsiManager
 import com.intellij.ui.treeStructure.SimpleNode
 import com.intellij.util.containers.hash.LinkedHashMap
 import spp.jetbrains.marker.js.JavascriptVariableRootNode
 import spp.jetbrains.marker.jvm.JVMVariableSimpleNode
 import spp.jetbrains.marker.py.PythonVariableRootNode
-import spp.jetbrains.sourcemarker.activities.PluginSourceMarkerStartupActivity.Companion.PYCHARM_PRODUCT_CODES
 import spp.jetbrains.sourcemarker.service.instrument.breakpoint.StackFrameManager
+import spp.protocol.artifact.ArtifactLanguage
 import spp.protocol.instrument.variable.LiveVariableScope
 
 /**
@@ -53,57 +48,48 @@ class VariableRootSimpleNode : SimpleNode() {
             NO_CHILDREN
         } else {
             val vars = stackFrameManager.currentFrame!!.variables
-
-            //todo: get language via LiveStackTrace
-            var language: Language? = null
-            val project = ProjectManager.getInstance().openProjects.first()
-            val manager = FileEditorManager.getInstance(project)
-            val selectedFile = manager.selectedFiles.firstOrNull()
-            if (selectedFile != null) {
-                val psiFile = PsiManager.getInstance(project).findFile(selectedFile)
-                if (psiFile != null) {
-                    language = psiFile.language
-                }
-            }
-            if (language?.id == "ECMAScript 6") {
-                return arrayOf(
-                    JavascriptVariableRootNode(
-                        vars.filter { it.scope == LiveVariableScope.LOCAL_VARIABLE },
-                        LiveVariableScope.LOCAL_VARIABLE
-                    ),
-                    JavascriptVariableRootNode(
-                        vars.filter { it.scope == LiveVariableScope.GLOBAL_VARIABLE },
-                        LiveVariableScope.GLOBAL_VARIABLE
+            when (stackFrameManager.stackTrace.language) {
+                ArtifactLanguage.NODEJS -> {
+                    return arrayOf(
+                        JavascriptVariableRootNode(
+                            vars.filter { it.scope == LiveVariableScope.LOCAL_VARIABLE },
+                            LiveVariableScope.LOCAL_VARIABLE
+                        ),
+                        JavascriptVariableRootNode(
+                            vars.filter { it.scope == LiveVariableScope.GLOBAL_VARIABLE },
+                            LiveVariableScope.GLOBAL_VARIABLE
+                        )
                     )
-                )
-            }
+                }
 
-            val productCode = ApplicationInfo.getInstance().build.productCode
-            if (PYCHARM_PRODUCT_CODES.contains(productCode)) {
-                return arrayOf(
-                    PythonVariableRootNode(
-                        vars.filter { it.scope == LiveVariableScope.GLOBAL_VARIABLE },
-                        LiveVariableScope.GLOBAL_VARIABLE
-                    ),
-                    PythonVariableRootNode(
-                        vars.filter { it.scope == LiveVariableScope.LOCAL_VARIABLE },
-                        LiveVariableScope.LOCAL_VARIABLE
+                ArtifactLanguage.PYTHON -> {
+                    return arrayOf(
+                        PythonVariableRootNode(
+                            vars.filter { it.scope == LiveVariableScope.GLOBAL_VARIABLE },
+                            LiveVariableScope.GLOBAL_VARIABLE
+                        ),
+                        PythonVariableRootNode(
+                            vars.filter { it.scope == LiveVariableScope.LOCAL_VARIABLE },
+                            LiveVariableScope.LOCAL_VARIABLE
+                        )
                     )
-                )
-            } else {
-                val simpleNodeMap: MutableMap<String, JVMVariableSimpleNode> = LinkedHashMap()
-                vars.forEach {
-                    if (it.name.isNotEmpty()) {
-                        simpleNodeMap[it.name] = JVMVariableSimpleNode(it, mutableMapOf())
-                    }
                 }
-                simpleNodeMap.values.sortedWith { p0, p1 ->
-                    when {
-                        p0.variable.name == "this" -> -1
-                        p1.variable.name == "this" -> 1
-                        else -> 0
+
+                else -> {
+                    val simpleNodeMap: MutableMap<String, JVMVariableSimpleNode> = LinkedHashMap()
+                    vars.forEach {
+                        if (it.name.isNotEmpty()) {
+                            simpleNodeMap[it.name] = JVMVariableSimpleNode(it, mutableMapOf())
+                        }
                     }
-                }.toTypedArray()
+                    simpleNodeMap.values.sortedWith { p0, p1 ->
+                        when {
+                            p0.variable.name == "this" -> -1
+                            p1.variable.name == "this" -> 1
+                            else -> 0
+                        }
+                    }.toTypedArray()
+                }
             }
         }
     }